Documentation ¶
Index ¶
- Variables
- type AllocationDescription
- func (*AllocationDescription) Descriptor() ([]byte, []int)deprecated
- func (x *AllocationDescription) GetAllocatedBytes() int64
- func (x *AllocationDescription) GetAllocationId() int64
- func (x *AllocationDescription) GetAllocatorName() string
- func (x *AllocationDescription) GetHasSingleReference() bool
- func (x *AllocationDescription) GetPtr() uint64
- func (x *AllocationDescription) GetRequestedBytes() int64
- func (*AllocationDescription) ProtoMessage()
- func (x *AllocationDescription) ProtoReflect() protoreflect.Message
- func (x *AllocationDescription) Reset()
- func (x *AllocationDescription) String() string
- type AllocationRecord
- func (*AllocationRecord) Descriptor() ([]byte, []int)deprecated
- func (x *AllocationRecord) GetAllocBytes() int64
- func (x *AllocationRecord) GetAllocMicros() int64
- func (*AllocationRecord) ProtoMessage()
- func (x *AllocationRecord) ProtoReflect() protoreflect.Message
- func (x *AllocationRecord) Reset()
- func (x *AllocationRecord) String() string
- type AllocatorMemoryUsed
- func (*AllocatorMemoryUsed) Descriptor() ([]byte, []int)deprecated
- func (x *AllocatorMemoryUsed) GetAllocationRecords() []*AllocationRecord
- func (x *AllocatorMemoryUsed) GetAllocatorBytesInUse() int64
- func (x *AllocatorMemoryUsed) GetAllocatorName() string
- func (x *AllocatorMemoryUsed) GetLiveBytes() int64
- func (x *AllocatorMemoryUsed) GetPeakBytes() int64
- func (x *AllocatorMemoryUsed) GetTotalBytes() int64
- func (*AllocatorMemoryUsed) ProtoMessage()
- func (x *AllocatorMemoryUsed) ProtoReflect() protoreflect.Message
- func (x *AllocatorMemoryUsed) Reset()
- func (x *AllocatorMemoryUsed) String() string
- type ApiDef
- func (*ApiDef) Descriptor() ([]byte, []int)deprecated
- func (x *ApiDef) GetArgOrder() []string
- func (x *ApiDef) GetAttr() []*ApiDef_Attr
- func (x *ApiDef) GetDeprecationMessage() string
- func (x *ApiDef) GetDeprecationVersion() int32
- func (x *ApiDef) GetDescription() string
- func (x *ApiDef) GetDescriptionPrefix() string
- func (x *ApiDef) GetDescriptionSuffix() string
- func (x *ApiDef) GetEndpoint() []*ApiDef_Endpoint
- func (x *ApiDef) GetGraphOpName() string
- func (x *ApiDef) GetInArg() []*ApiDef_Arg
- func (x *ApiDef) GetOutArg() []*ApiDef_Arg
- func (x *ApiDef) GetSummary() string
- func (x *ApiDef) GetVisibility() ApiDef_Visibility
- func (*ApiDef) ProtoMessage()
- func (x *ApiDef) ProtoReflect() protoreflect.Message
- func (x *ApiDef) Reset()
- func (x *ApiDef) String() string
- type ApiDef_Arg
- func (*ApiDef_Arg) Descriptor() ([]byte, []int)deprecated
- func (x *ApiDef_Arg) GetDescription() string
- func (x *ApiDef_Arg) GetName() string
- func (x *ApiDef_Arg) GetRenameTo() string
- func (*ApiDef_Arg) ProtoMessage()
- func (x *ApiDef_Arg) ProtoReflect() protoreflect.Message
- func (x *ApiDef_Arg) Reset()
- func (x *ApiDef_Arg) String() string
- type ApiDef_Attr
- func (*ApiDef_Attr) Descriptor() ([]byte, []int)deprecated
- func (x *ApiDef_Attr) GetDefaultValue() *AttrValue
- func (x *ApiDef_Attr) GetDescription() string
- func (x *ApiDef_Attr) GetName() string
- func (x *ApiDef_Attr) GetRenameTo() string
- func (*ApiDef_Attr) ProtoMessage()
- func (x *ApiDef_Attr) ProtoReflect() protoreflect.Message
- func (x *ApiDef_Attr) Reset()
- func (x *ApiDef_Attr) String() string
- type ApiDef_Endpoint
- func (*ApiDef_Endpoint) Descriptor() ([]byte, []int)deprecated
- func (x *ApiDef_Endpoint) GetDeprecated() bool
- func (x *ApiDef_Endpoint) GetDeprecationVersion() int32
- func (x *ApiDef_Endpoint) GetName() string
- func (*ApiDef_Endpoint) ProtoMessage()
- func (x *ApiDef_Endpoint) ProtoReflect() protoreflect.Message
- func (x *ApiDef_Endpoint) Reset()
- func (x *ApiDef_Endpoint) String() string
- type ApiDef_Visibility
- func (ApiDef_Visibility) Descriptor() protoreflect.EnumDescriptor
- func (x ApiDef_Visibility) Enum() *ApiDef_Visibility
- func (ApiDef_Visibility) EnumDescriptor() ([]byte, []int)deprecated
- func (x ApiDef_Visibility) Number() protoreflect.EnumNumber
- func (x ApiDef_Visibility) String() string
- func (ApiDef_Visibility) Type() protoreflect.EnumType
- type ApiDefs
- type AttrValue
- func (*AttrValue) Descriptor() ([]byte, []int)deprecated
- func (x *AttrValue) GetB() bool
- func (x *AttrValue) GetF() float32
- func (x *AttrValue) GetFunc() *NameAttrList
- func (x *AttrValue) GetI() int64
- func (x *AttrValue) GetList() *AttrValue_ListValue
- func (x *AttrValue) GetPlaceholder() string
- func (x *AttrValue) GetS() []byte
- func (x *AttrValue) GetShape() *TensorShapeProto
- func (x *AttrValue) GetTensor() *TensorProto
- func (x *AttrValue) GetType() DataType
- func (m *AttrValue) GetValue() isAttrValue_Value
- func (*AttrValue) ProtoMessage()
- func (x *AttrValue) ProtoReflect() protoreflect.Message
- func (x *AttrValue) Reset()
- func (x *AttrValue) String() string
- type AttrValue_B
- type AttrValue_F
- type AttrValue_Func
- type AttrValue_I
- type AttrValue_List
- type AttrValue_ListValue
- func (*AttrValue_ListValue) Descriptor() ([]byte, []int)deprecated
- func (x *AttrValue_ListValue) GetB() []bool
- func (x *AttrValue_ListValue) GetF() []float32
- func (x *AttrValue_ListValue) GetFunc() []*NameAttrList
- func (x *AttrValue_ListValue) GetI() []int64
- func (x *AttrValue_ListValue) GetS() [][]byte
- func (x *AttrValue_ListValue) GetShape() []*TensorShapeProto
- func (x *AttrValue_ListValue) GetTensor() []*TensorProto
- func (x *AttrValue_ListValue) GetType() []DataType
- func (*AttrValue_ListValue) ProtoMessage()
- func (x *AttrValue_ListValue) ProtoReflect() protoreflect.Message
- func (x *AttrValue_ListValue) Reset()
- func (x *AttrValue_ListValue) String() string
- type AttrValue_Placeholder
- type AttrValue_S
- type AttrValue_Shape
- type AttrValue_Tensor
- type AttrValue_Type
- type CostGraphDef
- type CostGraphDef_Node
- func (*CostGraphDef_Node) Descriptor() ([]byte, []int)deprecated
- func (x *CostGraphDef_Node) GetComputeCost() int64
- func (x *CostGraphDef_Node) GetComputeTime() int64
- func (x *CostGraphDef_Node) GetControlInput() []int32
- func (x *CostGraphDef_Node) GetDevice() string
- func (x *CostGraphDef_Node) GetDevicePersistentMemorySize() int64deprecated
- func (x *CostGraphDef_Node) GetDeviceTempMemorySize() int64deprecated
- func (x *CostGraphDef_Node) GetHostTempMemorySize() int64deprecated
- func (x *CostGraphDef_Node) GetId() int32
- func (x *CostGraphDef_Node) GetInaccurate() bool
- func (x *CostGraphDef_Node) GetInputInfo() []*CostGraphDef_Node_InputInfo
- func (x *CostGraphDef_Node) GetIsFinal() bool
- func (x *CostGraphDef_Node) GetMemoryTime() int64
- func (x *CostGraphDef_Node) GetName() string
- func (x *CostGraphDef_Node) GetOutputInfo() []*CostGraphDef_Node_OutputInfo
- func (x *CostGraphDef_Node) GetPersistentMemorySize() int64
- func (x *CostGraphDef_Node) GetTemporaryMemorySize() int64
- func (*CostGraphDef_Node) ProtoMessage()
- func (x *CostGraphDef_Node) ProtoReflect() protoreflect.Message
- func (x *CostGraphDef_Node) Reset()
- func (x *CostGraphDef_Node) String() string
- type CostGraphDef_Node_InputInfo
- func (*CostGraphDef_Node_InputInfo) Descriptor() ([]byte, []int)deprecated
- func (x *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32
- func (x *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32
- func (*CostGraphDef_Node_InputInfo) ProtoMessage()
- func (x *CostGraphDef_Node_InputInfo) ProtoReflect() protoreflect.Message
- func (x *CostGraphDef_Node_InputInfo) Reset()
- func (x *CostGraphDef_Node_InputInfo) String() string
- type CostGraphDef_Node_OutputInfo
- func (*CostGraphDef_Node_OutputInfo) Descriptor() ([]byte, []int)deprecated
- func (x *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64
- func (x *CostGraphDef_Node_OutputInfo) GetDtype() DataType
- func (x *CostGraphDef_Node_OutputInfo) GetShape() *TensorShapeProto
- func (x *CostGraphDef_Node_OutputInfo) GetSize() int64
- func (*CostGraphDef_Node_OutputInfo) ProtoMessage()
- func (x *CostGraphDef_Node_OutputInfo) ProtoReflect() protoreflect.Message
- func (x *CostGraphDef_Node_OutputInfo) Reset()
- func (x *CostGraphDef_Node_OutputInfo) String() string
- type DataType
- type DeviceAttributes
- func (*DeviceAttributes) Descriptor() ([]byte, []int)deprecated
- func (x *DeviceAttributes) GetDeviceType() string
- func (x *DeviceAttributes) GetIncarnation() uint64
- func (x *DeviceAttributes) GetLocality() *DeviceLocality
- func (x *DeviceAttributes) GetMemoryLimit() int64
- func (x *DeviceAttributes) GetName() string
- func (x *DeviceAttributes) GetPhysicalDeviceDesc() string
- func (*DeviceAttributes) ProtoMessage()
- func (x *DeviceAttributes) ProtoReflect() protoreflect.Message
- func (x *DeviceAttributes) Reset()
- func (x *DeviceAttributes) String() string
- type DeviceLocality
- func (*DeviceLocality) Descriptor() ([]byte, []int)deprecated
- func (x *DeviceLocality) GetBusId() int32
- func (x *DeviceLocality) GetLinks() *LocalLinks
- func (x *DeviceLocality) GetNumaNode() int32
- func (*DeviceLocality) ProtoMessage()
- func (x *DeviceLocality) ProtoReflect() protoreflect.Message
- func (x *DeviceLocality) Reset()
- func (x *DeviceLocality) String() string
- type DeviceStepStats
- func (*DeviceStepStats) Descriptor() ([]byte, []int)deprecated
- func (x *DeviceStepStats) GetDevice() string
- func (x *DeviceStepStats) GetNodeStats() []*NodeExecStats
- func (*DeviceStepStats) ProtoMessage()
- func (x *DeviceStepStats) ProtoReflect() protoreflect.Message
- func (x *DeviceStepStats) Reset()
- func (x *DeviceStepStats) String() string
- type FunctionDef
- func (*FunctionDef) Descriptor() ([]byte, []int)deprecated
- func (x *FunctionDef) GetAttr() map[string]*AttrValue
- func (x *FunctionDef) GetNodeDef() []*NodeDef
- func (x *FunctionDef) GetRet() map[string]string
- func (x *FunctionDef) GetSignature() *OpDef
- func (*FunctionDef) ProtoMessage()
- func (x *FunctionDef) ProtoReflect() protoreflect.Message
- func (x *FunctionDef) Reset()
- func (x *FunctionDef) String() string
- type FunctionDefLibrary
- func (*FunctionDefLibrary) Descriptor() ([]byte, []int)deprecated
- func (x *FunctionDefLibrary) GetFunction() []*FunctionDef
- func (x *FunctionDefLibrary) GetGradient() []*GradientDef
- func (*FunctionDefLibrary) ProtoMessage()
- func (x *FunctionDefLibrary) ProtoReflect() protoreflect.Message
- func (x *FunctionDefLibrary) Reset()
- func (x *FunctionDefLibrary) String() string
- type GradientDef
- func (*GradientDef) Descriptor() ([]byte, []int)deprecated
- func (x *GradientDef) GetFunctionName() string
- func (x *GradientDef) GetGradientFunc() string
- func (*GradientDef) ProtoMessage()
- func (x *GradientDef) ProtoReflect() protoreflect.Message
- func (x *GradientDef) Reset()
- func (x *GradientDef) String() string
- type GraphDef
- func (*GraphDef) Descriptor() ([]byte, []int)deprecated
- func (x *GraphDef) GetLibrary() *FunctionDefLibrary
- func (x *GraphDef) GetNode() []*NodeDef
- func (x *GraphDef) GetVersion() int32deprecated
- func (x *GraphDef) GetVersions() *VersionDef
- func (*GraphDef) ProtoMessage()
- func (x *GraphDef) ProtoReflect() protoreflect.Message
- func (x *GraphDef) Reset()
- func (x *GraphDef) String() string
- type GraphTransferConstNodeInfo
- func (*GraphTransferConstNodeInfo) Descriptor() ([]byte, []int)deprecated
- func (x *GraphTransferConstNodeInfo) GetData() []byte
- func (x *GraphTransferConstNodeInfo) GetDtype() DataType
- func (x *GraphTransferConstNodeInfo) GetName() string
- func (x *GraphTransferConstNodeInfo) GetNodeId() int32
- func (x *GraphTransferConstNodeInfo) GetShape() []int64
- func (*GraphTransferConstNodeInfo) ProtoMessage()
- func (x *GraphTransferConstNodeInfo) ProtoReflect() protoreflect.Message
- func (x *GraphTransferConstNodeInfo) Reset()
- func (x *GraphTransferConstNodeInfo) String() string
- type GraphTransferGraphInputNodeInfo
- func (*GraphTransferGraphInputNodeInfo) Descriptor() ([]byte, []int)deprecated
- func (x *GraphTransferGraphInputNodeInfo) GetDtype() DataType
- func (x *GraphTransferGraphInputNodeInfo) GetName() string
- func (x *GraphTransferGraphInputNodeInfo) GetShape() []int64
- func (*GraphTransferGraphInputNodeInfo) ProtoMessage()
- func (x *GraphTransferGraphInputNodeInfo) ProtoReflect() protoreflect.Message
- func (x *GraphTransferGraphInputNodeInfo) Reset()
- func (x *GraphTransferGraphInputNodeInfo) String() string
- type GraphTransferGraphOutputNodeInfo
- func (*GraphTransferGraphOutputNodeInfo) Descriptor() ([]byte, []int)deprecated
- func (x *GraphTransferGraphOutputNodeInfo) GetDtype() DataType
- func (x *GraphTransferGraphOutputNodeInfo) GetName() string
- func (x *GraphTransferGraphOutputNodeInfo) GetShape() []int64
- func (*GraphTransferGraphOutputNodeInfo) ProtoMessage()
- func (x *GraphTransferGraphOutputNodeInfo) ProtoReflect() protoreflect.Message
- func (x *GraphTransferGraphOutputNodeInfo) Reset()
- func (x *GraphTransferGraphOutputNodeInfo) String() string
- type GraphTransferInfo
- func (*GraphTransferInfo) Descriptor() ([]byte, []int)deprecated
- func (x *GraphTransferInfo) GetConstNodeInfo() []*GraphTransferConstNodeInfo
- func (x *GraphTransferInfo) GetDestination() GraphTransferInfo_Destination
- func (x *GraphTransferInfo) GetGraphInputNodeInfo() []*GraphTransferGraphInputNodeInfo
- func (x *GraphTransferInfo) GetGraphOutputNodeInfo() []*GraphTransferGraphOutputNodeInfo
- func (x *GraphTransferInfo) GetNodeInfo() []*GraphTransferNodeInfo
- func (x *GraphTransferInfo) GetNodeInputInfo() []*GraphTransferNodeInputInfo
- func (x *GraphTransferInfo) GetNodeOutputInfo() []*GraphTransferNodeOutputInfo
- func (*GraphTransferInfo) ProtoMessage()
- func (x *GraphTransferInfo) ProtoReflect() protoreflect.Message
- func (x *GraphTransferInfo) Reset()
- func (x *GraphTransferInfo) String() string
- type GraphTransferInfo_Destination
- func (GraphTransferInfo_Destination) Descriptor() protoreflect.EnumDescriptor
- func (x GraphTransferInfo_Destination) Enum() *GraphTransferInfo_Destination
- func (GraphTransferInfo_Destination) EnumDescriptor() ([]byte, []int)deprecated
- func (x GraphTransferInfo_Destination) Number() protoreflect.EnumNumber
- func (x GraphTransferInfo_Destination) String() string
- func (GraphTransferInfo_Destination) Type() protoreflect.EnumType
- type GraphTransferNodeInfo
- func (*GraphTransferNodeInfo) Descriptor() ([]byte, []int)deprecated
- func (x *GraphTransferNodeInfo) GetInputCount() int32
- func (x *GraphTransferNodeInfo) GetName() string
- func (x *GraphTransferNodeInfo) GetNodeId() int32
- func (x *GraphTransferNodeInfo) GetOutputCount() int32
- func (x *GraphTransferNodeInfo) GetPaddingId() int32
- func (x *GraphTransferNodeInfo) GetSocOpId() int32
- func (x *GraphTransferNodeInfo) GetTypeName() string
- func (*GraphTransferNodeInfo) ProtoMessage()
- func (x *GraphTransferNodeInfo) ProtoReflect() protoreflect.Message
- func (x *GraphTransferNodeInfo) Reset()
- func (x *GraphTransferNodeInfo) String() string
- type GraphTransferNodeInput
- func (*GraphTransferNodeInput) Descriptor() ([]byte, []int)deprecated
- func (x *GraphTransferNodeInput) GetNodeId() int32
- func (x *GraphTransferNodeInput) GetOutputPort() int32
- func (*GraphTransferNodeInput) ProtoMessage()
- func (x *GraphTransferNodeInput) ProtoReflect() protoreflect.Message
- func (x *GraphTransferNodeInput) Reset()
- func (x *GraphTransferNodeInput) String() string
- type GraphTransferNodeInputInfo
- func (*GraphTransferNodeInputInfo) Descriptor() ([]byte, []int)deprecated
- func (x *GraphTransferNodeInputInfo) GetNodeId() int32
- func (x *GraphTransferNodeInputInfo) GetNodeInput() []*GraphTransferNodeInput
- func (*GraphTransferNodeInputInfo) ProtoMessage()
- func (x *GraphTransferNodeInputInfo) ProtoReflect() protoreflect.Message
- func (x *GraphTransferNodeInputInfo) Reset()
- func (x *GraphTransferNodeInputInfo) String() string
- type GraphTransferNodeOutputInfo
- func (*GraphTransferNodeOutputInfo) Descriptor() ([]byte, []int)deprecated
- func (x *GraphTransferNodeOutputInfo) GetMaxByteSize() []int32
- func (x *GraphTransferNodeOutputInfo) GetNodeId() int32
- func (*GraphTransferNodeOutputInfo) ProtoMessage()
- func (x *GraphTransferNodeOutputInfo) ProtoReflect() protoreflect.Message
- func (x *GraphTransferNodeOutputInfo) Reset()
- func (x *GraphTransferNodeOutputInfo) String() string
- type HistogramProto
- func (*HistogramProto) Descriptor() ([]byte, []int)deprecated
- func (x *HistogramProto) GetBucket() []float64
- func (x *HistogramProto) GetBucketLimit() []float64
- func (x *HistogramProto) GetMax() float64
- func (x *HistogramProto) GetMin() float64
- func (x *HistogramProto) GetNum() float64
- func (x *HistogramProto) GetSum() float64
- func (x *HistogramProto) GetSumSquares() float64
- func (*HistogramProto) ProtoMessage()
- func (x *HistogramProto) ProtoReflect() protoreflect.Message
- func (x *HistogramProto) Reset()
- func (x *HistogramProto) String() string
- type InterconnectLink
- func (*InterconnectLink) Descriptor() ([]byte, []int)deprecated
- func (x *InterconnectLink) GetDeviceId() int32
- func (x *InterconnectLink) GetStrength() int32
- func (x *InterconnectLink) GetType() string
- func (*InterconnectLink) ProtoMessage()
- func (x *InterconnectLink) ProtoReflect() protoreflect.Message
- func (x *InterconnectLink) Reset()
- func (x *InterconnectLink) String() string
- type IteratorStateMetadata
- func (*IteratorStateMetadata) Descriptor() ([]byte, []int)deprecated
- func (x *IteratorStateMetadata) GetKeys() []string
- func (x *IteratorStateMetadata) GetVersion() string
- func (*IteratorStateMetadata) ProtoMessage()
- func (x *IteratorStateMetadata) ProtoReflect() protoreflect.Message
- func (x *IteratorStateMetadata) Reset()
- func (x *IteratorStateMetadata) String() string
- type KernelDef
- func (*KernelDef) Descriptor() ([]byte, []int)deprecated
- func (x *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
- func (x *KernelDef) GetDeviceType() string
- func (x *KernelDef) GetHostMemoryArg() []string
- func (x *KernelDef) GetLabel() string
- func (x *KernelDef) GetOp() string
- func (x *KernelDef) GetPriority() int32
- func (*KernelDef) ProtoMessage()
- func (x *KernelDef) ProtoReflect() protoreflect.Message
- func (x *KernelDef) Reset()
- func (x *KernelDef) String() string
- type KernelDef_AttrConstraint
- func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)deprecated
- func (x *KernelDef_AttrConstraint) GetAllowedValues() *AttrValue
- func (x *KernelDef_AttrConstraint) GetName() string
- func (*KernelDef_AttrConstraint) ProtoMessage()
- func (x *KernelDef_AttrConstraint) ProtoReflect() protoreflect.Message
- func (x *KernelDef_AttrConstraint) Reset()
- func (x *KernelDef_AttrConstraint) String() string
- type KernelList
- type LocalLinks
- type MemoryLogRawAllocation
- func (*MemoryLogRawAllocation) Descriptor() ([]byte, []int)deprecated
- func (x *MemoryLogRawAllocation) GetAllocationId() int64
- func (x *MemoryLogRawAllocation) GetAllocatorName() string
- func (x *MemoryLogRawAllocation) GetNumBytes() int64
- func (x *MemoryLogRawAllocation) GetOperation() string
- func (x *MemoryLogRawAllocation) GetPtr() uint64
- func (x *MemoryLogRawAllocation) GetStepId() int64
- func (*MemoryLogRawAllocation) ProtoMessage()
- func (x *MemoryLogRawAllocation) ProtoReflect() protoreflect.Message
- func (x *MemoryLogRawAllocation) Reset()
- func (x *MemoryLogRawAllocation) String() string
- type MemoryLogRawDeallocation
- func (*MemoryLogRawDeallocation) Descriptor() ([]byte, []int)deprecated
- func (x *MemoryLogRawDeallocation) GetAllocationId() int64
- func (x *MemoryLogRawDeallocation) GetAllocatorName() string
- func (x *MemoryLogRawDeallocation) GetDeferred() bool
- func (x *MemoryLogRawDeallocation) GetOperation() string
- func (x *MemoryLogRawDeallocation) GetStepId() int64
- func (*MemoryLogRawDeallocation) ProtoMessage()
- func (x *MemoryLogRawDeallocation) ProtoReflect() protoreflect.Message
- func (x *MemoryLogRawDeallocation) Reset()
- func (x *MemoryLogRawDeallocation) String() string
- type MemoryLogStep
- func (*MemoryLogStep) Descriptor() ([]byte, []int)deprecated
- func (x *MemoryLogStep) GetHandle() string
- func (x *MemoryLogStep) GetStepId() int64
- func (*MemoryLogStep) ProtoMessage()
- func (x *MemoryLogStep) ProtoReflect() protoreflect.Message
- func (x *MemoryLogStep) Reset()
- func (x *MemoryLogStep) String() string
- type MemoryLogTensorAllocation
- func (*MemoryLogTensorAllocation) Descriptor() ([]byte, []int)deprecated
- func (x *MemoryLogTensorAllocation) GetKernelName() string
- func (x *MemoryLogTensorAllocation) GetStepId() int64
- func (x *MemoryLogTensorAllocation) GetTensor() *TensorDescription
- func (*MemoryLogTensorAllocation) ProtoMessage()
- func (x *MemoryLogTensorAllocation) ProtoReflect() protoreflect.Message
- func (x *MemoryLogTensorAllocation) Reset()
- func (x *MemoryLogTensorAllocation) String() string
- type MemoryLogTensorDeallocation
- func (*MemoryLogTensorDeallocation) Descriptor() ([]byte, []int)deprecated
- func (x *MemoryLogTensorDeallocation) GetAllocationId() int64
- func (x *MemoryLogTensorDeallocation) GetAllocatorName() string
- func (*MemoryLogTensorDeallocation) ProtoMessage()
- func (x *MemoryLogTensorDeallocation) ProtoReflect() protoreflect.Message
- func (x *MemoryLogTensorDeallocation) Reset()
- func (x *MemoryLogTensorDeallocation) String() string
- type MemoryLogTensorOutput
- func (*MemoryLogTensorOutput) Descriptor() ([]byte, []int)deprecated
- func (x *MemoryLogTensorOutput) GetIndex() int32
- func (x *MemoryLogTensorOutput) GetKernelName() string
- func (x *MemoryLogTensorOutput) GetStepId() int64
- func (x *MemoryLogTensorOutput) GetTensor() *TensorDescription
- func (*MemoryLogTensorOutput) ProtoMessage()
- func (x *MemoryLogTensorOutput) ProtoReflect() protoreflect.Message
- func (x *MemoryLogTensorOutput) Reset()
- func (x *MemoryLogTensorOutput) String() string
- type MemoryStats
- func (*MemoryStats) Descriptor() ([]byte, []int)deprecated
- func (x *MemoryStats) GetDevicePersistentMemorySize() int64deprecated
- func (x *MemoryStats) GetDevicePersistentTensorAllocIds() []int64deprecated
- func (x *MemoryStats) GetDeviceTempMemorySize() int64deprecated
- func (x *MemoryStats) GetPersistentMemorySize() int64
- func (x *MemoryStats) GetPersistentTensorAllocIds() []int64
- func (x *MemoryStats) GetTempMemorySize() int64
- func (*MemoryStats) ProtoMessage()
- func (x *MemoryStats) ProtoReflect() protoreflect.Message
- func (x *MemoryStats) Reset()
- func (x *MemoryStats) String() string
- type NameAttrList
- func (*NameAttrList) Descriptor() ([]byte, []int)deprecated
- func (x *NameAttrList) GetAttr() map[string]*AttrValue
- func (x *NameAttrList) GetName() string
- func (*NameAttrList) ProtoMessage()
- func (x *NameAttrList) ProtoReflect() protoreflect.Message
- func (x *NameAttrList) Reset()
- func (x *NameAttrList) String() string
- type NodeDef
- func (*NodeDef) Descriptor() ([]byte, []int)deprecated
- func (x *NodeDef) GetAttr() map[string]*AttrValue
- func (x *NodeDef) GetDevice() string
- func (x *NodeDef) GetExperimentalDebugInfo() *NodeDef_ExperimentalDebugInfo
- func (x *NodeDef) GetInput() []string
- func (x *NodeDef) GetName() string
- func (x *NodeDef) GetOp() string
- func (*NodeDef) ProtoMessage()
- func (x *NodeDef) ProtoReflect() protoreflect.Message
- func (x *NodeDef) Reset()
- func (x *NodeDef) String() string
- type NodeDef_ExperimentalDebugInfo
- func (*NodeDef_ExperimentalDebugInfo) Descriptor() ([]byte, []int)deprecated
- func (x *NodeDef_ExperimentalDebugInfo) GetOriginalNodeNames() []string
- func (*NodeDef_ExperimentalDebugInfo) ProtoMessage()
- func (x *NodeDef_ExperimentalDebugInfo) ProtoReflect() protoreflect.Message
- func (x *NodeDef_ExperimentalDebugInfo) Reset()
- func (x *NodeDef_ExperimentalDebugInfo) String() string
- type NodeExecStats
- func (*NodeExecStats) Descriptor() ([]byte, []int)deprecated
- func (x *NodeExecStats) GetAllEndRelMicros() int64
- func (x *NodeExecStats) GetAllEndRelNanos() int64
- func (x *NodeExecStats) GetAllStartMicros() int64
- func (x *NodeExecStats) GetAllStartNanos() int64
- func (x *NodeExecStats) GetMemory() []*AllocatorMemoryUsed
- func (x *NodeExecStats) GetMemoryStats() *MemoryStats
- func (x *NodeExecStats) GetNodeName() string
- func (x *NodeExecStats) GetOpEndRelMicros() int64
- func (x *NodeExecStats) GetOpEndRelNanos() int64
- func (x *NodeExecStats) GetOpStartRelMicros() int64
- func (x *NodeExecStats) GetOpStartRelNanos() int64
- func (x *NodeExecStats) GetOutput() []*NodeOutput
- func (x *NodeExecStats) GetReferencedTensor() []*AllocationDescription
- func (x *NodeExecStats) GetScheduledMicros() int64
- func (x *NodeExecStats) GetScheduledNanos() int64
- func (x *NodeExecStats) GetThreadId() uint32
- func (x *NodeExecStats) GetTimelineLabel() string
- func (*NodeExecStats) ProtoMessage()
- func (x *NodeExecStats) ProtoReflect() protoreflect.Message
- func (x *NodeExecStats) Reset()
- func (x *NodeExecStats) String() string
- type NodeOutput
- func (*NodeOutput) Descriptor() ([]byte, []int)deprecated
- func (x *NodeOutput) GetSlot() int32
- func (x *NodeOutput) GetTensorDescription() *TensorDescription
- func (*NodeOutput) ProtoMessage()
- func (x *NodeOutput) ProtoReflect() protoreflect.Message
- func (x *NodeOutput) Reset()
- func (x *NodeOutput) String() string
- type OpDef
- func (*OpDef) Descriptor() ([]byte, []int)deprecated
- func (x *OpDef) GetAllowsUninitializedInput() bool
- func (x *OpDef) GetAttr() []*OpDef_AttrDef
- func (x *OpDef) GetDeprecation() *OpDeprecation
- func (x *OpDef) GetDescription() string
- func (x *OpDef) GetInputArg() []*OpDef_ArgDef
- func (x *OpDef) GetIsAggregate() bool
- func (x *OpDef) GetIsCommutative() bool
- func (x *OpDef) GetIsStateful() bool
- func (x *OpDef) GetName() string
- func (x *OpDef) GetOutputArg() []*OpDef_ArgDef
- func (x *OpDef) GetSummary() string
- func (*OpDef) ProtoMessage()
- func (x *OpDef) ProtoReflect() protoreflect.Message
- func (x *OpDef) Reset()
- func (x *OpDef) String() string
- type OpDef_ArgDef
- func (*OpDef_ArgDef) Descriptor() ([]byte, []int)deprecated
- func (x *OpDef_ArgDef) GetDescription() string
- func (x *OpDef_ArgDef) GetIsRef() bool
- func (x *OpDef_ArgDef) GetName() string
- func (x *OpDef_ArgDef) GetNumberAttr() string
- func (x *OpDef_ArgDef) GetType() DataType
- func (x *OpDef_ArgDef) GetTypeAttr() string
- func (x *OpDef_ArgDef) GetTypeListAttr() string
- func (*OpDef_ArgDef) ProtoMessage()
- func (x *OpDef_ArgDef) ProtoReflect() protoreflect.Message
- func (x *OpDef_ArgDef) Reset()
- func (x *OpDef_ArgDef) String() string
- type OpDef_AttrDef
- func (*OpDef_AttrDef) Descriptor() ([]byte, []int)deprecated
- func (x *OpDef_AttrDef) GetAllowedValues() *AttrValue
- func (x *OpDef_AttrDef) GetDefaultValue() *AttrValue
- func (x *OpDef_AttrDef) GetDescription() string
- func (x *OpDef_AttrDef) GetHasMinimum() bool
- func (x *OpDef_AttrDef) GetMinimum() int64
- func (x *OpDef_AttrDef) GetName() string
- func (x *OpDef_AttrDef) GetType() string
- func (*OpDef_AttrDef) ProtoMessage()
- func (x *OpDef_AttrDef) ProtoReflect() protoreflect.Message
- func (x *OpDef_AttrDef) Reset()
- func (x *OpDef_AttrDef) String() string
- type OpDeprecation
- func (*OpDeprecation) Descriptor() ([]byte, []int)deprecated
- func (x *OpDeprecation) GetExplanation() string
- func (x *OpDeprecation) GetVersion() int32
- func (*OpDeprecation) ProtoMessage()
- func (x *OpDeprecation) ProtoReflect() protoreflect.Message
- func (x *OpDeprecation) Reset()
- func (x *OpDeprecation) String() string
- type OpList
- type ReaderBaseState
- func (*ReaderBaseState) Descriptor() ([]byte, []int)deprecated
- func (x *ReaderBaseState) GetCurrentWork() []byte
- func (x *ReaderBaseState) GetNumRecordsProduced() int64
- func (x *ReaderBaseState) GetWorkFinished() int64
- func (x *ReaderBaseState) GetWorkStarted() int64
- func (*ReaderBaseState) ProtoMessage()
- func (x *ReaderBaseState) ProtoReflect() protoreflect.Message
- func (x *ReaderBaseState) Reset()
- func (x *ReaderBaseState) String() string
- type RemoteFusedGraphExecuteInfo
- func (*RemoteFusedGraphExecuteInfo) Descriptor() ([]byte, []int)deprecated
- func (x *RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (x *RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (x *RemoteFusedGraphExecuteInfo) GetExecutorName() string
- func (x *RemoteFusedGraphExecuteInfo) GetGraphInputNodeName() []string
- func (x *RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName() []string
- func (x *RemoteFusedGraphExecuteInfo) GetRemoteGraph() *GraphDef
- func (x *RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters() []byte
- func (*RemoteFusedGraphExecuteInfo) ProtoMessage()
- func (x *RemoteFusedGraphExecuteInfo) ProtoReflect() protoreflect.Message
- func (x *RemoteFusedGraphExecuteInfo) Reset()
- func (x *RemoteFusedGraphExecuteInfo) String() string
- type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor() ([]byte, []int)deprecated
- func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype() DataType
- func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape() *TensorShapeProto
- func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage()
- func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoReflect() protoreflect.Message
- func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset()
- func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) String() string
- type ResourceHandleProto
- func (*ResourceHandleProto) Descriptor() ([]byte, []int)deprecated
- func (x *ResourceHandleProto) GetContainer() string
- func (x *ResourceHandleProto) GetDevice() string
- func (x *ResourceHandleProto) GetHashCode() uint64
- func (x *ResourceHandleProto) GetMaybeTypeName() string
- func (x *ResourceHandleProto) GetName() string
- func (*ResourceHandleProto) ProtoMessage()
- func (x *ResourceHandleProto) ProtoReflect() protoreflect.Message
- func (x *ResourceHandleProto) Reset()
- func (x *ResourceHandleProto) String() string
- type SaveSliceInfoDef
- func (*SaveSliceInfoDef) Descriptor() ([]byte, []int)deprecated
- func (x *SaveSliceInfoDef) GetFullName() string
- func (x *SaveSliceInfoDef) GetFullShape() []int64
- func (x *SaveSliceInfoDef) GetVarOffset() []int64
- func (x *SaveSliceInfoDef) GetVarShape() []int64
- func (*SaveSliceInfoDef) ProtoMessage()
- func (x *SaveSliceInfoDef) ProtoReflect() protoreflect.Message
- func (x *SaveSliceInfoDef) Reset()
- func (x *SaveSliceInfoDef) String() string
- type StepStats
- type Summary
- type SummaryDescription
- func (*SummaryDescription) Descriptor() ([]byte, []int)deprecated
- func (x *SummaryDescription) GetTypeHint() string
- func (*SummaryDescription) ProtoMessage()
- func (x *SummaryDescription) ProtoReflect() protoreflect.Message
- func (x *SummaryDescription) Reset()
- func (x *SummaryDescription) String() string
- type SummaryMetadata
- func (*SummaryMetadata) Descriptor() ([]byte, []int)deprecated
- func (x *SummaryMetadata) GetDisplayName() string
- func (x *SummaryMetadata) GetPluginData() *SummaryMetadata_PluginData
- func (x *SummaryMetadata) GetSummaryDescription() string
- func (*SummaryMetadata) ProtoMessage()
- func (x *SummaryMetadata) ProtoReflect() protoreflect.Message
- func (x *SummaryMetadata) Reset()
- func (x *SummaryMetadata) String() string
- type SummaryMetadata_PluginData
- func (*SummaryMetadata_PluginData) Descriptor() ([]byte, []int)deprecated
- func (x *SummaryMetadata_PluginData) GetContent() []byte
- func (x *SummaryMetadata_PluginData) GetPluginName() string
- func (*SummaryMetadata_PluginData) ProtoMessage()
- func (x *SummaryMetadata_PluginData) ProtoReflect() protoreflect.Message
- func (x *SummaryMetadata_PluginData) Reset()
- func (x *SummaryMetadata_PluginData) String() string
- type Summary_Audio
- func (*Summary_Audio) Descriptor() ([]byte, []int)deprecated
- func (x *Summary_Audio) GetContentType() string
- func (x *Summary_Audio) GetEncodedAudioString() []byte
- func (x *Summary_Audio) GetLengthFrames() int64
- func (x *Summary_Audio) GetNumChannels() int64
- func (x *Summary_Audio) GetSampleRate() float32
- func (*Summary_Audio) ProtoMessage()
- func (x *Summary_Audio) ProtoReflect() protoreflect.Message
- func (x *Summary_Audio) Reset()
- func (x *Summary_Audio) String() string
- type Summary_Image
- func (*Summary_Image) Descriptor() ([]byte, []int)deprecated
- func (x *Summary_Image) GetColorspace() int32
- func (x *Summary_Image) GetEncodedImageString() []byte
- func (x *Summary_Image) GetHeight() int32
- func (x *Summary_Image) GetWidth() int32
- func (*Summary_Image) ProtoMessage()
- func (x *Summary_Image) ProtoReflect() protoreflect.Message
- func (x *Summary_Image) Reset()
- func (x *Summary_Image) String() string
- type Summary_Value
- func (*Summary_Value) Descriptor() ([]byte, []int)deprecated
- func (x *Summary_Value) GetAudio() *Summary_Audio
- func (x *Summary_Value) GetHisto() *HistogramProto
- func (x *Summary_Value) GetImage() *Summary_Image
- func (x *Summary_Value) GetMetadata() *SummaryMetadata
- func (x *Summary_Value) GetNodeName() string
- func (x *Summary_Value) GetObsoleteOldStyleHistogram() []byte
- func (x *Summary_Value) GetSimpleValue() float32
- func (x *Summary_Value) GetTag() string
- func (x *Summary_Value) GetTensor() *TensorProto
- func (m *Summary_Value) GetValue() isSummary_Value_Value
- func (*Summary_Value) ProtoMessage()
- func (x *Summary_Value) ProtoReflect() protoreflect.Message
- func (x *Summary_Value) Reset()
- func (x *Summary_Value) String() string
- type Summary_Value_Audio
- type Summary_Value_Histo
- type Summary_Value_Image
- type Summary_Value_ObsoleteOldStyleHistogram
- type Summary_Value_SimpleValue
- type Summary_Value_Tensor
- type TensorDescription
- func (*TensorDescription) Descriptor() ([]byte, []int)deprecated
- func (x *TensorDescription) GetAllocationDescription() *AllocationDescription
- func (x *TensorDescription) GetDtype() DataType
- func (x *TensorDescription) GetShape() *TensorShapeProto
- func (*TensorDescription) ProtoMessage()
- func (x *TensorDescription) ProtoReflect() protoreflect.Message
- func (x *TensorDescription) Reset()
- func (x *TensorDescription) String() string
- type TensorProto
- func (*TensorProto) Descriptor() ([]byte, []int)deprecated
- func (x *TensorProto) GetBoolVal() []bool
- func (x *TensorProto) GetDcomplexVal() []float64
- func (x *TensorProto) GetDoubleVal() []float64
- func (x *TensorProto) GetDtype() DataType
- func (x *TensorProto) GetFloatVal() []float32
- func (x *TensorProto) GetHalfVal() []int32
- func (x *TensorProto) GetInt64Val() []int64
- func (x *TensorProto) GetIntVal() []int32
- func (x *TensorProto) GetResourceHandleVal() []*ResourceHandleProto
- func (x *TensorProto) GetScomplexVal() []float32
- func (x *TensorProto) GetStringVal() [][]byte
- func (x *TensorProto) GetTensorContent() []byte
- func (x *TensorProto) GetTensorShape() *TensorShapeProto
- func (x *TensorProto) GetUint32Val() []uint32
- func (x *TensorProto) GetUint64Val() []uint64
- func (x *TensorProto) GetVariantVal() []*VariantTensorDataProto
- func (x *TensorProto) GetVersionNumber() int32
- func (*TensorProto) ProtoMessage()
- func (x *TensorProto) ProtoReflect() protoreflect.Message
- func (x *TensorProto) Reset()
- func (x *TensorProto) String() string
- type TensorShapeProto
- func (*TensorShapeProto) Descriptor() ([]byte, []int)deprecated
- func (x *TensorShapeProto) GetDim() []*TensorShapeProto_Dim
- func (x *TensorShapeProto) GetUnknownRank() bool
- func (*TensorShapeProto) ProtoMessage()
- func (x *TensorShapeProto) ProtoReflect() protoreflect.Message
- func (x *TensorShapeProto) Reset()
- func (x *TensorShapeProto) String() string
- type TensorShapeProto_Dim
- func (*TensorShapeProto_Dim) Descriptor() ([]byte, []int)deprecated
- func (x *TensorShapeProto_Dim) GetName() string
- func (x *TensorShapeProto_Dim) GetSize() int64
- func (*TensorShapeProto_Dim) ProtoMessage()
- func (x *TensorShapeProto_Dim) ProtoReflect() protoreflect.Message
- func (x *TensorShapeProto_Dim) Reset()
- func (x *TensorShapeProto_Dim) String() string
- type TensorSliceProto
- func (*TensorSliceProto) Descriptor() ([]byte, []int)deprecated
- func (x *TensorSliceProto) GetExtent() []*TensorSliceProto_Extent
- func (*TensorSliceProto) ProtoMessage()
- func (x *TensorSliceProto) ProtoReflect() protoreflect.Message
- func (x *TensorSliceProto) Reset()
- func (x *TensorSliceProto) String() string
- type TensorSliceProto_Extent
- func (*TensorSliceProto_Extent) Descriptor() ([]byte, []int)deprecated
- func (m *TensorSliceProto_Extent) GetHasLength() isTensorSliceProto_Extent_HasLength
- func (x *TensorSliceProto_Extent) GetLength() int64
- func (x *TensorSliceProto_Extent) GetStart() int64
- func (*TensorSliceProto_Extent) ProtoMessage()
- func (x *TensorSliceProto_Extent) ProtoReflect() protoreflect.Message
- func (x *TensorSliceProto_Extent) Reset()
- func (x *TensorSliceProto_Extent) String() string
- type TensorSliceProto_Extent_Length
- type VariableDef
- func (*VariableDef) Descriptor() ([]byte, []int)deprecated
- func (x *VariableDef) GetInitialValueName() string
- func (x *VariableDef) GetInitializerName() string
- func (x *VariableDef) GetIsResource() bool
- func (x *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef
- func (x *VariableDef) GetSnapshotName() string
- func (x *VariableDef) GetTrainable() bool
- func (x *VariableDef) GetVariableName() string
- func (*VariableDef) ProtoMessage()
- func (x *VariableDef) ProtoReflect() protoreflect.Message
- func (x *VariableDef) Reset()
- func (x *VariableDef) String() string
- type VariantTensorDataProto
- func (*VariantTensorDataProto) Descriptor() ([]byte, []int)deprecated
- func (x *VariantTensorDataProto) GetMetadata() []byte
- func (x *VariantTensorDataProto) GetTensors() []*TensorProto
- func (x *VariantTensorDataProto) GetTypeName() string
- func (*VariantTensorDataProto) ProtoMessage()
- func (x *VariantTensorDataProto) ProtoReflect() protoreflect.Message
- func (x *VariantTensorDataProto) Reset()
- func (x *VariantTensorDataProto) String() string
- type VersionDef
- func (*VersionDef) Descriptor() ([]byte, []int)deprecated
- func (x *VersionDef) GetBadConsumers() []int32
- func (x *VersionDef) GetMinConsumer() int32
- func (x *VersionDef) GetProducer() int32
- func (*VersionDef) ProtoMessage()
- func (x *VersionDef) ProtoReflect() protoreflect.Message
- func (x *VersionDef) Reset()
- func (x *VersionDef) String() string
Constants ¶
This section is empty.
Variables ¶
var ( ApiDef_Visibility_name = map[int32]string{ 0: "DEFAULT_VISIBILITY", 1: "VISIBLE", 2: "SKIP", 3: "HIDDEN", } ApiDef_Visibility_value = map[string]int32{ "DEFAULT_VISIBILITY": 0, "VISIBLE": 1, "SKIP": 2, "HIDDEN": 3, } )
Enum value maps for ApiDef_Visibility.
var ( GraphTransferInfo_Destination_name = map[int32]string{ 0: "NOP", 1: "HEXAGON", } GraphTransferInfo_Destination_value = map[string]int32{ "NOP": 0, "HEXAGON": 1, } )
Enum value maps for GraphTransferInfo_Destination.
var ( DataType_name = map[int32]string{ 0: "DT_INVALID", 1: "DT_FLOAT", 2: "DT_DOUBLE", 3: "DT_INT32", 4: "DT_UINT8", 5: "DT_INT16", 6: "DT_INT8", 7: "DT_STRING", 8: "DT_COMPLEX64", 9: "DT_INT64", 10: "DT_BOOL", 11: "DT_QINT8", 12: "DT_QUINT8", 13: "DT_QINT32", 14: "DT_BFLOAT16", 15: "DT_QINT16", 16: "DT_QUINT16", 17: "DT_UINT16", 18: "DT_COMPLEX128", 19: "DT_HALF", 20: "DT_RESOURCE", 21: "DT_VARIANT", 22: "DT_UINT32", 23: "DT_UINT64", 101: "DT_FLOAT_REF", 102: "DT_DOUBLE_REF", 103: "DT_INT32_REF", 104: "DT_UINT8_REF", 105: "DT_INT16_REF", 106: "DT_INT8_REF", 107: "DT_STRING_REF", 108: "DT_COMPLEX64_REF", 109: "DT_INT64_REF", 110: "DT_BOOL_REF", 111: "DT_QINT8_REF", 112: "DT_QUINT8_REF", 113: "DT_QINT32_REF", 114: "DT_BFLOAT16_REF", 115: "DT_QINT16_REF", 116: "DT_QUINT16_REF", 117: "DT_UINT16_REF", 118: "DT_COMPLEX128_REF", 119: "DT_HALF_REF", 120: "DT_RESOURCE_REF", 121: "DT_VARIANT_REF", 122: "DT_UINT32_REF", 123: "DT_UINT64_REF", } DataType_value = map[string]int32{ "DT_INVALID": 0, "DT_FLOAT": 1, "DT_DOUBLE": 2, "DT_INT32": 3, "DT_UINT8": 4, "DT_INT16": 5, "DT_INT8": 6, "DT_STRING": 7, "DT_COMPLEX64": 8, "DT_INT64": 9, "DT_BOOL": 10, "DT_QINT8": 11, "DT_QUINT8": 12, "DT_QINT32": 13, "DT_BFLOAT16": 14, "DT_QINT16": 15, "DT_QUINT16": 16, "DT_UINT16": 17, "DT_COMPLEX128": 18, "DT_HALF": 19, "DT_RESOURCE": 20, "DT_VARIANT": 21, "DT_UINT32": 22, "DT_UINT64": 23, "DT_FLOAT_REF": 101, "DT_DOUBLE_REF": 102, "DT_INT32_REF": 103, "DT_UINT8_REF": 104, "DT_INT16_REF": 105, "DT_INT8_REF": 106, "DT_STRING_REF": 107, "DT_COMPLEX64_REF": 108, "DT_INT64_REF": 109, "DT_BOOL_REF": 110, "DT_QINT8_REF": 111, "DT_QUINT8_REF": 112, "DT_QINT32_REF": 113, "DT_BFLOAT16_REF": 114, "DT_QINT16_REF": 115, "DT_QUINT16_REF": 116, "DT_UINT16_REF": 117, "DT_COMPLEX128_REF": 118, "DT_HALF_REF": 119, "DT_RESOURCE_REF": 120, "DT_VARIANT_REF": 121, "DT_UINT32_REF": 122, "DT_UINT64_REF": 123, } )
Enum value maps for DataType.
var File_tensorflow_core_framework_allocation_description_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_api_def_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_attr_value_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_cost_graph_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_device_attributes_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_function_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_graph_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_graph_transfer_info_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_iterator_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_kernel_def_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_log_memory_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_node_def_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_op_def_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_reader_base_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_remote_fused_graph_execute_info_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_resource_handle_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_step_stats_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_summary_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_tensor_description_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_tensor_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_tensor_shape_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_tensor_slice_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_types_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_variable_proto protoreflect.FileDescriptor
var File_tensorflow_core_framework_versions_proto protoreflect.FileDescriptor
Functions ¶
This section is empty.
Types ¶
type AllocationDescription ¶
type AllocationDescription struct { // Total number of bytes requested RequestedBytes int64 `protobuf:"varint,1,opt,name=requested_bytes,json=requestedBytes,proto3" json:"requested_bytes,omitempty"` // Total number of bytes allocated if known AllocatedBytes int64 `protobuf:"varint,2,opt,name=allocated_bytes,json=allocatedBytes,proto3" json:"allocated_bytes,omitempty"` // Name of the allocator used AllocatorName string `protobuf:"bytes,3,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"` // Identifier of the allocated buffer if known AllocationId int64 `protobuf:"varint,4,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"` // Set if this tensor only has one remaining reference HasSingleReference bool `protobuf:"varint,5,opt,name=has_single_reference,json=hasSingleReference,proto3" json:"has_single_reference,omitempty"` // Address of the allocation. Ptr uint64 `protobuf:"varint,6,opt,name=ptr,proto3" json:"ptr,omitempty"` // contains filtered or unexported fields }
func (*AllocationDescription) Descriptor
deprecated
func (*AllocationDescription) Descriptor() ([]byte, []int)
Deprecated: Use AllocationDescription.ProtoReflect.Descriptor instead.
func (*AllocationDescription) GetAllocatedBytes ¶
func (x *AllocationDescription) GetAllocatedBytes() int64
func (*AllocationDescription) GetAllocationId ¶
func (x *AllocationDescription) GetAllocationId() int64
func (*AllocationDescription) GetAllocatorName ¶
func (x *AllocationDescription) GetAllocatorName() string
func (*AllocationDescription) GetHasSingleReference ¶
func (x *AllocationDescription) GetHasSingleReference() bool
func (*AllocationDescription) GetPtr ¶
func (x *AllocationDescription) GetPtr() uint64
func (*AllocationDescription) GetRequestedBytes ¶
func (x *AllocationDescription) GetRequestedBytes() int64
func (*AllocationDescription) ProtoMessage ¶
func (*AllocationDescription) ProtoMessage()
func (*AllocationDescription) ProtoReflect ¶ added in v0.12.0
func (x *AllocationDescription) ProtoReflect() protoreflect.Message
func (*AllocationDescription) Reset ¶
func (x *AllocationDescription) Reset()
func (*AllocationDescription) String ¶
func (x *AllocationDescription) String() string
type AllocationRecord ¶
type AllocationRecord struct { // The timestamp of the operation. AllocMicros int64 `protobuf:"varint,1,opt,name=alloc_micros,json=allocMicros,proto3" json:"alloc_micros,omitempty"` // Number of bytes allocated, or de-allocated if negative. AllocBytes int64 `protobuf:"varint,2,opt,name=alloc_bytes,json=allocBytes,proto3" json:"alloc_bytes,omitempty"` // contains filtered or unexported fields }
An allocation/de-allocation operation performed by the allocator.
func (*AllocationRecord) Descriptor
deprecated
func (*AllocationRecord) Descriptor() ([]byte, []int)
Deprecated: Use AllocationRecord.ProtoReflect.Descriptor instead.
func (*AllocationRecord) GetAllocBytes ¶
func (x *AllocationRecord) GetAllocBytes() int64
func (*AllocationRecord) GetAllocMicros ¶
func (x *AllocationRecord) GetAllocMicros() int64
func (*AllocationRecord) ProtoMessage ¶
func (*AllocationRecord) ProtoMessage()
func (*AllocationRecord) ProtoReflect ¶ added in v0.12.0
func (x *AllocationRecord) ProtoReflect() protoreflect.Message
func (*AllocationRecord) Reset ¶
func (x *AllocationRecord) Reset()
func (*AllocationRecord) String ¶
func (x *AllocationRecord) String() string
type AllocatorMemoryUsed ¶
type AllocatorMemoryUsed struct { AllocatorName string `protobuf:"bytes,1,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"` // These are per-node allocator memory stats. TotalBytes int64 `protobuf:"varint,2,opt,name=total_bytes,json=totalBytes,proto3" json:"total_bytes,omitempty"` PeakBytes int64 `protobuf:"varint,3,opt,name=peak_bytes,json=peakBytes,proto3" json:"peak_bytes,omitempty"` // The bytes that are not deallocated. LiveBytes int64 `protobuf:"varint,4,opt,name=live_bytes,json=liveBytes,proto3" json:"live_bytes,omitempty"` // The allocation and deallocation timeline. AllocationRecords []*AllocationRecord `protobuf:"bytes,6,rep,name=allocation_records,json=allocationRecords,proto3" json:"allocation_records,omitempty"` // These are snapshots of the overall allocator memory stats. // The number of live bytes currently allocated by the allocator. AllocatorBytesInUse int64 `protobuf:"varint,5,opt,name=allocator_bytes_in_use,json=allocatorBytesInUse,proto3" json:"allocator_bytes_in_use,omitempty"` // contains filtered or unexported fields }
func (*AllocatorMemoryUsed) Descriptor
deprecated
func (*AllocatorMemoryUsed) Descriptor() ([]byte, []int)
Deprecated: Use AllocatorMemoryUsed.ProtoReflect.Descriptor instead.
func (*AllocatorMemoryUsed) GetAllocationRecords ¶
func (x *AllocatorMemoryUsed) GetAllocationRecords() []*AllocationRecord
func (*AllocatorMemoryUsed) GetAllocatorBytesInUse ¶
func (x *AllocatorMemoryUsed) GetAllocatorBytesInUse() int64
func (*AllocatorMemoryUsed) GetAllocatorName ¶
func (x *AllocatorMemoryUsed) GetAllocatorName() string
func (*AllocatorMemoryUsed) GetLiveBytes ¶
func (x *AllocatorMemoryUsed) GetLiveBytes() int64
func (*AllocatorMemoryUsed) GetPeakBytes ¶
func (x *AllocatorMemoryUsed) GetPeakBytes() int64
func (*AllocatorMemoryUsed) GetTotalBytes ¶
func (x *AllocatorMemoryUsed) GetTotalBytes() int64
func (*AllocatorMemoryUsed) ProtoMessage ¶
func (*AllocatorMemoryUsed) ProtoMessage()
func (*AllocatorMemoryUsed) ProtoReflect ¶ added in v0.12.0
func (x *AllocatorMemoryUsed) ProtoReflect() protoreflect.Message
func (*AllocatorMemoryUsed) Reset ¶
func (x *AllocatorMemoryUsed) Reset()
func (*AllocatorMemoryUsed) String ¶
func (x *AllocatorMemoryUsed) String() string
type ApiDef ¶
type ApiDef struct { // Name of the op (in the OpDef) to specify the API for. GraphOpName string `protobuf:"bytes,1,opt,name=graph_op_name,json=graphOpName,proto3" json:"graph_op_name,omitempty"` // If this op is deprecated, set deprecation message to the message // that should be logged when this op is used. // The message should indicate alternative op to use, if any. DeprecationMessage string `protobuf:"bytes,12,opt,name=deprecation_message,json=deprecationMessage,proto3" json:"deprecation_message,omitempty"` // Major version when the op will be deleted. For e.g. set this // value to 2 if op API should be removed in TensorFlow 2.0 and // deprecated in versions before that. DeprecationVersion int32 `protobuf:"varint,13,opt,name=deprecation_version,json=deprecationVersion,proto3" json:"deprecation_version,omitempty"` Visibility ApiDef_Visibility `protobuf:"varint,2,opt,name=visibility,proto3,enum=tensorflow.ApiDef_Visibility" json:"visibility,omitempty"` Endpoint []*ApiDef_Endpoint `protobuf:"bytes,3,rep,name=endpoint,proto3" json:"endpoint,omitempty"` InArg []*ApiDef_Arg `protobuf:"bytes,4,rep,name=in_arg,json=inArg,proto3" json:"in_arg,omitempty"` OutArg []*ApiDef_Arg `protobuf:"bytes,5,rep,name=out_arg,json=outArg,proto3" json:"out_arg,omitempty"` // List of original in_arg names to specify new argument order. // Length of arg_order should be either empty to keep current order // or match size of in_arg. ArgOrder []string `protobuf:"bytes,11,rep,name=arg_order,json=argOrder,proto3" json:"arg_order,omitempty"` Attr []*ApiDef_Attr `protobuf:"bytes,6,rep,name=attr,proto3" json:"attr,omitempty"` // One-line human-readable description of what the Op does. Summary string `protobuf:"bytes,7,opt,name=summary,proto3" json:"summary,omitempty"` // Additional, longer human-readable description of what the Op does. Description string `protobuf:"bytes,8,opt,name=description,proto3" json:"description,omitempty"` // Modify an existing/inherited description by adding text to the beginning // or end. DescriptionPrefix string `protobuf:"bytes,9,opt,name=description_prefix,json=descriptionPrefix,proto3" json:"description_prefix,omitempty"` DescriptionSuffix string `protobuf:"bytes,10,opt,name=description_suffix,json=descriptionSuffix,proto3" json:"description_suffix,omitempty"` // contains filtered or unexported fields }
Used to specify and override the default API & behavior in the generated code for client languages, from what you would get from the OpDef alone. There will be a set of ApiDefs that are common to all client languages, and another set per client language. The per-client-language ApiDefs will inherit values from the common ApiDefs which it can either replace or modify.
We separate the API definition from the OpDef so we can evolve the API while remaining backwards compatible when interpretting old graphs. Overrides go in an "api_def.pbtxt" file with a text-format ApiDefs message.
WARNING: Be *very* careful changing the API for any existing op -- you can change the semantics of existing code. These changes may need to wait until a major release of TensorFlow to avoid breaking our compatibility promises.
func (*ApiDef) Descriptor
deprecated
func (*ApiDef) GetArgOrder ¶
func (*ApiDef) GetAttr ¶
func (x *ApiDef) GetAttr() []*ApiDef_Attr
func (*ApiDef) GetDeprecationMessage ¶
func (*ApiDef) GetDeprecationVersion ¶
func (*ApiDef) GetDescription ¶
func (*ApiDef) GetDescriptionPrefix ¶
func (*ApiDef) GetDescriptionSuffix ¶
func (*ApiDef) GetEndpoint ¶
func (x *ApiDef) GetEndpoint() []*ApiDef_Endpoint
func (*ApiDef) GetGraphOpName ¶
func (*ApiDef) GetInArg ¶
func (x *ApiDef) GetInArg() []*ApiDef_Arg
func (*ApiDef) GetOutArg ¶
func (x *ApiDef) GetOutArg() []*ApiDef_Arg
func (*ApiDef) GetSummary ¶
func (*ApiDef) GetVisibility ¶
func (x *ApiDef) GetVisibility() ApiDef_Visibility
func (*ApiDef) ProtoMessage ¶
func (*ApiDef) ProtoMessage()
func (*ApiDef) ProtoReflect ¶ added in v0.12.0
func (x *ApiDef) ProtoReflect() protoreflect.Message
type ApiDef_Arg ¶
type ApiDef_Arg struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // Change the name used to access this arg in the API from what // is used in the GraphDef. Note that these names in `backticks` // will also be replaced in the summary & description fields. RenameTo string `protobuf:"bytes,2,opt,name=rename_to,json=renameTo,proto3" json:"rename_to,omitempty"` // Note: this will replace any inherited arg doc. There is no // current way of modifying arg descriptions (other than replacing // them entirely) as can be done with op descriptions. Description string `protobuf:"bytes,3,opt,name=description,proto3" json:"description,omitempty"` // contains filtered or unexported fields }
func (*ApiDef_Arg) Descriptor
deprecated
func (*ApiDef_Arg) Descriptor() ([]byte, []int)
Deprecated: Use ApiDef_Arg.ProtoReflect.Descriptor instead.
func (*ApiDef_Arg) GetDescription ¶
func (x *ApiDef_Arg) GetDescription() string
func (*ApiDef_Arg) GetName ¶
func (x *ApiDef_Arg) GetName() string
func (*ApiDef_Arg) GetRenameTo ¶
func (x *ApiDef_Arg) GetRenameTo() string
func (*ApiDef_Arg) ProtoMessage ¶
func (*ApiDef_Arg) ProtoMessage()
func (*ApiDef_Arg) ProtoReflect ¶ added in v0.12.0
func (x *ApiDef_Arg) ProtoReflect() protoreflect.Message
func (*ApiDef_Arg) Reset ¶
func (x *ApiDef_Arg) Reset()
func (*ApiDef_Arg) String ¶
func (x *ApiDef_Arg) String() string
type ApiDef_Attr ¶
type ApiDef_Attr struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // Change the name used to access this attr in the API from what // is used in the GraphDef. Note that these names in `backticks` // will also be replaced in the summary & description fields. RenameTo string `protobuf:"bytes,2,opt,name=rename_to,json=renameTo,proto3" json:"rename_to,omitempty"` // Specify a new default value to use for this attr. This default // will be used when creating new graphs, as opposed to the // default in the OpDef, which will be used when interpreting old // GraphDefs. DefaultValue *AttrValue `protobuf:"bytes,3,opt,name=default_value,json=defaultValue,proto3" json:"default_value,omitempty"` // Note: this will replace any inherited attr doc, there is no current // way of modifying attr descriptions as can be done with op descriptions. Description string `protobuf:"bytes,4,opt,name=description,proto3" json:"description,omitempty"` // contains filtered or unexported fields }
Description of the graph-construction-time configuration of this Op. That is to say, this describes the attr fields that will be specified in the NodeDef.
func (*ApiDef_Attr) Descriptor
deprecated
func (*ApiDef_Attr) Descriptor() ([]byte, []int)
Deprecated: Use ApiDef_Attr.ProtoReflect.Descriptor instead.
func (*ApiDef_Attr) GetDefaultValue ¶
func (x *ApiDef_Attr) GetDefaultValue() *AttrValue
func (*ApiDef_Attr) GetDescription ¶
func (x *ApiDef_Attr) GetDescription() string
func (*ApiDef_Attr) GetName ¶
func (x *ApiDef_Attr) GetName() string
func (*ApiDef_Attr) GetRenameTo ¶
func (x *ApiDef_Attr) GetRenameTo() string
func (*ApiDef_Attr) ProtoMessage ¶
func (*ApiDef_Attr) ProtoMessage()
func (*ApiDef_Attr) ProtoReflect ¶ added in v0.12.0
func (x *ApiDef_Attr) ProtoReflect() protoreflect.Message
func (*ApiDef_Attr) Reset ¶
func (x *ApiDef_Attr) Reset()
func (*ApiDef_Attr) String ¶
func (x *ApiDef_Attr) String() string
type ApiDef_Endpoint ¶
type ApiDef_Endpoint struct { // Name should be either like "CamelCaseName" or // "Package.CamelCaseName". Client-language-specific ApiDefs may // use a snake_case convention instead of CamelCase. Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // Set if this endpoint is deprecated. If set to true, a message suggesting // to use a non-deprecated endpoint instead will be printed. If all // endpoints are deprecated, set deprecation_message in ApiDef instead. Deprecated bool `protobuf:"varint,3,opt,name=deprecated,proto3" json:"deprecated,omitempty"` // Major version when an endpoint will be deleted. For e.g. set this // value to 2 if endpoint should be removed in TensorFlow 2.0 and // deprecated in versions before that. DeprecationVersion int32 `protobuf:"varint,4,opt,name=deprecation_version,json=deprecationVersion,proto3" json:"deprecation_version,omitempty"` // contains filtered or unexported fields }
If you specify any endpoint, this will replace all of the inherited endpoints. The first endpoint should be the "canonical" endpoint, and should not be deprecated (unless all endpoints are deprecated).
func (*ApiDef_Endpoint) Descriptor
deprecated
func (*ApiDef_Endpoint) Descriptor() ([]byte, []int)
Deprecated: Use ApiDef_Endpoint.ProtoReflect.Descriptor instead.
func (*ApiDef_Endpoint) GetDeprecated ¶
func (x *ApiDef_Endpoint) GetDeprecated() bool
func (*ApiDef_Endpoint) GetDeprecationVersion ¶
func (x *ApiDef_Endpoint) GetDeprecationVersion() int32
func (*ApiDef_Endpoint) GetName ¶
func (x *ApiDef_Endpoint) GetName() string
func (*ApiDef_Endpoint) ProtoMessage ¶
func (*ApiDef_Endpoint) ProtoMessage()
func (*ApiDef_Endpoint) ProtoReflect ¶ added in v0.12.0
func (x *ApiDef_Endpoint) ProtoReflect() protoreflect.Message
func (*ApiDef_Endpoint) Reset ¶
func (x *ApiDef_Endpoint) Reset()
func (*ApiDef_Endpoint) String ¶
func (x *ApiDef_Endpoint) String() string
type ApiDef_Visibility ¶
type ApiDef_Visibility int32
const ( // Normally this is "VISIBLE" unless you are inheriting a // different value from another ApiDef. ApiDef_DEFAULT_VISIBILITY ApiDef_Visibility = 0 // Publicly visible in the API. ApiDef_VISIBLE ApiDef_Visibility = 1 // Do not include this op in the generated API. If visibility is // set to 'SKIP', other fields are ignored for this op. ApiDef_SKIP ApiDef_Visibility = 2 // Hide this op by putting it into an internal namespace (or whatever // is appropriate in the target language). ApiDef_HIDDEN ApiDef_Visibility = 3 )
func (ApiDef_Visibility) Descriptor ¶ added in v0.12.0
func (ApiDef_Visibility) Descriptor() protoreflect.EnumDescriptor
func (ApiDef_Visibility) Enum ¶ added in v0.12.0
func (x ApiDef_Visibility) Enum() *ApiDef_Visibility
func (ApiDef_Visibility) EnumDescriptor
deprecated
func (ApiDef_Visibility) EnumDescriptor() ([]byte, []int)
Deprecated: Use ApiDef_Visibility.Descriptor instead.
func (ApiDef_Visibility) Number ¶ added in v0.12.0
func (x ApiDef_Visibility) Number() protoreflect.EnumNumber
func (ApiDef_Visibility) String ¶
func (x ApiDef_Visibility) String() string
func (ApiDef_Visibility) Type ¶ added in v0.12.0
func (ApiDef_Visibility) Type() protoreflect.EnumType
type ApiDefs ¶
type ApiDefs struct { Op []*ApiDef `protobuf:"bytes,1,rep,name=op,proto3" json:"op,omitempty"` // contains filtered or unexported fields }
func (*ApiDefs) Descriptor
deprecated
func (*ApiDefs) ProtoMessage ¶
func (*ApiDefs) ProtoMessage()
func (*ApiDefs) ProtoReflect ¶ added in v0.12.0
func (x *ApiDefs) ProtoReflect() protoreflect.Message
type AttrValue ¶
type AttrValue struct { // Types that are assignable to Value: // // *AttrValue_S // *AttrValue_I // *AttrValue_F // *AttrValue_B // *AttrValue_Type // *AttrValue_Shape // *AttrValue_Tensor // *AttrValue_List // *AttrValue_Func // *AttrValue_Placeholder Value isAttrValue_Value `protobuf_oneof:"value"` // contains filtered or unexported fields }
Protocol buffer representing the value for an attr used to configure an Op. Comment indicates the corresponding attr type. Only the field matching the attr type may be filled.
func (*AttrValue) Descriptor
deprecated
func (*AttrValue) GetFunc ¶
func (x *AttrValue) GetFunc() *NameAttrList
func (*AttrValue) GetList ¶
func (x *AttrValue) GetList() *AttrValue_ListValue
func (*AttrValue) GetPlaceholder ¶
func (*AttrValue) GetShape ¶
func (x *AttrValue) GetShape() *TensorShapeProto
func (*AttrValue) GetTensor ¶
func (x *AttrValue) GetTensor() *TensorProto
func (*AttrValue) ProtoMessage ¶
func (*AttrValue) ProtoMessage()
func (*AttrValue) ProtoReflect ¶ added in v0.12.0
func (x *AttrValue) ProtoReflect() protoreflect.Message
type AttrValue_B ¶
type AttrValue_B struct {
B bool `protobuf:"varint,5,opt,name=b,proto3,oneof"` // "bool"
}
type AttrValue_F ¶
type AttrValue_F struct {
F float32 `protobuf:"fixed32,4,opt,name=f,proto3,oneof"` // "float"
}
type AttrValue_Func ¶
type AttrValue_Func struct { // "func" represents a function. func.name is a function's name or // a primitive op's name. func.attr.first is the name of an attr // defined for that function. func.attr.second is the value for // that attr in the instantiation. Func *NameAttrList `protobuf:"bytes,10,opt,name=func,proto3,oneof"` }
type AttrValue_I ¶
type AttrValue_I struct {
I int64 `protobuf:"varint,3,opt,name=i,proto3,oneof"` // "int"
}
type AttrValue_List ¶
type AttrValue_List struct {
List *AttrValue_ListValue `protobuf:"bytes,1,opt,name=list,proto3,oneof"` // any "list(...)"
}
type AttrValue_ListValue ¶
type AttrValue_ListValue struct { S [][]byte `protobuf:"bytes,2,rep,name=s,proto3" json:"s,omitempty"` // "list(string)" I []int64 `protobuf:"varint,3,rep,packed,name=i,proto3" json:"i,omitempty"` // "list(int)" F []float32 `protobuf:"fixed32,4,rep,packed,name=f,proto3" json:"f,omitempty"` // "list(float)" B []bool `protobuf:"varint,5,rep,packed,name=b,proto3" json:"b,omitempty"` // "list(bool)" Type []DataType `protobuf:"varint,6,rep,packed,name=type,proto3,enum=tensorflow.DataType" json:"type,omitempty"` // "list(type)" Shape []*TensorShapeProto `protobuf:"bytes,7,rep,name=shape,proto3" json:"shape,omitempty"` // "list(shape)" Tensor []*TensorProto `protobuf:"bytes,8,rep,name=tensor,proto3" json:"tensor,omitempty"` // "list(tensor)" Func []*NameAttrList `protobuf:"bytes,9,rep,name=func,proto3" json:"func,omitempty"` // "list(attr)" // contains filtered or unexported fields }
LINT.IfChange
func (*AttrValue_ListValue) Descriptor
deprecated
func (*AttrValue_ListValue) Descriptor() ([]byte, []int)
Deprecated: Use AttrValue_ListValue.ProtoReflect.Descriptor instead.
func (*AttrValue_ListValue) GetB ¶
func (x *AttrValue_ListValue) GetB() []bool
func (*AttrValue_ListValue) GetF ¶
func (x *AttrValue_ListValue) GetF() []float32
func (*AttrValue_ListValue) GetFunc ¶
func (x *AttrValue_ListValue) GetFunc() []*NameAttrList
func (*AttrValue_ListValue) GetI ¶
func (x *AttrValue_ListValue) GetI() []int64
func (*AttrValue_ListValue) GetS ¶
func (x *AttrValue_ListValue) GetS() [][]byte
func (*AttrValue_ListValue) GetShape ¶
func (x *AttrValue_ListValue) GetShape() []*TensorShapeProto
func (*AttrValue_ListValue) GetTensor ¶
func (x *AttrValue_ListValue) GetTensor() []*TensorProto
func (*AttrValue_ListValue) GetType ¶
func (x *AttrValue_ListValue) GetType() []DataType
func (*AttrValue_ListValue) ProtoMessage ¶
func (*AttrValue_ListValue) ProtoMessage()
func (*AttrValue_ListValue) ProtoReflect ¶ added in v0.12.0
func (x *AttrValue_ListValue) ProtoReflect() protoreflect.Message
func (*AttrValue_ListValue) Reset ¶
func (x *AttrValue_ListValue) Reset()
func (*AttrValue_ListValue) String ¶
func (x *AttrValue_ListValue) String() string
type AttrValue_Placeholder ¶
type AttrValue_Placeholder struct { // This is a placeholder only used in nodes defined inside a // function. It indicates the attr value will be supplied when // the function is instantiated. For example, let us suppose a // node "N" in function "FN". "N" has an attr "A" with value // placeholder = "foo". When FN is instantiated with attr "foo" // set to "bar", the instantiated node N's attr A will have been // given the value "bar". Placeholder string `protobuf:"bytes,9,opt,name=placeholder,proto3,oneof"` }
type AttrValue_S ¶
type AttrValue_S struct {
S []byte `protobuf:"bytes,2,opt,name=s,proto3,oneof"` // "string"
}
type AttrValue_Shape ¶
type AttrValue_Shape struct {
Shape *TensorShapeProto `protobuf:"bytes,7,opt,name=shape,proto3,oneof"` // "shape"
}
type AttrValue_Tensor ¶
type AttrValue_Tensor struct {
Tensor *TensorProto `protobuf:"bytes,8,opt,name=tensor,proto3,oneof"` // "tensor"
}
type AttrValue_Type ¶
type AttrValue_Type struct {
Type DataType `protobuf:"varint,6,opt,name=type,proto3,enum=tensorflow.DataType,oneof"` // "type"
}
type CostGraphDef ¶
type CostGraphDef struct { Node []*CostGraphDef_Node `protobuf:"bytes,1,rep,name=node,proto3" json:"node,omitempty"` // contains filtered or unexported fields }
func (*CostGraphDef) Descriptor
deprecated
func (*CostGraphDef) Descriptor() ([]byte, []int)
Deprecated: Use CostGraphDef.ProtoReflect.Descriptor instead.
func (*CostGraphDef) GetNode ¶
func (x *CostGraphDef) GetNode() []*CostGraphDef_Node
func (*CostGraphDef) ProtoMessage ¶
func (*CostGraphDef) ProtoMessage()
func (*CostGraphDef) ProtoReflect ¶ added in v0.12.0
func (x *CostGraphDef) ProtoReflect() protoreflect.Message
func (*CostGraphDef) Reset ¶
func (x *CostGraphDef) Reset()
func (*CostGraphDef) String ¶
func (x *CostGraphDef) String() string
type CostGraphDef_Node ¶
type CostGraphDef_Node struct { // The name of the node. Names are globally unique. Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // The device of the node. Can be empty if the node is mapped to the // default partition or partitioning hasn't been run yet. Device string `protobuf:"bytes,2,opt,name=device,proto3" json:"device,omitempty"` // The id of the node. Node ids are only unique inside a partition. Id int32 `protobuf:"varint,3,opt,name=id,proto3" json:"id,omitempty"` InputInfo []*CostGraphDef_Node_InputInfo `protobuf:"bytes,4,rep,name=input_info,json=inputInfo,proto3" json:"input_info,omitempty"` OutputInfo []*CostGraphDef_Node_OutputInfo `protobuf:"bytes,5,rep,name=output_info,json=outputInfo,proto3" json:"output_info,omitempty"` // Temporary memory used by this node. TemporaryMemorySize int64 `protobuf:"varint,6,opt,name=temporary_memory_size,json=temporaryMemorySize,proto3" json:"temporary_memory_size,omitempty"` // Persistent memory used by this node. PersistentMemorySize int64 `protobuf:"varint,12,opt,name=persistent_memory_size,json=persistentMemorySize,proto3" json:"persistent_memory_size,omitempty"` // Deprecated: Do not use. HostTempMemorySize int64 `protobuf:"varint,10,opt,name=host_temp_memory_size,json=hostTempMemorySize,proto3" json:"host_temp_memory_size,omitempty"` // Deprecated: Do not use. DeviceTempMemorySize int64 `` /* 127-byte string literal not displayed */ // Deprecated: Do not use. DevicePersistentMemorySize int64 `` /* 145-byte string literal not displayed */ // Estimate of the computational cost of this node, in microseconds. ComputeCost int64 `protobuf:"varint,9,opt,name=compute_cost,json=computeCost,proto3" json:"compute_cost,omitempty"` // Analytical estimate of the computational cost of this node, in // microseconds. ComputeTime int64 `protobuf:"varint,14,opt,name=compute_time,json=computeTime,proto3" json:"compute_time,omitempty"` // Analytical estimate of the memory access cost of this node, in // microseconds. MemoryTime int64 `protobuf:"varint,15,opt,name=memory_time,json=memoryTime,proto3" json:"memory_time,omitempty"` // If true, the output is permanent: it can't be discarded, because this // node is part of the "final output". Nodes may depend on final nodes. IsFinal bool `protobuf:"varint,7,opt,name=is_final,json=isFinal,proto3" json:"is_final,omitempty"` // Ids of the control inputs for this node. ControlInput []int32 `protobuf:"varint,8,rep,packed,name=control_input,json=controlInput,proto3" json:"control_input,omitempty"` // Are the costs inaccurate? Inaccurate bool `protobuf:"varint,17,opt,name=inaccurate,proto3" json:"inaccurate,omitempty"` // contains filtered or unexported fields }
func (*CostGraphDef_Node) Descriptor
deprecated
func (*CostGraphDef_Node) Descriptor() ([]byte, []int)
Deprecated: Use CostGraphDef_Node.ProtoReflect.Descriptor instead.
func (*CostGraphDef_Node) GetComputeCost ¶
func (x *CostGraphDef_Node) GetComputeCost() int64
func (*CostGraphDef_Node) GetComputeTime ¶
func (x *CostGraphDef_Node) GetComputeTime() int64
func (*CostGraphDef_Node) GetControlInput ¶
func (x *CostGraphDef_Node) GetControlInput() []int32
func (*CostGraphDef_Node) GetDevice ¶
func (x *CostGraphDef_Node) GetDevice() string
func (*CostGraphDef_Node) GetDevicePersistentMemorySize
deprecated
func (x *CostGraphDef_Node) GetDevicePersistentMemorySize() int64
Deprecated: Do not use.
func (*CostGraphDef_Node) GetDeviceTempMemorySize
deprecated
func (x *CostGraphDef_Node) GetDeviceTempMemorySize() int64
Deprecated: Do not use.
func (*CostGraphDef_Node) GetHostTempMemorySize
deprecated
func (x *CostGraphDef_Node) GetHostTempMemorySize() int64
Deprecated: Do not use.
func (*CostGraphDef_Node) GetId ¶
func (x *CostGraphDef_Node) GetId() int32
func (*CostGraphDef_Node) GetInaccurate ¶
func (x *CostGraphDef_Node) GetInaccurate() bool
func (*CostGraphDef_Node) GetInputInfo ¶
func (x *CostGraphDef_Node) GetInputInfo() []*CostGraphDef_Node_InputInfo
func (*CostGraphDef_Node) GetIsFinal ¶
func (x *CostGraphDef_Node) GetIsFinal() bool
func (*CostGraphDef_Node) GetMemoryTime ¶
func (x *CostGraphDef_Node) GetMemoryTime() int64
func (*CostGraphDef_Node) GetName ¶
func (x *CostGraphDef_Node) GetName() string
func (*CostGraphDef_Node) GetOutputInfo ¶
func (x *CostGraphDef_Node) GetOutputInfo() []*CostGraphDef_Node_OutputInfo
func (*CostGraphDef_Node) GetPersistentMemorySize ¶
func (x *CostGraphDef_Node) GetPersistentMemorySize() int64
func (*CostGraphDef_Node) GetTemporaryMemorySize ¶
func (x *CostGraphDef_Node) GetTemporaryMemorySize() int64
func (*CostGraphDef_Node) ProtoMessage ¶
func (*CostGraphDef_Node) ProtoMessage()
func (*CostGraphDef_Node) ProtoReflect ¶ added in v0.12.0
func (x *CostGraphDef_Node) ProtoReflect() protoreflect.Message
func (*CostGraphDef_Node) Reset ¶
func (x *CostGraphDef_Node) Reset()
func (*CostGraphDef_Node) String ¶
func (x *CostGraphDef_Node) String() string
type CostGraphDef_Node_InputInfo ¶
type CostGraphDef_Node_InputInfo struct { PrecedingNode int32 `protobuf:"varint,1,opt,name=preceding_node,json=precedingNode,proto3" json:"preceding_node,omitempty"` PrecedingPort int32 `protobuf:"varint,2,opt,name=preceding_port,json=precedingPort,proto3" json:"preceding_port,omitempty"` // contains filtered or unexported fields }
Inputs of this node. They must be executed before this node can be executed. An input is a particular output of another node, specified by the node id and the output index.
func (*CostGraphDef_Node_InputInfo) Descriptor
deprecated
func (*CostGraphDef_Node_InputInfo) Descriptor() ([]byte, []int)
Deprecated: Use CostGraphDef_Node_InputInfo.ProtoReflect.Descriptor instead.
func (*CostGraphDef_Node_InputInfo) GetPrecedingNode ¶
func (x *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32
func (*CostGraphDef_Node_InputInfo) GetPrecedingPort ¶
func (x *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32
func (*CostGraphDef_Node_InputInfo) ProtoMessage ¶
func (*CostGraphDef_Node_InputInfo) ProtoMessage()
func (*CostGraphDef_Node_InputInfo) ProtoReflect ¶ added in v0.12.0
func (x *CostGraphDef_Node_InputInfo) ProtoReflect() protoreflect.Message
func (*CostGraphDef_Node_InputInfo) Reset ¶
func (x *CostGraphDef_Node_InputInfo) Reset()
func (*CostGraphDef_Node_InputInfo) String ¶
func (x *CostGraphDef_Node_InputInfo) String() string
type CostGraphDef_Node_OutputInfo ¶
type CostGraphDef_Node_OutputInfo struct { Size int64 `protobuf:"varint,1,opt,name=size,proto3" json:"size,omitempty"` // If >= 0, the output is an alias of an input. Note that an alias input // may itself be an alias. The algorithm will therefore need to follow // those pointers. AliasInputPort int64 `protobuf:"varint,2,opt,name=alias_input_port,json=aliasInputPort,proto3" json:"alias_input_port,omitempty"` Shape *TensorShapeProto `protobuf:"bytes,3,opt,name=shape,proto3" json:"shape,omitempty"` Dtype DataType `protobuf:"varint,4,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` // contains filtered or unexported fields }
Outputs of this node.
func (*CostGraphDef_Node_OutputInfo) Descriptor
deprecated
func (*CostGraphDef_Node_OutputInfo) Descriptor() ([]byte, []int)
Deprecated: Use CostGraphDef_Node_OutputInfo.ProtoReflect.Descriptor instead.
func (*CostGraphDef_Node_OutputInfo) GetAliasInputPort ¶
func (x *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64
func (*CostGraphDef_Node_OutputInfo) GetDtype ¶
func (x *CostGraphDef_Node_OutputInfo) GetDtype() DataType
func (*CostGraphDef_Node_OutputInfo) GetShape ¶
func (x *CostGraphDef_Node_OutputInfo) GetShape() *TensorShapeProto
func (*CostGraphDef_Node_OutputInfo) GetSize ¶
func (x *CostGraphDef_Node_OutputInfo) GetSize() int64
func (*CostGraphDef_Node_OutputInfo) ProtoMessage ¶
func (*CostGraphDef_Node_OutputInfo) ProtoMessage()
func (*CostGraphDef_Node_OutputInfo) ProtoReflect ¶ added in v0.12.0
func (x *CostGraphDef_Node_OutputInfo) ProtoReflect() protoreflect.Message
func (*CostGraphDef_Node_OutputInfo) Reset ¶
func (x *CostGraphDef_Node_OutputInfo) Reset()
func (*CostGraphDef_Node_OutputInfo) String ¶
func (x *CostGraphDef_Node_OutputInfo) String() string
type DataType ¶
type DataType int32
LINT.IfChange
const ( // Not a legal value for DataType. Used to indicate a DataType field // has not been set. DataType_DT_INVALID DataType = 0 // Data types that all computation devices are expected to be // capable to support. DataType_DT_FLOAT DataType = 1 DataType_DT_DOUBLE DataType = 2 DataType_DT_INT32 DataType = 3 DataType_DT_UINT8 DataType = 4 DataType_DT_INT16 DataType = 5 DataType_DT_INT8 DataType = 6 DataType_DT_STRING DataType = 7 DataType_DT_COMPLEX64 DataType = 8 // Single-precision complex DataType_DT_INT64 DataType = 9 DataType_DT_BOOL DataType = 10 DataType_DT_QINT8 DataType = 11 // Quantized int8 DataType_DT_QUINT8 DataType = 12 // Quantized uint8 DataType_DT_QINT32 DataType = 13 // Quantized int32 DataType_DT_BFLOAT16 DataType = 14 // Float32 truncated to 16 bits. Only for cast ops. DataType_DT_QINT16 DataType = 15 // Quantized int16 DataType_DT_QUINT16 DataType = 16 // Quantized uint16 DataType_DT_UINT16 DataType = 17 DataType_DT_COMPLEX128 DataType = 18 // Double-precision complex DataType_DT_HALF DataType = 19 DataType_DT_RESOURCE DataType = 20 DataType_DT_VARIANT DataType = 21 // Arbitrary C++ data types DataType_DT_UINT32 DataType = 22 DataType_DT_UINT64 DataType = 23 // Do not use! These are only for parameters. Every enum above // should have a corresponding value below (verified by types_test). DataType_DT_FLOAT_REF DataType = 101 DataType_DT_DOUBLE_REF DataType = 102 DataType_DT_INT32_REF DataType = 103 DataType_DT_UINT8_REF DataType = 104 DataType_DT_INT16_REF DataType = 105 DataType_DT_INT8_REF DataType = 106 DataType_DT_STRING_REF DataType = 107 DataType_DT_COMPLEX64_REF DataType = 108 DataType_DT_INT64_REF DataType = 109 DataType_DT_BOOL_REF DataType = 110 DataType_DT_QINT8_REF DataType = 111 DataType_DT_QUINT8_REF DataType = 112 DataType_DT_QINT32_REF DataType = 113 DataType_DT_BFLOAT16_REF DataType = 114 DataType_DT_QINT16_REF DataType = 115 DataType_DT_QUINT16_REF DataType = 116 DataType_DT_UINT16_REF DataType = 117 DataType_DT_COMPLEX128_REF DataType = 118 DataType_DT_HALF_REF DataType = 119 DataType_DT_RESOURCE_REF DataType = 120 DataType_DT_VARIANT_REF DataType = 121 DataType_DT_UINT32_REF DataType = 122 DataType_DT_UINT64_REF DataType = 123 )
func (DataType) Descriptor ¶ added in v0.12.0
func (DataType) Descriptor() protoreflect.EnumDescriptor
func (DataType) EnumDescriptor
deprecated
func (DataType) Number ¶ added in v0.12.0
func (x DataType) Number() protoreflect.EnumNumber
func (DataType) Type ¶ added in v0.12.0
func (DataType) Type() protoreflect.EnumType
type DeviceAttributes ¶
type DeviceAttributes struct { // Fully specified name of the device within a cluster. Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // String representation of device_type. DeviceType string `protobuf:"bytes,2,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"` // Memory capacity of device in bytes. MemoryLimit int64 `protobuf:"varint,4,opt,name=memory_limit,json=memoryLimit,proto3" json:"memory_limit,omitempty"` // Platform-specific data about device that may be useful // for supporting efficient data transfers. Locality *DeviceLocality `protobuf:"bytes,5,opt,name=locality,proto3" json:"locality,omitempty"` // A device is assigned a global unique number each time it is // initialized. "incarnation" should never be 0. Incarnation uint64 `protobuf:"fixed64,6,opt,name=incarnation,proto3" json:"incarnation,omitempty"` // String representation of the physical device that this device maps to. PhysicalDeviceDesc string `protobuf:"bytes,7,opt,name=physical_device_desc,json=physicalDeviceDesc,proto3" json:"physical_device_desc,omitempty"` // contains filtered or unexported fields }
func (*DeviceAttributes) Descriptor
deprecated
func (*DeviceAttributes) Descriptor() ([]byte, []int)
Deprecated: Use DeviceAttributes.ProtoReflect.Descriptor instead.
func (*DeviceAttributes) GetDeviceType ¶
func (x *DeviceAttributes) GetDeviceType() string
func (*DeviceAttributes) GetIncarnation ¶
func (x *DeviceAttributes) GetIncarnation() uint64
func (*DeviceAttributes) GetLocality ¶
func (x *DeviceAttributes) GetLocality() *DeviceLocality
func (*DeviceAttributes) GetMemoryLimit ¶
func (x *DeviceAttributes) GetMemoryLimit() int64
func (*DeviceAttributes) GetName ¶
func (x *DeviceAttributes) GetName() string
func (*DeviceAttributes) GetPhysicalDeviceDesc ¶
func (x *DeviceAttributes) GetPhysicalDeviceDesc() string
func (*DeviceAttributes) ProtoMessage ¶
func (*DeviceAttributes) ProtoMessage()
func (*DeviceAttributes) ProtoReflect ¶ added in v0.12.0
func (x *DeviceAttributes) ProtoReflect() protoreflect.Message
func (*DeviceAttributes) Reset ¶
func (x *DeviceAttributes) Reset()
func (*DeviceAttributes) String ¶
func (x *DeviceAttributes) String() string
type DeviceLocality ¶
type DeviceLocality struct { // Optional bus locality of device. Default value of 0 means // no specific locality. Specific localities are indexed from 1. BusId int32 `protobuf:"varint,1,opt,name=bus_id,json=busId,proto3" json:"bus_id,omitempty"` // Optional NUMA locality of device. NumaNode int32 `protobuf:"varint,2,opt,name=numa_node,json=numaNode,proto3" json:"numa_node,omitempty"` // Optional local interconnect links to other devices. Links *LocalLinks `protobuf:"bytes,3,opt,name=links,proto3" json:"links,omitempty"` // contains filtered or unexported fields }
func (*DeviceLocality) Descriptor
deprecated
func (*DeviceLocality) Descriptor() ([]byte, []int)
Deprecated: Use DeviceLocality.ProtoReflect.Descriptor instead.
func (*DeviceLocality) GetBusId ¶
func (x *DeviceLocality) GetBusId() int32
func (*DeviceLocality) GetLinks ¶
func (x *DeviceLocality) GetLinks() *LocalLinks
func (*DeviceLocality) GetNumaNode ¶
func (x *DeviceLocality) GetNumaNode() int32
func (*DeviceLocality) ProtoMessage ¶
func (*DeviceLocality) ProtoMessage()
func (*DeviceLocality) ProtoReflect ¶ added in v0.12.0
func (x *DeviceLocality) ProtoReflect() protoreflect.Message
func (*DeviceLocality) Reset ¶
func (x *DeviceLocality) Reset()
func (*DeviceLocality) String ¶
func (x *DeviceLocality) String() string
type DeviceStepStats ¶
type DeviceStepStats struct { Device string `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"` NodeStats []*NodeExecStats `protobuf:"bytes,2,rep,name=node_stats,json=nodeStats,proto3" json:"node_stats,omitempty"` // contains filtered or unexported fields }
func (*DeviceStepStats) Descriptor
deprecated
func (*DeviceStepStats) Descriptor() ([]byte, []int)
Deprecated: Use DeviceStepStats.ProtoReflect.Descriptor instead.
func (*DeviceStepStats) GetDevice ¶
func (x *DeviceStepStats) GetDevice() string
func (*DeviceStepStats) GetNodeStats ¶
func (x *DeviceStepStats) GetNodeStats() []*NodeExecStats
func (*DeviceStepStats) ProtoMessage ¶
func (*DeviceStepStats) ProtoMessage()
func (*DeviceStepStats) ProtoReflect ¶ added in v0.12.0
func (x *DeviceStepStats) ProtoReflect() protoreflect.Message
func (*DeviceStepStats) Reset ¶
func (x *DeviceStepStats) Reset()
func (*DeviceStepStats) String ¶
func (x *DeviceStepStats) String() string
type FunctionDef ¶
type FunctionDef struct { // The definition of the function's name, arguments, return values, // attrs etc. Signature *OpDef `protobuf:"bytes,1,opt,name=signature,proto3" json:"signature,omitempty"` // Attributes specific to this function definition. Attr map[string]*AttrValue `` /* 149-byte string literal not displayed */ // By convention, "op" in node_def is resolved by consulting with a // user-defined library first. If not resolved, "func" is assumed to // be a builtin op. NodeDef []*NodeDef `protobuf:"bytes,3,rep,name=node_def,json=nodeDef,proto3" json:"node_def,omitempty"` // A mapping from the output arg names from `signature` to the // outputs from `node_def` that should be returned by the function. Ret map[string]string `` /* 147-byte string literal not displayed */ // contains filtered or unexported fields }
A function can be instantiated when the runtime can bind every attr with a value. When a GraphDef has a call to a function, it must have binding for every attr defined in the signature.
TODO(zhifengc):
- device spec, etc.
func (*FunctionDef) Descriptor
deprecated
func (*FunctionDef) Descriptor() ([]byte, []int)
Deprecated: Use FunctionDef.ProtoReflect.Descriptor instead.
func (*FunctionDef) GetAttr ¶
func (x *FunctionDef) GetAttr() map[string]*AttrValue
func (*FunctionDef) GetNodeDef ¶
func (x *FunctionDef) GetNodeDef() []*NodeDef
func (*FunctionDef) GetRet ¶
func (x *FunctionDef) GetRet() map[string]string
func (*FunctionDef) GetSignature ¶
func (x *FunctionDef) GetSignature() *OpDef
func (*FunctionDef) ProtoMessage ¶
func (*FunctionDef) ProtoMessage()
func (*FunctionDef) ProtoReflect ¶ added in v0.12.0
func (x *FunctionDef) ProtoReflect() protoreflect.Message
func (*FunctionDef) Reset ¶
func (x *FunctionDef) Reset()
func (*FunctionDef) String ¶
func (x *FunctionDef) String() string
type FunctionDefLibrary ¶
type FunctionDefLibrary struct { Function []*FunctionDef `protobuf:"bytes,1,rep,name=function,proto3" json:"function,omitempty"` Gradient []*GradientDef `protobuf:"bytes,2,rep,name=gradient,proto3" json:"gradient,omitempty"` // contains filtered or unexported fields }
A library is a set of named functions.
func (*FunctionDefLibrary) Descriptor
deprecated
func (*FunctionDefLibrary) Descriptor() ([]byte, []int)
Deprecated: Use FunctionDefLibrary.ProtoReflect.Descriptor instead.
func (*FunctionDefLibrary) GetFunction ¶
func (x *FunctionDefLibrary) GetFunction() []*FunctionDef
func (*FunctionDefLibrary) GetGradient ¶
func (x *FunctionDefLibrary) GetGradient() []*GradientDef
func (*FunctionDefLibrary) ProtoMessage ¶
func (*FunctionDefLibrary) ProtoMessage()
func (*FunctionDefLibrary) ProtoReflect ¶ added in v0.12.0
func (x *FunctionDefLibrary) ProtoReflect() protoreflect.Message
func (*FunctionDefLibrary) Reset ¶
func (x *FunctionDefLibrary) Reset()
func (*FunctionDefLibrary) String ¶
func (x *FunctionDefLibrary) String() string
type GradientDef ¶
type GradientDef struct { FunctionName string `protobuf:"bytes,1,opt,name=function_name,json=functionName,proto3" json:"function_name,omitempty"` // The function name. GradientFunc string `protobuf:"bytes,2,opt,name=gradient_func,json=gradientFunc,proto3" json:"gradient_func,omitempty"` // The gradient function's name. // contains filtered or unexported fields }
GradientDef defines the gradient function of a function defined in a function library.
A gradient function g (specified by gradient_func) for a function f (specified by function_name) must follow the following:
The function 'f' must be a numerical function which takes N inputs and produces M outputs. Its gradient function 'g', which is a function taking N + M inputs and produces N outputs.
I.e. if we have
(y1, y2, ..., y_M) = f(x1, x2, ..., x_N),
then, g is
(dL/dx1, dL/dx2, ..., dL/dx_N) = g(x1, x2, ..., x_N, dL/dy1, dL/dy2, ..., dL/dy_M),
where L is a scalar-value function of (x1, x2, ..., xN) (e.g., the loss function). dL/dx_i is the partial derivative of L with respect to x_i.
func (*GradientDef) Descriptor
deprecated
func (*GradientDef) Descriptor() ([]byte, []int)
Deprecated: Use GradientDef.ProtoReflect.Descriptor instead.
func (*GradientDef) GetFunctionName ¶
func (x *GradientDef) GetFunctionName() string
func (*GradientDef) GetGradientFunc ¶
func (x *GradientDef) GetGradientFunc() string
func (*GradientDef) ProtoMessage ¶
func (*GradientDef) ProtoMessage()
func (*GradientDef) ProtoReflect ¶ added in v0.12.0
func (x *GradientDef) ProtoReflect() protoreflect.Message
func (*GradientDef) Reset ¶
func (x *GradientDef) Reset()
func (*GradientDef) String ¶
func (x *GradientDef) String() string
type GraphDef ¶
type GraphDef struct { Node []*NodeDef `protobuf:"bytes,1,rep,name=node,proto3" json:"node,omitempty"` // Compatibility versions of the graph. See core/public/version.h for version // history. The GraphDef version is distinct from the TensorFlow version, and // each release of TensorFlow will support a range of GraphDef versions. Versions *VersionDef `protobuf:"bytes,4,opt,name=versions,proto3" json:"versions,omitempty"` // Deprecated single version field; use versions above instead. Since all // GraphDef changes before "versions" was introduced were forward // compatible, this field is entirely ignored. // // Deprecated: Do not use. Version int32 `protobuf:"varint,3,opt,name=version,proto3" json:"version,omitempty"` // EXPERIMENTAL. DO NOT USE OR DEPEND ON THIS YET. // // "library" provides user-defined functions. // // Naming: // - library.function.name are in a flat namespace. // NOTE: We may need to change it to be hierarchical to support // different orgs. E.g., // { "/google/nn", { ... }}, // { "/google/vision", { ... }} // { "/org_foo/module_bar", { ... }} // map<string, FunctionDefLib> named_lib; // - If node[i].op is the name of one function in "library", // node[i] is deemed as a function call. Otherwise, node[i].op // must be a primitive operation supported by the runtime. // // Function call semantics: // // - The callee may start execution as soon as some of its inputs // are ready. The caller may want to use Tuple() mechanism to // ensure all inputs are ready in the same time. // // - The consumer of return values may start executing as soon as // the return values the consumer depends on are ready. The // consumer may want to use Tuple() mechanism to ensure the // consumer does not start until all return values of the callee // function are ready. Library *FunctionDefLibrary `protobuf:"bytes,2,opt,name=library,proto3" json:"library,omitempty"` // contains filtered or unexported fields }
Represents the graph of operations
func (*GraphDef) Descriptor
deprecated
func (*GraphDef) GetLibrary ¶
func (x *GraphDef) GetLibrary() *FunctionDefLibrary
func (*GraphDef) GetVersion
deprecated
func (*GraphDef) GetVersions ¶
func (x *GraphDef) GetVersions() *VersionDef
func (*GraphDef) ProtoMessage ¶
func (*GraphDef) ProtoMessage()
func (*GraphDef) ProtoReflect ¶ added in v0.12.0
func (x *GraphDef) ProtoReflect() protoreflect.Message
type GraphTransferConstNodeInfo ¶
type GraphTransferConstNodeInfo struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` NodeId int32 `protobuf:"varint,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"` Shape []int64 `protobuf:"varint,3,rep,packed,name=shape,proto3" json:"shape,omitempty"` Data []byte `protobuf:"bytes,4,opt,name=data,proto3" json:"data,omitempty"` Dtype DataType `protobuf:"varint,5,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` // contains filtered or unexported fields }
func (*GraphTransferConstNodeInfo) Descriptor
deprecated
func (*GraphTransferConstNodeInfo) Descriptor() ([]byte, []int)
Deprecated: Use GraphTransferConstNodeInfo.ProtoReflect.Descriptor instead.
func (*GraphTransferConstNodeInfo) GetData ¶
func (x *GraphTransferConstNodeInfo) GetData() []byte
func (*GraphTransferConstNodeInfo) GetDtype ¶
func (x *GraphTransferConstNodeInfo) GetDtype() DataType
func (*GraphTransferConstNodeInfo) GetName ¶
func (x *GraphTransferConstNodeInfo) GetName() string
func (*GraphTransferConstNodeInfo) GetNodeId ¶
func (x *GraphTransferConstNodeInfo) GetNodeId() int32
func (*GraphTransferConstNodeInfo) GetShape ¶
func (x *GraphTransferConstNodeInfo) GetShape() []int64
func (*GraphTransferConstNodeInfo) ProtoMessage ¶
func (*GraphTransferConstNodeInfo) ProtoMessage()
func (*GraphTransferConstNodeInfo) ProtoReflect ¶ added in v0.12.0
func (x *GraphTransferConstNodeInfo) ProtoReflect() protoreflect.Message
func (*GraphTransferConstNodeInfo) Reset ¶
func (x *GraphTransferConstNodeInfo) Reset()
func (*GraphTransferConstNodeInfo) String ¶
func (x *GraphTransferConstNodeInfo) String() string
type GraphTransferGraphInputNodeInfo ¶
type GraphTransferGraphInputNodeInfo struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` Shape []int64 `protobuf:"varint,2,rep,packed,name=shape,proto3" json:"shape,omitempty"` Dtype DataType `protobuf:"varint,3,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` // contains filtered or unexported fields }
func (*GraphTransferGraphInputNodeInfo) Descriptor
deprecated
func (*GraphTransferGraphInputNodeInfo) Descriptor() ([]byte, []int)
Deprecated: Use GraphTransferGraphInputNodeInfo.ProtoReflect.Descriptor instead.
func (*GraphTransferGraphInputNodeInfo) GetDtype ¶
func (x *GraphTransferGraphInputNodeInfo) GetDtype() DataType
func (*GraphTransferGraphInputNodeInfo) GetName ¶
func (x *GraphTransferGraphInputNodeInfo) GetName() string
func (*GraphTransferGraphInputNodeInfo) GetShape ¶
func (x *GraphTransferGraphInputNodeInfo) GetShape() []int64
func (*GraphTransferGraphInputNodeInfo) ProtoMessage ¶
func (*GraphTransferGraphInputNodeInfo) ProtoMessage()
func (*GraphTransferGraphInputNodeInfo) ProtoReflect ¶ added in v0.12.0
func (x *GraphTransferGraphInputNodeInfo) ProtoReflect() protoreflect.Message
func (*GraphTransferGraphInputNodeInfo) Reset ¶
func (x *GraphTransferGraphInputNodeInfo) Reset()
func (*GraphTransferGraphInputNodeInfo) String ¶
func (x *GraphTransferGraphInputNodeInfo) String() string
type GraphTransferGraphOutputNodeInfo ¶
type GraphTransferGraphOutputNodeInfo struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` Shape []int64 `protobuf:"varint,2,rep,packed,name=shape,proto3" json:"shape,omitempty"` Dtype DataType `protobuf:"varint,3,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` // contains filtered or unexported fields }
func (*GraphTransferGraphOutputNodeInfo) Descriptor
deprecated
func (*GraphTransferGraphOutputNodeInfo) Descriptor() ([]byte, []int)
Deprecated: Use GraphTransferGraphOutputNodeInfo.ProtoReflect.Descriptor instead.
func (*GraphTransferGraphOutputNodeInfo) GetDtype ¶
func (x *GraphTransferGraphOutputNodeInfo) GetDtype() DataType
func (*GraphTransferGraphOutputNodeInfo) GetName ¶
func (x *GraphTransferGraphOutputNodeInfo) GetName() string
func (*GraphTransferGraphOutputNodeInfo) GetShape ¶
func (x *GraphTransferGraphOutputNodeInfo) GetShape() []int64
func (*GraphTransferGraphOutputNodeInfo) ProtoMessage ¶
func (*GraphTransferGraphOutputNodeInfo) ProtoMessage()
func (*GraphTransferGraphOutputNodeInfo) ProtoReflect ¶ added in v0.12.0
func (x *GraphTransferGraphOutputNodeInfo) ProtoReflect() protoreflect.Message
func (*GraphTransferGraphOutputNodeInfo) Reset ¶
func (x *GraphTransferGraphOutputNodeInfo) Reset()
func (*GraphTransferGraphOutputNodeInfo) String ¶
func (x *GraphTransferGraphOutputNodeInfo) String() string
type GraphTransferInfo ¶
type GraphTransferInfo struct { NodeInfo []*GraphTransferNodeInfo `protobuf:"bytes,1,rep,name=node_info,json=nodeInfo,proto3" json:"node_info,omitempty"` ConstNodeInfo []*GraphTransferConstNodeInfo `protobuf:"bytes,2,rep,name=const_node_info,json=constNodeInfo,proto3" json:"const_node_info,omitempty"` NodeInputInfo []*GraphTransferNodeInputInfo `protobuf:"bytes,3,rep,name=node_input_info,json=nodeInputInfo,proto3" json:"node_input_info,omitempty"` NodeOutputInfo []*GraphTransferNodeOutputInfo `protobuf:"bytes,4,rep,name=node_output_info,json=nodeOutputInfo,proto3" json:"node_output_info,omitempty"` // Input Node parameters of transferred graph GraphInputNodeInfo []*GraphTransferGraphInputNodeInfo `protobuf:"bytes,5,rep,name=graph_input_node_info,json=graphInputNodeInfo,proto3" json:"graph_input_node_info,omitempty"` GraphOutputNodeInfo []*GraphTransferGraphOutputNodeInfo `protobuf:"bytes,6,rep,name=graph_output_node_info,json=graphOutputNodeInfo,proto3" json:"graph_output_node_info,omitempty"` // Destination of graph transfer Destination GraphTransferInfo_Destination `protobuf:"varint,7,opt,name=destination,proto3,enum=tensorflow.GraphTransferInfo_Destination" json:"destination,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing a handle to a tensorflow resource. Handles are not valid across executions, but can be serialized back and forth from within a single run.
func (*GraphTransferInfo) Descriptor
deprecated
func (*GraphTransferInfo) Descriptor() ([]byte, []int)
Deprecated: Use GraphTransferInfo.ProtoReflect.Descriptor instead.
func (*GraphTransferInfo) GetConstNodeInfo ¶
func (x *GraphTransferInfo) GetConstNodeInfo() []*GraphTransferConstNodeInfo
func (*GraphTransferInfo) GetDestination ¶
func (x *GraphTransferInfo) GetDestination() GraphTransferInfo_Destination
func (*GraphTransferInfo) GetGraphInputNodeInfo ¶
func (x *GraphTransferInfo) GetGraphInputNodeInfo() []*GraphTransferGraphInputNodeInfo
func (*GraphTransferInfo) GetGraphOutputNodeInfo ¶
func (x *GraphTransferInfo) GetGraphOutputNodeInfo() []*GraphTransferGraphOutputNodeInfo
func (*GraphTransferInfo) GetNodeInfo ¶
func (x *GraphTransferInfo) GetNodeInfo() []*GraphTransferNodeInfo
func (*GraphTransferInfo) GetNodeInputInfo ¶
func (x *GraphTransferInfo) GetNodeInputInfo() []*GraphTransferNodeInputInfo
func (*GraphTransferInfo) GetNodeOutputInfo ¶
func (x *GraphTransferInfo) GetNodeOutputInfo() []*GraphTransferNodeOutputInfo
func (*GraphTransferInfo) ProtoMessage ¶
func (*GraphTransferInfo) ProtoMessage()
func (*GraphTransferInfo) ProtoReflect ¶ added in v0.12.0
func (x *GraphTransferInfo) ProtoReflect() protoreflect.Message
func (*GraphTransferInfo) Reset ¶
func (x *GraphTransferInfo) Reset()
func (*GraphTransferInfo) String ¶
func (x *GraphTransferInfo) String() string
type GraphTransferInfo_Destination ¶
type GraphTransferInfo_Destination int32
const ( GraphTransferInfo_NOP GraphTransferInfo_Destination = 0 GraphTransferInfo_HEXAGON GraphTransferInfo_Destination = 1 )
func (GraphTransferInfo_Destination) Descriptor ¶ added in v0.12.0
func (GraphTransferInfo_Destination) Descriptor() protoreflect.EnumDescriptor
func (GraphTransferInfo_Destination) Enum ¶ added in v0.12.0
func (x GraphTransferInfo_Destination) Enum() *GraphTransferInfo_Destination
func (GraphTransferInfo_Destination) EnumDescriptor
deprecated
func (GraphTransferInfo_Destination) EnumDescriptor() ([]byte, []int)
Deprecated: Use GraphTransferInfo_Destination.Descriptor instead.
func (GraphTransferInfo_Destination) Number ¶ added in v0.12.0
func (x GraphTransferInfo_Destination) Number() protoreflect.EnumNumber
func (GraphTransferInfo_Destination) String ¶
func (x GraphTransferInfo_Destination) String() string
func (GraphTransferInfo_Destination) Type ¶ added in v0.12.0
func (GraphTransferInfo_Destination) Type() protoreflect.EnumType
type GraphTransferNodeInfo ¶
type GraphTransferNodeInfo struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` NodeId int32 `protobuf:"varint,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"` TypeName string `protobuf:"bytes,3,opt,name=type_name,json=typeName,proto3" json:"type_name,omitempty"` SocOpId int32 `protobuf:"varint,4,opt,name=soc_op_id,json=socOpId,proto3" json:"soc_op_id,omitempty"` PaddingId int32 `protobuf:"varint,5,opt,name=padding_id,json=paddingId,proto3" json:"padding_id,omitempty"` InputCount int32 `protobuf:"varint,6,opt,name=input_count,json=inputCount,proto3" json:"input_count,omitempty"` OutputCount int32 `protobuf:"varint,7,opt,name=output_count,json=outputCount,proto3" json:"output_count,omitempty"` // contains filtered or unexported fields }
func (*GraphTransferNodeInfo) Descriptor
deprecated
func (*GraphTransferNodeInfo) Descriptor() ([]byte, []int)
Deprecated: Use GraphTransferNodeInfo.ProtoReflect.Descriptor instead.
func (*GraphTransferNodeInfo) GetInputCount ¶
func (x *GraphTransferNodeInfo) GetInputCount() int32
func (*GraphTransferNodeInfo) GetName ¶
func (x *GraphTransferNodeInfo) GetName() string
func (*GraphTransferNodeInfo) GetNodeId ¶
func (x *GraphTransferNodeInfo) GetNodeId() int32
func (*GraphTransferNodeInfo) GetOutputCount ¶
func (x *GraphTransferNodeInfo) GetOutputCount() int32
func (*GraphTransferNodeInfo) GetPaddingId ¶
func (x *GraphTransferNodeInfo) GetPaddingId() int32
func (*GraphTransferNodeInfo) GetSocOpId ¶
func (x *GraphTransferNodeInfo) GetSocOpId() int32
func (*GraphTransferNodeInfo) GetTypeName ¶
func (x *GraphTransferNodeInfo) GetTypeName() string
func (*GraphTransferNodeInfo) ProtoMessage ¶
func (*GraphTransferNodeInfo) ProtoMessage()
func (*GraphTransferNodeInfo) ProtoReflect ¶ added in v0.12.0
func (x *GraphTransferNodeInfo) ProtoReflect() protoreflect.Message
func (*GraphTransferNodeInfo) Reset ¶
func (x *GraphTransferNodeInfo) Reset()
func (*GraphTransferNodeInfo) String ¶
func (x *GraphTransferNodeInfo) String() string
type GraphTransferNodeInput ¶
type GraphTransferNodeInput struct { NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"` OutputPort int32 `protobuf:"varint,2,opt,name=output_port,json=outputPort,proto3" json:"output_port,omitempty"` // contains filtered or unexported fields }
func (*GraphTransferNodeInput) Descriptor
deprecated
func (*GraphTransferNodeInput) Descriptor() ([]byte, []int)
Deprecated: Use GraphTransferNodeInput.ProtoReflect.Descriptor instead.
func (*GraphTransferNodeInput) GetNodeId ¶
func (x *GraphTransferNodeInput) GetNodeId() int32
func (*GraphTransferNodeInput) GetOutputPort ¶
func (x *GraphTransferNodeInput) GetOutputPort() int32
func (*GraphTransferNodeInput) ProtoMessage ¶
func (*GraphTransferNodeInput) ProtoMessage()
func (*GraphTransferNodeInput) ProtoReflect ¶ added in v0.12.0
func (x *GraphTransferNodeInput) ProtoReflect() protoreflect.Message
func (*GraphTransferNodeInput) Reset ¶
func (x *GraphTransferNodeInput) Reset()
func (*GraphTransferNodeInput) String ¶
func (x *GraphTransferNodeInput) String() string
type GraphTransferNodeInputInfo ¶
type GraphTransferNodeInputInfo struct { NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"` NodeInput []*GraphTransferNodeInput `protobuf:"bytes,2,rep,name=node_input,json=nodeInput,proto3" json:"node_input,omitempty"` // contains filtered or unexported fields }
func (*GraphTransferNodeInputInfo) Descriptor
deprecated
func (*GraphTransferNodeInputInfo) Descriptor() ([]byte, []int)
Deprecated: Use GraphTransferNodeInputInfo.ProtoReflect.Descriptor instead.
func (*GraphTransferNodeInputInfo) GetNodeId ¶
func (x *GraphTransferNodeInputInfo) GetNodeId() int32
func (*GraphTransferNodeInputInfo) GetNodeInput ¶
func (x *GraphTransferNodeInputInfo) GetNodeInput() []*GraphTransferNodeInput
func (*GraphTransferNodeInputInfo) ProtoMessage ¶
func (*GraphTransferNodeInputInfo) ProtoMessage()
func (*GraphTransferNodeInputInfo) ProtoReflect ¶ added in v0.12.0
func (x *GraphTransferNodeInputInfo) ProtoReflect() protoreflect.Message
func (*GraphTransferNodeInputInfo) Reset ¶
func (x *GraphTransferNodeInputInfo) Reset()
func (*GraphTransferNodeInputInfo) String ¶
func (x *GraphTransferNodeInputInfo) String() string
type GraphTransferNodeOutputInfo ¶
type GraphTransferNodeOutputInfo struct { NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"` MaxByteSize []int32 `protobuf:"varint,2,rep,packed,name=max_byte_size,json=maxByteSize,proto3" json:"max_byte_size,omitempty"` // contains filtered or unexported fields }
func (*GraphTransferNodeOutputInfo) Descriptor
deprecated
func (*GraphTransferNodeOutputInfo) Descriptor() ([]byte, []int)
Deprecated: Use GraphTransferNodeOutputInfo.ProtoReflect.Descriptor instead.
func (*GraphTransferNodeOutputInfo) GetMaxByteSize ¶
func (x *GraphTransferNodeOutputInfo) GetMaxByteSize() []int32
func (*GraphTransferNodeOutputInfo) GetNodeId ¶
func (x *GraphTransferNodeOutputInfo) GetNodeId() int32
func (*GraphTransferNodeOutputInfo) ProtoMessage ¶
func (*GraphTransferNodeOutputInfo) ProtoMessage()
func (*GraphTransferNodeOutputInfo) ProtoReflect ¶ added in v0.12.0
func (x *GraphTransferNodeOutputInfo) ProtoReflect() protoreflect.Message
func (*GraphTransferNodeOutputInfo) Reset ¶
func (x *GraphTransferNodeOutputInfo) Reset()
func (*GraphTransferNodeOutputInfo) String ¶
func (x *GraphTransferNodeOutputInfo) String() string
type HistogramProto ¶
type HistogramProto struct { Min float64 `protobuf:"fixed64,1,opt,name=min,proto3" json:"min,omitempty"` Max float64 `protobuf:"fixed64,2,opt,name=max,proto3" json:"max,omitempty"` Num float64 `protobuf:"fixed64,3,opt,name=num,proto3" json:"num,omitempty"` Sum float64 `protobuf:"fixed64,4,opt,name=sum,proto3" json:"sum,omitempty"` SumSquares float64 `protobuf:"fixed64,5,opt,name=sum_squares,json=sumSquares,proto3" json:"sum_squares,omitempty"` // Parallel arrays encoding the bucket boundaries and the bucket values. // bucket(i) is the count for the bucket i. The range for // a bucket is: // // i == 0: -DBL_MAX .. bucket_limit(0) // i != 0: bucket_limit(i-1) .. bucket_limit(i) BucketLimit []float64 `protobuf:"fixed64,6,rep,packed,name=bucket_limit,json=bucketLimit,proto3" json:"bucket_limit,omitempty"` Bucket []float64 `protobuf:"fixed64,7,rep,packed,name=bucket,proto3" json:"bucket,omitempty"` // contains filtered or unexported fields }
Serialization format for histogram module in core/lib/histogram/histogram.h
func (*HistogramProto) Descriptor
deprecated
func (*HistogramProto) Descriptor() ([]byte, []int)
Deprecated: Use HistogramProto.ProtoReflect.Descriptor instead.
func (*HistogramProto) GetBucket ¶
func (x *HistogramProto) GetBucket() []float64
func (*HistogramProto) GetBucketLimit ¶
func (x *HistogramProto) GetBucketLimit() []float64
func (*HistogramProto) GetMax ¶
func (x *HistogramProto) GetMax() float64
func (*HistogramProto) GetMin ¶
func (x *HistogramProto) GetMin() float64
func (*HistogramProto) GetNum ¶
func (x *HistogramProto) GetNum() float64
func (*HistogramProto) GetSum ¶
func (x *HistogramProto) GetSum() float64
func (*HistogramProto) GetSumSquares ¶
func (x *HistogramProto) GetSumSquares() float64
func (*HistogramProto) ProtoMessage ¶
func (*HistogramProto) ProtoMessage()
func (*HistogramProto) ProtoReflect ¶ added in v0.12.0
func (x *HistogramProto) ProtoReflect() protoreflect.Message
func (*HistogramProto) Reset ¶
func (x *HistogramProto) Reset()
func (*HistogramProto) String ¶
func (x *HistogramProto) String() string
type InterconnectLink ¶
type InterconnectLink struct { DeviceId int32 `protobuf:"varint,1,opt,name=device_id,json=deviceId,proto3" json:"device_id,omitempty"` Type string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"` Strength int32 `protobuf:"varint,3,opt,name=strength,proto3" json:"strength,omitempty"` // contains filtered or unexported fields }
func (*InterconnectLink) Descriptor
deprecated
func (*InterconnectLink) Descriptor() ([]byte, []int)
Deprecated: Use InterconnectLink.ProtoReflect.Descriptor instead.
func (*InterconnectLink) GetDeviceId ¶
func (x *InterconnectLink) GetDeviceId() int32
func (*InterconnectLink) GetStrength ¶
func (x *InterconnectLink) GetStrength() int32
func (*InterconnectLink) GetType ¶
func (x *InterconnectLink) GetType() string
func (*InterconnectLink) ProtoMessage ¶
func (*InterconnectLink) ProtoMessage()
func (*InterconnectLink) ProtoReflect ¶ added in v0.12.0
func (x *InterconnectLink) ProtoReflect() protoreflect.Message
func (*InterconnectLink) Reset ¶
func (x *InterconnectLink) Reset()
func (*InterconnectLink) String ¶
func (x *InterconnectLink) String() string
type IteratorStateMetadata ¶
type IteratorStateMetadata struct { // A user-specified version string. Version string `protobuf:"bytes,1,opt,name=version,proto3" json:"version,omitempty"` // Keys for tensors in the VariantTensorDataProto. Keys []string `protobuf:"bytes,2,rep,name=keys,proto3" json:"keys,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing the metadata for an iterator's state stored as a Variant tensor.
func (*IteratorStateMetadata) Descriptor
deprecated
func (*IteratorStateMetadata) Descriptor() ([]byte, []int)
Deprecated: Use IteratorStateMetadata.ProtoReflect.Descriptor instead.
func (*IteratorStateMetadata) GetKeys ¶
func (x *IteratorStateMetadata) GetKeys() []string
func (*IteratorStateMetadata) GetVersion ¶
func (x *IteratorStateMetadata) GetVersion() string
func (*IteratorStateMetadata) ProtoMessage ¶
func (*IteratorStateMetadata) ProtoMessage()
func (*IteratorStateMetadata) ProtoReflect ¶ added in v0.12.0
func (x *IteratorStateMetadata) ProtoReflect() protoreflect.Message
func (*IteratorStateMetadata) Reset ¶
func (x *IteratorStateMetadata) Reset()
func (*IteratorStateMetadata) String ¶
func (x *IteratorStateMetadata) String() string
type KernelDef ¶
type KernelDef struct { // Must match the name of an Op. Op string `protobuf:"bytes,1,opt,name=op,proto3" json:"op,omitempty"` // Type of device this kernel runs on. DeviceType string `protobuf:"bytes,2,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"` Constraint []*KernelDef_AttrConstraint `protobuf:"bytes,3,rep,name=constraint,proto3" json:"constraint,omitempty"` // Names of the Op's input_/output_args that reside in host memory // instead of device memory. HostMemoryArg []string `protobuf:"bytes,4,rep,name=host_memory_arg,json=hostMemoryArg,proto3" json:"host_memory_arg,omitempty"` // This allows experimental kernels to be registered for an op that // won't be used unless the user specifies a "_kernel" attr with // value matching this. Label string `protobuf:"bytes,5,opt,name=label,proto3" json:"label,omitempty"` // Prioritization of kernel amongst different devices. By default we assume // priority is 0. The higher the priority the better. By default (i.e. if // this is not set), we prefer GPU kernels over CPU. Priority int32 `protobuf:"varint,6,opt,name=priority,proto3" json:"priority,omitempty"` // contains filtered or unexported fields }
func (*KernelDef) Descriptor
deprecated
func (*KernelDef) GetConstraint ¶
func (x *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
func (*KernelDef) GetDeviceType ¶
func (*KernelDef) GetHostMemoryArg ¶
func (*KernelDef) GetPriority ¶
func (*KernelDef) ProtoMessage ¶
func (*KernelDef) ProtoMessage()
func (*KernelDef) ProtoReflect ¶ added in v0.12.0
func (x *KernelDef) ProtoReflect() protoreflect.Message
type KernelDef_AttrConstraint ¶
type KernelDef_AttrConstraint struct { // Name of an attr from the Op. Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // A list of values that this kernel supports for this attr. // Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops. AllowedValues *AttrValue `protobuf:"bytes,2,opt,name=allowed_values,json=allowedValues,proto3" json:"allowed_values,omitempty"` // contains filtered or unexported fields }
func (*KernelDef_AttrConstraint) Descriptor
deprecated
func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)
Deprecated: Use KernelDef_AttrConstraint.ProtoReflect.Descriptor instead.
func (*KernelDef_AttrConstraint) GetAllowedValues ¶
func (x *KernelDef_AttrConstraint) GetAllowedValues() *AttrValue
func (*KernelDef_AttrConstraint) GetName ¶
func (x *KernelDef_AttrConstraint) GetName() string
func (*KernelDef_AttrConstraint) ProtoMessage ¶
func (*KernelDef_AttrConstraint) ProtoMessage()
func (*KernelDef_AttrConstraint) ProtoReflect ¶ added in v0.12.0
func (x *KernelDef_AttrConstraint) ProtoReflect() protoreflect.Message
func (*KernelDef_AttrConstraint) Reset ¶
func (x *KernelDef_AttrConstraint) Reset()
func (*KernelDef_AttrConstraint) String ¶
func (x *KernelDef_AttrConstraint) String() string
type KernelList ¶
type KernelList struct { Kernel []*KernelDef `protobuf:"bytes,1,rep,name=kernel,proto3" json:"kernel,omitempty"` // contains filtered or unexported fields }
A collection of KernelDefs
func (*KernelList) Descriptor
deprecated
func (*KernelList) Descriptor() ([]byte, []int)
Deprecated: Use KernelList.ProtoReflect.Descriptor instead.
func (*KernelList) GetKernel ¶
func (x *KernelList) GetKernel() []*KernelDef
func (*KernelList) ProtoMessage ¶
func (*KernelList) ProtoMessage()
func (*KernelList) ProtoReflect ¶ added in v0.12.0
func (x *KernelList) ProtoReflect() protoreflect.Message
func (*KernelList) Reset ¶
func (x *KernelList) Reset()
func (*KernelList) String ¶
func (x *KernelList) String() string
type LocalLinks ¶
type LocalLinks struct { Link []*InterconnectLink `protobuf:"bytes,1,rep,name=link,proto3" json:"link,omitempty"` // contains filtered or unexported fields }
func (*LocalLinks) Descriptor
deprecated
func (*LocalLinks) Descriptor() ([]byte, []int)
Deprecated: Use LocalLinks.ProtoReflect.Descriptor instead.
func (*LocalLinks) GetLink ¶
func (x *LocalLinks) GetLink() []*InterconnectLink
func (*LocalLinks) ProtoMessage ¶
func (*LocalLinks) ProtoMessage()
func (*LocalLinks) ProtoReflect ¶ added in v0.12.0
func (x *LocalLinks) ProtoReflect() protoreflect.Message
func (*LocalLinks) Reset ¶
func (x *LocalLinks) Reset()
func (*LocalLinks) String ¶
func (x *LocalLinks) String() string
type MemoryLogRawAllocation ¶
type MemoryLogRawAllocation struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // Name of the operation making the allocation. Operation string `protobuf:"bytes,2,opt,name=operation,proto3" json:"operation,omitempty"` // Number of bytes in the allocation. NumBytes int64 `protobuf:"varint,3,opt,name=num_bytes,json=numBytes,proto3" json:"num_bytes,omitempty"` // Address of the allocation. Ptr uint64 `protobuf:"varint,4,opt,name=ptr,proto3" json:"ptr,omitempty"` // Id of the tensor buffer being allocated, used to match to a // corresponding deallocation. AllocationId int64 `protobuf:"varint,5,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"` // Name of the allocator used. AllocatorName string `protobuf:"bytes,6,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"` // contains filtered or unexported fields }
func (*MemoryLogRawAllocation) Descriptor
deprecated
func (*MemoryLogRawAllocation) Descriptor() ([]byte, []int)
Deprecated: Use MemoryLogRawAllocation.ProtoReflect.Descriptor instead.
func (*MemoryLogRawAllocation) GetAllocationId ¶
func (x *MemoryLogRawAllocation) GetAllocationId() int64
func (*MemoryLogRawAllocation) GetAllocatorName ¶
func (x *MemoryLogRawAllocation) GetAllocatorName() string
func (*MemoryLogRawAllocation) GetNumBytes ¶
func (x *MemoryLogRawAllocation) GetNumBytes() int64
func (*MemoryLogRawAllocation) GetOperation ¶
func (x *MemoryLogRawAllocation) GetOperation() string
func (*MemoryLogRawAllocation) GetPtr ¶
func (x *MemoryLogRawAllocation) GetPtr() uint64
func (*MemoryLogRawAllocation) GetStepId ¶
func (x *MemoryLogRawAllocation) GetStepId() int64
func (*MemoryLogRawAllocation) ProtoMessage ¶
func (*MemoryLogRawAllocation) ProtoMessage()
func (*MemoryLogRawAllocation) ProtoReflect ¶ added in v0.12.0
func (x *MemoryLogRawAllocation) ProtoReflect() protoreflect.Message
func (*MemoryLogRawAllocation) Reset ¶
func (x *MemoryLogRawAllocation) Reset()
func (*MemoryLogRawAllocation) String ¶
func (x *MemoryLogRawAllocation) String() string
type MemoryLogRawDeallocation ¶
type MemoryLogRawDeallocation struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // Name of the operation making the deallocation. Operation string `protobuf:"bytes,2,opt,name=operation,proto3" json:"operation,omitempty"` // Id of the tensor buffer being deallocated, used to match to a // corresponding allocation. AllocationId int64 `protobuf:"varint,3,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"` // Name of the allocator used. AllocatorName string `protobuf:"bytes,4,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"` // True if the deallocation is queued and will be performed later, // e.g. for GPU lazy freeing of buffers. Deferred bool `protobuf:"varint,5,opt,name=deferred,proto3" json:"deferred,omitempty"` // contains filtered or unexported fields }
func (*MemoryLogRawDeallocation) Descriptor
deprecated
func (*MemoryLogRawDeallocation) Descriptor() ([]byte, []int)
Deprecated: Use MemoryLogRawDeallocation.ProtoReflect.Descriptor instead.
func (*MemoryLogRawDeallocation) GetAllocationId ¶
func (x *MemoryLogRawDeallocation) GetAllocationId() int64
func (*MemoryLogRawDeallocation) GetAllocatorName ¶
func (x *MemoryLogRawDeallocation) GetAllocatorName() string
func (*MemoryLogRawDeallocation) GetDeferred ¶
func (x *MemoryLogRawDeallocation) GetDeferred() bool
func (*MemoryLogRawDeallocation) GetOperation ¶
func (x *MemoryLogRawDeallocation) GetOperation() string
func (*MemoryLogRawDeallocation) GetStepId ¶
func (x *MemoryLogRawDeallocation) GetStepId() int64
func (*MemoryLogRawDeallocation) ProtoMessage ¶
func (*MemoryLogRawDeallocation) ProtoMessage()
func (*MemoryLogRawDeallocation) ProtoReflect ¶ added in v0.12.0
func (x *MemoryLogRawDeallocation) ProtoReflect() protoreflect.Message
func (*MemoryLogRawDeallocation) Reset ¶
func (x *MemoryLogRawDeallocation) Reset()
func (*MemoryLogRawDeallocation) String ¶
func (x *MemoryLogRawDeallocation) String() string
type MemoryLogStep ¶
type MemoryLogStep struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // Handle describing the feeds and fetches of the step. Handle string `protobuf:"bytes,2,opt,name=handle,proto3" json:"handle,omitempty"` // contains filtered or unexported fields }
func (*MemoryLogStep) Descriptor
deprecated
func (*MemoryLogStep) Descriptor() ([]byte, []int)
Deprecated: Use MemoryLogStep.ProtoReflect.Descriptor instead.
func (*MemoryLogStep) GetHandle ¶
func (x *MemoryLogStep) GetHandle() string
func (*MemoryLogStep) GetStepId ¶
func (x *MemoryLogStep) GetStepId() int64
func (*MemoryLogStep) ProtoMessage ¶
func (*MemoryLogStep) ProtoMessage()
func (*MemoryLogStep) ProtoReflect ¶ added in v0.12.0
func (x *MemoryLogStep) ProtoReflect() protoreflect.Message
func (*MemoryLogStep) Reset ¶
func (x *MemoryLogStep) Reset()
func (*MemoryLogStep) String ¶
func (x *MemoryLogStep) String() string
type MemoryLogTensorAllocation ¶
type MemoryLogTensorAllocation struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // Name of the kernel making the allocation as set in GraphDef, // e.g., "affine2/weights/Assign". KernelName string `protobuf:"bytes,2,opt,name=kernel_name,json=kernelName,proto3" json:"kernel_name,omitempty"` // Allocated tensor details. Tensor *TensorDescription `protobuf:"bytes,3,opt,name=tensor,proto3" json:"tensor,omitempty"` // contains filtered or unexported fields }
func (*MemoryLogTensorAllocation) Descriptor
deprecated
func (*MemoryLogTensorAllocation) Descriptor() ([]byte, []int)
Deprecated: Use MemoryLogTensorAllocation.ProtoReflect.Descriptor instead.
func (*MemoryLogTensorAllocation) GetKernelName ¶
func (x *MemoryLogTensorAllocation) GetKernelName() string
func (*MemoryLogTensorAllocation) GetStepId ¶
func (x *MemoryLogTensorAllocation) GetStepId() int64
func (*MemoryLogTensorAllocation) GetTensor ¶
func (x *MemoryLogTensorAllocation) GetTensor() *TensorDescription
func (*MemoryLogTensorAllocation) ProtoMessage ¶
func (*MemoryLogTensorAllocation) ProtoMessage()
func (*MemoryLogTensorAllocation) ProtoReflect ¶ added in v0.12.0
func (x *MemoryLogTensorAllocation) ProtoReflect() protoreflect.Message
func (*MemoryLogTensorAllocation) Reset ¶
func (x *MemoryLogTensorAllocation) Reset()
func (*MemoryLogTensorAllocation) String ¶
func (x *MemoryLogTensorAllocation) String() string
type MemoryLogTensorDeallocation ¶
type MemoryLogTensorDeallocation struct { // Id of the tensor buffer being deallocated, used to match to a // corresponding allocation. AllocationId int64 `protobuf:"varint,1,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"` // Name of the allocator used. AllocatorName string `protobuf:"bytes,2,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"` // contains filtered or unexported fields }
func (*MemoryLogTensorDeallocation) Descriptor
deprecated
func (*MemoryLogTensorDeallocation) Descriptor() ([]byte, []int)
Deprecated: Use MemoryLogTensorDeallocation.ProtoReflect.Descriptor instead.
func (*MemoryLogTensorDeallocation) GetAllocationId ¶
func (x *MemoryLogTensorDeallocation) GetAllocationId() int64
func (*MemoryLogTensorDeallocation) GetAllocatorName ¶
func (x *MemoryLogTensorDeallocation) GetAllocatorName() string
func (*MemoryLogTensorDeallocation) ProtoMessage ¶
func (*MemoryLogTensorDeallocation) ProtoMessage()
func (*MemoryLogTensorDeallocation) ProtoReflect ¶ added in v0.12.0
func (x *MemoryLogTensorDeallocation) ProtoReflect() protoreflect.Message
func (*MemoryLogTensorDeallocation) Reset ¶
func (x *MemoryLogTensorDeallocation) Reset()
func (*MemoryLogTensorDeallocation) String ¶
func (x *MemoryLogTensorDeallocation) String() string
type MemoryLogTensorOutput ¶
type MemoryLogTensorOutput struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // Name of the kernel producing an output as set in GraphDef, e.g., // "affine2/weights/Assign". KernelName string `protobuf:"bytes,2,opt,name=kernel_name,json=kernelName,proto3" json:"kernel_name,omitempty"` // Index of the output being set. Index int32 `protobuf:"varint,3,opt,name=index,proto3" json:"index,omitempty"` // Output tensor details. Tensor *TensorDescription `protobuf:"bytes,4,opt,name=tensor,proto3" json:"tensor,omitempty"` // contains filtered or unexported fields }
func (*MemoryLogTensorOutput) Descriptor
deprecated
func (*MemoryLogTensorOutput) Descriptor() ([]byte, []int)
Deprecated: Use MemoryLogTensorOutput.ProtoReflect.Descriptor instead.
func (*MemoryLogTensorOutput) GetIndex ¶
func (x *MemoryLogTensorOutput) GetIndex() int32
func (*MemoryLogTensorOutput) GetKernelName ¶
func (x *MemoryLogTensorOutput) GetKernelName() string
func (*MemoryLogTensorOutput) GetStepId ¶
func (x *MemoryLogTensorOutput) GetStepId() int64
func (*MemoryLogTensorOutput) GetTensor ¶
func (x *MemoryLogTensorOutput) GetTensor() *TensorDescription
func (*MemoryLogTensorOutput) ProtoMessage ¶
func (*MemoryLogTensorOutput) ProtoMessage()
func (*MemoryLogTensorOutput) ProtoReflect ¶ added in v0.12.0
func (x *MemoryLogTensorOutput) ProtoReflect() protoreflect.Message
func (*MemoryLogTensorOutput) Reset ¶
func (x *MemoryLogTensorOutput) Reset()
func (*MemoryLogTensorOutput) String ¶
func (x *MemoryLogTensorOutput) String() string
type MemoryStats ¶
type MemoryStats struct { TempMemorySize int64 `protobuf:"varint,1,opt,name=temp_memory_size,json=tempMemorySize,proto3" json:"temp_memory_size,omitempty"` PersistentMemorySize int64 `protobuf:"varint,3,opt,name=persistent_memory_size,json=persistentMemorySize,proto3" json:"persistent_memory_size,omitempty"` PersistentTensorAllocIds []int64 `` /* 145-byte string literal not displayed */ // Deprecated: Do not use. DeviceTempMemorySize int64 `` /* 126-byte string literal not displayed */ // Deprecated: Do not use. DevicePersistentMemorySize int64 `` /* 144-byte string literal not displayed */ // Deprecated: Do not use. DevicePersistentTensorAllocIds []int64 `` /* 165-byte string literal not displayed */ // contains filtered or unexported fields }
For memory tracking.
func (*MemoryStats) Descriptor
deprecated
func (*MemoryStats) Descriptor() ([]byte, []int)
Deprecated: Use MemoryStats.ProtoReflect.Descriptor instead.
func (*MemoryStats) GetDevicePersistentMemorySize
deprecated
func (x *MemoryStats) GetDevicePersistentMemorySize() int64
Deprecated: Do not use.
func (*MemoryStats) GetDevicePersistentTensorAllocIds
deprecated
func (x *MemoryStats) GetDevicePersistentTensorAllocIds() []int64
Deprecated: Do not use.
func (*MemoryStats) GetDeviceTempMemorySize
deprecated
func (x *MemoryStats) GetDeviceTempMemorySize() int64
Deprecated: Do not use.
func (*MemoryStats) GetPersistentMemorySize ¶
func (x *MemoryStats) GetPersistentMemorySize() int64
func (*MemoryStats) GetPersistentTensorAllocIds ¶
func (x *MemoryStats) GetPersistentTensorAllocIds() []int64
func (*MemoryStats) GetTempMemorySize ¶
func (x *MemoryStats) GetTempMemorySize() int64
func (*MemoryStats) ProtoMessage ¶
func (*MemoryStats) ProtoMessage()
func (*MemoryStats) ProtoReflect ¶ added in v0.12.0
func (x *MemoryStats) ProtoReflect() protoreflect.Message
func (*MemoryStats) Reset ¶
func (x *MemoryStats) Reset()
func (*MemoryStats) String ¶
func (x *MemoryStats) String() string
type NameAttrList ¶
type NameAttrList struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` Attr map[string]*AttrValue `` /* 149-byte string literal not displayed */ // contains filtered or unexported fields }
A list of attr names and their values. The whole list is attached with a string name. E.g., MatMul[T=float].
func (*NameAttrList) Descriptor
deprecated
func (*NameAttrList) Descriptor() ([]byte, []int)
Deprecated: Use NameAttrList.ProtoReflect.Descriptor instead.
func (*NameAttrList) GetAttr ¶
func (x *NameAttrList) GetAttr() map[string]*AttrValue
func (*NameAttrList) GetName ¶
func (x *NameAttrList) GetName() string
func (*NameAttrList) ProtoMessage ¶
func (*NameAttrList) ProtoMessage()
func (*NameAttrList) ProtoReflect ¶ added in v0.12.0
func (x *NameAttrList) ProtoReflect() protoreflect.Message
func (*NameAttrList) Reset ¶
func (x *NameAttrList) Reset()
func (*NameAttrList) String ¶
func (x *NameAttrList) String() string
type NodeDef ¶
type NodeDef struct { // The name given to this operator. Used for naming inputs, // logging, visualization, etc. Unique within a single GraphDef. // Must match the regexp "[A-Za-z0-9.][A-Za-z0-9_./]*". Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // The operation name. There may be custom parameters in attrs. // Op names starting with an underscore are reserved for internal use. Op string `protobuf:"bytes,2,opt,name=op,proto3" json:"op,omitempty"` // Each input is "node:src_output" with "node" being a string name and // "src_output" indicating which output tensor to use from "node". If // "src_output" is 0 the ":0" suffix can be omitted. Regular inputs // may optionally be followed by control inputs that have the format // "^node". Input []string `protobuf:"bytes,3,rep,name=input,proto3" json:"input,omitempty"` // A (possibly partial) specification for the device on which this // node should be placed. // The expected syntax for this string is as follows: // // DEVICE_SPEC ::= PARTIAL_SPEC // // PARTIAL_SPEC ::= ("/" CONSTRAINT) * // CONSTRAINT ::= ("job:" JOB_NAME) // // | ("replica:" [1-9][0-9]*) // | ("task:" [1-9][0-9]*) // | ("device:" [A-Za-z]* ":" ([1-9][0-9]* | "*") ) // // Valid values for this string include: // * "/job:worker/replica:0/task:1/device:GPU:3" (full specification) // * "/job:worker/device:GPU:3" (partial specification) // * "" (no specification) // // If the constraints do not resolve to a single device (or if this // field is empty or not present), the runtime will attempt to // choose a device automatically. Device string `protobuf:"bytes,4,opt,name=device,proto3" json:"device,omitempty"` // Operation-specific graph-construction-time configuration. // Note that this should include all attrs defined in the // corresponding OpDef, including those with a value matching // the default -- this allows the default to change and makes // NodeDefs easier to interpret on their own. However, if // an attr with a default is not specified in this list, the // default will be used. // The "names" (keys) must match the regexp "[a-z][a-z0-9_]+" (and // one of the names from the corresponding OpDef's attr field). // The values must have a type matching the corresponding OpDef // attr's type field. // TODO(josh11b): Add some examples here showing best practices. Attr map[string]*AttrValue `` /* 149-byte string literal not displayed */ // This stores debug information associated with the node. ExperimentalDebugInfo *NodeDef_ExperimentalDebugInfo `` /* 126-byte string literal not displayed */ // contains filtered or unexported fields }
func (*NodeDef) Descriptor
deprecated
func (*NodeDef) GetExperimentalDebugInfo ¶
func (x *NodeDef) GetExperimentalDebugInfo() *NodeDef_ExperimentalDebugInfo
func (*NodeDef) ProtoMessage ¶
func (*NodeDef) ProtoMessage()
func (*NodeDef) ProtoReflect ¶ added in v0.12.0
func (x *NodeDef) ProtoReflect() protoreflect.Message
type NodeDef_ExperimentalDebugInfo ¶
type NodeDef_ExperimentalDebugInfo struct { // Opaque string inserted into error messages created by the runtime. // // This is intended to store the list of names of the nodes from the // original graph that this node was derived. For example if this node, say // C, was result of a fusion of 2 nodes A and B, then 'original_node' would // be {A, B}. This information can be used to map errors originating at the // current node to some top level source code. OriginalNodeNames []string `protobuf:"bytes,1,rep,name=original_node_names,json=originalNodeNames,proto3" json:"original_node_names,omitempty"` // contains filtered or unexported fields }
func (*NodeDef_ExperimentalDebugInfo) Descriptor
deprecated
func (*NodeDef_ExperimentalDebugInfo) Descriptor() ([]byte, []int)
Deprecated: Use NodeDef_ExperimentalDebugInfo.ProtoReflect.Descriptor instead.
func (*NodeDef_ExperimentalDebugInfo) GetOriginalNodeNames ¶
func (x *NodeDef_ExperimentalDebugInfo) GetOriginalNodeNames() []string
func (*NodeDef_ExperimentalDebugInfo) ProtoMessage ¶
func (*NodeDef_ExperimentalDebugInfo) ProtoMessage()
func (*NodeDef_ExperimentalDebugInfo) ProtoReflect ¶ added in v0.12.0
func (x *NodeDef_ExperimentalDebugInfo) ProtoReflect() protoreflect.Message
func (*NodeDef_ExperimentalDebugInfo) Reset ¶
func (x *NodeDef_ExperimentalDebugInfo) Reset()
func (*NodeDef_ExperimentalDebugInfo) String ¶
func (x *NodeDef_ExperimentalDebugInfo) String() string
type NodeExecStats ¶
type NodeExecStats struct { // TODO(tucker): Use some more compact form of node identity than // the full string name. Either all processes should agree on a // global id (cost_id?) for each node, or we should use a hash of // the name. NodeName string `protobuf:"bytes,1,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"` AllStartMicros int64 `protobuf:"varint,2,opt,name=all_start_micros,json=allStartMicros,proto3" json:"all_start_micros,omitempty"` OpStartRelMicros int64 `protobuf:"varint,3,opt,name=op_start_rel_micros,json=opStartRelMicros,proto3" json:"op_start_rel_micros,omitempty"` OpEndRelMicros int64 `protobuf:"varint,4,opt,name=op_end_rel_micros,json=opEndRelMicros,proto3" json:"op_end_rel_micros,omitempty"` AllEndRelMicros int64 `protobuf:"varint,5,opt,name=all_end_rel_micros,json=allEndRelMicros,proto3" json:"all_end_rel_micros,omitempty"` Memory []*AllocatorMemoryUsed `protobuf:"bytes,6,rep,name=memory,proto3" json:"memory,omitempty"` Output []*NodeOutput `protobuf:"bytes,7,rep,name=output,proto3" json:"output,omitempty"` TimelineLabel string `protobuf:"bytes,8,opt,name=timeline_label,json=timelineLabel,proto3" json:"timeline_label,omitempty"` ScheduledMicros int64 `protobuf:"varint,9,opt,name=scheduled_micros,json=scheduledMicros,proto3" json:"scheduled_micros,omitempty"` ThreadId uint32 `protobuf:"varint,10,opt,name=thread_id,json=threadId,proto3" json:"thread_id,omitempty"` ReferencedTensor []*AllocationDescription `protobuf:"bytes,11,rep,name=referenced_tensor,json=referencedTensor,proto3" json:"referenced_tensor,omitempty"` MemoryStats *MemoryStats `protobuf:"bytes,12,opt,name=memory_stats,json=memoryStats,proto3" json:"memory_stats,omitempty"` AllStartNanos int64 `protobuf:"varint,13,opt,name=all_start_nanos,json=allStartNanos,proto3" json:"all_start_nanos,omitempty"` OpStartRelNanos int64 `protobuf:"varint,14,opt,name=op_start_rel_nanos,json=opStartRelNanos,proto3" json:"op_start_rel_nanos,omitempty"` OpEndRelNanos int64 `protobuf:"varint,15,opt,name=op_end_rel_nanos,json=opEndRelNanos,proto3" json:"op_end_rel_nanos,omitempty"` AllEndRelNanos int64 `protobuf:"varint,16,opt,name=all_end_rel_nanos,json=allEndRelNanos,proto3" json:"all_end_rel_nanos,omitempty"` ScheduledNanos int64 `protobuf:"varint,17,opt,name=scheduled_nanos,json=scheduledNanos,proto3" json:"scheduled_nanos,omitempty"` // contains filtered or unexported fields }
Time/size stats recorded for a single execution of a graph node.
func (*NodeExecStats) Descriptor
deprecated
func (*NodeExecStats) Descriptor() ([]byte, []int)
Deprecated: Use NodeExecStats.ProtoReflect.Descriptor instead.
func (*NodeExecStats) GetAllEndRelMicros ¶
func (x *NodeExecStats) GetAllEndRelMicros() int64
func (*NodeExecStats) GetAllEndRelNanos ¶
func (x *NodeExecStats) GetAllEndRelNanos() int64
func (*NodeExecStats) GetAllStartMicros ¶
func (x *NodeExecStats) GetAllStartMicros() int64
func (*NodeExecStats) GetAllStartNanos ¶
func (x *NodeExecStats) GetAllStartNanos() int64
func (*NodeExecStats) GetMemory ¶
func (x *NodeExecStats) GetMemory() []*AllocatorMemoryUsed
func (*NodeExecStats) GetMemoryStats ¶
func (x *NodeExecStats) GetMemoryStats() *MemoryStats
func (*NodeExecStats) GetNodeName ¶
func (x *NodeExecStats) GetNodeName() string
func (*NodeExecStats) GetOpEndRelMicros ¶
func (x *NodeExecStats) GetOpEndRelMicros() int64
func (*NodeExecStats) GetOpEndRelNanos ¶
func (x *NodeExecStats) GetOpEndRelNanos() int64
func (*NodeExecStats) GetOpStartRelMicros ¶
func (x *NodeExecStats) GetOpStartRelMicros() int64
func (*NodeExecStats) GetOpStartRelNanos ¶
func (x *NodeExecStats) GetOpStartRelNanos() int64
func (*NodeExecStats) GetOutput ¶
func (x *NodeExecStats) GetOutput() []*NodeOutput
func (*NodeExecStats) GetReferencedTensor ¶
func (x *NodeExecStats) GetReferencedTensor() []*AllocationDescription
func (*NodeExecStats) GetScheduledMicros ¶
func (x *NodeExecStats) GetScheduledMicros() int64
func (*NodeExecStats) GetScheduledNanos ¶
func (x *NodeExecStats) GetScheduledNanos() int64
func (*NodeExecStats) GetThreadId ¶
func (x *NodeExecStats) GetThreadId() uint32
func (*NodeExecStats) GetTimelineLabel ¶
func (x *NodeExecStats) GetTimelineLabel() string
func (*NodeExecStats) ProtoMessage ¶
func (*NodeExecStats) ProtoMessage()
func (*NodeExecStats) ProtoReflect ¶ added in v0.12.0
func (x *NodeExecStats) ProtoReflect() protoreflect.Message
func (*NodeExecStats) Reset ¶
func (x *NodeExecStats) Reset()
func (*NodeExecStats) String ¶
func (x *NodeExecStats) String() string
type NodeOutput ¶
type NodeOutput struct { Slot int32 `protobuf:"varint,1,opt,name=slot,proto3" json:"slot,omitempty"` TensorDescription *TensorDescription `protobuf:"bytes,3,opt,name=tensor_description,json=tensorDescription,proto3" json:"tensor_description,omitempty"` // contains filtered or unexported fields }
Output sizes recorded for a single execution of a graph node.
func (*NodeOutput) Descriptor
deprecated
func (*NodeOutput) Descriptor() ([]byte, []int)
Deprecated: Use NodeOutput.ProtoReflect.Descriptor instead.
func (*NodeOutput) GetSlot ¶
func (x *NodeOutput) GetSlot() int32
func (*NodeOutput) GetTensorDescription ¶
func (x *NodeOutput) GetTensorDescription() *TensorDescription
func (*NodeOutput) ProtoMessage ¶
func (*NodeOutput) ProtoMessage()
func (*NodeOutput) ProtoReflect ¶ added in v0.12.0
func (x *NodeOutput) ProtoReflect() protoreflect.Message
func (*NodeOutput) Reset ¶
func (x *NodeOutput) Reset()
func (*NodeOutput) String ¶
func (x *NodeOutput) String() string
type OpDef ¶
type OpDef struct { // Op names starting with an underscore are reserved for internal use. // Names should be CamelCase and match the regexp "[A-Z][a-zA-Z0-9_]*". Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // Description of the input(s). InputArg []*OpDef_ArgDef `protobuf:"bytes,2,rep,name=input_arg,json=inputArg,proto3" json:"input_arg,omitempty"` // Description of the output(s). OutputArg []*OpDef_ArgDef `protobuf:"bytes,3,rep,name=output_arg,json=outputArg,proto3" json:"output_arg,omitempty"` Attr []*OpDef_AttrDef `protobuf:"bytes,4,rep,name=attr,proto3" json:"attr,omitempty"` // Optional deprecation based on GraphDef versions. Deprecation *OpDeprecation `protobuf:"bytes,8,opt,name=deprecation,proto3" json:"deprecation,omitempty"` // One-line human-readable description of what the Op does. Summary string `protobuf:"bytes,5,opt,name=summary,proto3" json:"summary,omitempty"` // Additional, longer human-readable description of what the Op does. Description string `protobuf:"bytes,6,opt,name=description,proto3" json:"description,omitempty"` // True if the operation is commutative ("op(a,b) == op(b,a)" for all inputs) IsCommutative bool `protobuf:"varint,18,opt,name=is_commutative,json=isCommutative,proto3" json:"is_commutative,omitempty"` // If is_aggregate is true, then this operation accepts N >= 2 // inputs and produces 1 output all of the same type. Should be // associative and commutative, and produce output with the same // shape as the input. The optimizer may replace an aggregate op // taking input from multiple devices with a tree of aggregate ops // that aggregate locally within each device (and possibly within // groups of nearby devices) before communicating. // TODO(josh11b): Implement that optimization. IsAggregate bool `protobuf:"varint,16,opt,name=is_aggregate,json=isAggregate,proto3" json:"is_aggregate,omitempty"` // for things like add // Ops are marked as stateful if their behavior depends on some state beyond // their input tensors (e.g. variable reading op) or if they have // a side-effect (e.g. printing or asserting ops). Equivalently, stateless ops // must always produce the same output for the same input and have // no side-effects. // // By default Ops may be moved between devices. Stateful ops should // either not be moved, or should only be moved if that state can also // be moved (e.g. via some sort of save / restore). // Stateful ops are guaranteed to never be optimized away by Common // Subexpression Elimination (CSE). IsStateful bool `protobuf:"varint,17,opt,name=is_stateful,json=isStateful,proto3" json:"is_stateful,omitempty"` // for things like variables, queue // By default, all inputs to an Op must be initialized Tensors. Ops // that may initialize tensors for the first time should set this // field to true, to allow the Op to take an uninitialized Tensor as // input. AllowsUninitializedInput bool `` // for Assign, etc. /* 137-byte string literal not displayed */ // contains filtered or unexported fields }
Defines an operation. A NodeDef in a GraphDef specifies an Op by using the "op" field which should match the name of a OpDef. LINT.IfChange
func (*OpDef) Descriptor
deprecated
func (*OpDef) GetAllowsUninitializedInput ¶
func (*OpDef) GetAttr ¶
func (x *OpDef) GetAttr() []*OpDef_AttrDef
func (*OpDef) GetDeprecation ¶
func (x *OpDef) GetDeprecation() *OpDeprecation
func (*OpDef) GetDescription ¶
func (*OpDef) GetInputArg ¶
func (x *OpDef) GetInputArg() []*OpDef_ArgDef
func (*OpDef) GetIsAggregate ¶
func (*OpDef) GetIsCommutative ¶
func (*OpDef) GetIsStateful ¶
func (*OpDef) GetOutputArg ¶
func (x *OpDef) GetOutputArg() []*OpDef_ArgDef
func (*OpDef) GetSummary ¶
func (*OpDef) ProtoMessage ¶
func (*OpDef) ProtoMessage()
func (*OpDef) ProtoReflect ¶ added in v0.12.0
func (x *OpDef) ProtoReflect() protoreflect.Message
type OpDef_ArgDef ¶
type OpDef_ArgDef struct { // Name for the input/output. Should match the regexp "[a-z][a-z0-9_]*". Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // Human readable description. Description string `protobuf:"bytes,2,opt,name=description,proto3" json:"description,omitempty"` // Describes the type of one or more tensors that are accepted/produced // by this input/output arg. The only legal combinations are: // - For a single tensor: either the "type" field is set or the // "type_attr" field is set to the name of an attr with type "type". // - For a sequence of tensors with the same type: the "number_attr" // field will be set to the name of an attr with type "int", and // either the "type" or "type_attr" field will be set as for // single tensors. // - For a sequence of tensors, the "type_list_attr" field will be set // to the name of an attr with type "list(type)". Type DataType `protobuf:"varint,3,opt,name=type,proto3,enum=tensorflow.DataType" json:"type,omitempty"` TypeAttr string `protobuf:"bytes,4,opt,name=type_attr,json=typeAttr,proto3" json:"type_attr,omitempty"` // if specified, attr must have type "type" NumberAttr string `protobuf:"bytes,5,opt,name=number_attr,json=numberAttr,proto3" json:"number_attr,omitempty"` // if specified, attr must have type "int" // If specified, attr must have type "list(type)", and none of // type, type_attr, and number_attr may be specified. TypeListAttr string `protobuf:"bytes,6,opt,name=type_list_attr,json=typeListAttr,proto3" json:"type_list_attr,omitempty"` // For inputs: if true, the inputs are required to be refs. // // By default, inputs can be either refs or non-refs. // // For outputs: if true, outputs are refs, otherwise they are not. IsRef bool `protobuf:"varint,16,opt,name=is_ref,json=isRef,proto3" json:"is_ref,omitempty"` // contains filtered or unexported fields }
For describing inputs and outputs.
func (*OpDef_ArgDef) Descriptor
deprecated
func (*OpDef_ArgDef) Descriptor() ([]byte, []int)
Deprecated: Use OpDef_ArgDef.ProtoReflect.Descriptor instead.
func (*OpDef_ArgDef) GetDescription ¶
func (x *OpDef_ArgDef) GetDescription() string
func (*OpDef_ArgDef) GetIsRef ¶
func (x *OpDef_ArgDef) GetIsRef() bool
func (*OpDef_ArgDef) GetName ¶
func (x *OpDef_ArgDef) GetName() string
func (*OpDef_ArgDef) GetNumberAttr ¶
func (x *OpDef_ArgDef) GetNumberAttr() string
func (*OpDef_ArgDef) GetType ¶
func (x *OpDef_ArgDef) GetType() DataType
func (*OpDef_ArgDef) GetTypeAttr ¶
func (x *OpDef_ArgDef) GetTypeAttr() string
func (*OpDef_ArgDef) GetTypeListAttr ¶
func (x *OpDef_ArgDef) GetTypeListAttr() string
func (*OpDef_ArgDef) ProtoMessage ¶
func (*OpDef_ArgDef) ProtoMessage()
func (*OpDef_ArgDef) ProtoReflect ¶ added in v0.12.0
func (x *OpDef_ArgDef) ProtoReflect() protoreflect.Message
func (*OpDef_ArgDef) Reset ¶
func (x *OpDef_ArgDef) Reset()
func (*OpDef_ArgDef) String ¶
func (x *OpDef_ArgDef) String() string
type OpDef_AttrDef ¶
type OpDef_AttrDef struct { // A descriptive name for the argument. May be used, e.g. by the // Python client, as a keyword argument name, and so should match // the regexp "[a-z][a-z0-9_]+". Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // One of the type names from attr_value.proto ("string", "list(string)", // "int", etc.). Type string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"` // A reasonable default for this attribute if the user does not supply // a value. If not specified, the user must supply a value. DefaultValue *AttrValue `protobuf:"bytes,3,opt,name=default_value,json=defaultValue,proto3" json:"default_value,omitempty"` // Human-readable description. Description string `protobuf:"bytes,4,opt,name=description,proto3" json:"description,omitempty"` // For type == "int", this is a minimum value. For "list(___)" // types, this is the minimum length. HasMinimum bool `protobuf:"varint,5,opt,name=has_minimum,json=hasMinimum,proto3" json:"has_minimum,omitempty"` Minimum int64 `protobuf:"varint,6,opt,name=minimum,proto3" json:"minimum,omitempty"` // The set of allowed values. Has type that is the "list" version // of the "type" field above (uses the "list" field of AttrValue). // If type == "type" or "list(type)" above, then the "type" field // of "allowed_values.list" has the set of allowed DataTypes. // If type == "string" or "list(string)", then the "s" field of // "allowed_values.list" has the set of allowed strings. AllowedValues *AttrValue `protobuf:"bytes,7,opt,name=allowed_values,json=allowedValues,proto3" json:"allowed_values,omitempty"` // contains filtered or unexported fields }
Description of the graph-construction-time configuration of this Op. That is to say, this describes the attr fields that will be specified in the NodeDef.
func (*OpDef_AttrDef) Descriptor
deprecated
func (*OpDef_AttrDef) Descriptor() ([]byte, []int)
Deprecated: Use OpDef_AttrDef.ProtoReflect.Descriptor instead.
func (*OpDef_AttrDef) GetAllowedValues ¶
func (x *OpDef_AttrDef) GetAllowedValues() *AttrValue
func (*OpDef_AttrDef) GetDefaultValue ¶
func (x *OpDef_AttrDef) GetDefaultValue() *AttrValue
func (*OpDef_AttrDef) GetDescription ¶
func (x *OpDef_AttrDef) GetDescription() string
func (*OpDef_AttrDef) GetHasMinimum ¶
func (x *OpDef_AttrDef) GetHasMinimum() bool
func (*OpDef_AttrDef) GetMinimum ¶
func (x *OpDef_AttrDef) GetMinimum() int64
func (*OpDef_AttrDef) GetName ¶
func (x *OpDef_AttrDef) GetName() string
func (*OpDef_AttrDef) GetType ¶
func (x *OpDef_AttrDef) GetType() string
func (*OpDef_AttrDef) ProtoMessage ¶
func (*OpDef_AttrDef) ProtoMessage()
func (*OpDef_AttrDef) ProtoReflect ¶ added in v0.12.0
func (x *OpDef_AttrDef) ProtoReflect() protoreflect.Message
func (*OpDef_AttrDef) Reset ¶
func (x *OpDef_AttrDef) Reset()
func (*OpDef_AttrDef) String ¶
func (x *OpDef_AttrDef) String() string
type OpDeprecation ¶
type OpDeprecation struct { // First GraphDef version at which the op is disallowed. Version int32 `protobuf:"varint,1,opt,name=version,proto3" json:"version,omitempty"` // Explanation of why it was deprecated and what to use instead. Explanation string `protobuf:"bytes,2,opt,name=explanation,proto3" json:"explanation,omitempty"` // contains filtered or unexported fields }
Information about version-dependent deprecation of an op
func (*OpDeprecation) Descriptor
deprecated
func (*OpDeprecation) Descriptor() ([]byte, []int)
Deprecated: Use OpDeprecation.ProtoReflect.Descriptor instead.
func (*OpDeprecation) GetExplanation ¶
func (x *OpDeprecation) GetExplanation() string
func (*OpDeprecation) GetVersion ¶
func (x *OpDeprecation) GetVersion() int32
func (*OpDeprecation) ProtoMessage ¶
func (*OpDeprecation) ProtoMessage()
func (*OpDeprecation) ProtoReflect ¶ added in v0.12.0
func (x *OpDeprecation) ProtoReflect() protoreflect.Message
func (*OpDeprecation) Reset ¶
func (x *OpDeprecation) Reset()
func (*OpDeprecation) String ¶
func (x *OpDeprecation) String() string
type OpList ¶
type OpList struct { Op []*OpDef `protobuf:"bytes,1,rep,name=op,proto3" json:"op,omitempty"` // contains filtered or unexported fields }
A collection of OpDefs
func (*OpList) Descriptor
deprecated
func (*OpList) ProtoMessage ¶
func (*OpList) ProtoMessage()
func (*OpList) ProtoReflect ¶ added in v0.12.0
func (x *OpList) ProtoReflect() protoreflect.Message
type ReaderBaseState ¶
type ReaderBaseState struct { WorkStarted int64 `protobuf:"varint,1,opt,name=work_started,json=workStarted,proto3" json:"work_started,omitempty"` WorkFinished int64 `protobuf:"varint,2,opt,name=work_finished,json=workFinished,proto3" json:"work_finished,omitempty"` NumRecordsProduced int64 `protobuf:"varint,3,opt,name=num_records_produced,json=numRecordsProduced,proto3" json:"num_records_produced,omitempty"` CurrentWork []byte `protobuf:"bytes,4,opt,name=current_work,json=currentWork,proto3" json:"current_work,omitempty"` // contains filtered or unexported fields }
For serializing and restoring the state of ReaderBase, see reader_base.h for details.
func (*ReaderBaseState) Descriptor
deprecated
func (*ReaderBaseState) Descriptor() ([]byte, []int)
Deprecated: Use ReaderBaseState.ProtoReflect.Descriptor instead.
func (*ReaderBaseState) GetCurrentWork ¶
func (x *ReaderBaseState) GetCurrentWork() []byte
func (*ReaderBaseState) GetNumRecordsProduced ¶
func (x *ReaderBaseState) GetNumRecordsProduced() int64
func (*ReaderBaseState) GetWorkFinished ¶
func (x *ReaderBaseState) GetWorkFinished() int64
func (*ReaderBaseState) GetWorkStarted ¶
func (x *ReaderBaseState) GetWorkStarted() int64
func (*ReaderBaseState) ProtoMessage ¶
func (*ReaderBaseState) ProtoMessage()
func (*ReaderBaseState) ProtoReflect ¶ added in v0.12.0
func (x *ReaderBaseState) ProtoReflect() protoreflect.Message
func (*ReaderBaseState) Reset ¶
func (x *ReaderBaseState) Reset()
func (*ReaderBaseState) String ¶
func (x *ReaderBaseState) String() string
type RemoteFusedGraphExecuteInfo ¶
type RemoteFusedGraphExecuteInfo struct { // Definition of remote graph RemoteGraph *GraphDef `protobuf:"bytes,1,opt,name=remote_graph,json=remoteGraph,proto3" json:"remote_graph,omitempty"` // Remote fused graph input node name GraphInputNodeName []string `protobuf:"bytes,2,rep,name=graph_input_node_name,json=graphInputNodeName,proto3" json:"graph_input_node_name,omitempty"` // Remote fused graph output node name GraphOutputNodeName []string `protobuf:"bytes,3,rep,name=graph_output_node_name,json=graphOutputNodeName,proto3" json:"graph_output_node_name,omitempty"` // Executor's name ExecutorName string `protobuf:"bytes,4,opt,name=executor_name,json=executorName,proto3" json:"executor_name,omitempty"` // Optional: Parameters given to the executor SerializedExecutorParameters []byte `` /* 147-byte string literal not displayed */ // Optional: Default graph input tensor shape used to allocate memory // before executing op DefaultGraphInputTensorShape []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto `` /* 151-byte string literal not displayed */ // Optional: Default graph input tensor shape used to allocate memory // before executing op // TODO(satok): Remote output tensor shape once shape information is stored // in NodeDef DefaultGraphOutputTensorShape []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto `` /* 154-byte string literal not displayed */ // contains filtered or unexported fields }
Protocol buffer representing a handle to a tensorflow resource. Handles are not valid across executions, but can be serialized back and forth from within a single run.
func (*RemoteFusedGraphExecuteInfo) Descriptor
deprecated
func (*RemoteFusedGraphExecuteInfo) Descriptor() ([]byte, []int)
Deprecated: Use RemoteFusedGraphExecuteInfo.ProtoReflect.Descriptor instead.
func (*RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape ¶
func (x *RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
func (*RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape ¶
func (x *RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
func (*RemoteFusedGraphExecuteInfo) GetExecutorName ¶
func (x *RemoteFusedGraphExecuteInfo) GetExecutorName() string
func (*RemoteFusedGraphExecuteInfo) GetGraphInputNodeName ¶
func (x *RemoteFusedGraphExecuteInfo) GetGraphInputNodeName() []string
func (*RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName ¶
func (x *RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName() []string
func (*RemoteFusedGraphExecuteInfo) GetRemoteGraph ¶
func (x *RemoteFusedGraphExecuteInfo) GetRemoteGraph() *GraphDef
func (*RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters ¶
func (x *RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters() []byte
func (*RemoteFusedGraphExecuteInfo) ProtoMessage ¶
func (*RemoteFusedGraphExecuteInfo) ProtoMessage()
func (*RemoteFusedGraphExecuteInfo) ProtoReflect ¶ added in v0.12.0
func (x *RemoteFusedGraphExecuteInfo) ProtoReflect() protoreflect.Message
func (*RemoteFusedGraphExecuteInfo) Reset ¶
func (x *RemoteFusedGraphExecuteInfo) Reset()
func (*RemoteFusedGraphExecuteInfo) String ¶
func (x *RemoteFusedGraphExecuteInfo) String() string
type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto ¶
type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto struct { Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"` // contains filtered or unexported fields }
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor
deprecated
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor() ([]byte, []int)
Deprecated: Use RemoteFusedGraphExecuteInfo_TensorShapeTypeProto.ProtoReflect.Descriptor instead.
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype ¶
func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype() DataType
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape ¶
func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape() *TensorShapeProto
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage ¶
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage()
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoReflect ¶ added in v0.12.0
func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoReflect() protoreflect.Message
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset ¶
func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset()
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) String ¶
func (x *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) String() string
type ResourceHandleProto ¶
type ResourceHandleProto struct { // Unique name for the device containing the resource. Device string `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"` // Container in which this resource is placed. Container string `protobuf:"bytes,2,opt,name=container,proto3" json:"container,omitempty"` // Unique name of this resource. Name string `protobuf:"bytes,3,opt,name=name,proto3" json:"name,omitempty"` // Hash code for the type of the resource. Is only valid in the same device // and in the same execution. HashCode uint64 `protobuf:"varint,4,opt,name=hash_code,json=hashCode,proto3" json:"hash_code,omitempty"` // For debug-only, the name of the type pointed to by this handle, if // available. MaybeTypeName string `protobuf:"bytes,5,opt,name=maybe_type_name,json=maybeTypeName,proto3" json:"maybe_type_name,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing a handle to a tensorflow resource. Handles are not valid across executions, but can be serialized back and forth from within a single run.
func (*ResourceHandleProto) Descriptor
deprecated
func (*ResourceHandleProto) Descriptor() ([]byte, []int)
Deprecated: Use ResourceHandleProto.ProtoReflect.Descriptor instead.
func (*ResourceHandleProto) GetContainer ¶
func (x *ResourceHandleProto) GetContainer() string
func (*ResourceHandleProto) GetDevice ¶
func (x *ResourceHandleProto) GetDevice() string
func (*ResourceHandleProto) GetHashCode ¶
func (x *ResourceHandleProto) GetHashCode() uint64
func (*ResourceHandleProto) GetMaybeTypeName ¶
func (x *ResourceHandleProto) GetMaybeTypeName() string
func (*ResourceHandleProto) GetName ¶
func (x *ResourceHandleProto) GetName() string
func (*ResourceHandleProto) ProtoMessage ¶
func (*ResourceHandleProto) ProtoMessage()
func (*ResourceHandleProto) ProtoReflect ¶ added in v0.12.0
func (x *ResourceHandleProto) ProtoReflect() protoreflect.Message
func (*ResourceHandleProto) Reset ¶
func (x *ResourceHandleProto) Reset()
func (*ResourceHandleProto) String ¶
func (x *ResourceHandleProto) String() string
type SaveSliceInfoDef ¶
type SaveSliceInfoDef struct { // Name of the full variable of which this is a slice. FullName string `protobuf:"bytes,1,opt,name=full_name,json=fullName,proto3" json:"full_name,omitempty"` // Shape of the full variable. FullShape []int64 `protobuf:"varint,2,rep,packed,name=full_shape,json=fullShape,proto3" json:"full_shape,omitempty"` // Offset of this variable into the full variable. VarOffset []int64 `protobuf:"varint,3,rep,packed,name=var_offset,json=varOffset,proto3" json:"var_offset,omitempty"` // Shape of this variable. VarShape []int64 `protobuf:"varint,4,rep,packed,name=var_shape,json=varShape,proto3" json:"var_shape,omitempty"` // contains filtered or unexported fields }
func (*SaveSliceInfoDef) Descriptor
deprecated
func (*SaveSliceInfoDef) Descriptor() ([]byte, []int)
Deprecated: Use SaveSliceInfoDef.ProtoReflect.Descriptor instead.
func (*SaveSliceInfoDef) GetFullName ¶
func (x *SaveSliceInfoDef) GetFullName() string
func (*SaveSliceInfoDef) GetFullShape ¶
func (x *SaveSliceInfoDef) GetFullShape() []int64
func (*SaveSliceInfoDef) GetVarOffset ¶
func (x *SaveSliceInfoDef) GetVarOffset() []int64
func (*SaveSliceInfoDef) GetVarShape ¶
func (x *SaveSliceInfoDef) GetVarShape() []int64
func (*SaveSliceInfoDef) ProtoMessage ¶
func (*SaveSliceInfoDef) ProtoMessage()
func (*SaveSliceInfoDef) ProtoReflect ¶ added in v0.12.0
func (x *SaveSliceInfoDef) ProtoReflect() protoreflect.Message
func (*SaveSliceInfoDef) Reset ¶
func (x *SaveSliceInfoDef) Reset()
func (*SaveSliceInfoDef) String ¶
func (x *SaveSliceInfoDef) String() string
type StepStats ¶
type StepStats struct { DevStats []*DeviceStepStats `protobuf:"bytes,1,rep,name=dev_stats,json=devStats,proto3" json:"dev_stats,omitempty"` // contains filtered or unexported fields }
func (*StepStats) Descriptor
deprecated
func (*StepStats) GetDevStats ¶
func (x *StepStats) GetDevStats() []*DeviceStepStats
func (*StepStats) ProtoMessage ¶
func (*StepStats) ProtoMessage()
func (*StepStats) ProtoReflect ¶ added in v0.12.0
func (x *StepStats) ProtoReflect() protoreflect.Message
type Summary ¶
type Summary struct { // Set of values for the summary. Value []*Summary_Value `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"` // contains filtered or unexported fields }
A Summary is a set of named values to be displayed by the visualizer.
Summaries are produced regularly during training, as controlled by the "summary_interval_secs" attribute of the training operation. Summaries are also produced at the end of an evaluation.
func (*Summary) Descriptor
deprecated
func (*Summary) GetValue ¶
func (x *Summary) GetValue() []*Summary_Value
func (*Summary) ProtoMessage ¶
func (*Summary) ProtoMessage()
func (*Summary) ProtoReflect ¶ added in v0.12.0
func (x *Summary) ProtoReflect() protoreflect.Message
type SummaryDescription ¶
type SummaryDescription struct { // Hint on how plugins should process the data in this series. // Supported values include "scalar", "histogram", "image", "audio" TypeHint string `protobuf:"bytes,1,opt,name=type_hint,json=typeHint,proto3" json:"type_hint,omitempty"` // contains filtered or unexported fields }
Metadata associated with a series of Summary data
func (*SummaryDescription) Descriptor
deprecated
func (*SummaryDescription) Descriptor() ([]byte, []int)
Deprecated: Use SummaryDescription.ProtoReflect.Descriptor instead.
func (*SummaryDescription) GetTypeHint ¶
func (x *SummaryDescription) GetTypeHint() string
func (*SummaryDescription) ProtoMessage ¶
func (*SummaryDescription) ProtoMessage()
func (*SummaryDescription) ProtoReflect ¶ added in v0.12.0
func (x *SummaryDescription) ProtoReflect() protoreflect.Message
func (*SummaryDescription) Reset ¶
func (x *SummaryDescription) Reset()
func (*SummaryDescription) String ¶
func (x *SummaryDescription) String() string
type SummaryMetadata ¶
type SummaryMetadata struct { // Data that associates a summary with a certain plugin. PluginData *SummaryMetadata_PluginData `protobuf:"bytes,1,opt,name=plugin_data,json=pluginData,proto3" json:"plugin_data,omitempty"` // Display name for viewing in TensorBoard. DisplayName string `protobuf:"bytes,2,opt,name=display_name,json=displayName,proto3" json:"display_name,omitempty"` // Longform readable description of the summary sequence. Markdown supported. SummaryDescription string `protobuf:"bytes,3,opt,name=summary_description,json=summaryDescription,proto3" json:"summary_description,omitempty"` // contains filtered or unexported fields }
A SummaryMetadata encapsulates information on which plugins are able to make use of a certain summary value.
func (*SummaryMetadata) Descriptor
deprecated
func (*SummaryMetadata) Descriptor() ([]byte, []int)
Deprecated: Use SummaryMetadata.ProtoReflect.Descriptor instead.
func (*SummaryMetadata) GetDisplayName ¶
func (x *SummaryMetadata) GetDisplayName() string
func (*SummaryMetadata) GetPluginData ¶
func (x *SummaryMetadata) GetPluginData() *SummaryMetadata_PluginData
func (*SummaryMetadata) GetSummaryDescription ¶
func (x *SummaryMetadata) GetSummaryDescription() string
func (*SummaryMetadata) ProtoMessage ¶
func (*SummaryMetadata) ProtoMessage()
func (*SummaryMetadata) ProtoReflect ¶ added in v0.12.0
func (x *SummaryMetadata) ProtoReflect() protoreflect.Message
func (*SummaryMetadata) Reset ¶
func (x *SummaryMetadata) Reset()
func (*SummaryMetadata) String ¶
func (x *SummaryMetadata) String() string
type SummaryMetadata_PluginData ¶
type SummaryMetadata_PluginData struct { // The name of the plugin this data pertains to. PluginName string `protobuf:"bytes,1,opt,name=plugin_name,json=pluginName,proto3" json:"plugin_name,omitempty"` // The content to store for the plugin. The best practice is for this to be // a binary serialized protocol buffer. Content []byte `protobuf:"bytes,2,opt,name=content,proto3" json:"content,omitempty"` // contains filtered or unexported fields }
func (*SummaryMetadata_PluginData) Descriptor
deprecated
func (*SummaryMetadata_PluginData) Descriptor() ([]byte, []int)
Deprecated: Use SummaryMetadata_PluginData.ProtoReflect.Descriptor instead.
func (*SummaryMetadata_PluginData) GetContent ¶
func (x *SummaryMetadata_PluginData) GetContent() []byte
func (*SummaryMetadata_PluginData) GetPluginName ¶
func (x *SummaryMetadata_PluginData) GetPluginName() string
func (*SummaryMetadata_PluginData) ProtoMessage ¶
func (*SummaryMetadata_PluginData) ProtoMessage()
func (*SummaryMetadata_PluginData) ProtoReflect ¶ added in v0.12.0
func (x *SummaryMetadata_PluginData) ProtoReflect() protoreflect.Message
func (*SummaryMetadata_PluginData) Reset ¶
func (x *SummaryMetadata_PluginData) Reset()
func (*SummaryMetadata_PluginData) String ¶
func (x *SummaryMetadata_PluginData) String() string
type Summary_Audio ¶
type Summary_Audio struct { // Sample rate of the audio in Hz. SampleRate float32 `protobuf:"fixed32,1,opt,name=sample_rate,json=sampleRate,proto3" json:"sample_rate,omitempty"` // Number of channels of audio. NumChannels int64 `protobuf:"varint,2,opt,name=num_channels,json=numChannels,proto3" json:"num_channels,omitempty"` // Length of the audio in frames (samples per channel). LengthFrames int64 `protobuf:"varint,3,opt,name=length_frames,json=lengthFrames,proto3" json:"length_frames,omitempty"` // Encoded audio data and its associated RFC 2045 content type (e.g. // "audio/wav"). EncodedAudioString []byte `protobuf:"bytes,4,opt,name=encoded_audio_string,json=encodedAudioString,proto3" json:"encoded_audio_string,omitempty"` ContentType string `protobuf:"bytes,5,opt,name=content_type,json=contentType,proto3" json:"content_type,omitempty"` // contains filtered or unexported fields }
func (*Summary_Audio) Descriptor
deprecated
func (*Summary_Audio) Descriptor() ([]byte, []int)
Deprecated: Use Summary_Audio.ProtoReflect.Descriptor instead.
func (*Summary_Audio) GetContentType ¶
func (x *Summary_Audio) GetContentType() string
func (*Summary_Audio) GetEncodedAudioString ¶
func (x *Summary_Audio) GetEncodedAudioString() []byte
func (*Summary_Audio) GetLengthFrames ¶
func (x *Summary_Audio) GetLengthFrames() int64
func (*Summary_Audio) GetNumChannels ¶
func (x *Summary_Audio) GetNumChannels() int64
func (*Summary_Audio) GetSampleRate ¶
func (x *Summary_Audio) GetSampleRate() float32
func (*Summary_Audio) ProtoMessage ¶
func (*Summary_Audio) ProtoMessage()
func (*Summary_Audio) ProtoReflect ¶ added in v0.12.0
func (x *Summary_Audio) ProtoReflect() protoreflect.Message
func (*Summary_Audio) Reset ¶
func (x *Summary_Audio) Reset()
func (*Summary_Audio) String ¶
func (x *Summary_Audio) String() string
type Summary_Image ¶
type Summary_Image struct { // Dimensions of the image. Height int32 `protobuf:"varint,1,opt,name=height,proto3" json:"height,omitempty"` Width int32 `protobuf:"varint,2,opt,name=width,proto3" json:"width,omitempty"` // Valid colorspace values are // // 1 - grayscale // 2 - grayscale + alpha // 3 - RGB // 4 - RGBA // 5 - DIGITAL_YUV // 6 - BGRA Colorspace int32 `protobuf:"varint,3,opt,name=colorspace,proto3" json:"colorspace,omitempty"` // Image data in encoded format. All image formats supported by // image_codec::CoderUtil can be stored here. EncodedImageString []byte `protobuf:"bytes,4,opt,name=encoded_image_string,json=encodedImageString,proto3" json:"encoded_image_string,omitempty"` // contains filtered or unexported fields }
func (*Summary_Image) Descriptor
deprecated
func (*Summary_Image) Descriptor() ([]byte, []int)
Deprecated: Use Summary_Image.ProtoReflect.Descriptor instead.
func (*Summary_Image) GetColorspace ¶
func (x *Summary_Image) GetColorspace() int32
func (*Summary_Image) GetEncodedImageString ¶
func (x *Summary_Image) GetEncodedImageString() []byte
func (*Summary_Image) GetHeight ¶
func (x *Summary_Image) GetHeight() int32
func (*Summary_Image) GetWidth ¶
func (x *Summary_Image) GetWidth() int32
func (*Summary_Image) ProtoMessage ¶
func (*Summary_Image) ProtoMessage()
func (*Summary_Image) ProtoReflect ¶ added in v0.12.0
func (x *Summary_Image) ProtoReflect() protoreflect.Message
func (*Summary_Image) Reset ¶
func (x *Summary_Image) Reset()
func (*Summary_Image) String ¶
func (x *Summary_Image) String() string
type Summary_Value ¶
type Summary_Value struct { // This field is deprecated and will not be set. NodeName string `protobuf:"bytes,7,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"` // Tag name for the data. Used by TensorBoard plugins to organize data. Tags // are often organized by scope (which contains slashes to convey // hierarchy). For example: foo/bar/0 Tag string `protobuf:"bytes,1,opt,name=tag,proto3" json:"tag,omitempty"` // Contains metadata on the summary value such as which plugins may use it. // Take note that many summary values may lack a metadata field. This is // because the FileWriter only keeps a metadata object on the first summary // value with a certain tag for each tag. TensorBoard then remembers which // tags are associated with which plugins. This saves space. Metadata *SummaryMetadata `protobuf:"bytes,9,opt,name=metadata,proto3" json:"metadata,omitempty"` // Value associated with the tag. // // Types that are assignable to Value: // // *Summary_Value_SimpleValue // *Summary_Value_ObsoleteOldStyleHistogram // *Summary_Value_Image // *Summary_Value_Histo // *Summary_Value_Audio // *Summary_Value_Tensor Value isSummary_Value_Value `protobuf_oneof:"value"` // contains filtered or unexported fields }
func (*Summary_Value) Descriptor
deprecated
func (*Summary_Value) Descriptor() ([]byte, []int)
Deprecated: Use Summary_Value.ProtoReflect.Descriptor instead.
func (*Summary_Value) GetAudio ¶
func (x *Summary_Value) GetAudio() *Summary_Audio
func (*Summary_Value) GetHisto ¶
func (x *Summary_Value) GetHisto() *HistogramProto
func (*Summary_Value) GetImage ¶
func (x *Summary_Value) GetImage() *Summary_Image
func (*Summary_Value) GetMetadata ¶
func (x *Summary_Value) GetMetadata() *SummaryMetadata
func (*Summary_Value) GetNodeName ¶
func (x *Summary_Value) GetNodeName() string
func (*Summary_Value) GetObsoleteOldStyleHistogram ¶
func (x *Summary_Value) GetObsoleteOldStyleHistogram() []byte
func (*Summary_Value) GetSimpleValue ¶
func (x *Summary_Value) GetSimpleValue() float32
func (*Summary_Value) GetTag ¶
func (x *Summary_Value) GetTag() string
func (*Summary_Value) GetTensor ¶
func (x *Summary_Value) GetTensor() *TensorProto
func (*Summary_Value) GetValue ¶
func (m *Summary_Value) GetValue() isSummary_Value_Value
func (*Summary_Value) ProtoMessage ¶
func (*Summary_Value) ProtoMessage()
func (*Summary_Value) ProtoReflect ¶ added in v0.12.0
func (x *Summary_Value) ProtoReflect() protoreflect.Message
func (*Summary_Value) Reset ¶
func (x *Summary_Value) Reset()
func (*Summary_Value) String ¶
func (x *Summary_Value) String() string
type Summary_Value_Audio ¶
type Summary_Value_Audio struct {
Audio *Summary_Audio `protobuf:"bytes,6,opt,name=audio,proto3,oneof"`
}
type Summary_Value_Histo ¶
type Summary_Value_Histo struct {
Histo *HistogramProto `protobuf:"bytes,5,opt,name=histo,proto3,oneof"`
}
type Summary_Value_Image ¶
type Summary_Value_Image struct {
Image *Summary_Image `protobuf:"bytes,4,opt,name=image,proto3,oneof"`
}
type Summary_Value_ObsoleteOldStyleHistogram ¶
type Summary_Value_ObsoleteOldStyleHistogram struct {
ObsoleteOldStyleHistogram []byte `protobuf:"bytes,3,opt,name=obsolete_old_style_histogram,json=obsoleteOldStyleHistogram,proto3,oneof"`
}
type Summary_Value_SimpleValue ¶
type Summary_Value_SimpleValue struct {
SimpleValue float32 `protobuf:"fixed32,2,opt,name=simple_value,json=simpleValue,proto3,oneof"`
}
type Summary_Value_Tensor ¶
type Summary_Value_Tensor struct {
Tensor *TensorProto `protobuf:"bytes,8,opt,name=tensor,proto3,oneof"`
}
type TensorDescription ¶
type TensorDescription struct { // Data type of tensor elements Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` // Shape of the tensor. Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"` // Information about the size and allocator used for the data AllocationDescription *AllocationDescription `protobuf:"bytes,4,opt,name=allocation_description,json=allocationDescription,proto3" json:"allocation_description,omitempty"` // contains filtered or unexported fields }
func (*TensorDescription) Descriptor
deprecated
func (*TensorDescription) Descriptor() ([]byte, []int)
Deprecated: Use TensorDescription.ProtoReflect.Descriptor instead.
func (*TensorDescription) GetAllocationDescription ¶
func (x *TensorDescription) GetAllocationDescription() *AllocationDescription
func (*TensorDescription) GetDtype ¶
func (x *TensorDescription) GetDtype() DataType
func (*TensorDescription) GetShape ¶
func (x *TensorDescription) GetShape() *TensorShapeProto
func (*TensorDescription) ProtoMessage ¶
func (*TensorDescription) ProtoMessage()
func (*TensorDescription) ProtoReflect ¶ added in v0.12.0
func (x *TensorDescription) ProtoReflect() protoreflect.Message
func (*TensorDescription) Reset ¶
func (x *TensorDescription) Reset()
func (*TensorDescription) String ¶
func (x *TensorDescription) String() string
type TensorProto ¶
type TensorProto struct { Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` // Shape of the tensor. TODO(touts): sort out the 0-rank issues. TensorShape *TensorShapeProto `protobuf:"bytes,2,opt,name=tensor_shape,json=tensorShape,proto3" json:"tensor_shape,omitempty"` // Version number. // // In version 0, if the "repeated xxx" representations contain only one // element, that element is repeated to fill the shape. This makes it easy // to represent a constant Tensor with a single value. VersionNumber int32 `protobuf:"varint,3,opt,name=version_number,json=versionNumber,proto3" json:"version_number,omitempty"` // Serialized raw tensor content from either Tensor::AsProtoTensorContent or // memcpy in tensorflow::grpc::EncodeTensorToByteBuffer. This representation // can be used for all tensor types. The purpose of this representation is to // reduce serialization overhead during RPC call by avoiding serialization of // many repeated small items. TensorContent []byte `protobuf:"bytes,4,opt,name=tensor_content,json=tensorContent,proto3" json:"tensor_content,omitempty"` // DT_HALF, DT_BFLOAT16. Note that since protobuf has no int16 type, we'll // have some pointless zero padding for each value here. HalfVal []int32 `protobuf:"varint,13,rep,packed,name=half_val,json=halfVal,proto3" json:"half_val,omitempty"` // DT_FLOAT. FloatVal []float32 `protobuf:"fixed32,5,rep,packed,name=float_val,json=floatVal,proto3" json:"float_val,omitempty"` // DT_DOUBLE. DoubleVal []float64 `protobuf:"fixed64,6,rep,packed,name=double_val,json=doubleVal,proto3" json:"double_val,omitempty"` // DT_INT32, DT_INT16, DT_INT8, DT_UINT8. IntVal []int32 `protobuf:"varint,7,rep,packed,name=int_val,json=intVal,proto3" json:"int_val,omitempty"` // DT_STRING StringVal [][]byte `protobuf:"bytes,8,rep,name=string_val,json=stringVal,proto3" json:"string_val,omitempty"` // DT_COMPLEX64. scomplex_val(2*i) and scomplex_val(2*i+1) are real // and imaginary parts of i-th single precision complex. ScomplexVal []float32 `protobuf:"fixed32,9,rep,packed,name=scomplex_val,json=scomplexVal,proto3" json:"scomplex_val,omitempty"` // DT_INT64 Int64Val []int64 `protobuf:"varint,10,rep,packed,name=int64_val,json=int64Val,proto3" json:"int64_val,omitempty"` // DT_BOOL BoolVal []bool `protobuf:"varint,11,rep,packed,name=bool_val,json=boolVal,proto3" json:"bool_val,omitempty"` // DT_COMPLEX128. dcomplex_val(2*i) and dcomplex_val(2*i+1) are real // and imaginary parts of i-th double precision complex. DcomplexVal []float64 `protobuf:"fixed64,12,rep,packed,name=dcomplex_val,json=dcomplexVal,proto3" json:"dcomplex_val,omitempty"` // DT_RESOURCE ResourceHandleVal []*ResourceHandleProto `protobuf:"bytes,14,rep,name=resource_handle_val,json=resourceHandleVal,proto3" json:"resource_handle_val,omitempty"` // DT_VARIANT VariantVal []*VariantTensorDataProto `protobuf:"bytes,15,rep,name=variant_val,json=variantVal,proto3" json:"variant_val,omitempty"` // DT_UINT32 Uint32Val []uint32 `protobuf:"varint,16,rep,packed,name=uint32_val,json=uint32Val,proto3" json:"uint32_val,omitempty"` // DT_UINT64 Uint64Val []uint64 `protobuf:"varint,17,rep,packed,name=uint64_val,json=uint64Val,proto3" json:"uint64_val,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing a tensor.
func (*TensorProto) Descriptor
deprecated
func (*TensorProto) Descriptor() ([]byte, []int)
Deprecated: Use TensorProto.ProtoReflect.Descriptor instead.
func (*TensorProto) GetBoolVal ¶
func (x *TensorProto) GetBoolVal() []bool
func (*TensorProto) GetDcomplexVal ¶
func (x *TensorProto) GetDcomplexVal() []float64
func (*TensorProto) GetDoubleVal ¶
func (x *TensorProto) GetDoubleVal() []float64
func (*TensorProto) GetDtype ¶
func (x *TensorProto) GetDtype() DataType
func (*TensorProto) GetFloatVal ¶
func (x *TensorProto) GetFloatVal() []float32
func (*TensorProto) GetHalfVal ¶
func (x *TensorProto) GetHalfVal() []int32
func (*TensorProto) GetInt64Val ¶
func (x *TensorProto) GetInt64Val() []int64
func (*TensorProto) GetIntVal ¶
func (x *TensorProto) GetIntVal() []int32
func (*TensorProto) GetResourceHandleVal ¶
func (x *TensorProto) GetResourceHandleVal() []*ResourceHandleProto
func (*TensorProto) GetScomplexVal ¶
func (x *TensorProto) GetScomplexVal() []float32
func (*TensorProto) GetStringVal ¶
func (x *TensorProto) GetStringVal() [][]byte
func (*TensorProto) GetTensorContent ¶
func (x *TensorProto) GetTensorContent() []byte
func (*TensorProto) GetTensorShape ¶
func (x *TensorProto) GetTensorShape() *TensorShapeProto
func (*TensorProto) GetUint32Val ¶
func (x *TensorProto) GetUint32Val() []uint32
func (*TensorProto) GetUint64Val ¶
func (x *TensorProto) GetUint64Val() []uint64
func (*TensorProto) GetVariantVal ¶
func (x *TensorProto) GetVariantVal() []*VariantTensorDataProto
func (*TensorProto) GetVersionNumber ¶
func (x *TensorProto) GetVersionNumber() int32
func (*TensorProto) ProtoMessage ¶
func (*TensorProto) ProtoMessage()
func (*TensorProto) ProtoReflect ¶ added in v0.12.0
func (x *TensorProto) ProtoReflect() protoreflect.Message
func (*TensorProto) Reset ¶
func (x *TensorProto) Reset()
func (*TensorProto) String ¶
func (x *TensorProto) String() string
type TensorShapeProto ¶
type TensorShapeProto struct { // Dimensions of the tensor, such as {"input", 30}, {"output", 40} // for a 30 x 40 2D tensor. If an entry has size -1, this // corresponds to a dimension of unknown size. The names are // optional. // // The order of entries in "dim" matters: It indicates the layout of the // values in the tensor in-memory representation. // // The first entry in "dim" is the outermost dimension used to layout the // values, the last entry is the innermost dimension. This matches the // in-memory layout of RowMajor Eigen tensors. // // If "dim.size()" > 0, "unknown_rank" must be false. Dim []*TensorShapeProto_Dim `protobuf:"bytes,2,rep,name=dim,proto3" json:"dim,omitempty"` // If true, the number of dimensions in the shape is unknown. // // If true, "dim.size()" must be 0. UnknownRank bool `protobuf:"varint,3,opt,name=unknown_rank,json=unknownRank,proto3" json:"unknown_rank,omitempty"` // contains filtered or unexported fields }
Dimensions of a tensor.
func (*TensorShapeProto) Descriptor
deprecated
func (*TensorShapeProto) Descriptor() ([]byte, []int)
Deprecated: Use TensorShapeProto.ProtoReflect.Descriptor instead.
func (*TensorShapeProto) GetDim ¶
func (x *TensorShapeProto) GetDim() []*TensorShapeProto_Dim
func (*TensorShapeProto) GetUnknownRank ¶
func (x *TensorShapeProto) GetUnknownRank() bool
func (*TensorShapeProto) ProtoMessage ¶
func (*TensorShapeProto) ProtoMessage()
func (*TensorShapeProto) ProtoReflect ¶ added in v0.12.0
func (x *TensorShapeProto) ProtoReflect() protoreflect.Message
func (*TensorShapeProto) Reset ¶
func (x *TensorShapeProto) Reset()
func (*TensorShapeProto) String ¶
func (x *TensorShapeProto) String() string
type TensorShapeProto_Dim ¶
type TensorShapeProto_Dim struct { // Size of the tensor in that dimension. // This value must be >= -1, but values of -1 are reserved for "unknown" // shapes (values of -1 mean "unknown" dimension). Certain wrappers // that work with TensorShapeProto may fail at runtime when deserializing // a TensorShapeProto containing a dim value of -1. Size int64 `protobuf:"varint,1,opt,name=size,proto3" json:"size,omitempty"` // Optional name of the tensor dimension. Name string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"` // contains filtered or unexported fields }
One dimension of the tensor.
func (*TensorShapeProto_Dim) Descriptor
deprecated
func (*TensorShapeProto_Dim) Descriptor() ([]byte, []int)
Deprecated: Use TensorShapeProto_Dim.ProtoReflect.Descriptor instead.
func (*TensorShapeProto_Dim) GetName ¶
func (x *TensorShapeProto_Dim) GetName() string
func (*TensorShapeProto_Dim) GetSize ¶
func (x *TensorShapeProto_Dim) GetSize() int64
func (*TensorShapeProto_Dim) ProtoMessage ¶
func (*TensorShapeProto_Dim) ProtoMessage()
func (*TensorShapeProto_Dim) ProtoReflect ¶ added in v0.12.0
func (x *TensorShapeProto_Dim) ProtoReflect() protoreflect.Message
func (*TensorShapeProto_Dim) Reset ¶
func (x *TensorShapeProto_Dim) Reset()
func (*TensorShapeProto_Dim) String ¶
func (x *TensorShapeProto_Dim) String() string
type TensorSliceProto ¶
type TensorSliceProto struct { // Extent of the slice in all tensor dimensions. // // Must have one entry for each of the dimension of the tensor that this // slice belongs to. The order of sizes is the same as the order of // dimensions in the TensorShape. Extent []*TensorSliceProto_Extent `protobuf:"bytes,1,rep,name=extent,proto3" json:"extent,omitempty"` // contains filtered or unexported fields }
Can only be interpreted if you know the corresponding TensorShape.
func (*TensorSliceProto) Descriptor
deprecated
func (*TensorSliceProto) Descriptor() ([]byte, []int)
Deprecated: Use TensorSliceProto.ProtoReflect.Descriptor instead.
func (*TensorSliceProto) GetExtent ¶
func (x *TensorSliceProto) GetExtent() []*TensorSliceProto_Extent
func (*TensorSliceProto) ProtoMessage ¶
func (*TensorSliceProto) ProtoMessage()
func (*TensorSliceProto) ProtoReflect ¶ added in v0.12.0
func (x *TensorSliceProto) ProtoReflect() protoreflect.Message
func (*TensorSliceProto) Reset ¶
func (x *TensorSliceProto) Reset()
func (*TensorSliceProto) String ¶
func (x *TensorSliceProto) String() string
type TensorSliceProto_Extent ¶
type TensorSliceProto_Extent struct { // Start index of the slice, starting at 0. Start int64 `protobuf:"varint,1,opt,name=start,proto3" json:"start,omitempty"` // Length of the slice: if the length is missing or -1 we will // interpret this as "everything in this dimension". We use // "oneof" to preserve information about whether the length is // present without changing the serialization format from the // prior proto2 version of this proto. // // Types that are assignable to HasLength: // // *TensorSliceProto_Extent_Length HasLength isTensorSliceProto_Extent_HasLength `protobuf_oneof:"has_length"` // contains filtered or unexported fields }
Extent of the slice in one dimension.
func (*TensorSliceProto_Extent) Descriptor
deprecated
func (*TensorSliceProto_Extent) Descriptor() ([]byte, []int)
Deprecated: Use TensorSliceProto_Extent.ProtoReflect.Descriptor instead.
func (*TensorSliceProto_Extent) GetHasLength ¶
func (m *TensorSliceProto_Extent) GetHasLength() isTensorSliceProto_Extent_HasLength
func (*TensorSliceProto_Extent) GetLength ¶
func (x *TensorSliceProto_Extent) GetLength() int64
func (*TensorSliceProto_Extent) GetStart ¶
func (x *TensorSliceProto_Extent) GetStart() int64
func (*TensorSliceProto_Extent) ProtoMessage ¶
func (*TensorSliceProto_Extent) ProtoMessage()
func (*TensorSliceProto_Extent) ProtoReflect ¶ added in v0.12.0
func (x *TensorSliceProto_Extent) ProtoReflect() protoreflect.Message
func (*TensorSliceProto_Extent) Reset ¶
func (x *TensorSliceProto_Extent) Reset()
func (*TensorSliceProto_Extent) String ¶
func (x *TensorSliceProto_Extent) String() string
type TensorSliceProto_Extent_Length ¶
type TensorSliceProto_Extent_Length struct {
Length int64 `protobuf:"varint,2,opt,name=length,proto3,oneof"`
}
type VariableDef ¶
type VariableDef struct { // Name of the variable tensor. VariableName string `protobuf:"bytes,1,opt,name=variable_name,json=variableName,proto3" json:"variable_name,omitempty"` // Name of the tensor holding the variable's initial value. InitialValueName string `protobuf:"bytes,6,opt,name=initial_value_name,json=initialValueName,proto3" json:"initial_value_name,omitempty"` // Name of the initializer op. InitializerName string `protobuf:"bytes,2,opt,name=initializer_name,json=initializerName,proto3" json:"initializer_name,omitempty"` // Name of the snapshot tensor. SnapshotName string `protobuf:"bytes,3,opt,name=snapshot_name,json=snapshotName,proto3" json:"snapshot_name,omitempty"` // Support for saving variables as slices of a larger variable. SaveSliceInfoDef *SaveSliceInfoDef `protobuf:"bytes,4,opt,name=save_slice_info_def,json=saveSliceInfoDef,proto3" json:"save_slice_info_def,omitempty"` // Whether to represent this as a ResourceVariable. IsResource bool `protobuf:"varint,5,opt,name=is_resource,json=isResource,proto3" json:"is_resource,omitempty"` // Whether this variable should be trained. Trainable bool `protobuf:"varint,7,opt,name=trainable,proto3" json:"trainable,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing a Variable.
func (*VariableDef) Descriptor
deprecated
func (*VariableDef) Descriptor() ([]byte, []int)
Deprecated: Use VariableDef.ProtoReflect.Descriptor instead.
func (*VariableDef) GetInitialValueName ¶
func (x *VariableDef) GetInitialValueName() string
func (*VariableDef) GetInitializerName ¶
func (x *VariableDef) GetInitializerName() string
func (*VariableDef) GetIsResource ¶
func (x *VariableDef) GetIsResource() bool
func (*VariableDef) GetSaveSliceInfoDef ¶
func (x *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef
func (*VariableDef) GetSnapshotName ¶
func (x *VariableDef) GetSnapshotName() string
func (*VariableDef) GetTrainable ¶
func (x *VariableDef) GetTrainable() bool
func (*VariableDef) GetVariableName ¶
func (x *VariableDef) GetVariableName() string
func (*VariableDef) ProtoMessage ¶
func (*VariableDef) ProtoMessage()
func (*VariableDef) ProtoReflect ¶ added in v0.12.0
func (x *VariableDef) ProtoReflect() protoreflect.Message
func (*VariableDef) Reset ¶
func (x *VariableDef) Reset()
func (*VariableDef) String ¶
func (x *VariableDef) String() string
type VariantTensorDataProto ¶
type VariantTensorDataProto struct { // Name of the type of objects being serialized. TypeName string `protobuf:"bytes,1,opt,name=type_name,json=typeName,proto3" json:"type_name,omitempty"` // Portions of the object that are not Tensors. Metadata []byte `protobuf:"bytes,2,opt,name=metadata,proto3" json:"metadata,omitempty"` // Tensors contained within objects being serialized. Tensors []*TensorProto `protobuf:"bytes,3,rep,name=tensors,proto3" json:"tensors,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing the serialization format of DT_VARIANT tensors.
func (*VariantTensorDataProto) Descriptor
deprecated
func (*VariantTensorDataProto) Descriptor() ([]byte, []int)
Deprecated: Use VariantTensorDataProto.ProtoReflect.Descriptor instead.
func (*VariantTensorDataProto) GetMetadata ¶
func (x *VariantTensorDataProto) GetMetadata() []byte
func (*VariantTensorDataProto) GetTensors ¶
func (x *VariantTensorDataProto) GetTensors() []*TensorProto
func (*VariantTensorDataProto) GetTypeName ¶
func (x *VariantTensorDataProto) GetTypeName() string
func (*VariantTensorDataProto) ProtoMessage ¶
func (*VariantTensorDataProto) ProtoMessage()
func (*VariantTensorDataProto) ProtoReflect ¶ added in v0.12.0
func (x *VariantTensorDataProto) ProtoReflect() protoreflect.Message
func (*VariantTensorDataProto) Reset ¶
func (x *VariantTensorDataProto) Reset()
func (*VariantTensorDataProto) String ¶
func (x *VariantTensorDataProto) String() string
type VersionDef ¶
type VersionDef struct { // The version of the code that produced this data. Producer int32 `protobuf:"varint,1,opt,name=producer,proto3" json:"producer,omitempty"` // Any consumer below this version is not allowed to consume this data. MinConsumer int32 `protobuf:"varint,2,opt,name=min_consumer,json=minConsumer,proto3" json:"min_consumer,omitempty"` // Specific consumer versions which are disallowed (e.g. due to bugs). BadConsumers []int32 `protobuf:"varint,3,rep,packed,name=bad_consumers,json=badConsumers,proto3" json:"bad_consumers,omitempty"` // contains filtered or unexported fields }
Version information for a piece of serialized data
There are different types of versions for each type of data (GraphDef, etc.), but they all have the same common shape described here.
Each consumer has "consumer" and "min_producer" versions (specified elsewhere). A consumer is allowed to consume this data if
producer >= min_producer consumer >= min_consumer consumer not in bad_consumers
func (*VersionDef) Descriptor
deprecated
func (*VersionDef) Descriptor() ([]byte, []int)
Deprecated: Use VersionDef.ProtoReflect.Descriptor instead.
func (*VersionDef) GetBadConsumers ¶
func (x *VersionDef) GetBadConsumers() []int32
func (*VersionDef) GetMinConsumer ¶
func (x *VersionDef) GetMinConsumer() int32
func (*VersionDef) GetProducer ¶
func (x *VersionDef) GetProducer() int32
func (*VersionDef) ProtoMessage ¶
func (*VersionDef) ProtoMessage()
func (*VersionDef) ProtoReflect ¶ added in v0.12.0
func (x *VersionDef) ProtoReflect() protoreflect.Message
func (*VersionDef) Reset ¶
func (x *VersionDef) Reset()
func (*VersionDef) String ¶
func (x *VersionDef) String() string
Source Files ¶
- allocation_description.pb.go
- api_def.pb.go
- attr_value.pb.go
- cost_graph.pb.go
- device_attributes.pb.go
- function.pb.go
- graph.pb.go
- graph_transfer_info.pb.go
- iterator.pb.go
- kernel_def.pb.go
- log_memory.pb.go
- node_def.pb.go
- op_def.pb.go
- reader_base.pb.go
- remote_fused_graph_execute_info.pb.go
- resource_handle.pb.go
- step_stats.pb.go
- summary.pb.go
- tensor.pb.go
- tensor_description.pb.go
- tensor_shape.pb.go
- tensor_slice.pb.go
- types.pb.go
- variable.pb.go
- versions.pb.go