| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
TensorFlow.OpGen.ParsedOp
Description
This module helps parse the proto OpDef into a Haskell type which is more descriptive of how the attributes and arguments will be used in the generated code.
- data ParsedOp = ParsedOp {
- parsedOpName :: Name
- parsedOpSummary :: Text
- parsedOpDescription :: Text
- parsedInputs :: [ParsedArg]
- parsedOutputs :: [ParsedArg]
- explicitInputAttrs :: [Attr AttrType]
- inferredTypeAttrs :: [Attr TypeParam]
- inferredListSizeAttrs :: [Attr (NonEmpty Name)]
- parsedOpIsMonadic :: Bool
- data Name = Name {}
- newtype HaskellName = HaskellName {
- unHaskellName :: Text
- newtype TFName = TFName {
- unTFName :: Text
- data Attr a = Attr {
- attrName :: Name
- attrDescription :: Text
- attrInfo :: a
- data AttrType
- data AttrBaseType
- data TypeParam = TypeParam {}
- data ParsedArg = ParsedArg {
- parsedArgName :: Name
- parsedArgDescription :: Text
- parsedArgCase :: ParsedArgCase
- data ParsedArgCase
- data ArgType
- data ArgKind
- parseOp :: OpDef -> ParsedOp
- camelCase :: Text -> Text
Documentation
Constructors
| ParsedOp | |
Fields
| |
Constructors
| Name | |
Fields
| |
newtype HaskellName Source #
A name that's appropriate for a variable in a Haskell source file.
Constructors
| HaskellName | |
Fields
| |
A raw name as specified in the OpDef proto.
A named attribute, associated with some information about it.
Constructors
| Attr | |
Fields
| |
The type of an attribute.
Constructors
| AttrSingle AttrBaseType | |
| AttrList AttrBaseType |
data AttrBaseType Source #
Instances
Constructors
| TypeParam | |
Fields
| |
An input or output argument (Tensor) for an op.
Constructors
| ParsedArg | |
Fields
| |
data ParsedArgCase Source #
The type of an argument.
Constructors
| ArgTypeFixed DataType | A fixed type. |
| ArgTypeAttr Name | A type that depends on an attribute. |
Constructors
| ArgTensorRef | |
| ArgTensorValue | |
| ArgTensorBuild | |
| ArgSomeTensor Text |