diff --git a/tensorflow-core-ops/Setup.hs b/tensorflow-core-ops/Setup.hs index aebd49d..73f2e22 100644 --- a/tensorflow-core-ops/Setup.hs +++ b/tensorflow-core-ops/Setup.hs @@ -128,10 +128,12 @@ blackList = , "ScanDataset" , "SnapshotDatasetV2" , "StatefulPartitionedCall" + , "StatelessCase" , "StatelessIf" , "StatelessWhile" , "SymbolicGradient" , "TakeWhileDataset" + , "TPUCompile" , "TPUPartitionedCall" , "TPUReplicate" , "While" diff --git a/tensorflow-opgen/src/TensorFlow/OpGen.hs b/tensorflow-opgen/src/TensorFlow/OpGen.hs index 421b147..0d8a746 100644 --- a/tensorflow-opgen/src/TensorFlow/OpGen.hs +++ b/tensorflow-opgen/src/TensorFlow/OpGen.hs @@ -321,6 +321,7 @@ typeSig pre pOp = constraints AttrType -> "DataType" AttrShape -> "Shape" AttrTensor -> "TensorProto" + AttrFunc -> error "AttrFunc not supported" tensorArgAndComment t = tensorArg t <+> hang 0 ("-- ^" <+> argComment t) outputs = case parsedOutputs pOp of diff --git a/tensorflow-opgen/src/TensorFlow/OpGen/ParsedOp.hs b/tensorflow-opgen/src/TensorFlow/OpGen/ParsedOp.hs index e0fba89..b995713 100644 --- a/tensorflow-opgen/src/TensorFlow/OpGen/ParsedOp.hs +++ b/tensorflow-opgen/src/TensorFlow/OpGen/ParsedOp.hs @@ -101,7 +101,7 @@ data AttrType = AttrSingle AttrBaseType deriving Eq data AttrBaseType = AttrBytes | AttrInt64 | AttrFloat | AttrBool - | AttrType | AttrShape | AttrTensor + | AttrType | AttrShape | AttrTensor | AttrFunc deriving Eq data TypeParam = TypeParam @@ -334,6 +334,7 @@ parseAttrType o = \case "type" -> AttrSingle AttrType "shape" -> AttrSingle AttrShape "tensor" -> AttrSingle AttrTensor + "func" -> AttrSingle AttrFunc "list(string)" -> AttrList AttrBytes "list(int)" -> AttrList AttrInt64 "list(float)" -> AttrList AttrFloat @@ -341,5 +342,6 @@ parseAttrType o = \case "list(type)" -> AttrList AttrType "list(shape)" -> AttrList AttrShape "list(tensor)" -> AttrList AttrTensor + "list(func)" -> AttrList AttrFunc t -> error $ "parseAttrType: unrecognized type " ++ show t ++ " for op " ++ show (o ^. name)