eeva/tensorflow-haskell
1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-14 06:59:43 +01:00
Code Issues Releases Wiki Activity

Add StatelessCase, TPUCompile to blacklist.

Browse source 
Needed to support Tensorflow 2.4.

Also, add AttrFunc to unbreak build.
This commit is contained in:
Mike Sperber 2021-01-29 10:51:12 +01:00
parent 6b19e54722
commit 29b6729d68
3 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tensorflow-core-ops/Setup.hs
View file

@ -128,10 +128,12 @@ blackList =
, "ScanDataset"
, "SnapshotDatasetV2"
, "StatefulPartitionedCall"
, "StatelessCase"
, "StatelessIf"
, "StatelessWhile"
, "SymbolicGradient"
, "TakeWhileDataset"
, "TPUCompile"
, "TPUPartitionedCall"
, "TPUReplicate"
, "While"

1
tensorflow-opgen/src/TensorFlow/OpGen.hs
View file

@ -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

4
tensorflow-opgen/src/TensorFlow/OpGen/ParsedOp.hs
View file

@ -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)