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.

Synopsis

Documentation

data ParsedOp Source

Constructors

ParsedOp

Fields

parsedOpName :: Name
parsedOpSummary :: Text
parsedOpDescription :: Text
parsedInputs :: [ParsedArg]
parsedOutputs :: [ParsedArg]
explicitInputAttrs :: [Attr AttrType]: Attributes that must be set explicitly when creating the op. Associated with the type of the attribute.
inferredTypeAttrs :: [Attr [DataType]]: Attributes that are type parameters. Associated with the list of allowed types (see: TensorFlow.Types.OneOf). If this list is empty, then any type is acceptable.
inferredListSizeAttrs :: [Attr (NonEmpty Name)]
parsedOpIsMonadic :: Bool: Whether this op is stateful or takes a stateful input. Such ops should not be CSE'd and must be monadic in our API (i.e., return a Build action).

data Name Source

Constructors

Name
Fields haskellName :: HaskellName tfName :: TFName

newtype HaskellName Source

A name that's appropriate for a variable in a Haskell source file.

Constructors

HaskellName
Fields unHaskellName :: Text

newtype TFName Source

A raw name as specified in the OpDef proto.

Constructors

TFName
Fields unTFName :: Text

Instances

Eq TFName Source
Ord TFName Source

data Attr a Source

A named attribute, associated with some information about it.

Constructors

Attr
Fields attrName :: Name attrDescription :: Text attrInfo :: a

data AttrType Source

The type of an attribute.

Constructors

AttrSingle AttrBaseType
AttrList AttrBaseType

Instances

Eq AttrType Source

data AttrBaseType Source

Constructors

AttrBytes
AttrInt64
AttrFloat
AttrBool
AttrType
AttrShape
AttrTensor

Instances

Eq AttrBaseType Source

data ParsedArg Source

An input or output argument (Tensor) for an op.

Constructors

ParsedArg
Fields parsedArgName :: Name parsedArgDescription :: Text parsedArgCase :: ParsedArgCase parsedArgKind :: ArgKind

data ParsedArgCase Source

Constructors

SimpleArg
Fields argType :: ArgType
ListArg
Fields argLength :: Name The attribute that specifies this list's length. argType :: ArgType
MixedListArg	A heterogeneous list. TODO(judahjacobson): Implement this.
Fields argTypeAttr :: Name

data ArgType Source

The type of an argument.

Constructors

ArgTypeFixed DataType	A fixed type.
ArgTypeAttr Name	A type that depends on an attribute.

data ArgKind Source

Constructors

ArgTensorRef
ArgTensorValue
ArgTensorEither Text
ArgResource

parseOp :: OpDef -> ParsedOp Source

camelCase :: Text -> Text Source