| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
TensorFlow.OpGen
Description
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 eachAis 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:TensorTypeorOneOf).{input tensors}is of the formT_1 -> ... -> T_N, where eachTis of the formTensor Ref aorTensor v a(or a list of one of those types), andais 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 ReforTensor v ResourceHandleas input, or if it's explicitly marked "Stateful" in itsREGISTER_OPdefinition. (If there are no outputs, it is eitherControlNodeorBuild ControlNode.)
Documentation
data OpGenFlags Source #
Constructors
| OpGenFlags | |
Fields
| |
docOpList :: OpGenFlags -> OpList -> Doc Source #
flagParser :: Parser OpGenFlags Source #