Documentation ¶
Overview ¶
Package tensorflow is a generated protocol buffer package.
It is generated from these files:
tensorflow/core/framework/allocation_description.proto tensorflow/core/framework/api_def.proto tensorflow/core/framework/attr_value.proto tensorflow/core/framework/cost_graph.proto tensorflow/core/framework/device_attributes.proto tensorflow/core/framework/function.proto tensorflow/core/framework/graph.proto tensorflow/core/framework/graph_transfer_info.proto tensorflow/core/framework/kernel_def.proto tensorflow/core/framework/log_memory.proto tensorflow/core/framework/node_def.proto tensorflow/core/framework/op_def.proto tensorflow/core/framework/op_gen_overrides.proto tensorflow/core/framework/reader_base.proto tensorflow/core/framework/remote_fused_graph_execute_info.proto tensorflow/core/framework/resource_handle.proto tensorflow/core/framework/step_stats.proto tensorflow/core/framework/summary.proto tensorflow/core/framework/tensor.proto tensorflow/core/framework/tensor_description.proto tensorflow/core/framework/tensor_shape.proto tensorflow/core/framework/tensor_slice.proto tensorflow/core/framework/types.proto tensorflow/core/framework/variable.proto tensorflow/core/framework/versions.proto
It has these top-level messages:
AllocationDescription ApiDef ApiDefs AttrValue NameAttrList CostGraphDef DeviceLocality DeviceAttributes FunctionDefLibrary FunctionDef GradientDef GraphDef GraphTransferInfo KernelDef MemoryLogStep MemoryLogTensorAllocation MemoryLogTensorDeallocation MemoryLogTensorOutput MemoryLogRawAllocation MemoryLogRawDeallocation NodeDef OpDef OpDeprecation OpList OpGenOverride OpGenOverrides ReaderBaseState RemoteFusedGraphExecuteInfo ResourceHandleProto AllocationRecord AllocatorMemoryUsed NodeOutput MemoryStats NodeExecStats DeviceStepStats StepStats SummaryDescription HistogramProto SummaryMetadata Summary TensorProto VariantTensorDataProto TensorDescription TensorShapeProto TensorSliceProto VariableDef SaveSliceInfoDef VersionDef
Index ¶
- Variables
- type AllocationDescription
- func (*AllocationDescription) Descriptor() ([]byte, []int)
- func (m *AllocationDescription) GetAllocatedBytes() int64
- func (m *AllocationDescription) GetAllocationId() int64
- func (m *AllocationDescription) GetAllocatorName() string
- func (m *AllocationDescription) GetHasSingleReference() bool
- func (m *AllocationDescription) GetPtr() uint64
- func (m *AllocationDescription) GetRequestedBytes() int64
- func (*AllocationDescription) ProtoMessage()
- func (m *AllocationDescription) Reset()
- func (m *AllocationDescription) String() string
- type AllocationRecord
- type AllocatorMemoryUsed
- func (*AllocatorMemoryUsed) Descriptor() ([]byte, []int)
- func (m *AllocatorMemoryUsed) GetAllocationRecords() []*AllocationRecord
- func (m *AllocatorMemoryUsed) GetAllocatorBytesInUse() int64
- func (m *AllocatorMemoryUsed) GetAllocatorName() string
- func (m *AllocatorMemoryUsed) GetLiveBytes() int64
- func (m *AllocatorMemoryUsed) GetPeakBytes() int64
- func (m *AllocatorMemoryUsed) GetTotalBytes() int64
- func (*AllocatorMemoryUsed) ProtoMessage()
- func (m *AllocatorMemoryUsed) Reset()
- func (m *AllocatorMemoryUsed) String() string
- type ApiDef
- func (*ApiDef) Descriptor() ([]byte, []int)
- func (m *ApiDef) GetArgOrder() []string
- func (m *ApiDef) GetAttr() []*ApiDef_Attr
- func (m *ApiDef) GetDescription() string
- func (m *ApiDef) GetDescriptionPrefix() string
- func (m *ApiDef) GetDescriptionSuffix() string
- func (m *ApiDef) GetEndpoint() []*ApiDef_Endpoint
- func (m *ApiDef) GetGraphOpName() string
- func (m *ApiDef) GetInArg() []*ApiDef_Arg
- func (m *ApiDef) GetOutArg() []*ApiDef_Arg
- func (m *ApiDef) GetSummary() string
- func (m *ApiDef) GetVisibility() ApiDef_Visibility
- func (*ApiDef) ProtoMessage()
- func (m *ApiDef) Reset()
- func (m *ApiDef) String() string
- type ApiDef_Arg
- type ApiDef_Attr
- func (*ApiDef_Attr) Descriptor() ([]byte, []int)
- func (m *ApiDef_Attr) GetDefaultValue() *AttrValue
- func (m *ApiDef_Attr) GetDescription() string
- func (m *ApiDef_Attr) GetName() string
- func (m *ApiDef_Attr) GetRenameTo() string
- func (*ApiDef_Attr) ProtoMessage()
- func (m *ApiDef_Attr) Reset()
- func (m *ApiDef_Attr) String() string
- type ApiDef_Endpoint
- type ApiDef_Visibility
- type ApiDefs
- type AttrValue
- func (*AttrValue) Descriptor() ([]byte, []int)
- func (m *AttrValue) GetB() bool
- func (m *AttrValue) GetF() float32
- func (m *AttrValue) GetFunc() *NameAttrList
- func (m *AttrValue) GetI() int64
- func (m *AttrValue) GetList() *AttrValue_ListValue
- func (m *AttrValue) GetPlaceholder() string
- func (m *AttrValue) GetS() []byte
- func (m *AttrValue) GetShape() *TensorShapeProto
- func (m *AttrValue) GetTensor() *TensorProto
- func (m *AttrValue) GetType() DataType
- func (m *AttrValue) GetValue() isAttrValue_Value
- func (*AttrValue) ProtoMessage()
- func (m *AttrValue) Reset()
- func (m *AttrValue) String() string
- func (*AttrValue) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, ...)
- type AttrValue_B
- type AttrValue_F
- type AttrValue_Func
- type AttrValue_I
- type AttrValue_List
- type AttrValue_ListValue
- func (*AttrValue_ListValue) Descriptor() ([]byte, []int)
- func (m *AttrValue_ListValue) GetB() []bool
- func (m *AttrValue_ListValue) GetF() []float32
- func (m *AttrValue_ListValue) GetFunc() []*NameAttrList
- func (m *AttrValue_ListValue) GetI() []int64
- func (m *AttrValue_ListValue) GetS() [][]byte
- func (m *AttrValue_ListValue) GetShape() []*TensorShapeProto
- func (m *AttrValue_ListValue) GetTensor() []*TensorProto
- func (m *AttrValue_ListValue) GetType() []DataType
- func (*AttrValue_ListValue) ProtoMessage()
- func (m *AttrValue_ListValue) Reset()
- func (m *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)
- func (m *CostGraphDef_Node) GetComputeCost() int64
- func (m *CostGraphDef_Node) GetComputeTime() int64
- func (m *CostGraphDef_Node) GetControlInput() []int32
- func (m *CostGraphDef_Node) GetDevice() string
- func (m *CostGraphDef_Node) GetDevicePersistentMemorySize() int64
- func (m *CostGraphDef_Node) GetDeviceTempMemorySize() int64
- func (m *CostGraphDef_Node) GetHostPersistentMemorySize() int64
- func (m *CostGraphDef_Node) GetHostTempMemorySize() int64
- func (m *CostGraphDef_Node) GetId() int32
- func (m *CostGraphDef_Node) GetInputInfo() []*CostGraphDef_Node_InputInfo
- func (m *CostGraphDef_Node) GetIsFinal() bool
- func (m *CostGraphDef_Node) GetMemoryTime() int64
- func (m *CostGraphDef_Node) GetName() string
- func (m *CostGraphDef_Node) GetOutputInfo() []*CostGraphDef_Node_OutputInfo
- func (m *CostGraphDef_Node) GetTemporaryMemorySize() int64
- func (*CostGraphDef_Node) ProtoMessage()
- func (m *CostGraphDef_Node) Reset()
- func (m *CostGraphDef_Node) String() string
- type CostGraphDef_Node_InputInfo
- func (*CostGraphDef_Node_InputInfo) Descriptor() ([]byte, []int)
- func (m *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32
- func (m *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32
- func (*CostGraphDef_Node_InputInfo) ProtoMessage()
- func (m *CostGraphDef_Node_InputInfo) Reset()
- func (m *CostGraphDef_Node_InputInfo) String() string
- type CostGraphDef_Node_OutputInfo
- func (*CostGraphDef_Node_OutputInfo) Descriptor() ([]byte, []int)
- func (m *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64
- func (m *CostGraphDef_Node_OutputInfo) GetDtype() DataType
- func (m *CostGraphDef_Node_OutputInfo) GetShape() *TensorShapeProto
- func (m *CostGraphDef_Node_OutputInfo) GetSize() int64
- func (*CostGraphDef_Node_OutputInfo) ProtoMessage()
- func (m *CostGraphDef_Node_OutputInfo) Reset()
- func (m *CostGraphDef_Node_OutputInfo) String() string
- type DataType
- type DeviceAttributes
- func (*DeviceAttributes) Descriptor() ([]byte, []int)
- func (m *DeviceAttributes) GetDeviceType() string
- func (m *DeviceAttributes) GetIncarnation() uint64
- func (m *DeviceAttributes) GetLocality() *DeviceLocality
- func (m *DeviceAttributes) GetMemoryLimit() int64
- func (m *DeviceAttributes) GetName() string
- func (m *DeviceAttributes) GetPhysicalDeviceDesc() string
- func (*DeviceAttributes) ProtoMessage()
- func (m *DeviceAttributes) Reset()
- func (m *DeviceAttributes) String() string
- type DeviceLocality
- type DeviceStepStats
- type FunctionDef
- func (*FunctionDef) Descriptor() ([]byte, []int)
- func (m *FunctionDef) GetAttr() map[string]*AttrValue
- func (m *FunctionDef) GetNodeDef() []*NodeDef
- func (m *FunctionDef) GetRet() map[string]string
- func (m *FunctionDef) GetSignature() *OpDef
- func (*FunctionDef) ProtoMessage()
- func (m *FunctionDef) Reset()
- func (m *FunctionDef) String() string
- type FunctionDefLibrary
- type GradientDef
- type GraphDef
- func (*GraphDef) Descriptor() ([]byte, []int)
- func (m *GraphDef) GetLibrary() *FunctionDefLibrary
- func (m *GraphDef) GetNode() []*NodeDef
- func (m *GraphDef) GetVersion() int32
- func (m *GraphDef) GetVersions() *VersionDef
- func (*GraphDef) ProtoMessage()
- func (m *GraphDef) Reset()
- func (m *GraphDef) String() string
- type GraphTransferInfo
- func (*GraphTransferInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo) GetConstNodeInfo() []*GraphTransferInfo_ConstNodeInfo
- func (m *GraphTransferInfo) GetDestination() GraphTransferInfo_Destination
- func (m *GraphTransferInfo) GetGraphInputNodeInfo() []*GraphTransferInfo_GraphInputNodeInfo
- func (m *GraphTransferInfo) GetGraphOutputNodeInfo() []*GraphTransferInfo_GraphOutputNodeInfo
- func (m *GraphTransferInfo) GetNodeInfo() []*GraphTransferInfo_NodeInfo
- func (m *GraphTransferInfo) GetNodeInputInfo() []*GraphTransferInfo_NodeInputInfo
- func (m *GraphTransferInfo) GetNodeOutputInfo() []*GraphTransferInfo_NodeOutputInfo
- func (*GraphTransferInfo) ProtoMessage()
- func (m *GraphTransferInfo) Reset()
- func (m *GraphTransferInfo) String() string
- type GraphTransferInfo_ConstNodeInfo
- func (*GraphTransferInfo_ConstNodeInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_ConstNodeInfo) GetData() []byte
- func (m *GraphTransferInfo_ConstNodeInfo) GetDtype() DataType
- func (m *GraphTransferInfo_ConstNodeInfo) GetName() string
- func (m *GraphTransferInfo_ConstNodeInfo) GetNodeId() int32
- func (m *GraphTransferInfo_ConstNodeInfo) GetShape() []int64
- func (*GraphTransferInfo_ConstNodeInfo) ProtoMessage()
- func (m *GraphTransferInfo_ConstNodeInfo) Reset()
- func (m *GraphTransferInfo_ConstNodeInfo) String() string
- type GraphTransferInfo_Destination
- type GraphTransferInfo_GraphInputNodeInfo
- func (*GraphTransferInfo_GraphInputNodeInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_GraphInputNodeInfo) GetDtype() DataType
- func (m *GraphTransferInfo_GraphInputNodeInfo) GetName() string
- func (m *GraphTransferInfo_GraphInputNodeInfo) GetShape() []int64
- func (*GraphTransferInfo_GraphInputNodeInfo) ProtoMessage()
- func (m *GraphTransferInfo_GraphInputNodeInfo) Reset()
- func (m *GraphTransferInfo_GraphInputNodeInfo) String() string
- type GraphTransferInfo_GraphOutputNodeInfo
- func (*GraphTransferInfo_GraphOutputNodeInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_GraphOutputNodeInfo) GetDtype() DataType
- func (m *GraphTransferInfo_GraphOutputNodeInfo) GetName() string
- func (m *GraphTransferInfo_GraphOutputNodeInfo) GetShape() []int64
- func (*GraphTransferInfo_GraphOutputNodeInfo) ProtoMessage()
- func (m *GraphTransferInfo_GraphOutputNodeInfo) Reset()
- func (m *GraphTransferInfo_GraphOutputNodeInfo) String() string
- type GraphTransferInfo_NodeInfo
- func (*GraphTransferInfo_NodeInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_NodeInfo) GetInputCount() int32
- func (m *GraphTransferInfo_NodeInfo) GetName() string
- func (m *GraphTransferInfo_NodeInfo) GetNodeId() int32
- func (m *GraphTransferInfo_NodeInfo) GetOutputCount() int32
- func (m *GraphTransferInfo_NodeInfo) GetPaddingId() int32
- func (m *GraphTransferInfo_NodeInfo) GetSocOpId() int32
- func (m *GraphTransferInfo_NodeInfo) GetTypeName() string
- func (*GraphTransferInfo_NodeInfo) ProtoMessage()
- func (m *GraphTransferInfo_NodeInfo) Reset()
- func (m *GraphTransferInfo_NodeInfo) String() string
- type GraphTransferInfo_NodeInput
- func (*GraphTransferInfo_NodeInput) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_NodeInput) GetNodeId() int32
- func (m *GraphTransferInfo_NodeInput) GetOutputPort() int32
- func (*GraphTransferInfo_NodeInput) ProtoMessage()
- func (m *GraphTransferInfo_NodeInput) Reset()
- func (m *GraphTransferInfo_NodeInput) String() string
- type GraphTransferInfo_NodeInputInfo
- func (*GraphTransferInfo_NodeInputInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_NodeInputInfo) GetNodeId() int32
- func (m *GraphTransferInfo_NodeInputInfo) GetNodeInput() []*GraphTransferInfo_NodeInput
- func (*GraphTransferInfo_NodeInputInfo) ProtoMessage()
- func (m *GraphTransferInfo_NodeInputInfo) Reset()
- func (m *GraphTransferInfo_NodeInputInfo) String() string
- type GraphTransferInfo_NodeOutputInfo
- func (*GraphTransferInfo_NodeOutputInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_NodeOutputInfo) GetMaxByteSize() []int32
- func (m *GraphTransferInfo_NodeOutputInfo) GetNodeId() int32
- func (*GraphTransferInfo_NodeOutputInfo) ProtoMessage()
- func (m *GraphTransferInfo_NodeOutputInfo) Reset()
- func (m *GraphTransferInfo_NodeOutputInfo) String() string
- type HistogramProto
- func (*HistogramProto) Descriptor() ([]byte, []int)
- func (m *HistogramProto) GetBucket() []float64
- func (m *HistogramProto) GetBucketLimit() []float64
- func (m *HistogramProto) GetMax() float64
- func (m *HistogramProto) GetMin() float64
- func (m *HistogramProto) GetNum() float64
- func (m *HistogramProto) GetSum() float64
- func (m *HistogramProto) GetSumSquares() float64
- func (*HistogramProto) ProtoMessage()
- func (m *HistogramProto) Reset()
- func (m *HistogramProto) String() string
- type KernelDef
- func (*KernelDef) Descriptor() ([]byte, []int)
- func (m *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
- func (m *KernelDef) GetDeviceType() string
- func (m *KernelDef) GetHostMemoryArg() []string
- func (m *KernelDef) GetLabel() string
- func (m *KernelDef) GetOp() string
- func (*KernelDef) ProtoMessage()
- func (m *KernelDef) Reset()
- func (m *KernelDef) String() string
- type KernelDef_AttrConstraint
- func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)
- func (m *KernelDef_AttrConstraint) GetAllowedValues() *AttrValue
- func (m *KernelDef_AttrConstraint) GetName() string
- func (*KernelDef_AttrConstraint) ProtoMessage()
- func (m *KernelDef_AttrConstraint) Reset()
- func (m *KernelDef_AttrConstraint) String() string
- type MemoryLogRawAllocation
- func (*MemoryLogRawAllocation) Descriptor() ([]byte, []int)
- func (m *MemoryLogRawAllocation) GetAllocationId() int64
- func (m *MemoryLogRawAllocation) GetAllocatorName() string
- func (m *MemoryLogRawAllocation) GetNumBytes() int64
- func (m *MemoryLogRawAllocation) GetOperation() string
- func (m *MemoryLogRawAllocation) GetPtr() uint64
- func (m *MemoryLogRawAllocation) GetStepId() int64
- func (*MemoryLogRawAllocation) ProtoMessage()
- func (m *MemoryLogRawAllocation) Reset()
- func (m *MemoryLogRawAllocation) String() string
- type MemoryLogRawDeallocation
- func (*MemoryLogRawDeallocation) Descriptor() ([]byte, []int)
- func (m *MemoryLogRawDeallocation) GetAllocationId() int64
- func (m *MemoryLogRawDeallocation) GetAllocatorName() string
- func (m *MemoryLogRawDeallocation) GetDeferred() bool
- func (m *MemoryLogRawDeallocation) GetOperation() string
- func (m *MemoryLogRawDeallocation) GetStepId() int64
- func (*MemoryLogRawDeallocation) ProtoMessage()
- func (m *MemoryLogRawDeallocation) Reset()
- func (m *MemoryLogRawDeallocation) String() string
- type MemoryLogStep
- type MemoryLogTensorAllocation
- func (*MemoryLogTensorAllocation) Descriptor() ([]byte, []int)
- func (m *MemoryLogTensorAllocation) GetKernelName() string
- func (m *MemoryLogTensorAllocation) GetStepId() int64
- func (m *MemoryLogTensorAllocation) GetTensor() *TensorDescription
- func (*MemoryLogTensorAllocation) ProtoMessage()
- func (m *MemoryLogTensorAllocation) Reset()
- func (m *MemoryLogTensorAllocation) String() string
- type MemoryLogTensorDeallocation
- func (*MemoryLogTensorDeallocation) Descriptor() ([]byte, []int)
- func (m *MemoryLogTensorDeallocation) GetAllocationId() int64
- func (m *MemoryLogTensorDeallocation) GetAllocatorName() string
- func (*MemoryLogTensorDeallocation) ProtoMessage()
- func (m *MemoryLogTensorDeallocation) Reset()
- func (m *MemoryLogTensorDeallocation) String() string
- type MemoryLogTensorOutput
- func (*MemoryLogTensorOutput) Descriptor() ([]byte, []int)
- func (m *MemoryLogTensorOutput) GetIndex() int32
- func (m *MemoryLogTensorOutput) GetKernelName() string
- func (m *MemoryLogTensorOutput) GetStepId() int64
- func (m *MemoryLogTensorOutput) GetTensor() *TensorDescription
- func (*MemoryLogTensorOutput) ProtoMessage()
- func (m *MemoryLogTensorOutput) Reset()
- func (m *MemoryLogTensorOutput) String() string
- type MemoryStats
- func (*MemoryStats) Descriptor() ([]byte, []int)
- func (m *MemoryStats) GetDevicePersistentMemorySize() int64
- func (m *MemoryStats) GetDevicePersistentTensorAllocIds() []int64
- func (m *MemoryStats) GetDeviceTempMemorySize() int64
- func (m *MemoryStats) GetHostPersistentMemorySize() int64
- func (m *MemoryStats) GetHostPersistentTensorAllocIds() []int64
- func (m *MemoryStats) GetHostTempMemorySize() int64
- func (*MemoryStats) ProtoMessage()
- func (m *MemoryStats) Reset()
- func (m *MemoryStats) String() string
- type NameAttrList
- type NodeDef
- func (*NodeDef) Descriptor() ([]byte, []int)
- func (m *NodeDef) GetAttr() map[string]*AttrValue
- func (m *NodeDef) GetDevice() string
- func (m *NodeDef) GetInput() []string
- func (m *NodeDef) GetName() string
- func (m *NodeDef) GetOp() string
- func (*NodeDef) ProtoMessage()
- func (m *NodeDef) Reset()
- func (m *NodeDef) String() string
- type NodeExecStats
- func (*NodeExecStats) Descriptor() ([]byte, []int)
- func (m *NodeExecStats) GetAllEndRelMicros() int64
- func (m *NodeExecStats) GetAllStartMicros() int64
- func (m *NodeExecStats) GetMemory() []*AllocatorMemoryUsed
- func (m *NodeExecStats) GetMemoryStats() *MemoryStats
- func (m *NodeExecStats) GetNodeName() string
- func (m *NodeExecStats) GetOpEndRelMicros() int64
- func (m *NodeExecStats) GetOpStartRelMicros() int64
- func (m *NodeExecStats) GetOutput() []*NodeOutput
- func (m *NodeExecStats) GetReferencedTensor() []*AllocationDescription
- func (m *NodeExecStats) GetScheduledMicros() int64
- func (m *NodeExecStats) GetThreadId() uint32
- func (m *NodeExecStats) GetTimelineLabel() string
- func (*NodeExecStats) ProtoMessage()
- func (m *NodeExecStats) Reset()
- func (m *NodeExecStats) String() string
- type NodeOutput
- type OpDef
- func (*OpDef) Descriptor() ([]byte, []int)
- func (m *OpDef) GetAllowsUninitializedInput() bool
- func (m *OpDef) GetAttr() []*OpDef_AttrDef
- func (m *OpDef) GetDeprecation() *OpDeprecation
- func (m *OpDef) GetDescription() string
- func (m *OpDef) GetInputArg() []*OpDef_ArgDef
- func (m *OpDef) GetIsAggregate() bool
- func (m *OpDef) GetIsCommutative() bool
- func (m *OpDef) GetIsStateful() bool
- func (m *OpDef) GetName() string
- func (m *OpDef) GetOutputArg() []*OpDef_ArgDef
- func (m *OpDef) GetSummary() string
- func (*OpDef) ProtoMessage()
- func (m *OpDef) Reset()
- func (m *OpDef) String() string
- type OpDef_ArgDef
- func (*OpDef_ArgDef) Descriptor() ([]byte, []int)
- func (m *OpDef_ArgDef) GetDescription() string
- func (m *OpDef_ArgDef) GetIsRef() bool
- func (m *OpDef_ArgDef) GetName() string
- func (m *OpDef_ArgDef) GetNumberAttr() string
- func (m *OpDef_ArgDef) GetType() DataType
- func (m *OpDef_ArgDef) GetTypeAttr() string
- func (m *OpDef_ArgDef) GetTypeListAttr() string
- func (*OpDef_ArgDef) ProtoMessage()
- func (m *OpDef_ArgDef) Reset()
- func (m *OpDef_ArgDef) String() string
- type OpDef_AttrDef
- func (*OpDef_AttrDef) Descriptor() ([]byte, []int)
- func (m *OpDef_AttrDef) GetAllowedValues() *AttrValue
- func (m *OpDef_AttrDef) GetDefaultValue() *AttrValue
- func (m *OpDef_AttrDef) GetDescription() string
- func (m *OpDef_AttrDef) GetHasMinimum() bool
- func (m *OpDef_AttrDef) GetMinimum() int64
- func (m *OpDef_AttrDef) GetName() string
- func (m *OpDef_AttrDef) GetType() string
- func (*OpDef_AttrDef) ProtoMessage()
- func (m *OpDef_AttrDef) Reset()
- func (m *OpDef_AttrDef) String() string
- type OpDeprecation
- type OpGenOverride
- func (*OpGenOverride) Descriptor() ([]byte, []int)
- func (m *OpGenOverride) GetAlias() []string
- func (m *OpGenOverride) GetAttrDefault() []*OpGenOverride_AttrDefault
- func (m *OpGenOverride) GetAttrRename() []*OpGenOverride_Rename
- func (m *OpGenOverride) GetHide() bool
- func (m *OpGenOverride) GetInputRename() []*OpGenOverride_Rename
- func (m *OpGenOverride) GetName() string
- func (m *OpGenOverride) GetOutputRename() []*OpGenOverride_Rename
- func (m *OpGenOverride) GetRenameTo() string
- func (m *OpGenOverride) GetSkip() bool
- func (*OpGenOverride) ProtoMessage()
- func (m *OpGenOverride) Reset()
- func (m *OpGenOverride) String() string
- type OpGenOverride_AttrDefault
- func (*OpGenOverride_AttrDefault) Descriptor() ([]byte, []int)
- func (m *OpGenOverride_AttrDefault) GetName() string
- func (m *OpGenOverride_AttrDefault) GetValue() *AttrValue
- func (*OpGenOverride_AttrDefault) ProtoMessage()
- func (m *OpGenOverride_AttrDefault) Reset()
- func (m *OpGenOverride_AttrDefault) String() string
- type OpGenOverride_Rename
- type OpGenOverrides
- type OpList
- type ReaderBaseState
- func (*ReaderBaseState) Descriptor() ([]byte, []int)
- func (m *ReaderBaseState) GetCurrentWork() []byte
- func (m *ReaderBaseState) GetNumRecordsProduced() int64
- func (m *ReaderBaseState) GetWorkFinished() int64
- func (m *ReaderBaseState) GetWorkStarted() int64
- func (*ReaderBaseState) ProtoMessage()
- func (m *ReaderBaseState) Reset()
- func (m *ReaderBaseState) String() string
- type RemoteFusedGraphExecuteInfo
- func (*RemoteFusedGraphExecuteInfo) Descriptor() ([]byte, []int)
- func (m *RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (m *RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (m *RemoteFusedGraphExecuteInfo) GetExecutorName() string
- func (m *RemoteFusedGraphExecuteInfo) GetGraphInputNodeName() []string
- func (m *RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName() []string
- func (m *RemoteFusedGraphExecuteInfo) GetRemoteGraph() *GraphDef
- func (m *RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters() []byte
- func (*RemoteFusedGraphExecuteInfo) ProtoMessage()
- func (m *RemoteFusedGraphExecuteInfo) Reset()
- func (m *RemoteFusedGraphExecuteInfo) String() string
- type RemoteFusedGraphExecuteInfo_NodeType
- type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor() ([]byte, []int)
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype() DataType
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape() *TensorShapeProto
- func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage()
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset()
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) String() string
- type ResourceHandleProto
- func (*ResourceHandleProto) Descriptor() ([]byte, []int)
- func (m *ResourceHandleProto) GetContainer() string
- func (m *ResourceHandleProto) GetDevice() string
- func (m *ResourceHandleProto) GetHashCode() uint64
- func (m *ResourceHandleProto) GetMaybeTypeName() string
- func (m *ResourceHandleProto) GetName() string
- func (*ResourceHandleProto) ProtoMessage()
- func (m *ResourceHandleProto) Reset()
- func (m *ResourceHandleProto) String() string
- type SaveSliceInfoDef
- func (*SaveSliceInfoDef) Descriptor() ([]byte, []int)
- func (m *SaveSliceInfoDef) GetFullName() string
- func (m *SaveSliceInfoDef) GetFullShape() []int64
- func (m *SaveSliceInfoDef) GetVarOffset() []int64
- func (m *SaveSliceInfoDef) GetVarShape() []int64
- func (*SaveSliceInfoDef) ProtoMessage()
- func (m *SaveSliceInfoDef) Reset()
- func (m *SaveSliceInfoDef) String() string
- type StepStats
- type Summary
- type SummaryDescription
- type SummaryMetadata
- func (*SummaryMetadata) Descriptor() ([]byte, []int)
- func (m *SummaryMetadata) GetDisplayName() string
- func (m *SummaryMetadata) GetPluginData() *SummaryMetadata_PluginData
- func (m *SummaryMetadata) GetSummaryDescription() string
- func (*SummaryMetadata) ProtoMessage()
- func (m *SummaryMetadata) Reset()
- func (m *SummaryMetadata) String() string
- type SummaryMetadata_PluginData
- func (*SummaryMetadata_PluginData) Descriptor() ([]byte, []int)
- func (m *SummaryMetadata_PluginData) GetContent() []byte
- func (m *SummaryMetadata_PluginData) GetPluginName() string
- func (*SummaryMetadata_PluginData) ProtoMessage()
- func (m *SummaryMetadata_PluginData) Reset()
- func (m *SummaryMetadata_PluginData) String() string
- type Summary_Audio
- func (*Summary_Audio) Descriptor() ([]byte, []int)
- func (m *Summary_Audio) GetContentType() string
- func (m *Summary_Audio) GetEncodedAudioString() []byte
- func (m *Summary_Audio) GetLengthFrames() int64
- func (m *Summary_Audio) GetNumChannels() int64
- func (m *Summary_Audio) GetSampleRate() float32
- func (*Summary_Audio) ProtoMessage()
- func (m *Summary_Audio) Reset()
- func (m *Summary_Audio) String() string
- type Summary_Image
- func (*Summary_Image) Descriptor() ([]byte, []int)
- func (m *Summary_Image) GetColorspace() int32
- func (m *Summary_Image) GetEncodedImageString() []byte
- func (m *Summary_Image) GetHeight() int32
- func (m *Summary_Image) GetWidth() int32
- func (*Summary_Image) ProtoMessage()
- func (m *Summary_Image) Reset()
- func (m *Summary_Image) String() string
- type Summary_Value
- func (*Summary_Value) Descriptor() ([]byte, []int)
- func (m *Summary_Value) GetAudio() *Summary_Audio
- func (m *Summary_Value) GetHisto() *HistogramProto
- func (m *Summary_Value) GetImage() *Summary_Image
- func (m *Summary_Value) GetMetadata() *SummaryMetadata
- func (m *Summary_Value) GetNodeName() string
- func (m *Summary_Value) GetObsoleteOldStyleHistogram() []byte
- func (m *Summary_Value) GetSimpleValue() float32
- func (m *Summary_Value) GetTag() string
- func (m *Summary_Value) GetTensor() *TensorProto
- func (m *Summary_Value) GetValue() isSummary_Value_Value
- func (*Summary_Value) ProtoMessage()
- func (m *Summary_Value) Reset()
- func (m *Summary_Value) String() string
- func (*Summary_Value) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, ...)
- 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)
- func (m *TensorDescription) GetAllocationDescription() *AllocationDescription
- func (m *TensorDescription) GetDtype() DataType
- func (m *TensorDescription) GetShape() *TensorShapeProto
- func (*TensorDescription) ProtoMessage()
- func (m *TensorDescription) Reset()
- func (m *TensorDescription) String() string
- type TensorProto
- func (*TensorProto) Descriptor() ([]byte, []int)
- func (m *TensorProto) GetBoolVal() []bool
- func (m *TensorProto) GetDcomplexVal() []float64
- func (m *TensorProto) GetDoubleVal() []float64
- func (m *TensorProto) GetDtype() DataType
- func (m *TensorProto) GetFloatVal() []float32
- func (m *TensorProto) GetHalfVal() []int32
- func (m *TensorProto) GetInt64Val() []int64
- func (m *TensorProto) GetIntVal() []int32
- func (m *TensorProto) GetResourceHandleVal() []*ResourceHandleProto
- func (m *TensorProto) GetScomplexVal() []float32
- func (m *TensorProto) GetStringVal() [][]byte
- func (m *TensorProto) GetTensorContent() []byte
- func (m *TensorProto) GetTensorShape() *TensorShapeProto
- func (m *TensorProto) GetUint32Val() []uint32
- func (m *TensorProto) GetUint64Val() []uint64
- func (m *TensorProto) GetVariantVal() []*VariantTensorDataProto
- func (m *TensorProto) GetVersionNumber() int32
- func (*TensorProto) ProtoMessage()
- func (m *TensorProto) Reset()
- func (m *TensorProto) String() string
- type TensorShapeProto
- type TensorShapeProto_Dim
- type TensorSliceProto
- type TensorSliceProto_Extent
- func (*TensorSliceProto_Extent) Descriptor() ([]byte, []int)
- func (m *TensorSliceProto_Extent) GetHasLength() isTensorSliceProto_Extent_HasLength
- func (m *TensorSliceProto_Extent) GetLength() int64
- func (m *TensorSliceProto_Extent) GetStart() int64
- func (*TensorSliceProto_Extent) ProtoMessage()
- func (m *TensorSliceProto_Extent) Reset()
- func (m *TensorSliceProto_Extent) String() string
- func (*TensorSliceProto_Extent) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, ...)
- type TensorSliceProto_Extent_Length
- type VariableDef
- func (*VariableDef) Descriptor() ([]byte, []int)
- func (m *VariableDef) GetInitialValueName() string
- func (m *VariableDef) GetInitializerName() string
- func (m *VariableDef) GetIsResource() bool
- func (m *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef
- func (m *VariableDef) GetSnapshotName() string
- func (m *VariableDef) GetVariableName() string
- func (*VariableDef) ProtoMessage()
- func (m *VariableDef) Reset()
- func (m *VariableDef) String() string
- type VariantTensorDataProto
- func (*VariantTensorDataProto) Descriptor() ([]byte, []int)
- func (m *VariantTensorDataProto) GetMetadata() []byte
- func (m *VariantTensorDataProto) GetTensors() []*TensorProto
- func (m *VariantTensorDataProto) GetTypeName() string
- func (*VariantTensorDataProto) ProtoMessage()
- func (m *VariantTensorDataProto) Reset()
- func (m *VariantTensorDataProto) String() string
- type VersionDef
Constants ¶
This section is empty.
Variables ¶
var ApiDef_Visibility_name = map[int32]string{
0: "DEFAULT_VISIBILITY",
1: "VISIBLE",
2: "SKIP",
3: "HIDDEN",
}
var ApiDef_Visibility_value = map[string]int32{
"DEFAULT_VISIBILITY": 0,
"VISIBLE": 1,
"SKIP": 2,
"HIDDEN": 3,
}
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",
}
var 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,
}
var GraphTransferInfo_Destination_name = map[int32]string{
0: "NOP",
1: "HEXAGON",
}
var GraphTransferInfo_Destination_value = map[string]int32{
"NOP": 0,
"HEXAGON": 1,
}
var RemoteFusedGraphExecuteInfo_NodeType_name = map[int32]string{
0: "UNUSED",
1: "GRAPH_INPUT",
2: "GRAPH_OUTPUT",
3: "FUSED_NODE",
4: "BORDER_INPUT",
5: "BORDER_OUTPUT",
}
var RemoteFusedGraphExecuteInfo_NodeType_value = map[string]int32{
"UNUSED": 0,
"GRAPH_INPUT": 1,
"GRAPH_OUTPUT": 2,
"FUSED_NODE": 3,
"BORDER_INPUT": 4,
"BORDER_OUTPUT": 5,
}
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" json:"requested_bytes,omitempty"` // Total number of bytes allocated if known AllocatedBytes int64 `protobuf:"varint,2,opt,name=allocated_bytes,json=allocatedBytes" json:"allocated_bytes,omitempty"` // Name of the allocator used AllocatorName string `protobuf:"bytes,3,opt,name=allocator_name,json=allocatorName" json:"allocator_name,omitempty"` // Identifier of the allocated buffer if known AllocationId int64 `protobuf:"varint,4,opt,name=allocation_id,json=allocationId" 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" json:"has_single_reference,omitempty"` // Address of the allocation. Ptr uint64 `protobuf:"varint,6,opt,name=ptr" json:"ptr,omitempty"` }
func (*AllocationDescription) Descriptor ¶
func (*AllocationDescription) Descriptor() ([]byte, []int)
func (*AllocationDescription) GetAllocatedBytes ¶
func (m *AllocationDescription) GetAllocatedBytes() int64
func (*AllocationDescription) GetAllocationId ¶
func (m *AllocationDescription) GetAllocationId() int64
func (*AllocationDescription) GetAllocatorName ¶
func (m *AllocationDescription) GetAllocatorName() string
func (*AllocationDescription) GetHasSingleReference ¶
func (m *AllocationDescription) GetHasSingleReference() bool
func (*AllocationDescription) GetPtr ¶
func (m *AllocationDescription) GetPtr() uint64
func (*AllocationDescription) GetRequestedBytes ¶
func (m *AllocationDescription) GetRequestedBytes() int64
func (*AllocationDescription) ProtoMessage ¶
func (*AllocationDescription) ProtoMessage()
func (*AllocationDescription) Reset ¶
func (m *AllocationDescription) Reset()
func (*AllocationDescription) String ¶
func (m *AllocationDescription) String() string
type AllocationRecord ¶
type AllocationRecord struct { // The timestamp of the operation. AllocMicros int64 `protobuf:"varint,1,opt,name=alloc_micros,json=allocMicros" json:"alloc_micros,omitempty"` // Number of bytes allocated, or de-allocated if negative. AllocBytes int64 `protobuf:"varint,2,opt,name=alloc_bytes,json=allocBytes" json:"alloc_bytes,omitempty"` }
An allocation/de-allocation operation performed by the allocator.
func (*AllocationRecord) Descriptor ¶
func (*AllocationRecord) Descriptor() ([]byte, []int)
func (*AllocationRecord) GetAllocBytes ¶
func (m *AllocationRecord) GetAllocBytes() int64
func (*AllocationRecord) GetAllocMicros ¶
func (m *AllocationRecord) GetAllocMicros() int64
func (*AllocationRecord) ProtoMessage ¶
func (*AllocationRecord) ProtoMessage()
func (*AllocationRecord) Reset ¶
func (m *AllocationRecord) Reset()
func (*AllocationRecord) String ¶
func (m *AllocationRecord) String() string
type AllocatorMemoryUsed ¶
type AllocatorMemoryUsed struct { AllocatorName string `protobuf:"bytes,1,opt,name=allocator_name,json=allocatorName" json:"allocator_name,omitempty"` // These are per-node allocator memory stats. TotalBytes int64 `protobuf:"varint,2,opt,name=total_bytes,json=totalBytes" json:"total_bytes,omitempty"` PeakBytes int64 `protobuf:"varint,3,opt,name=peak_bytes,json=peakBytes" json:"peak_bytes,omitempty"` // The bytes that are not deallocated. LiveBytes int64 `protobuf:"varint,4,opt,name=live_bytes,json=liveBytes" json:"live_bytes,omitempty"` // The allocation and deallocation timeline. AllocationRecords []*AllocationRecord `protobuf:"bytes,6,rep,name=allocation_records,json=allocationRecords" 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" json:"allocator_bytes_in_use,omitempty"` }
func (*AllocatorMemoryUsed) Descriptor ¶
func (*AllocatorMemoryUsed) Descriptor() ([]byte, []int)
func (*AllocatorMemoryUsed) GetAllocationRecords ¶
func (m *AllocatorMemoryUsed) GetAllocationRecords() []*AllocationRecord
func (*AllocatorMemoryUsed) GetAllocatorBytesInUse ¶
func (m *AllocatorMemoryUsed) GetAllocatorBytesInUse() int64
func (*AllocatorMemoryUsed) GetAllocatorName ¶
func (m *AllocatorMemoryUsed) GetAllocatorName() string
func (*AllocatorMemoryUsed) GetLiveBytes ¶
func (m *AllocatorMemoryUsed) GetLiveBytes() int64
func (*AllocatorMemoryUsed) GetPeakBytes ¶
func (m *AllocatorMemoryUsed) GetPeakBytes() int64
func (*AllocatorMemoryUsed) GetTotalBytes ¶
func (m *AllocatorMemoryUsed) GetTotalBytes() int64
func (*AllocatorMemoryUsed) ProtoMessage ¶
func (*AllocatorMemoryUsed) ProtoMessage()
func (*AllocatorMemoryUsed) Reset ¶
func (m *AllocatorMemoryUsed) Reset()
func (*AllocatorMemoryUsed) String ¶
func (m *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" json:"graph_op_name,omitempty"` Visibility ApiDef_Visibility `protobuf:"varint,2,opt,name=visibility,enum=tensorflow.ApiDef_Visibility" json:"visibility,omitempty"` Endpoint []*ApiDef_Endpoint `protobuf:"bytes,3,rep,name=endpoint" json:"endpoint,omitempty"` InArg []*ApiDef_Arg `protobuf:"bytes,4,rep,name=in_arg,json=inArg" json:"in_arg,omitempty"` OutArg []*ApiDef_Arg `protobuf:"bytes,5,rep,name=out_arg,json=outArg" json:"out_arg,omitempty"` // List of post-rename 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" json:"arg_order,omitempty"` Attr []*ApiDef_Attr `protobuf:"bytes,6,rep,name=attr" json:"attr,omitempty"` // One-line human-readable description of what the Op does. Summary string `protobuf:"bytes,7,opt,name=summary" json:"summary,omitempty"` // Additional, longer human-readable description of what the Op does. Description string `protobuf:"bytes,8,opt,name=description" 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" json:"description_prefix,omitempty"` DescriptionSuffix string `protobuf:"bytes,10,opt,name=description_suffix,json=descriptionSuffix" json:"description_suffix,omitempty"` }
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 ¶
func (*ApiDef) GetArgOrder ¶
func (*ApiDef) GetAttr ¶
func (m *ApiDef) GetAttr() []*ApiDef_Attr
func (*ApiDef) GetDescription ¶
func (*ApiDef) GetDescriptionPrefix ¶
func (*ApiDef) GetDescriptionSuffix ¶
func (*ApiDef) GetEndpoint ¶
func (m *ApiDef) GetEndpoint() []*ApiDef_Endpoint
func (*ApiDef) GetGraphOpName ¶
func (*ApiDef) GetInArg ¶
func (m *ApiDef) GetInArg() []*ApiDef_Arg
func (*ApiDef) GetOutArg ¶
func (m *ApiDef) GetOutArg() []*ApiDef_Arg
func (*ApiDef) GetSummary ¶
func (*ApiDef) GetVisibility ¶
func (m *ApiDef) GetVisibility() ApiDef_Visibility
func (*ApiDef) ProtoMessage ¶
func (*ApiDef) ProtoMessage()
type ApiDef_Arg ¶
type ApiDef_Arg struct { Name string `protobuf:"bytes,1,opt,name=name" 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" 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" json:"description,omitempty"` }
func (*ApiDef_Arg) Descriptor ¶
func (*ApiDef_Arg) Descriptor() ([]byte, []int)
func (*ApiDef_Arg) GetDescription ¶
func (m *ApiDef_Arg) GetDescription() string
func (*ApiDef_Arg) GetName ¶
func (m *ApiDef_Arg) GetName() string
func (*ApiDef_Arg) GetRenameTo ¶
func (m *ApiDef_Arg) GetRenameTo() string
func (*ApiDef_Arg) ProtoMessage ¶
func (*ApiDef_Arg) ProtoMessage()
func (*ApiDef_Arg) Reset ¶
func (m *ApiDef_Arg) Reset()
func (*ApiDef_Arg) String ¶
func (m *ApiDef_Arg) String() string
type ApiDef_Attr ¶
type ApiDef_Attr struct { Name string `protobuf:"bytes,1,opt,name=name" 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" 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" 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" json:"description,omitempty"` }
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 ¶
func (*ApiDef_Attr) Descriptor() ([]byte, []int)
func (*ApiDef_Attr) GetDefaultValue ¶
func (m *ApiDef_Attr) GetDefaultValue() *AttrValue
func (*ApiDef_Attr) GetDescription ¶
func (m *ApiDef_Attr) GetDescription() string
func (*ApiDef_Attr) GetName ¶
func (m *ApiDef_Attr) GetName() string
func (*ApiDef_Attr) GetRenameTo ¶
func (m *ApiDef_Attr) GetRenameTo() string
func (*ApiDef_Attr) ProtoMessage ¶
func (*ApiDef_Attr) ProtoMessage()
func (*ApiDef_Attr) Reset ¶
func (m *ApiDef_Attr) Reset()
func (*ApiDef_Attr) String ¶
func (m *ApiDef_Attr) String() string
type ApiDef_Endpoint ¶
type ApiDef_Endpoint struct { // Name should be either like "CamelCaseName" or // "Package.CamelCaseName". Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` // First GraphDef version at which the op is disallowed. DeprecationVersion int32 `protobuf:"varint,2,opt,name=deprecation_version,json=deprecationVersion" json:"deprecation_version,omitempty"` }
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 ¶
func (*ApiDef_Endpoint) Descriptor() ([]byte, []int)
func (*ApiDef_Endpoint) GetDeprecationVersion ¶
func (m *ApiDef_Endpoint) GetDeprecationVersion() int32
func (*ApiDef_Endpoint) GetName ¶
func (m *ApiDef_Endpoint) GetName() string
func (*ApiDef_Endpoint) ProtoMessage ¶
func (*ApiDef_Endpoint) ProtoMessage()
func (*ApiDef_Endpoint) Reset ¶
func (m *ApiDef_Endpoint) Reset()
func (*ApiDef_Endpoint) String ¶
func (m *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) EnumDescriptor ¶
func (ApiDef_Visibility) EnumDescriptor() ([]byte, []int)
func (ApiDef_Visibility) String ¶
func (x ApiDef_Visibility) String() string
type ApiDefs ¶
type ApiDefs struct {
Op []*ApiDef `protobuf:"bytes,1,rep,name=op" json:"op,omitempty"`
}
func (*ApiDefs) Descriptor ¶
func (*ApiDefs) ProtoMessage ¶
func (*ApiDefs) ProtoMessage()
type AttrValue ¶
type AttrValue struct { // Types that are valid to be assigned 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"` }
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 ¶
func (*AttrValue) GetFunc ¶
func (m *AttrValue) GetFunc() *NameAttrList
func (*AttrValue) GetList ¶
func (m *AttrValue) GetList() *AttrValue_ListValue
func (*AttrValue) GetPlaceholder ¶
func (*AttrValue) GetShape ¶
func (m *AttrValue) GetShape() *TensorShapeProto
func (*AttrValue) GetTensor ¶
func (m *AttrValue) GetTensor() *TensorProto
func (*AttrValue) ProtoMessage ¶
func (*AttrValue) ProtoMessage()
type AttrValue_B ¶
type AttrValue_B struct {
B bool `protobuf:"varint,5,opt,name=b,oneof"`
}
type AttrValue_F ¶
type AttrValue_F struct {
F float32 `protobuf:"fixed32,4,opt,name=f,oneof"`
}
type AttrValue_Func ¶
type AttrValue_Func struct {
Func *NameAttrList `protobuf:"bytes,10,opt,name=func,oneof"`
}
type AttrValue_I ¶
type AttrValue_I struct {
I int64 `protobuf:"varint,3,opt,name=i,oneof"`
}
type AttrValue_List ¶
type AttrValue_List struct {
List *AttrValue_ListValue `protobuf:"bytes,1,opt,name=list,oneof"`
}
type AttrValue_ListValue ¶
type AttrValue_ListValue struct { S [][]byte `protobuf:"bytes,2,rep,name=s,proto3" json:"s,omitempty"` I []int64 `protobuf:"varint,3,rep,packed,name=i" json:"i,omitempty"` F []float32 `protobuf:"fixed32,4,rep,packed,name=f" json:"f,omitempty"` B []bool `protobuf:"varint,5,rep,packed,name=b" json:"b,omitempty"` Type []DataType `protobuf:"varint,6,rep,packed,name=type,enum=tensorflow.DataType" json:"type,omitempty"` Shape []*TensorShapeProto `protobuf:"bytes,7,rep,name=shape" json:"shape,omitempty"` Tensor []*TensorProto `protobuf:"bytes,8,rep,name=tensor" json:"tensor,omitempty"` Func []*NameAttrList `protobuf:"bytes,9,rep,name=func" json:"func,omitempty"` }
LINT.IfChange
func (*AttrValue_ListValue) Descriptor ¶
func (*AttrValue_ListValue) Descriptor() ([]byte, []int)
func (*AttrValue_ListValue) GetB ¶
func (m *AttrValue_ListValue) GetB() []bool
func (*AttrValue_ListValue) GetF ¶
func (m *AttrValue_ListValue) GetF() []float32
func (*AttrValue_ListValue) GetFunc ¶
func (m *AttrValue_ListValue) GetFunc() []*NameAttrList
func (*AttrValue_ListValue) GetI ¶
func (m *AttrValue_ListValue) GetI() []int64
func (*AttrValue_ListValue) GetS ¶
func (m *AttrValue_ListValue) GetS() [][]byte
func (*AttrValue_ListValue) GetShape ¶
func (m *AttrValue_ListValue) GetShape() []*TensorShapeProto
func (*AttrValue_ListValue) GetTensor ¶
func (m *AttrValue_ListValue) GetTensor() []*TensorProto
func (*AttrValue_ListValue) GetType ¶
func (m *AttrValue_ListValue) GetType() []DataType
func (*AttrValue_ListValue) ProtoMessage ¶
func (*AttrValue_ListValue) ProtoMessage()
func (*AttrValue_ListValue) Reset ¶
func (m *AttrValue_ListValue) Reset()
func (*AttrValue_ListValue) String ¶
func (m *AttrValue_ListValue) String() string
type AttrValue_Placeholder ¶
type AttrValue_Placeholder struct {
Placeholder string `protobuf:"bytes,9,opt,name=placeholder,oneof"`
}
type AttrValue_S ¶
type AttrValue_S struct {
S []byte `protobuf:"bytes,2,opt,name=s,proto3,oneof"`
}
type AttrValue_Shape ¶
type AttrValue_Shape struct {
Shape *TensorShapeProto `protobuf:"bytes,7,opt,name=shape,oneof"`
}
type AttrValue_Tensor ¶
type AttrValue_Tensor struct {
Tensor *TensorProto `protobuf:"bytes,8,opt,name=tensor,oneof"`
}
type AttrValue_Type ¶
type AttrValue_Type struct {
Type DataType `protobuf:"varint,6,opt,name=type,enum=tensorflow.DataType,oneof"`
}
type CostGraphDef ¶
type CostGraphDef struct {
Node []*CostGraphDef_Node `protobuf:"bytes,1,rep,name=node" json:"node,omitempty"`
}
func (*CostGraphDef) Descriptor ¶
func (*CostGraphDef) Descriptor() ([]byte, []int)
func (*CostGraphDef) GetNode ¶
func (m *CostGraphDef) GetNode() []*CostGraphDef_Node
func (*CostGraphDef) ProtoMessage ¶
func (*CostGraphDef) ProtoMessage()
func (*CostGraphDef) Reset ¶
func (m *CostGraphDef) Reset()
func (*CostGraphDef) String ¶
func (m *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" 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" json:"device,omitempty"` // The id of the node. Node ids are only unique inside a partition. Id int32 `protobuf:"varint,3,opt,name=id" json:"id,omitempty"` InputInfo []*CostGraphDef_Node_InputInfo `protobuf:"bytes,4,rep,name=input_info,json=inputInfo" json:"input_info,omitempty"` OutputInfo []*CostGraphDef_Node_OutputInfo `protobuf:"bytes,5,rep,name=output_info,json=outputInfo" json:"output_info,omitempty"` // Temporary memory used by this node. TemporaryMemorySize int64 `protobuf:"varint,6,opt,name=temporary_memory_size,json=temporaryMemorySize" json:"temporary_memory_size,omitempty"` HostTempMemorySize int64 `protobuf:"varint,10,opt,name=host_temp_memory_size,json=hostTempMemorySize" json:"host_temp_memory_size,omitempty"` DeviceTempMemorySize int64 `protobuf:"varint,11,opt,name=device_temp_memory_size,json=deviceTempMemorySize" json:"device_temp_memory_size,omitempty"` HostPersistentMemorySize int64 `` /* 132-byte string literal not displayed */ DevicePersistentMemorySize int64 `` /* 138-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" 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" 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" 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" json:"is_final,omitempty"` // Ids of the control inputs for this node. ControlInput []int32 `protobuf:"varint,8,rep,packed,name=control_input,json=controlInput" json:"control_input,omitempty"` }
func (*CostGraphDef_Node) Descriptor ¶
func (*CostGraphDef_Node) Descriptor() ([]byte, []int)
func (*CostGraphDef_Node) GetComputeCost ¶
func (m *CostGraphDef_Node) GetComputeCost() int64
func (*CostGraphDef_Node) GetComputeTime ¶
func (m *CostGraphDef_Node) GetComputeTime() int64
func (*CostGraphDef_Node) GetControlInput ¶
func (m *CostGraphDef_Node) GetControlInput() []int32
func (*CostGraphDef_Node) GetDevice ¶
func (m *CostGraphDef_Node) GetDevice() string
func (*CostGraphDef_Node) GetDevicePersistentMemorySize ¶
func (m *CostGraphDef_Node) GetDevicePersistentMemorySize() int64
func (*CostGraphDef_Node) GetDeviceTempMemorySize ¶
func (m *CostGraphDef_Node) GetDeviceTempMemorySize() int64
func (*CostGraphDef_Node) GetHostPersistentMemorySize ¶
func (m *CostGraphDef_Node) GetHostPersistentMemorySize() int64
func (*CostGraphDef_Node) GetHostTempMemorySize ¶
func (m *CostGraphDef_Node) GetHostTempMemorySize() int64
func (*CostGraphDef_Node) GetId ¶
func (m *CostGraphDef_Node) GetId() int32
func (*CostGraphDef_Node) GetInputInfo ¶
func (m *CostGraphDef_Node) GetInputInfo() []*CostGraphDef_Node_InputInfo
func (*CostGraphDef_Node) GetIsFinal ¶
func (m *CostGraphDef_Node) GetIsFinal() bool
func (*CostGraphDef_Node) GetMemoryTime ¶
func (m *CostGraphDef_Node) GetMemoryTime() int64
func (*CostGraphDef_Node) GetName ¶
func (m *CostGraphDef_Node) GetName() string
func (*CostGraphDef_Node) GetOutputInfo ¶
func (m *CostGraphDef_Node) GetOutputInfo() []*CostGraphDef_Node_OutputInfo
func (*CostGraphDef_Node) GetTemporaryMemorySize ¶
func (m *CostGraphDef_Node) GetTemporaryMemorySize() int64
func (*CostGraphDef_Node) ProtoMessage ¶
func (*CostGraphDef_Node) ProtoMessage()
func (*CostGraphDef_Node) Reset ¶
func (m *CostGraphDef_Node) Reset()
func (*CostGraphDef_Node) String ¶
func (m *CostGraphDef_Node) String() string
type CostGraphDef_Node_InputInfo ¶
type CostGraphDef_Node_InputInfo struct { PrecedingNode int32 `protobuf:"varint,1,opt,name=preceding_node,json=precedingNode" json:"preceding_node,omitempty"` PrecedingPort int32 `protobuf:"varint,2,opt,name=preceding_port,json=precedingPort" json:"preceding_port,omitempty"` }
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 ¶
func (*CostGraphDef_Node_InputInfo) Descriptor() ([]byte, []int)
func (*CostGraphDef_Node_InputInfo) GetPrecedingNode ¶
func (m *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32
func (*CostGraphDef_Node_InputInfo) GetPrecedingPort ¶
func (m *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32
func (*CostGraphDef_Node_InputInfo) ProtoMessage ¶
func (*CostGraphDef_Node_InputInfo) ProtoMessage()
func (*CostGraphDef_Node_InputInfo) Reset ¶
func (m *CostGraphDef_Node_InputInfo) Reset()
func (*CostGraphDef_Node_InputInfo) String ¶
func (m *CostGraphDef_Node_InputInfo) String() string
type CostGraphDef_Node_OutputInfo ¶
type CostGraphDef_Node_OutputInfo struct { Size int64 `protobuf:"varint,1,opt,name=size" 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" json:"alias_input_port,omitempty"` Shape *TensorShapeProto `protobuf:"bytes,3,opt,name=shape" json:"shape,omitempty"` Dtype DataType `protobuf:"varint,4,opt,name=dtype,enum=tensorflow.DataType" json:"dtype,omitempty"` }
Outputs of this node.
func (*CostGraphDef_Node_OutputInfo) Descriptor ¶
func (*CostGraphDef_Node_OutputInfo) Descriptor() ([]byte, []int)
func (*CostGraphDef_Node_OutputInfo) GetAliasInputPort ¶
func (m *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64
func (*CostGraphDef_Node_OutputInfo) GetDtype ¶
func (m *CostGraphDef_Node_OutputInfo) GetDtype() DataType
func (*CostGraphDef_Node_OutputInfo) GetShape ¶
func (m *CostGraphDef_Node_OutputInfo) GetShape() *TensorShapeProto
func (*CostGraphDef_Node_OutputInfo) GetSize ¶
func (m *CostGraphDef_Node_OutputInfo) GetSize() int64
func (*CostGraphDef_Node_OutputInfo) ProtoMessage ¶
func (*CostGraphDef_Node_OutputInfo) ProtoMessage()
func (*CostGraphDef_Node_OutputInfo) Reset ¶
func (m *CostGraphDef_Node_OutputInfo) Reset()
func (*CostGraphDef_Node_OutputInfo) String ¶
func (m *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 DataType_DT_INT64 DataType = 9 DataType_DT_BOOL DataType = 10 DataType_DT_QINT8 DataType = 11 DataType_DT_QUINT8 DataType = 12 DataType_DT_QINT32 DataType = 13 DataType_DT_BFLOAT16 DataType = 14 DataType_DT_QINT16 DataType = 15 DataType_DT_QUINT16 DataType = 16 DataType_DT_UINT16 DataType = 17 DataType_DT_COMPLEX128 DataType = 18 DataType_DT_HALF DataType = 19 DataType_DT_RESOURCE DataType = 20 DataType_DT_VARIANT DataType = 21 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) EnumDescriptor ¶
type DeviceAttributes ¶
type DeviceAttributes struct { // Fully specified name of the device within a cluster. Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` // String representation of device_type. DeviceType string `protobuf:"bytes,2,opt,name=device_type,json=deviceType" json:"device_type,omitempty"` // Memory capacity of device in bytes. MemoryLimit int64 `protobuf:"varint,4,opt,name=memory_limit,json=memoryLimit" 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" 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" 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" json:"physical_device_desc,omitempty"` }
func (*DeviceAttributes) Descriptor ¶
func (*DeviceAttributes) Descriptor() ([]byte, []int)
func (*DeviceAttributes) GetDeviceType ¶
func (m *DeviceAttributes) GetDeviceType() string
func (*DeviceAttributes) GetIncarnation ¶
func (m *DeviceAttributes) GetIncarnation() uint64
func (*DeviceAttributes) GetLocality ¶
func (m *DeviceAttributes) GetLocality() *DeviceLocality
func (*DeviceAttributes) GetMemoryLimit ¶
func (m *DeviceAttributes) GetMemoryLimit() int64
func (*DeviceAttributes) GetName ¶
func (m *DeviceAttributes) GetName() string
func (*DeviceAttributes) GetPhysicalDeviceDesc ¶
func (m *DeviceAttributes) GetPhysicalDeviceDesc() string
func (*DeviceAttributes) ProtoMessage ¶
func (*DeviceAttributes) ProtoMessage()
func (*DeviceAttributes) Reset ¶
func (m *DeviceAttributes) Reset()
func (*DeviceAttributes) String ¶
func (m *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" json:"bus_id,omitempty"` }
func (*DeviceLocality) Descriptor ¶
func (*DeviceLocality) Descriptor() ([]byte, []int)
func (*DeviceLocality) GetBusId ¶
func (m *DeviceLocality) GetBusId() int32
func (*DeviceLocality) ProtoMessage ¶
func (*DeviceLocality) ProtoMessage()
func (*DeviceLocality) Reset ¶
func (m *DeviceLocality) Reset()
func (*DeviceLocality) String ¶
func (m *DeviceLocality) String() string
type DeviceStepStats ¶
type DeviceStepStats struct { Device string `protobuf:"bytes,1,opt,name=device" json:"device,omitempty"` NodeStats []*NodeExecStats `protobuf:"bytes,2,rep,name=node_stats,json=nodeStats" json:"node_stats,omitempty"` }
func (*DeviceStepStats) Descriptor ¶
func (*DeviceStepStats) Descriptor() ([]byte, []int)
func (*DeviceStepStats) GetDevice ¶
func (m *DeviceStepStats) GetDevice() string
func (*DeviceStepStats) GetNodeStats ¶
func (m *DeviceStepStats) GetNodeStats() []*NodeExecStats
func (*DeviceStepStats) ProtoMessage ¶
func (*DeviceStepStats) ProtoMessage()
func (*DeviceStepStats) Reset ¶
func (m *DeviceStepStats) Reset()
func (*DeviceStepStats) String ¶
func (m *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" json:"signature,omitempty"` // Attributes specific to this function definition. Attr map[string]*AttrValue `` /* 128-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" 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 `` /* 126-byte string literal not displayed */ }
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 ¶
func (*FunctionDef) Descriptor() ([]byte, []int)
func (*FunctionDef) GetAttr ¶
func (m *FunctionDef) GetAttr() map[string]*AttrValue
func (*FunctionDef) GetNodeDef ¶
func (m *FunctionDef) GetNodeDef() []*NodeDef
func (*FunctionDef) GetRet ¶
func (m *FunctionDef) GetRet() map[string]string
func (*FunctionDef) GetSignature ¶
func (m *FunctionDef) GetSignature() *OpDef
func (*FunctionDef) ProtoMessage ¶
func (*FunctionDef) ProtoMessage()
func (*FunctionDef) Reset ¶
func (m *FunctionDef) Reset()
func (*FunctionDef) String ¶
func (m *FunctionDef) String() string
type FunctionDefLibrary ¶
type FunctionDefLibrary struct { Function []*FunctionDef `protobuf:"bytes,1,rep,name=function" json:"function,omitempty"` Gradient []*GradientDef `protobuf:"bytes,2,rep,name=gradient" json:"gradient,omitempty"` }
A library is a set of named functions.
func (*FunctionDefLibrary) Descriptor ¶
func (*FunctionDefLibrary) Descriptor() ([]byte, []int)
func (*FunctionDefLibrary) GetFunction ¶
func (m *FunctionDefLibrary) GetFunction() []*FunctionDef
func (*FunctionDefLibrary) GetGradient ¶
func (m *FunctionDefLibrary) GetGradient() []*GradientDef
func (*FunctionDefLibrary) ProtoMessage ¶
func (*FunctionDefLibrary) ProtoMessage()
func (*FunctionDefLibrary) Reset ¶
func (m *FunctionDefLibrary) Reset()
func (*FunctionDefLibrary) String ¶
func (m *FunctionDefLibrary) String() string
type GradientDef ¶
type GradientDef struct { FunctionName string `protobuf:"bytes,1,opt,name=function_name,json=functionName" json:"function_name,omitempty"` GradientFunc string `protobuf:"bytes,2,opt,name=gradient_func,json=gradientFunc" json:"gradient_func,omitempty"` }
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 ¶
func (*GradientDef) Descriptor() ([]byte, []int)
func (*GradientDef) GetFunctionName ¶
func (m *GradientDef) GetFunctionName() string
func (*GradientDef) GetGradientFunc ¶
func (m *GradientDef) GetGradientFunc() string
func (*GradientDef) ProtoMessage ¶
func (*GradientDef) ProtoMessage()
func (*GradientDef) Reset ¶
func (m *GradientDef) Reset()
func (*GradientDef) String ¶
func (m *GradientDef) String() string
type GraphDef ¶
type GraphDef struct { Node []*NodeDef `protobuf:"bytes,1,rep,name=node" 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" 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. Version int32 `protobuf:"varint,3,opt,name=version" 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" json:"library,omitempty"` }
Represents the graph of operations
func (*GraphDef) Descriptor ¶
func (*GraphDef) GetLibrary ¶
func (m *GraphDef) GetLibrary() *FunctionDefLibrary
func (*GraphDef) GetVersion ¶
func (*GraphDef) GetVersions ¶
func (m *GraphDef) GetVersions() *VersionDef
func (*GraphDef) ProtoMessage ¶
func (*GraphDef) ProtoMessage()
type GraphTransferInfo ¶
type GraphTransferInfo struct { NodeInfo []*GraphTransferInfo_NodeInfo `protobuf:"bytes,1,rep,name=node_info,json=nodeInfo" json:"node_info,omitempty"` ConstNodeInfo []*GraphTransferInfo_ConstNodeInfo `protobuf:"bytes,2,rep,name=const_node_info,json=constNodeInfo" json:"const_node_info,omitempty"` NodeInputInfo []*GraphTransferInfo_NodeInputInfo `protobuf:"bytes,3,rep,name=node_input_info,json=nodeInputInfo" json:"node_input_info,omitempty"` NodeOutputInfo []*GraphTransferInfo_NodeOutputInfo `protobuf:"bytes,4,rep,name=node_output_info,json=nodeOutputInfo" json:"node_output_info,omitempty"` // Input Node parameters of transferred graph GraphInputNodeInfo []*GraphTransferInfo_GraphInputNodeInfo `protobuf:"bytes,5,rep,name=graph_input_node_info,json=graphInputNodeInfo" json:"graph_input_node_info,omitempty"` GraphOutputNodeInfo []*GraphTransferInfo_GraphOutputNodeInfo `protobuf:"bytes,6,rep,name=graph_output_node_info,json=graphOutputNodeInfo" json:"graph_output_node_info,omitempty"` // Destination of graph transfer Destination GraphTransferInfo_Destination `protobuf:"varint,7,opt,name=destination,enum=tensorflow.GraphTransferInfo_Destination" json:"destination,omitempty"` }
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 ¶
func (*GraphTransferInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo) GetConstNodeInfo ¶
func (m *GraphTransferInfo) GetConstNodeInfo() []*GraphTransferInfo_ConstNodeInfo
func (*GraphTransferInfo) GetDestination ¶
func (m *GraphTransferInfo) GetDestination() GraphTransferInfo_Destination
func (*GraphTransferInfo) GetGraphInputNodeInfo ¶
func (m *GraphTransferInfo) GetGraphInputNodeInfo() []*GraphTransferInfo_GraphInputNodeInfo
func (*GraphTransferInfo) GetGraphOutputNodeInfo ¶
func (m *GraphTransferInfo) GetGraphOutputNodeInfo() []*GraphTransferInfo_GraphOutputNodeInfo
func (*GraphTransferInfo) GetNodeInfo ¶
func (m *GraphTransferInfo) GetNodeInfo() []*GraphTransferInfo_NodeInfo
func (*GraphTransferInfo) GetNodeInputInfo ¶
func (m *GraphTransferInfo) GetNodeInputInfo() []*GraphTransferInfo_NodeInputInfo
func (*GraphTransferInfo) GetNodeOutputInfo ¶
func (m *GraphTransferInfo) GetNodeOutputInfo() []*GraphTransferInfo_NodeOutputInfo
func (*GraphTransferInfo) ProtoMessage ¶
func (*GraphTransferInfo) ProtoMessage()
func (*GraphTransferInfo) Reset ¶
func (m *GraphTransferInfo) Reset()
func (*GraphTransferInfo) String ¶
func (m *GraphTransferInfo) String() string
type GraphTransferInfo_ConstNodeInfo ¶
type GraphTransferInfo_ConstNodeInfo struct { Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` NodeId int32 `protobuf:"varint,2,opt,name=node_id,json=nodeId" json:"node_id,omitempty"` Shape []int64 `protobuf:"varint,3,rep,packed,name=shape" json:"shape,omitempty"` Data []byte `protobuf:"bytes,4,opt,name=data,proto3" json:"data,omitempty"` Dtype DataType `protobuf:"varint,5,opt,name=dtype,enum=tensorflow.DataType" json:"dtype,omitempty"` }
func (*GraphTransferInfo_ConstNodeInfo) Descriptor ¶
func (*GraphTransferInfo_ConstNodeInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_ConstNodeInfo) GetData ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetData() []byte
func (*GraphTransferInfo_ConstNodeInfo) GetDtype ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetDtype() DataType
func (*GraphTransferInfo_ConstNodeInfo) GetName ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetName() string
func (*GraphTransferInfo_ConstNodeInfo) GetNodeId ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetNodeId() int32
func (*GraphTransferInfo_ConstNodeInfo) GetShape ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetShape() []int64
func (*GraphTransferInfo_ConstNodeInfo) ProtoMessage ¶
func (*GraphTransferInfo_ConstNodeInfo) ProtoMessage()
func (*GraphTransferInfo_ConstNodeInfo) Reset ¶
func (m *GraphTransferInfo_ConstNodeInfo) Reset()
func (*GraphTransferInfo_ConstNodeInfo) String ¶
func (m *GraphTransferInfo_ConstNodeInfo) String() string
type GraphTransferInfo_Destination ¶
type GraphTransferInfo_Destination int32
const ( GraphTransferInfo_NOP GraphTransferInfo_Destination = 0 GraphTransferInfo_HEXAGON GraphTransferInfo_Destination = 1 )
func (GraphTransferInfo_Destination) EnumDescriptor ¶
func (GraphTransferInfo_Destination) EnumDescriptor() ([]byte, []int)
func (GraphTransferInfo_Destination) String ¶
func (x GraphTransferInfo_Destination) String() string
type GraphTransferInfo_GraphInputNodeInfo ¶
type GraphTransferInfo_GraphInputNodeInfo struct { Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` Shape []int64 `protobuf:"varint,2,rep,packed,name=shape" json:"shape,omitempty"` Dtype DataType `protobuf:"varint,3,opt,name=dtype,enum=tensorflow.DataType" json:"dtype,omitempty"` }
func (*GraphTransferInfo_GraphInputNodeInfo) Descriptor ¶
func (*GraphTransferInfo_GraphInputNodeInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_GraphInputNodeInfo) GetDtype ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) GetDtype() DataType
func (*GraphTransferInfo_GraphInputNodeInfo) GetName ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) GetName() string
func (*GraphTransferInfo_GraphInputNodeInfo) GetShape ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) GetShape() []int64
func (*GraphTransferInfo_GraphInputNodeInfo) ProtoMessage ¶
func (*GraphTransferInfo_GraphInputNodeInfo) ProtoMessage()
func (*GraphTransferInfo_GraphInputNodeInfo) Reset ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) Reset()
func (*GraphTransferInfo_GraphInputNodeInfo) String ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) String() string
type GraphTransferInfo_GraphOutputNodeInfo ¶
type GraphTransferInfo_GraphOutputNodeInfo struct { Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` Shape []int64 `protobuf:"varint,2,rep,packed,name=shape" json:"shape,omitempty"` Dtype DataType `protobuf:"varint,3,opt,name=dtype,enum=tensorflow.DataType" json:"dtype,omitempty"` }
func (*GraphTransferInfo_GraphOutputNodeInfo) Descriptor ¶
func (*GraphTransferInfo_GraphOutputNodeInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_GraphOutputNodeInfo) GetDtype ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) GetDtype() DataType
func (*GraphTransferInfo_GraphOutputNodeInfo) GetName ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) GetName() string
func (*GraphTransferInfo_GraphOutputNodeInfo) GetShape ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) GetShape() []int64
func (*GraphTransferInfo_GraphOutputNodeInfo) ProtoMessage ¶
func (*GraphTransferInfo_GraphOutputNodeInfo) ProtoMessage()
func (*GraphTransferInfo_GraphOutputNodeInfo) Reset ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) Reset()
func (*GraphTransferInfo_GraphOutputNodeInfo) String ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) String() string
type GraphTransferInfo_NodeInfo ¶
type GraphTransferInfo_NodeInfo struct { Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` NodeId int32 `protobuf:"varint,2,opt,name=node_id,json=nodeId" json:"node_id,omitempty"` TypeName string `protobuf:"bytes,3,opt,name=type_name,json=typeName" json:"type_name,omitempty"` SocOpId int32 `protobuf:"varint,4,opt,name=soc_op_id,json=socOpId" json:"soc_op_id,omitempty"` PaddingId int32 `protobuf:"varint,5,opt,name=padding_id,json=paddingId" json:"padding_id,omitempty"` InputCount int32 `protobuf:"varint,6,opt,name=input_count,json=inputCount" json:"input_count,omitempty"` OutputCount int32 `protobuf:"varint,7,opt,name=output_count,json=outputCount" json:"output_count,omitempty"` }
func (*GraphTransferInfo_NodeInfo) Descriptor ¶
func (*GraphTransferInfo_NodeInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_NodeInfo) GetInputCount ¶
func (m *GraphTransferInfo_NodeInfo) GetInputCount() int32
func (*GraphTransferInfo_NodeInfo) GetName ¶
func (m *GraphTransferInfo_NodeInfo) GetName() string
func (*GraphTransferInfo_NodeInfo) GetNodeId ¶
func (m *GraphTransferInfo_NodeInfo) GetNodeId() int32
func (*GraphTransferInfo_NodeInfo) GetOutputCount ¶
func (m *GraphTransferInfo_NodeInfo) GetOutputCount() int32
func (*GraphTransferInfo_NodeInfo) GetPaddingId ¶
func (m *GraphTransferInfo_NodeInfo) GetPaddingId() int32
func (*GraphTransferInfo_NodeInfo) GetSocOpId ¶
func (m *GraphTransferInfo_NodeInfo) GetSocOpId() int32
func (*GraphTransferInfo_NodeInfo) GetTypeName ¶
func (m *GraphTransferInfo_NodeInfo) GetTypeName() string
func (*GraphTransferInfo_NodeInfo) ProtoMessage ¶
func (*GraphTransferInfo_NodeInfo) ProtoMessage()
func (*GraphTransferInfo_NodeInfo) Reset ¶
func (m *GraphTransferInfo_NodeInfo) Reset()
func (*GraphTransferInfo_NodeInfo) String ¶
func (m *GraphTransferInfo_NodeInfo) String() string
type GraphTransferInfo_NodeInput ¶
type GraphTransferInfo_NodeInput struct { NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId" json:"node_id,omitempty"` OutputPort int32 `protobuf:"varint,2,opt,name=output_port,json=outputPort" json:"output_port,omitempty"` }
func (*GraphTransferInfo_NodeInput) Descriptor ¶
func (*GraphTransferInfo_NodeInput) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_NodeInput) GetNodeId ¶
func (m *GraphTransferInfo_NodeInput) GetNodeId() int32
func (*GraphTransferInfo_NodeInput) GetOutputPort ¶
func (m *GraphTransferInfo_NodeInput) GetOutputPort() int32
func (*GraphTransferInfo_NodeInput) ProtoMessage ¶
func (*GraphTransferInfo_NodeInput) ProtoMessage()
func (*GraphTransferInfo_NodeInput) Reset ¶
func (m *GraphTransferInfo_NodeInput) Reset()
func (*GraphTransferInfo_NodeInput) String ¶
func (m *GraphTransferInfo_NodeInput) String() string
type GraphTransferInfo_NodeInputInfo ¶
type GraphTransferInfo_NodeInputInfo struct { NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId" json:"node_id,omitempty"` NodeInput []*GraphTransferInfo_NodeInput `protobuf:"bytes,2,rep,name=node_input,json=nodeInput" json:"node_input,omitempty"` }
func (*GraphTransferInfo_NodeInputInfo) Descriptor ¶
func (*GraphTransferInfo_NodeInputInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_NodeInputInfo) GetNodeId ¶
func (m *GraphTransferInfo_NodeInputInfo) GetNodeId() int32
func (*GraphTransferInfo_NodeInputInfo) GetNodeInput ¶
func (m *GraphTransferInfo_NodeInputInfo) GetNodeInput() []*GraphTransferInfo_NodeInput
func (*GraphTransferInfo_NodeInputInfo) ProtoMessage ¶
func (*GraphTransferInfo_NodeInputInfo) ProtoMessage()
func (*GraphTransferInfo_NodeInputInfo) Reset ¶
func (m *GraphTransferInfo_NodeInputInfo) Reset()
func (*GraphTransferInfo_NodeInputInfo) String ¶
func (m *GraphTransferInfo_NodeInputInfo) String() string
type GraphTransferInfo_NodeOutputInfo ¶
type GraphTransferInfo_NodeOutputInfo struct { NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId" json:"node_id,omitempty"` MaxByteSize []int32 `protobuf:"varint,2,rep,packed,name=max_byte_size,json=maxByteSize" json:"max_byte_size,omitempty"` }
func (*GraphTransferInfo_NodeOutputInfo) Descriptor ¶
func (*GraphTransferInfo_NodeOutputInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_NodeOutputInfo) GetMaxByteSize ¶
func (m *GraphTransferInfo_NodeOutputInfo) GetMaxByteSize() []int32
func (*GraphTransferInfo_NodeOutputInfo) GetNodeId ¶
func (m *GraphTransferInfo_NodeOutputInfo) GetNodeId() int32
func (*GraphTransferInfo_NodeOutputInfo) ProtoMessage ¶
func (*GraphTransferInfo_NodeOutputInfo) ProtoMessage()
func (*GraphTransferInfo_NodeOutputInfo) Reset ¶
func (m *GraphTransferInfo_NodeOutputInfo) Reset()
func (*GraphTransferInfo_NodeOutputInfo) String ¶
func (m *GraphTransferInfo_NodeOutputInfo) String() string
type HistogramProto ¶
type HistogramProto struct { Min float64 `protobuf:"fixed64,1,opt,name=min" json:"min,omitempty"` Max float64 `protobuf:"fixed64,2,opt,name=max" json:"max,omitempty"` Num float64 `protobuf:"fixed64,3,opt,name=num" json:"num,omitempty"` Sum float64 `protobuf:"fixed64,4,opt,name=sum" json:"sum,omitempty"` SumSquares float64 `protobuf:"fixed64,5,opt,name=sum_squares,json=sumSquares" 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" json:"bucket_limit,omitempty"` Bucket []float64 `protobuf:"fixed64,7,rep,packed,name=bucket" json:"bucket,omitempty"` }
Serialization format for histogram module in core/lib/histogram/histogram.h
func (*HistogramProto) Descriptor ¶
func (*HistogramProto) Descriptor() ([]byte, []int)
func (*HistogramProto) GetBucket ¶
func (m *HistogramProto) GetBucket() []float64
func (*HistogramProto) GetBucketLimit ¶
func (m *HistogramProto) GetBucketLimit() []float64
func (*HistogramProto) GetMax ¶
func (m *HistogramProto) GetMax() float64
func (*HistogramProto) GetMin ¶
func (m *HistogramProto) GetMin() float64
func (*HistogramProto) GetNum ¶
func (m *HistogramProto) GetNum() float64
func (*HistogramProto) GetSum ¶
func (m *HistogramProto) GetSum() float64
func (*HistogramProto) GetSumSquares ¶
func (m *HistogramProto) GetSumSquares() float64
func (*HistogramProto) ProtoMessage ¶
func (*HistogramProto) ProtoMessage()
func (*HistogramProto) Reset ¶
func (m *HistogramProto) Reset()
func (*HistogramProto) String ¶
func (m *HistogramProto) String() string
type KernelDef ¶
type KernelDef struct { // Must match the name of an Op. Op string `protobuf:"bytes,1,opt,name=op" json:"op,omitempty"` // Type of device this kernel runs on. DeviceType string `protobuf:"bytes,2,opt,name=device_type,json=deviceType" json:"device_type,omitempty"` Constraint []*KernelDef_AttrConstraint `protobuf:"bytes,3,rep,name=constraint" 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" 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" json:"label,omitempty"` }
func (*KernelDef) Descriptor ¶
func (*KernelDef) GetConstraint ¶
func (m *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
func (*KernelDef) GetDeviceType ¶
func (*KernelDef) GetHostMemoryArg ¶
func (*KernelDef) ProtoMessage ¶
func (*KernelDef) ProtoMessage()
type KernelDef_AttrConstraint ¶
type KernelDef_AttrConstraint struct { // Name of an attr from the Op. Name string `protobuf:"bytes,1,opt,name=name" 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" json:"allowed_values,omitempty"` }
func (*KernelDef_AttrConstraint) Descriptor ¶
func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)
func (*KernelDef_AttrConstraint) GetAllowedValues ¶
func (m *KernelDef_AttrConstraint) GetAllowedValues() *AttrValue
func (*KernelDef_AttrConstraint) GetName ¶
func (m *KernelDef_AttrConstraint) GetName() string
func (*KernelDef_AttrConstraint) ProtoMessage ¶
func (*KernelDef_AttrConstraint) ProtoMessage()
func (*KernelDef_AttrConstraint) Reset ¶
func (m *KernelDef_AttrConstraint) Reset()
func (*KernelDef_AttrConstraint) String ¶
func (m *KernelDef_AttrConstraint) String() string
type MemoryLogRawAllocation ¶
type MemoryLogRawAllocation struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId" json:"step_id,omitempty"` // Name of the operation making the allocation. Operation string `protobuf:"bytes,2,opt,name=operation" json:"operation,omitempty"` // Number of bytes in the allocation. NumBytes int64 `protobuf:"varint,3,opt,name=num_bytes,json=numBytes" json:"num_bytes,omitempty"` // Address of the allocation. Ptr uint64 `protobuf:"varint,4,opt,name=ptr" 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" json:"allocation_id,omitempty"` // Name of the allocator used. AllocatorName string `protobuf:"bytes,6,opt,name=allocator_name,json=allocatorName" json:"allocator_name,omitempty"` }
func (*MemoryLogRawAllocation) Descriptor ¶
func (*MemoryLogRawAllocation) Descriptor() ([]byte, []int)
func (*MemoryLogRawAllocation) GetAllocationId ¶
func (m *MemoryLogRawAllocation) GetAllocationId() int64
func (*MemoryLogRawAllocation) GetAllocatorName ¶
func (m *MemoryLogRawAllocation) GetAllocatorName() string
func (*MemoryLogRawAllocation) GetNumBytes ¶
func (m *MemoryLogRawAllocation) GetNumBytes() int64
func (*MemoryLogRawAllocation) GetOperation ¶
func (m *MemoryLogRawAllocation) GetOperation() string
func (*MemoryLogRawAllocation) GetPtr ¶
func (m *MemoryLogRawAllocation) GetPtr() uint64
func (*MemoryLogRawAllocation) GetStepId ¶
func (m *MemoryLogRawAllocation) GetStepId() int64
func (*MemoryLogRawAllocation) ProtoMessage ¶
func (*MemoryLogRawAllocation) ProtoMessage()
func (*MemoryLogRawAllocation) Reset ¶
func (m *MemoryLogRawAllocation) Reset()
func (*MemoryLogRawAllocation) String ¶
func (m *MemoryLogRawAllocation) String() string
type MemoryLogRawDeallocation ¶
type MemoryLogRawDeallocation struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId" json:"step_id,omitempty"` // Name of the operation making the deallocation. Operation string `protobuf:"bytes,2,opt,name=operation" 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" json:"allocation_id,omitempty"` // Name of the allocator used. AllocatorName string `protobuf:"bytes,4,opt,name=allocator_name,json=allocatorName" 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" json:"deferred,omitempty"` }
func (*MemoryLogRawDeallocation) Descriptor ¶
func (*MemoryLogRawDeallocation) Descriptor() ([]byte, []int)
func (*MemoryLogRawDeallocation) GetAllocationId ¶
func (m *MemoryLogRawDeallocation) GetAllocationId() int64
func (*MemoryLogRawDeallocation) GetAllocatorName ¶
func (m *MemoryLogRawDeallocation) GetAllocatorName() string
func (*MemoryLogRawDeallocation) GetDeferred ¶
func (m *MemoryLogRawDeallocation) GetDeferred() bool
func (*MemoryLogRawDeallocation) GetOperation ¶
func (m *MemoryLogRawDeallocation) GetOperation() string
func (*MemoryLogRawDeallocation) GetStepId ¶
func (m *MemoryLogRawDeallocation) GetStepId() int64
func (*MemoryLogRawDeallocation) ProtoMessage ¶
func (*MemoryLogRawDeallocation) ProtoMessage()
func (*MemoryLogRawDeallocation) Reset ¶
func (m *MemoryLogRawDeallocation) Reset()
func (*MemoryLogRawDeallocation) String ¶
func (m *MemoryLogRawDeallocation) String() string
type MemoryLogStep ¶
type MemoryLogStep struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId" json:"step_id,omitempty"` // Handle describing the feeds and fetches of the step. Handle string `protobuf:"bytes,2,opt,name=handle" json:"handle,omitempty"` }
func (*MemoryLogStep) Descriptor ¶
func (*MemoryLogStep) Descriptor() ([]byte, []int)
func (*MemoryLogStep) GetHandle ¶
func (m *MemoryLogStep) GetHandle() string
func (*MemoryLogStep) GetStepId ¶
func (m *MemoryLogStep) GetStepId() int64
func (*MemoryLogStep) ProtoMessage ¶
func (*MemoryLogStep) ProtoMessage()
func (*MemoryLogStep) Reset ¶
func (m *MemoryLogStep) Reset()
func (*MemoryLogStep) String ¶
func (m *MemoryLogStep) String() string
type MemoryLogTensorAllocation ¶
type MemoryLogTensorAllocation struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId" 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" json:"kernel_name,omitempty"` // Allocated tensor details. Tensor *TensorDescription `protobuf:"bytes,3,opt,name=tensor" json:"tensor,omitempty"` }
func (*MemoryLogTensorAllocation) Descriptor ¶
func (*MemoryLogTensorAllocation) Descriptor() ([]byte, []int)
func (*MemoryLogTensorAllocation) GetKernelName ¶
func (m *MemoryLogTensorAllocation) GetKernelName() string
func (*MemoryLogTensorAllocation) GetStepId ¶
func (m *MemoryLogTensorAllocation) GetStepId() int64
func (*MemoryLogTensorAllocation) GetTensor ¶
func (m *MemoryLogTensorAllocation) GetTensor() *TensorDescription
func (*MemoryLogTensorAllocation) ProtoMessage ¶
func (*MemoryLogTensorAllocation) ProtoMessage()
func (*MemoryLogTensorAllocation) Reset ¶
func (m *MemoryLogTensorAllocation) Reset()
func (*MemoryLogTensorAllocation) String ¶
func (m *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" json:"allocation_id,omitempty"` // Name of the allocator used. AllocatorName string `protobuf:"bytes,2,opt,name=allocator_name,json=allocatorName" json:"allocator_name,omitempty"` }
func (*MemoryLogTensorDeallocation) Descriptor ¶
func (*MemoryLogTensorDeallocation) Descriptor() ([]byte, []int)
func (*MemoryLogTensorDeallocation) GetAllocationId ¶
func (m *MemoryLogTensorDeallocation) GetAllocationId() int64
func (*MemoryLogTensorDeallocation) GetAllocatorName ¶
func (m *MemoryLogTensorDeallocation) GetAllocatorName() string
func (*MemoryLogTensorDeallocation) ProtoMessage ¶
func (*MemoryLogTensorDeallocation) ProtoMessage()
func (*MemoryLogTensorDeallocation) Reset ¶
func (m *MemoryLogTensorDeallocation) Reset()
func (*MemoryLogTensorDeallocation) String ¶
func (m *MemoryLogTensorDeallocation) String() string
type MemoryLogTensorOutput ¶
type MemoryLogTensorOutput struct { // Process-unique step id. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId" 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" json:"kernel_name,omitempty"` // Index of the output being set. Index int32 `protobuf:"varint,3,opt,name=index" json:"index,omitempty"` // Output tensor details. Tensor *TensorDescription `protobuf:"bytes,4,opt,name=tensor" json:"tensor,omitempty"` }
func (*MemoryLogTensorOutput) Descriptor ¶
func (*MemoryLogTensorOutput) Descriptor() ([]byte, []int)
func (*MemoryLogTensorOutput) GetIndex ¶
func (m *MemoryLogTensorOutput) GetIndex() int32
func (*MemoryLogTensorOutput) GetKernelName ¶
func (m *MemoryLogTensorOutput) GetKernelName() string
func (*MemoryLogTensorOutput) GetStepId ¶
func (m *MemoryLogTensorOutput) GetStepId() int64
func (*MemoryLogTensorOutput) GetTensor ¶
func (m *MemoryLogTensorOutput) GetTensor() *TensorDescription
func (*MemoryLogTensorOutput) ProtoMessage ¶
func (*MemoryLogTensorOutput) ProtoMessage()
func (*MemoryLogTensorOutput) Reset ¶
func (m *MemoryLogTensorOutput) Reset()
func (*MemoryLogTensorOutput) String ¶
func (m *MemoryLogTensorOutput) String() string
type MemoryStats ¶
type MemoryStats struct { HostTempMemorySize int64 `protobuf:"varint,1,opt,name=host_temp_memory_size,json=hostTempMemorySize" json:"host_temp_memory_size,omitempty"` DeviceTempMemorySize int64 `protobuf:"varint,2,opt,name=device_temp_memory_size,json=deviceTempMemorySize" json:"device_temp_memory_size,omitempty"` HostPersistentMemorySize int64 `` /* 131-byte string literal not displayed */ DevicePersistentMemorySize int64 `` /* 137-byte string literal not displayed */ HostPersistentTensorAllocIds []int64 `` /* 152-byte string literal not displayed */ DevicePersistentTensorAllocIds []int64 `` /* 158-byte string literal not displayed */ }
For memory tracking.
func (*MemoryStats) Descriptor ¶
func (*MemoryStats) Descriptor() ([]byte, []int)
func (*MemoryStats) GetDevicePersistentMemorySize ¶
func (m *MemoryStats) GetDevicePersistentMemorySize() int64
func (*MemoryStats) GetDevicePersistentTensorAllocIds ¶
func (m *MemoryStats) GetDevicePersistentTensorAllocIds() []int64
func (*MemoryStats) GetDeviceTempMemorySize ¶
func (m *MemoryStats) GetDeviceTempMemorySize() int64
func (*MemoryStats) GetHostPersistentMemorySize ¶
func (m *MemoryStats) GetHostPersistentMemorySize() int64
func (*MemoryStats) GetHostPersistentTensorAllocIds ¶
func (m *MemoryStats) GetHostPersistentTensorAllocIds() []int64
func (*MemoryStats) GetHostTempMemorySize ¶
func (m *MemoryStats) GetHostTempMemorySize() int64
func (*MemoryStats) ProtoMessage ¶
func (*MemoryStats) ProtoMessage()
func (*MemoryStats) Reset ¶
func (m *MemoryStats) Reset()
func (*MemoryStats) String ¶
func (m *MemoryStats) String() string
type NameAttrList ¶
type NameAttrList struct { Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` Attr map[string]*AttrValue `` /* 128-byte string literal not displayed */ }
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 ¶
func (*NameAttrList) Descriptor() ([]byte, []int)
func (*NameAttrList) GetAttr ¶
func (m *NameAttrList) GetAttr() map[string]*AttrValue
func (*NameAttrList) GetName ¶
func (m *NameAttrList) GetName() string
func (*NameAttrList) ProtoMessage ¶
func (*NameAttrList) ProtoMessage()
func (*NameAttrList) Reset ¶
func (m *NameAttrList) Reset()
func (*NameAttrList) String ¶
func (m *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" 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" 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" 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]*) // | ( ("gpu" | "cpu") ":" ([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" 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 `` /* 128-byte string literal not displayed */ }
func (*NodeDef) Descriptor ¶
func (*NodeDef) ProtoMessage ¶
func (*NodeDef) ProtoMessage()
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" json:"node_name,omitempty"` AllStartMicros int64 `protobuf:"varint,2,opt,name=all_start_micros,json=allStartMicros" json:"all_start_micros,omitempty"` OpStartRelMicros int64 `protobuf:"varint,3,opt,name=op_start_rel_micros,json=opStartRelMicros" json:"op_start_rel_micros,omitempty"` OpEndRelMicros int64 `protobuf:"varint,4,opt,name=op_end_rel_micros,json=opEndRelMicros" json:"op_end_rel_micros,omitempty"` AllEndRelMicros int64 `protobuf:"varint,5,opt,name=all_end_rel_micros,json=allEndRelMicros" json:"all_end_rel_micros,omitempty"` Memory []*AllocatorMemoryUsed `protobuf:"bytes,6,rep,name=memory" json:"memory,omitempty"` Output []*NodeOutput `protobuf:"bytes,7,rep,name=output" json:"output,omitempty"` TimelineLabel string `protobuf:"bytes,8,opt,name=timeline_label,json=timelineLabel" json:"timeline_label,omitempty"` ScheduledMicros int64 `protobuf:"varint,9,opt,name=scheduled_micros,json=scheduledMicros" json:"scheduled_micros,omitempty"` ThreadId uint32 `protobuf:"varint,10,opt,name=thread_id,json=threadId" json:"thread_id,omitempty"` ReferencedTensor []*AllocationDescription `protobuf:"bytes,11,rep,name=referenced_tensor,json=referencedTensor" json:"referenced_tensor,omitempty"` MemoryStats *MemoryStats `protobuf:"bytes,12,opt,name=memory_stats,json=memoryStats" json:"memory_stats,omitempty"` }
Time/size stats recorded for a single execution of a graph node.
func (*NodeExecStats) Descriptor ¶
func (*NodeExecStats) Descriptor() ([]byte, []int)
func (*NodeExecStats) GetAllEndRelMicros ¶
func (m *NodeExecStats) GetAllEndRelMicros() int64
func (*NodeExecStats) GetAllStartMicros ¶
func (m *NodeExecStats) GetAllStartMicros() int64
func (*NodeExecStats) GetMemory ¶
func (m *NodeExecStats) GetMemory() []*AllocatorMemoryUsed
func (*NodeExecStats) GetMemoryStats ¶
func (m *NodeExecStats) GetMemoryStats() *MemoryStats
func (*NodeExecStats) GetNodeName ¶
func (m *NodeExecStats) GetNodeName() string
func (*NodeExecStats) GetOpEndRelMicros ¶
func (m *NodeExecStats) GetOpEndRelMicros() int64
func (*NodeExecStats) GetOpStartRelMicros ¶
func (m *NodeExecStats) GetOpStartRelMicros() int64
func (*NodeExecStats) GetOutput ¶
func (m *NodeExecStats) GetOutput() []*NodeOutput
func (*NodeExecStats) GetReferencedTensor ¶
func (m *NodeExecStats) GetReferencedTensor() []*AllocationDescription
func (*NodeExecStats) GetScheduledMicros ¶
func (m *NodeExecStats) GetScheduledMicros() int64
func (*NodeExecStats) GetThreadId ¶
func (m *NodeExecStats) GetThreadId() uint32
func (*NodeExecStats) GetTimelineLabel ¶
func (m *NodeExecStats) GetTimelineLabel() string
func (*NodeExecStats) ProtoMessage ¶
func (*NodeExecStats) ProtoMessage()
func (*NodeExecStats) Reset ¶
func (m *NodeExecStats) Reset()
func (*NodeExecStats) String ¶
func (m *NodeExecStats) String() string
type NodeOutput ¶
type NodeOutput struct { Slot int32 `protobuf:"varint,1,opt,name=slot" json:"slot,omitempty"` TensorDescription *TensorDescription `protobuf:"bytes,3,opt,name=tensor_description,json=tensorDescription" json:"tensor_description,omitempty"` }
Output sizes recorded for a single execution of a graph node.
func (*NodeOutput) Descriptor ¶
func (*NodeOutput) Descriptor() ([]byte, []int)
func (*NodeOutput) GetSlot ¶
func (m *NodeOutput) GetSlot() int32
func (*NodeOutput) GetTensorDescription ¶
func (m *NodeOutput) GetTensorDescription() *TensorDescription
func (*NodeOutput) ProtoMessage ¶
func (*NodeOutput) ProtoMessage()
func (*NodeOutput) Reset ¶
func (m *NodeOutput) Reset()
func (*NodeOutput) String ¶
func (m *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" json:"name,omitempty"` // Description of the input(s). InputArg []*OpDef_ArgDef `protobuf:"bytes,2,rep,name=input_arg,json=inputArg" json:"input_arg,omitempty"` // Description of the output(s). OutputArg []*OpDef_ArgDef `protobuf:"bytes,3,rep,name=output_arg,json=outputArg" json:"output_arg,omitempty"` Attr []*OpDef_AttrDef `protobuf:"bytes,4,rep,name=attr" json:"attr,omitempty"` // Optional deprecation based on GraphDef versions. Deprecation *OpDeprecation `protobuf:"bytes,8,opt,name=deprecation" json:"deprecation,omitempty"` // One-line human-readable description of what the Op does. Summary string `protobuf:"bytes,5,opt,name=summary" json:"summary,omitempty"` // Additional, longer human-readable description of what the Op does. Description string `protobuf:"bytes,6,opt,name=description" 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" 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" json:"is_aggregate,omitempty"` // 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" json:"is_stateful,omitempty"` // 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 `` /* 130-byte string literal not displayed */ }
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 ¶
func (*OpDef) GetAllowsUninitializedInput ¶
func (*OpDef) GetAttr ¶
func (m *OpDef) GetAttr() []*OpDef_AttrDef
func (*OpDef) GetDeprecation ¶
func (m *OpDef) GetDeprecation() *OpDeprecation
func (*OpDef) GetDescription ¶
func (*OpDef) GetInputArg ¶
func (m *OpDef) GetInputArg() []*OpDef_ArgDef
func (*OpDef) GetIsAggregate ¶
func (*OpDef) GetIsCommutative ¶
func (*OpDef) GetIsStateful ¶
func (*OpDef) GetOutputArg ¶
func (m *OpDef) GetOutputArg() []*OpDef_ArgDef
func (*OpDef) GetSummary ¶
func (*OpDef) ProtoMessage ¶
func (*OpDef) ProtoMessage()
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" json:"name,omitempty"` // Human readable description. Description string `protobuf:"bytes,2,opt,name=description" 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,enum=tensorflow.DataType" json:"type,omitempty"` TypeAttr string `protobuf:"bytes,4,opt,name=type_attr,json=typeAttr" json:"type_attr,omitempty"` NumberAttr string `protobuf:"bytes,5,opt,name=number_attr,json=numberAttr" json:"number_attr,omitempty"` // 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" 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" json:"is_ref,omitempty"` }
For describing inputs and outputs.
func (*OpDef_ArgDef) Descriptor ¶
func (*OpDef_ArgDef) Descriptor() ([]byte, []int)
func (*OpDef_ArgDef) GetDescription ¶
func (m *OpDef_ArgDef) GetDescription() string
func (*OpDef_ArgDef) GetIsRef ¶
func (m *OpDef_ArgDef) GetIsRef() bool
func (*OpDef_ArgDef) GetName ¶
func (m *OpDef_ArgDef) GetName() string
func (*OpDef_ArgDef) GetNumberAttr ¶
func (m *OpDef_ArgDef) GetNumberAttr() string
func (*OpDef_ArgDef) GetType ¶
func (m *OpDef_ArgDef) GetType() DataType
func (*OpDef_ArgDef) GetTypeAttr ¶
func (m *OpDef_ArgDef) GetTypeAttr() string
func (*OpDef_ArgDef) GetTypeListAttr ¶
func (m *OpDef_ArgDef) GetTypeListAttr() string
func (*OpDef_ArgDef) ProtoMessage ¶
func (*OpDef_ArgDef) ProtoMessage()
func (*OpDef_ArgDef) Reset ¶
func (m *OpDef_ArgDef) Reset()
func (*OpDef_ArgDef) String ¶
func (m *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" 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" 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" json:"default_value,omitempty"` // Human-readable description. Description string `protobuf:"bytes,4,opt,name=description" 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" json:"has_minimum,omitempty"` Minimum int64 `protobuf:"varint,6,opt,name=minimum" 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" json:"allowed_values,omitempty"` }
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 ¶
func (*OpDef_AttrDef) Descriptor() ([]byte, []int)
func (*OpDef_AttrDef) GetAllowedValues ¶
func (m *OpDef_AttrDef) GetAllowedValues() *AttrValue
func (*OpDef_AttrDef) GetDefaultValue ¶
func (m *OpDef_AttrDef) GetDefaultValue() *AttrValue
func (*OpDef_AttrDef) GetDescription ¶
func (m *OpDef_AttrDef) GetDescription() string
func (*OpDef_AttrDef) GetHasMinimum ¶
func (m *OpDef_AttrDef) GetHasMinimum() bool
func (*OpDef_AttrDef) GetMinimum ¶
func (m *OpDef_AttrDef) GetMinimum() int64
func (*OpDef_AttrDef) GetName ¶
func (m *OpDef_AttrDef) GetName() string
func (*OpDef_AttrDef) GetType ¶
func (m *OpDef_AttrDef) GetType() string
func (*OpDef_AttrDef) ProtoMessage ¶
func (*OpDef_AttrDef) ProtoMessage()
func (*OpDef_AttrDef) Reset ¶
func (m *OpDef_AttrDef) Reset()
func (*OpDef_AttrDef) String ¶
func (m *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" json:"version,omitempty"` // Explanation of why it was deprecated and what to use instead. Explanation string `protobuf:"bytes,2,opt,name=explanation" json:"explanation,omitempty"` }
Information about version-dependent deprecation of an op
func (*OpDeprecation) Descriptor ¶
func (*OpDeprecation) Descriptor() ([]byte, []int)
func (*OpDeprecation) GetExplanation ¶
func (m *OpDeprecation) GetExplanation() string
func (*OpDeprecation) GetVersion ¶
func (m *OpDeprecation) GetVersion() int32
func (*OpDeprecation) ProtoMessage ¶
func (*OpDeprecation) ProtoMessage()
func (*OpDeprecation) Reset ¶
func (m *OpDeprecation) Reset()
func (*OpDeprecation) String ¶
func (m *OpDeprecation) String() string
type OpGenOverride ¶
type OpGenOverride struct { // Name of the op to apply overrides to. Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` // Do not include this op in the generated API. // If `skip` is true, all other overrides are ignored for this op. Skip bool `protobuf:"varint,2,opt,name=skip" json:"skip,omitempty"` // Hide this op by putting it into an internal namespace (or whatever // is appropriate in the target language). Hide bool `protobuf:"varint,3,opt,name=hide" json:"hide,omitempty"` // Use a different name in the API than the op's name. Note that // the op's name in `backticks` will also be replaced in the docs. RenameTo string `protobuf:"bytes,4,opt,name=rename_to,json=renameTo" json:"rename_to,omitempty"` // Create *additional* API endpoints with different names (contrast // with rename_to, which affects the original name). Alias []string `protobuf:"bytes,5,rep,name=alias" json:"alias,omitempty"` AttrDefault []*OpGenOverride_AttrDefault `protobuf:"bytes,6,rep,name=attr_default,json=attrDefault" json:"attr_default,omitempty"` AttrRename []*OpGenOverride_Rename `protobuf:"bytes,7,rep,name=attr_rename,json=attrRename" json:"attr_rename,omitempty"` InputRename []*OpGenOverride_Rename `protobuf:"bytes,8,rep,name=input_rename,json=inputRename" json:"input_rename,omitempty"` OutputRename []*OpGenOverride_Rename `protobuf:"bytes,9,rep,name=output_rename,json=outputRename" json:"output_rename,omitempty"` }
Used to override the default API & behavior in the generated code for client languages, from what you would get from the OpDef alone. This is so we can evolve the API while remaining backwards compatible when interpretting old graphs. Overrides go in an "op_gen_overrides.pbtxt" file with a text-format OpGenOverrides message. Right now these only apply to the C++ API. TODO(josh11b): In the future there will be a common set of overrides and per-client-language overrides.
WARNING: Be *very* careful using these features -- these overrides 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 (*OpGenOverride) Descriptor ¶
func (*OpGenOverride) Descriptor() ([]byte, []int)
func (*OpGenOverride) GetAlias ¶
func (m *OpGenOverride) GetAlias() []string
func (*OpGenOverride) GetAttrDefault ¶
func (m *OpGenOverride) GetAttrDefault() []*OpGenOverride_AttrDefault
func (*OpGenOverride) GetAttrRename ¶
func (m *OpGenOverride) GetAttrRename() []*OpGenOverride_Rename
func (*OpGenOverride) GetHide ¶
func (m *OpGenOverride) GetHide() bool
func (*OpGenOverride) GetInputRename ¶
func (m *OpGenOverride) GetInputRename() []*OpGenOverride_Rename
func (*OpGenOverride) GetName ¶
func (m *OpGenOverride) GetName() string
func (*OpGenOverride) GetOutputRename ¶
func (m *OpGenOverride) GetOutputRename() []*OpGenOverride_Rename
func (*OpGenOverride) GetRenameTo ¶
func (m *OpGenOverride) GetRenameTo() string
func (*OpGenOverride) GetSkip ¶
func (m *OpGenOverride) GetSkip() bool
func (*OpGenOverride) ProtoMessage ¶
func (*OpGenOverride) ProtoMessage()
func (*OpGenOverride) Reset ¶
func (m *OpGenOverride) Reset()
func (*OpGenOverride) String ¶
func (m *OpGenOverride) String() string
type OpGenOverride_AttrDefault ¶
type OpGenOverride_AttrDefault struct { Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"` Value *AttrValue `protobuf:"bytes,2,opt,name=value" json:"value,omitempty"` }
Map the name of an attr to a new default value to use. 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. If this attr is also renamed (using attr_rename below), use the original name of the attr.
func (*OpGenOverride_AttrDefault) Descriptor ¶
func (*OpGenOverride_AttrDefault) Descriptor() ([]byte, []int)
func (*OpGenOverride_AttrDefault) GetName ¶
func (m *OpGenOverride_AttrDefault) GetName() string
func (*OpGenOverride_AttrDefault) GetValue ¶
func (m *OpGenOverride_AttrDefault) GetValue() *AttrValue
func (*OpGenOverride_AttrDefault) ProtoMessage ¶
func (*OpGenOverride_AttrDefault) ProtoMessage()
func (*OpGenOverride_AttrDefault) Reset ¶
func (m *OpGenOverride_AttrDefault) Reset()
func (*OpGenOverride_AttrDefault) String ¶
func (m *OpGenOverride_AttrDefault) String() string
type OpGenOverride_Rename ¶
type OpGenOverride_Rename struct { From string `protobuf:"bytes,1,opt,name=from" json:"from,omitempty"` To string `protobuf:"bytes,2,opt,name=to" json:"to,omitempty"` }
Change the name used to access attrs/inputs/outputs in the API from what is used in the GraphDef. Note that these names in `backticks` will also be replaced in the docs.
func (*OpGenOverride_Rename) Descriptor ¶
func (*OpGenOverride_Rename) Descriptor() ([]byte, []int)
func (*OpGenOverride_Rename) GetFrom ¶
func (m *OpGenOverride_Rename) GetFrom() string
func (*OpGenOverride_Rename) GetTo ¶
func (m *OpGenOverride_Rename) GetTo() string
func (*OpGenOverride_Rename) ProtoMessage ¶
func (*OpGenOverride_Rename) ProtoMessage()
func (*OpGenOverride_Rename) Reset ¶
func (m *OpGenOverride_Rename) Reset()
func (*OpGenOverride_Rename) String ¶
func (m *OpGenOverride_Rename) String() string
type OpGenOverrides ¶
type OpGenOverrides struct {
Op []*OpGenOverride `protobuf:"bytes,1,rep,name=op" json:"op,omitempty"`
}
func (*OpGenOverrides) Descriptor ¶
func (*OpGenOverrides) Descriptor() ([]byte, []int)
func (*OpGenOverrides) GetOp ¶
func (m *OpGenOverrides) GetOp() []*OpGenOverride
func (*OpGenOverrides) ProtoMessage ¶
func (*OpGenOverrides) ProtoMessage()
func (*OpGenOverrides) Reset ¶
func (m *OpGenOverrides) Reset()
func (*OpGenOverrides) String ¶
func (m *OpGenOverrides) String() string
type OpList ¶
type OpList struct {
Op []*OpDef `protobuf:"bytes,1,rep,name=op" json:"op,omitempty"`
}
A collection of OpDefs
func (*OpList) Descriptor ¶
func (*OpList) ProtoMessage ¶
func (*OpList) ProtoMessage()
type ReaderBaseState ¶
type ReaderBaseState struct { WorkStarted int64 `protobuf:"varint,1,opt,name=work_started,json=workStarted" json:"work_started,omitempty"` WorkFinished int64 `protobuf:"varint,2,opt,name=work_finished,json=workFinished" json:"work_finished,omitempty"` NumRecordsProduced int64 `protobuf:"varint,3,opt,name=num_records_produced,json=numRecordsProduced" json:"num_records_produced,omitempty"` CurrentWork []byte `protobuf:"bytes,4,opt,name=current_work,json=currentWork,proto3" json:"current_work,omitempty"` }
For serializing and restoring the state of ReaderBase, see reader_base.h for details.
func (*ReaderBaseState) Descriptor ¶
func (*ReaderBaseState) Descriptor() ([]byte, []int)
func (*ReaderBaseState) GetCurrentWork ¶
func (m *ReaderBaseState) GetCurrentWork() []byte
func (*ReaderBaseState) GetNumRecordsProduced ¶
func (m *ReaderBaseState) GetNumRecordsProduced() int64
func (*ReaderBaseState) GetWorkFinished ¶
func (m *ReaderBaseState) GetWorkFinished() int64
func (*ReaderBaseState) GetWorkStarted ¶
func (m *ReaderBaseState) GetWorkStarted() int64
func (*ReaderBaseState) ProtoMessage ¶
func (*ReaderBaseState) ProtoMessage()
func (*ReaderBaseState) Reset ¶
func (m *ReaderBaseState) Reset()
func (*ReaderBaseState) String ¶
func (m *ReaderBaseState) String() string
type RemoteFusedGraphExecuteInfo ¶
type RemoteFusedGraphExecuteInfo struct { // Definition of remote graph RemoteGraph *GraphDef `protobuf:"bytes,1,opt,name=remote_graph,json=remoteGraph" json:"remote_graph,omitempty"` // Remote fused graph input node name GraphInputNodeName []string `protobuf:"bytes,2,rep,name=graph_input_node_name,json=graphInputNodeName" 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" json:"graph_output_node_name,omitempty"` // Executor's name ExecutorName string `protobuf:"bytes,4,opt,name=executor_name,json=executorName" 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 `` /* 144-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 `` /* 147-byte string literal not displayed */ }
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 ¶
func (*RemoteFusedGraphExecuteInfo) Descriptor() ([]byte, []int)
func (*RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape ¶
func (m *RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
func (*RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape ¶
func (m *RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
func (*RemoteFusedGraphExecuteInfo) GetExecutorName ¶
func (m *RemoteFusedGraphExecuteInfo) GetExecutorName() string
func (*RemoteFusedGraphExecuteInfo) GetGraphInputNodeName ¶
func (m *RemoteFusedGraphExecuteInfo) GetGraphInputNodeName() []string
func (*RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName ¶
func (m *RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName() []string
func (*RemoteFusedGraphExecuteInfo) GetRemoteGraph ¶
func (m *RemoteFusedGraphExecuteInfo) GetRemoteGraph() *GraphDef
func (*RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters ¶
func (m *RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters() []byte
func (*RemoteFusedGraphExecuteInfo) ProtoMessage ¶
func (*RemoteFusedGraphExecuteInfo) ProtoMessage()
func (*RemoteFusedGraphExecuteInfo) Reset ¶
func (m *RemoteFusedGraphExecuteInfo) Reset()
func (*RemoteFusedGraphExecuteInfo) String ¶
func (m *RemoteFusedGraphExecuteInfo) String() string
type RemoteFusedGraphExecuteInfo_NodeType ¶
type RemoteFusedGraphExecuteInfo_NodeType int32
const ( RemoteFusedGraphExecuteInfo_UNUSED RemoteFusedGraphExecuteInfo_NodeType = 0 RemoteFusedGraphExecuteInfo_GRAPH_INPUT RemoteFusedGraphExecuteInfo_NodeType = 1 RemoteFusedGraphExecuteInfo_GRAPH_OUTPUT RemoteFusedGraphExecuteInfo_NodeType = 2 RemoteFusedGraphExecuteInfo_FUSED_NODE RemoteFusedGraphExecuteInfo_NodeType = 3 RemoteFusedGraphExecuteInfo_BORDER_INPUT RemoteFusedGraphExecuteInfo_NodeType = 4 RemoteFusedGraphExecuteInfo_BORDER_OUTPUT RemoteFusedGraphExecuteInfo_NodeType = 5 )
func (RemoteFusedGraphExecuteInfo_NodeType) EnumDescriptor ¶
func (RemoteFusedGraphExecuteInfo_NodeType) EnumDescriptor() ([]byte, []int)
func (RemoteFusedGraphExecuteInfo_NodeType) String ¶
func (x RemoteFusedGraphExecuteInfo_NodeType) String() string
type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto ¶
type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto struct { Dtype DataType `protobuf:"varint,1,opt,name=dtype,enum=tensorflow.DataType" json:"dtype,omitempty"` Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape" json:"shape,omitempty"` }
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor ¶
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor() ([]byte, []int)
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype() DataType
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape() *TensorShapeProto
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage ¶
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage()
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset()
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) String ¶
func (m *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" json:"device,omitempty"` // Container in which this resource is placed. Container string `protobuf:"bytes,2,opt,name=container" json:"container,omitempty"` // Unique name of this resource. Name string `protobuf:"bytes,3,opt,name=name" 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" 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" json:"maybe_type_name,omitempty"` }
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 ¶
func (*ResourceHandleProto) Descriptor() ([]byte, []int)
func (*ResourceHandleProto) GetContainer ¶
func (m *ResourceHandleProto) GetContainer() string
func (*ResourceHandleProto) GetDevice ¶
func (m *ResourceHandleProto) GetDevice() string
func (*ResourceHandleProto) GetHashCode ¶
func (m *ResourceHandleProto) GetHashCode() uint64
func (*ResourceHandleProto) GetMaybeTypeName ¶
func (m *ResourceHandleProto) GetMaybeTypeName() string
func (*ResourceHandleProto) GetName ¶
func (m *ResourceHandleProto) GetName() string
func (*ResourceHandleProto) ProtoMessage ¶
func (*ResourceHandleProto) ProtoMessage()
func (*ResourceHandleProto) Reset ¶
func (m *ResourceHandleProto) Reset()
func (*ResourceHandleProto) String ¶
func (m *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" json:"full_name,omitempty"` // Shape of the full variable. FullShape []int64 `protobuf:"varint,2,rep,packed,name=full_shape,json=fullShape" json:"full_shape,omitempty"` // Offset of this variable into the full variable. VarOffset []int64 `protobuf:"varint,3,rep,packed,name=var_offset,json=varOffset" json:"var_offset,omitempty"` // Shape of this variable. VarShape []int64 `protobuf:"varint,4,rep,packed,name=var_shape,json=varShape" json:"var_shape,omitempty"` }
func (*SaveSliceInfoDef) Descriptor ¶
func (*SaveSliceInfoDef) Descriptor() ([]byte, []int)
func (*SaveSliceInfoDef) GetFullName ¶
func (m *SaveSliceInfoDef) GetFullName() string
func (*SaveSliceInfoDef) GetFullShape ¶
func (m *SaveSliceInfoDef) GetFullShape() []int64
func (*SaveSliceInfoDef) GetVarOffset ¶
func (m *SaveSliceInfoDef) GetVarOffset() []int64
func (*SaveSliceInfoDef) GetVarShape ¶
func (m *SaveSliceInfoDef) GetVarShape() []int64
func (*SaveSliceInfoDef) ProtoMessage ¶
func (*SaveSliceInfoDef) ProtoMessage()
func (*SaveSliceInfoDef) Reset ¶
func (m *SaveSliceInfoDef) Reset()
func (*SaveSliceInfoDef) String ¶
func (m *SaveSliceInfoDef) String() string
type StepStats ¶
type StepStats struct {
DevStats []*DeviceStepStats `protobuf:"bytes,1,rep,name=dev_stats,json=devStats" json:"dev_stats,omitempty"`
}
func (*StepStats) Descriptor ¶
func (*StepStats) GetDevStats ¶
func (m *StepStats) GetDevStats() []*DeviceStepStats
func (*StepStats) ProtoMessage ¶
func (*StepStats) ProtoMessage()
type Summary ¶
type Summary struct { // Set of values for the summary. Value []*Summary_Value `protobuf:"bytes,1,rep,name=value" json:"value,omitempty"` }
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 ¶
func (*Summary) GetValue ¶
func (m *Summary) GetValue() []*Summary_Value
func (*Summary) ProtoMessage ¶
func (*Summary) ProtoMessage()
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" json:"type_hint,omitempty"` }
Metadata associated with a series of Summary data
func (*SummaryDescription) Descriptor ¶
func (*SummaryDescription) Descriptor() ([]byte, []int)
func (*SummaryDescription) GetTypeHint ¶
func (m *SummaryDescription) GetTypeHint() string
func (*SummaryDescription) ProtoMessage ¶
func (*SummaryDescription) ProtoMessage()
func (*SummaryDescription) Reset ¶
func (m *SummaryDescription) Reset()
func (*SummaryDescription) String ¶
func (m *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" json:"plugin_data,omitempty"` // Display name for viewing in TensorBoard. DisplayName string `protobuf:"bytes,2,opt,name=display_name,json=displayName" json:"display_name,omitempty"` // Longform readable description of the summary sequence. Markdown supported. SummaryDescription string `protobuf:"bytes,3,opt,name=summary_description,json=summaryDescription" json:"summary_description,omitempty"` }
A SummaryMetadata encapsulates information on which plugins are able to make use of a certain summary value.
func (*SummaryMetadata) Descriptor ¶
func (*SummaryMetadata) Descriptor() ([]byte, []int)
func (*SummaryMetadata) GetDisplayName ¶
func (m *SummaryMetadata) GetDisplayName() string
func (*SummaryMetadata) GetPluginData ¶
func (m *SummaryMetadata) GetPluginData() *SummaryMetadata_PluginData
func (*SummaryMetadata) GetSummaryDescription ¶
func (m *SummaryMetadata) GetSummaryDescription() string
func (*SummaryMetadata) ProtoMessage ¶
func (*SummaryMetadata) ProtoMessage()
func (*SummaryMetadata) Reset ¶
func (m *SummaryMetadata) Reset()
func (*SummaryMetadata) String ¶
func (m *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" 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"` }
func (*SummaryMetadata_PluginData) Descriptor ¶
func (*SummaryMetadata_PluginData) Descriptor() ([]byte, []int)
func (*SummaryMetadata_PluginData) GetContent ¶
func (m *SummaryMetadata_PluginData) GetContent() []byte
func (*SummaryMetadata_PluginData) GetPluginName ¶
func (m *SummaryMetadata_PluginData) GetPluginName() string
func (*SummaryMetadata_PluginData) ProtoMessage ¶
func (*SummaryMetadata_PluginData) ProtoMessage()
func (*SummaryMetadata_PluginData) Reset ¶
func (m *SummaryMetadata_PluginData) Reset()
func (*SummaryMetadata_PluginData) String ¶
func (m *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" json:"sample_rate,omitempty"` // Number of channels of audio. NumChannels int64 `protobuf:"varint,2,opt,name=num_channels,json=numChannels" json:"num_channels,omitempty"` // Length of the audio in frames (samples per channel). LengthFrames int64 `protobuf:"varint,3,opt,name=length_frames,json=lengthFrames" 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" json:"content_type,omitempty"` }
func (*Summary_Audio) Descriptor ¶
func (*Summary_Audio) Descriptor() ([]byte, []int)
func (*Summary_Audio) GetContentType ¶
func (m *Summary_Audio) GetContentType() string
func (*Summary_Audio) GetEncodedAudioString ¶
func (m *Summary_Audio) GetEncodedAudioString() []byte
func (*Summary_Audio) GetLengthFrames ¶
func (m *Summary_Audio) GetLengthFrames() int64
func (*Summary_Audio) GetNumChannels ¶
func (m *Summary_Audio) GetNumChannels() int64
func (*Summary_Audio) GetSampleRate ¶
func (m *Summary_Audio) GetSampleRate() float32
func (*Summary_Audio) ProtoMessage ¶
func (*Summary_Audio) ProtoMessage()
func (*Summary_Audio) Reset ¶
func (m *Summary_Audio) Reset()
func (*Summary_Audio) String ¶
func (m *Summary_Audio) String() string
type Summary_Image ¶
type Summary_Image struct { // Dimensions of the image. Height int32 `protobuf:"varint,1,opt,name=height" json:"height,omitempty"` Width int32 `protobuf:"varint,2,opt,name=width" 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" 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"` }
func (*Summary_Image) Descriptor ¶
func (*Summary_Image) Descriptor() ([]byte, []int)
func (*Summary_Image) GetColorspace ¶
func (m *Summary_Image) GetColorspace() int32
func (*Summary_Image) GetEncodedImageString ¶
func (m *Summary_Image) GetEncodedImageString() []byte
func (*Summary_Image) GetHeight ¶
func (m *Summary_Image) GetHeight() int32
func (*Summary_Image) GetWidth ¶
func (m *Summary_Image) GetWidth() int32
func (*Summary_Image) ProtoMessage ¶
func (*Summary_Image) ProtoMessage()
func (*Summary_Image) Reset ¶
func (m *Summary_Image) Reset()
func (*Summary_Image) String ¶
func (m *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" 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" 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" json:"metadata,omitempty"` // Value associated with the tag. // // Types that are valid to be assigned 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"` }
func (*Summary_Value) Descriptor ¶
func (*Summary_Value) Descriptor() ([]byte, []int)
func (*Summary_Value) GetAudio ¶
func (m *Summary_Value) GetAudio() *Summary_Audio
func (*Summary_Value) GetHisto ¶
func (m *Summary_Value) GetHisto() *HistogramProto
func (*Summary_Value) GetImage ¶
func (m *Summary_Value) GetImage() *Summary_Image
func (*Summary_Value) GetMetadata ¶
func (m *Summary_Value) GetMetadata() *SummaryMetadata
func (*Summary_Value) GetNodeName ¶
func (m *Summary_Value) GetNodeName() string
func (*Summary_Value) GetObsoleteOldStyleHistogram ¶
func (m *Summary_Value) GetObsoleteOldStyleHistogram() []byte
func (*Summary_Value) GetSimpleValue ¶
func (m *Summary_Value) GetSimpleValue() float32
func (*Summary_Value) GetTag ¶
func (m *Summary_Value) GetTag() string
func (*Summary_Value) GetTensor ¶
func (m *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) Reset ¶
func (m *Summary_Value) Reset()
func (*Summary_Value) String ¶
func (m *Summary_Value) String() string
func (*Summary_Value) XXX_OneofFuncs ¶
func (*Summary_Value) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, func(msg proto.Message, tag, wire int, b *proto.Buffer) (bool, error), func(msg proto.Message) (n int), []interface{})
XXX_OneofFuncs is for the internal use of the proto package.
type Summary_Value_Audio ¶
type Summary_Value_Audio struct {
Audio *Summary_Audio `protobuf:"bytes,6,opt,name=audio,oneof"`
}
type Summary_Value_Histo ¶
type Summary_Value_Histo struct {
Histo *HistogramProto `protobuf:"bytes,5,opt,name=histo,oneof"`
}
type Summary_Value_Image ¶
type Summary_Value_Image struct {
Image *Summary_Image `protobuf:"bytes,4,opt,name=image,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,oneof"`
}
type Summary_Value_Tensor ¶
type Summary_Value_Tensor struct {
Tensor *TensorProto `protobuf:"bytes,8,opt,name=tensor,oneof"`
}
type TensorDescription ¶
type TensorDescription struct { // Data type of tensor elements Dtype DataType `protobuf:"varint,1,opt,name=dtype,enum=tensorflow.DataType" json:"dtype,omitempty"` // Shape of the tensor. Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape" json:"shape,omitempty"` // Information about the size and allocator used for the data AllocationDescription *AllocationDescription `protobuf:"bytes,4,opt,name=allocation_description,json=allocationDescription" json:"allocation_description,omitempty"` }
func (*TensorDescription) Descriptor ¶
func (*TensorDescription) Descriptor() ([]byte, []int)
func (*TensorDescription) GetAllocationDescription ¶
func (m *TensorDescription) GetAllocationDescription() *AllocationDescription
func (*TensorDescription) GetDtype ¶
func (m *TensorDescription) GetDtype() DataType
func (*TensorDescription) GetShape ¶
func (m *TensorDescription) GetShape() *TensorShapeProto
func (*TensorDescription) ProtoMessage ¶
func (*TensorDescription) ProtoMessage()
func (*TensorDescription) Reset ¶
func (m *TensorDescription) Reset()
func (*TensorDescription) String ¶
func (m *TensorDescription) String() string
type TensorProto ¶
type TensorProto struct { Dtype DataType `protobuf:"varint,1,opt,name=dtype,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" 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" 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. 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" json:"half_val,omitempty"` // DT_FLOAT. FloatVal []float32 `protobuf:"fixed32,5,rep,packed,name=float_val,json=floatVal" json:"float_val,omitempty"` // DT_DOUBLE. DoubleVal []float64 `protobuf:"fixed64,6,rep,packed,name=double_val,json=doubleVal" json:"double_val,omitempty"` // DT_INT32, DT_INT16, DT_INT8, DT_UINT8. IntVal []int32 `protobuf:"varint,7,rep,packed,name=int_val,json=intVal" 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" json:"scomplex_val,omitempty"` // DT_INT64 Int64Val []int64 `protobuf:"varint,10,rep,packed,name=int64_val,json=int64Val" json:"int64_val,omitempty"` // DT_BOOL BoolVal []bool `protobuf:"varint,11,rep,packed,name=bool_val,json=boolVal" 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" json:"dcomplex_val,omitempty"` // DT_RESOURCE ResourceHandleVal []*ResourceHandleProto `protobuf:"bytes,14,rep,name=resource_handle_val,json=resourceHandleVal" json:"resource_handle_val,omitempty"` // DT_VARIANT VariantVal []*VariantTensorDataProto `protobuf:"bytes,15,rep,name=variant_val,json=variantVal" json:"variant_val,omitempty"` // DT_UINT32 Uint32Val []uint32 `protobuf:"varint,16,rep,packed,name=uint32_val,json=uint32Val" json:"uint32_val,omitempty"` // DT_UINT64 Uint64Val []uint64 `protobuf:"varint,17,rep,packed,name=uint64_val,json=uint64Val" json:"uint64_val,omitempty"` }
Protocol buffer representing a tensor.
func (*TensorProto) Descriptor ¶
func (*TensorProto) Descriptor() ([]byte, []int)
func (*TensorProto) GetBoolVal ¶
func (m *TensorProto) GetBoolVal() []bool
func (*TensorProto) GetDcomplexVal ¶
func (m *TensorProto) GetDcomplexVal() []float64
func (*TensorProto) GetDoubleVal ¶
func (m *TensorProto) GetDoubleVal() []float64
func (*TensorProto) GetDtype ¶
func (m *TensorProto) GetDtype() DataType
func (*TensorProto) GetFloatVal ¶
func (m *TensorProto) GetFloatVal() []float32
func (*TensorProto) GetHalfVal ¶
func (m *TensorProto) GetHalfVal() []int32
func (*TensorProto) GetInt64Val ¶
func (m *TensorProto) GetInt64Val() []int64
func (*TensorProto) GetIntVal ¶
func (m *TensorProto) GetIntVal() []int32
func (*TensorProto) GetResourceHandleVal ¶
func (m *TensorProto) GetResourceHandleVal() []*ResourceHandleProto
func (*TensorProto) GetScomplexVal ¶
func (m *TensorProto) GetScomplexVal() []float32
func (*TensorProto) GetStringVal ¶
func (m *TensorProto) GetStringVal() [][]byte
func (*TensorProto) GetTensorContent ¶
func (m *TensorProto) GetTensorContent() []byte
func (*TensorProto) GetTensorShape ¶
func (m *TensorProto) GetTensorShape() *TensorShapeProto
func (*TensorProto) GetUint32Val ¶
func (m *TensorProto) GetUint32Val() []uint32
func (*TensorProto) GetUint64Val ¶
func (m *TensorProto) GetUint64Val() []uint64
func (*TensorProto) GetVariantVal ¶
func (m *TensorProto) GetVariantVal() []*VariantTensorDataProto
func (*TensorProto) GetVersionNumber ¶
func (m *TensorProto) GetVersionNumber() int32
func (*TensorProto) ProtoMessage ¶
func (*TensorProto) ProtoMessage()
func (*TensorProto) Reset ¶
func (m *TensorProto) Reset()
func (*TensorProto) String ¶
func (m *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" 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" json:"unknown_rank,omitempty"` }
Dimensions of a tensor.
func (*TensorShapeProto) Descriptor ¶
func (*TensorShapeProto) Descriptor() ([]byte, []int)
func (*TensorShapeProto) GetDim ¶
func (m *TensorShapeProto) GetDim() []*TensorShapeProto_Dim
func (*TensorShapeProto) GetUnknownRank ¶
func (m *TensorShapeProto) GetUnknownRank() bool
func (*TensorShapeProto) ProtoMessage ¶
func (*TensorShapeProto) ProtoMessage()
func (*TensorShapeProto) Reset ¶
func (m *TensorShapeProto) Reset()
func (*TensorShapeProto) String ¶
func (m *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" json:"size,omitempty"` // Optional name of the tensor dimension. Name string `protobuf:"bytes,2,opt,name=name" json:"name,omitempty"` }
One dimension of the tensor.
func (*TensorShapeProto_Dim) Descriptor ¶
func (*TensorShapeProto_Dim) Descriptor() ([]byte, []int)
func (*TensorShapeProto_Dim) GetName ¶
func (m *TensorShapeProto_Dim) GetName() string
func (*TensorShapeProto_Dim) GetSize ¶
func (m *TensorShapeProto_Dim) GetSize() int64
func (*TensorShapeProto_Dim) ProtoMessage ¶
func (*TensorShapeProto_Dim) ProtoMessage()
func (*TensorShapeProto_Dim) Reset ¶
func (m *TensorShapeProto_Dim) Reset()
func (*TensorShapeProto_Dim) String ¶
func (m *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" json:"extent,omitempty"` }
Can only be interpreted if you know the corresponding TensorShape.
func (*TensorSliceProto) Descriptor ¶
func (*TensorSliceProto) Descriptor() ([]byte, []int)
func (*TensorSliceProto) GetExtent ¶
func (m *TensorSliceProto) GetExtent() []*TensorSliceProto_Extent
func (*TensorSliceProto) ProtoMessage ¶
func (*TensorSliceProto) ProtoMessage()
func (*TensorSliceProto) Reset ¶
func (m *TensorSliceProto) Reset()
func (*TensorSliceProto) String ¶
func (m *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" 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 valid to be assigned to HasLength: // *TensorSliceProto_Extent_Length HasLength isTensorSliceProto_Extent_HasLength `protobuf_oneof:"has_length"` }
Extent of the slice in one dimension.
func (*TensorSliceProto_Extent) Descriptor ¶
func (*TensorSliceProto_Extent) Descriptor() ([]byte, []int)
func (*TensorSliceProto_Extent) GetHasLength ¶
func (m *TensorSliceProto_Extent) GetHasLength() isTensorSliceProto_Extent_HasLength
func (*TensorSliceProto_Extent) GetLength ¶
func (m *TensorSliceProto_Extent) GetLength() int64
func (*TensorSliceProto_Extent) GetStart ¶
func (m *TensorSliceProto_Extent) GetStart() int64
func (*TensorSliceProto_Extent) ProtoMessage ¶
func (*TensorSliceProto_Extent) ProtoMessage()
func (*TensorSliceProto_Extent) Reset ¶
func (m *TensorSliceProto_Extent) Reset()
func (*TensorSliceProto_Extent) String ¶
func (m *TensorSliceProto_Extent) String() string
func (*TensorSliceProto_Extent) XXX_OneofFuncs ¶
func (*TensorSliceProto_Extent) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, func(msg proto.Message, tag, wire int, b *proto.Buffer) (bool, error), func(msg proto.Message) (n int), []interface{})
XXX_OneofFuncs is for the internal use of the proto package.
type TensorSliceProto_Extent_Length ¶
type TensorSliceProto_Extent_Length struct {
Length int64 `protobuf:"varint,2,opt,name=length,oneof"`
}
type VariableDef ¶
type VariableDef struct { // Name of the variable tensor. VariableName string `protobuf:"bytes,1,opt,name=variable_name,json=variableName" 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" json:"initial_value_name,omitempty"` // Name of the initializer op. InitializerName string `protobuf:"bytes,2,opt,name=initializer_name,json=initializerName" json:"initializer_name,omitempty"` // Name of the snapshot tensor. SnapshotName string `protobuf:"bytes,3,opt,name=snapshot_name,json=snapshotName" 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" json:"save_slice_info_def,omitempty"` // Whether to represent this as a ResourceVariable. IsResource bool `protobuf:"varint,5,opt,name=is_resource,json=isResource" json:"is_resource,omitempty"` }
Protocol buffer representing a Variable.
func (*VariableDef) Descriptor ¶
func (*VariableDef) Descriptor() ([]byte, []int)
func (*VariableDef) GetInitialValueName ¶
func (m *VariableDef) GetInitialValueName() string
func (*VariableDef) GetInitializerName ¶
func (m *VariableDef) GetInitializerName() string
func (*VariableDef) GetIsResource ¶
func (m *VariableDef) GetIsResource() bool
func (*VariableDef) GetSaveSliceInfoDef ¶
func (m *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef
func (*VariableDef) GetSnapshotName ¶
func (m *VariableDef) GetSnapshotName() string
func (*VariableDef) GetVariableName ¶
func (m *VariableDef) GetVariableName() string
func (*VariableDef) ProtoMessage ¶
func (*VariableDef) ProtoMessage()
func (*VariableDef) Reset ¶
func (m *VariableDef) Reset()
func (*VariableDef) String ¶
func (m *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" 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" json:"tensors,omitempty"` }
Protocol buffer representing the serialization format of DT_VARIANT tensors.
func (*VariantTensorDataProto) Descriptor ¶
func (*VariantTensorDataProto) Descriptor() ([]byte, []int)
func (*VariantTensorDataProto) GetMetadata ¶
func (m *VariantTensorDataProto) GetMetadata() []byte
func (*VariantTensorDataProto) GetTensors ¶
func (m *VariantTensorDataProto) GetTensors() []*TensorProto
func (*VariantTensorDataProto) GetTypeName ¶
func (m *VariantTensorDataProto) GetTypeName() string
func (*VariantTensorDataProto) ProtoMessage ¶
func (*VariantTensorDataProto) ProtoMessage()
func (*VariantTensorDataProto) Reset ¶
func (m *VariantTensorDataProto) Reset()
func (*VariantTensorDataProto) String ¶
func (m *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" 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" 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" json:"bad_consumers,omitempty"` }
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 ¶
func (*VersionDef) Descriptor() ([]byte, []int)
func (*VersionDef) GetBadConsumers ¶
func (m *VersionDef) GetBadConsumers() []int32
func (*VersionDef) GetMinConsumer ¶
func (m *VersionDef) GetMinConsumer() int32
func (*VersionDef) GetProducer ¶
func (m *VersionDef) GetProducer() int32
func (*VersionDef) ProtoMessage ¶
func (*VersionDef) ProtoMessage()
func (*VersionDef) Reset ¶
func (m *VersionDef) Reset()
func (*VersionDef) String ¶
func (m *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
- kernel_def.pb.go
- log_memory.pb.go
- node_def.pb.go
- op_def.pb.go
- op_gen_overrides.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