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

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 {
- haskellName :: HaskellName
- tfName :: TFName
}
newtype HaskellName = HaskellName {
- unHaskellName :: Text
}
newtype TFName = TFName {
- unTFName :: Text
}
data Attr a = Attr {
- attrName :: Name
- attrDescription :: Text
- attrInfo :: a
}
data AttrType
- = AttrSingle AttrBaseType
- | AttrList AttrBaseType
data AttrBaseType
- = AttrBytes
- | AttrInt64
- | AttrFloat
- | AttrBool
- | AttrType
- | AttrShape
- | AttrTensor
data TypeParam = TypeParam {
- typeParamIsList :: Bool
- typeParamRestrictions :: Maybe (NonEmpty DataType)
}
data ParsedArg = ParsedArg {
- parsedArgName :: Name
- parsedArgDescription :: Text
- parsedArgCase :: ParsedArgCase
}
data ParsedArgCase
- = SimpleArg {
  - argType :: ArgType
  - argKind :: ArgKind
  }
- | ListArg {
  - argLength :: Name
  - argType :: ArgType
  - argKind :: ArgKind
  }
- | MixedListArg {
  - argTypeAttr :: Name
  - argKind :: ArgKind
  }
data ArgType
- = ArgTypeFixed DataType
- | ArgTypeAttr Name
data ArgKind
- = ArgTensorRef
- | ArgTensorValue
- | ArgTensorBuild
- | ArgSomeTensor Text
parseOp :: OpDef -> ParsedOp
camelCase :: Text -> Text

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 TypeParam]
Attributes that are type parameters.
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

Instances details

Eq TFName Source #
Instance details Defined in TensorFlow.OpGen.ParsedOp Methods (==) :: TFName -> TFName -> Bool # (/=) :: TFName -> TFName -> Bool #
Ord TFName Source #
Instance details Defined in TensorFlow.OpGen.ParsedOp Methods compare :: TFName -> TFName -> Ordering # (<) :: TFName -> TFName -> Bool # (<=) :: TFName -> TFName -> Bool # (>) :: TFName -> TFName -> Bool # (>=) :: TFName -> TFName -> Bool # max :: TFName -> TFName -> TFName # min :: TFName -> TFName -> TFName #

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

Instances details

Eq AttrType Source #
Instance details Defined in TensorFlow.OpGen.ParsedOp Methods (==) :: AttrType -> AttrType -> Bool # (/=) :: AttrType -> AttrType -> Bool #

data AttrBaseType Source #

Constructors

AttrBytes
AttrInt64
AttrFloat
AttrBool
AttrType
AttrShape
AttrTensor

Instances

Instances details

Eq AttrBaseType Source #
Instance details Defined in TensorFlow.OpGen.ParsedOp Methods (==) :: AttrBaseType -> AttrBaseType -> Bool # (/=) :: AttrBaseType -> AttrBaseType -> Bool #

data TypeParam Source #

Constructors

TypeParam
Fields typeParamIsList :: Bool typeParamRestrictions :: Maybe (NonEmpty DataType) The list of allowed types (see: TensorFlow.Types.OneOf). If `Nothing`, then any type is acceptable.

data ParsedArg Source #

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

Constructors

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

data ParsedArgCase Source #

Constructors

SimpleArg
Fields argType :: ArgType argKind :: ArgKind
ListArg
Fields argLength :: Name The attribute that specifies this list's length. argType :: ArgType argKind :: ArgKind
MixedListArg	A heterogeneous list.
Fields argTypeAttr :: Name argKind :: ArgKind

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
ArgTensorBuild
ArgSomeTensor Text

Instances

Instances details

Eq ArgKind Source #
Instance details Defined in TensorFlow.OpGen.ParsedOp Methods (==) :: ArgKind -> ArgKind -> Bool # (/=) :: ArgKind -> ArgKind -> Bool #

parseOp :: OpDef -> ParsedOp Source #

camelCase :: Text -> Text Source #