Safe Haskell | None |
---|---|
Language | Haskell2010 |
Rendering of TensorFlow operations as Haskell functions.
The basic type signature generated for each op is:
{constraints} => {mandatory attrs} -> {input tensors} -> {output tensors}
where:
{mandatory attrs}
is of the formA_1 -> ... -> A_N
, where eachA
is an op attribute that doesn't have a default and can't be inferred from other inputs.{constraints}
restrict the type parameters of the input and output tensors (for example:TensorType
orOneOf
).{input tensors}
is of the formT_1 -> ... -> T_N
, where eachT
is of the formTensor Ref a
,Tensor v a
orResourceHandle
(or a list of one of those types), anda
is either a concrete type or a (constrained) type variable.{output tensors}
is of the form(T_1,...,T_N)
for "pure" ops, andBuild (T_1,...,T_N)
for "stateful" ops. An op is considered "stateful" if it takes aTensor Ref
orResourceHandle
as input, or if it's explicitly marked "Stateful" in itsREGISTER_OP
definition. (If there are no outputs, it is eitherControlNode
orBuild ControlNode
.)
Documentation
data OpGenFlags
OpGenFlags | |
|
docOpList :: OpGenFlags -> OpList -> Doc
flagParser :: Parser OpGenFlags