Documentation ¶
Index ¶
- Variables
- type AssetFileDef
- func (*AssetFileDef) Descriptor() ([]byte, []int)deprecated
- func (x *AssetFileDef) GetFilename() string
- func (x *AssetFileDef) GetTensorInfo() *TensorInfo
- func (*AssetFileDef) ProtoMessage()
- func (x *AssetFileDef) ProtoReflect() protoreflect.Message
- func (x *AssetFileDef) Reset()
- func (x *AssetFileDef) String() string
- type AutoParallelOptions
- func (*AutoParallelOptions) Descriptor() ([]byte, []int)deprecated
- func (x *AutoParallelOptions) GetEnable() bool
- func (x *AutoParallelOptions) GetNumReplicas() int32
- func (*AutoParallelOptions) ProtoMessage()
- func (x *AutoParallelOptions) ProtoReflect() protoreflect.Message
- func (x *AutoParallelOptions) Reset()
- func (x *AutoParallelOptions) String() string
- type BundleEntryProto
- func (*BundleEntryProto) Descriptor() ([]byte, []int)deprecated
- func (x *BundleEntryProto) GetCrc32C() uint32
- func (x *BundleEntryProto) GetDtype() framework.DataType
- func (x *BundleEntryProto) GetOffset() int64
- func (x *BundleEntryProto) GetShape() *framework.TensorShapeProto
- func (x *BundleEntryProto) GetShardId() int32
- func (x *BundleEntryProto) GetSize() int64
- func (x *BundleEntryProto) GetSlices() []*framework.TensorSliceProto
- func (*BundleEntryProto) ProtoMessage()
- func (x *BundleEntryProto) ProtoReflect() protoreflect.Message
- func (x *BundleEntryProto) Reset()
- func (x *BundleEntryProto) String() string
- type BundleHeaderProto
- func (*BundleHeaderProto) Descriptor() ([]byte, []int)deprecated
- func (x *BundleHeaderProto) GetEndianness() BundleHeaderProto_Endianness
- func (x *BundleHeaderProto) GetNumShards() int32
- func (x *BundleHeaderProto) GetVersion() *framework.VersionDef
- func (*BundleHeaderProto) ProtoMessage()
- func (x *BundleHeaderProto) ProtoReflect() protoreflect.Message
- func (x *BundleHeaderProto) Reset()
- func (x *BundleHeaderProto) String() string
- type BundleHeaderProto_Endianness
- func (BundleHeaderProto_Endianness) Descriptor() protoreflect.EnumDescriptor
- func (x BundleHeaderProto_Endianness) Enum() *BundleHeaderProto_Endianness
- func (BundleHeaderProto_Endianness) EnumDescriptor() ([]byte, []int)deprecated
- func (x BundleHeaderProto_Endianness) Number() protoreflect.EnumNumber
- func (x BundleHeaderProto_Endianness) String() string
- func (BundleHeaderProto_Endianness) Type() protoreflect.EnumType
- type CallableOptions
- func (*CallableOptions) Descriptor() ([]byte, []int)deprecated
- func (x *CallableOptions) GetFeed() []string
- func (x *CallableOptions) GetFeedDevices() map[string]string
- func (x *CallableOptions) GetFetch() []string
- func (x *CallableOptions) GetFetchDevices() map[string]string
- func (x *CallableOptions) GetFetchSkipSync() bool
- func (x *CallableOptions) GetRunOptions() *RunOptions
- func (x *CallableOptions) GetTarget() []string
- func (x *CallableOptions) GetTensorConnection() []*TensorConnection
- func (*CallableOptions) ProtoMessage()
- func (x *CallableOptions) ProtoReflect() protoreflect.Message
- func (x *CallableOptions) Reset()
- func (x *CallableOptions) String() string
- type CheckpointableObjectGraph
- func (*CheckpointableObjectGraph) Descriptor() ([]byte, []int)deprecated
- func (x *CheckpointableObjectGraph) GetNodes() []*CheckpointableObjectGraph_CheckpointableObject
- func (*CheckpointableObjectGraph) ProtoMessage()
- func (x *CheckpointableObjectGraph) ProtoReflect() protoreflect.Message
- func (x *CheckpointableObjectGraph) Reset()
- func (x *CheckpointableObjectGraph) String() string
- type CheckpointableObjectGraph_CheckpointableObject
- func (*CheckpointableObjectGraph_CheckpointableObject) Descriptor() ([]byte, []int)deprecated
- func (x *CheckpointableObjectGraph_CheckpointableObject) GetAttributes() []*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor
- func (x *CheckpointableObjectGraph_CheckpointableObject) GetChildren() []*CheckpointableObjectGraph_CheckpointableObject_ObjectReference
- func (x *CheckpointableObjectGraph_CheckpointableObject) GetSlotVariables() []*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference
- func (*CheckpointableObjectGraph_CheckpointableObject) ProtoMessage()
- func (x *CheckpointableObjectGraph_CheckpointableObject) ProtoReflect() protoreflect.Message
- func (x *CheckpointableObjectGraph_CheckpointableObject) Reset()
- func (x *CheckpointableObjectGraph_CheckpointableObject) String() string
- type CheckpointableObjectGraph_CheckpointableObject_ObjectReference
- func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) Descriptor() ([]byte, []int)deprecated
- func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) GetLocalName() string
- func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) GetNodeId() int32
- func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) ProtoMessage()
- func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) ProtoReflect() protoreflect.Message
- func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) Reset()
- func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) String() string
- type CheckpointableObjectGraph_CheckpointableObject_SerializedTensor
- func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) Descriptor() ([]byte, []int)deprecated
- func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetCheckpointKey() string
- func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetFullName() string
- func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetName() string
- func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) ProtoMessage()
- func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) ProtoReflect() protoreflect.Message
- func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) Reset()
- func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) String() string
- type CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference
- func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) Descriptor() ([]byte, []int)deprecated
- func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetOriginalVariableNodeId() int32
- func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetSlotName() string
- func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetSlotVariableNodeId() int32
- func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) ProtoMessage()
- func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) ProtoReflect() protoreflect.Message
- func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) Reset()
- func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) String() string
- type CleanupAllRequest
- func (*CleanupAllRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CleanupAllRequest) GetContainer() []string
- func (*CleanupAllRequest) ProtoMessage()
- func (x *CleanupAllRequest) ProtoReflect() protoreflect.Message
- func (x *CleanupAllRequest) Reset()
- func (x *CleanupAllRequest) String() string
- type CleanupAllResponse
- type CleanupGraphRequest
- func (*CleanupGraphRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CleanupGraphRequest) GetStepId() int64
- func (*CleanupGraphRequest) ProtoMessage()
- func (x *CleanupGraphRequest) ProtoReflect() protoreflect.Message
- func (x *CleanupGraphRequest) Reset()
- func (x *CleanupGraphRequest) String() string
- type CleanupGraphResponse
- type CloseContextRequest
- func (*CloseContextRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CloseContextRequest) GetContextId() uint64
- func (*CloseContextRequest) ProtoMessage()
- func (x *CloseContextRequest) ProtoReflect() protoreflect.Message
- func (x *CloseContextRequest) Reset()
- func (x *CloseContextRequest) String() string
- type CloseContextResponse
- type CloseSessionRequest
- func (*CloseSessionRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CloseSessionRequest) GetSessionHandle() string
- func (*CloseSessionRequest) ProtoMessage()
- func (x *CloseSessionRequest) ProtoReflect() protoreflect.Message
- func (x *CloseSessionRequest) Reset()
- func (x *CloseSessionRequest) String() string
- type CloseSessionResponse
- type ClusterDef
- type CollectionDef
- func (*CollectionDef) Descriptor() ([]byte, []int)deprecated
- func (x *CollectionDef) GetAnyList() *CollectionDef_AnyList
- func (x *CollectionDef) GetBytesList() *CollectionDef_BytesList
- func (x *CollectionDef) GetFloatList() *CollectionDef_FloatList
- func (x *CollectionDef) GetInt64List() *CollectionDef_Int64List
- func (m *CollectionDef) GetKind() isCollectionDef_Kind
- func (x *CollectionDef) GetNodeList() *CollectionDef_NodeList
- func (*CollectionDef) ProtoMessage()
- func (x *CollectionDef) ProtoReflect() protoreflect.Message
- func (x *CollectionDef) Reset()
- func (x *CollectionDef) String() string
- type CollectionDef_AnyList
- func (*CollectionDef_AnyList) Descriptor() ([]byte, []int)deprecated
- func (x *CollectionDef_AnyList) GetValue() []*any.Any
- func (*CollectionDef_AnyList) ProtoMessage()
- func (x *CollectionDef_AnyList) ProtoReflect() protoreflect.Message
- func (x *CollectionDef_AnyList) Reset()
- func (x *CollectionDef_AnyList) String() string
- type CollectionDef_AnyList_
- type CollectionDef_BytesList
- func (*CollectionDef_BytesList) Descriptor() ([]byte, []int)deprecated
- func (x *CollectionDef_BytesList) GetValue() [][]byte
- func (*CollectionDef_BytesList) ProtoMessage()
- func (x *CollectionDef_BytesList) ProtoReflect() protoreflect.Message
- func (x *CollectionDef_BytesList) Reset()
- func (x *CollectionDef_BytesList) String() string
- type CollectionDef_BytesList_
- type CollectionDef_FloatList
- func (*CollectionDef_FloatList) Descriptor() ([]byte, []int)deprecated
- func (x *CollectionDef_FloatList) GetValue() []float32
- func (*CollectionDef_FloatList) ProtoMessage()
- func (x *CollectionDef_FloatList) ProtoReflect() protoreflect.Message
- func (x *CollectionDef_FloatList) Reset()
- func (x *CollectionDef_FloatList) String() string
- type CollectionDef_FloatList_
- type CollectionDef_Int64List
- func (*CollectionDef_Int64List) Descriptor() ([]byte, []int)deprecated
- func (x *CollectionDef_Int64List) GetValue() []int64
- func (*CollectionDef_Int64List) ProtoMessage()
- func (x *CollectionDef_Int64List) ProtoReflect() protoreflect.Message
- func (x *CollectionDef_Int64List) Reset()
- func (x *CollectionDef_Int64List) String() string
- type CollectionDef_Int64List_
- type CollectionDef_NodeList
- func (*CollectionDef_NodeList) Descriptor() ([]byte, []int)deprecated
- func (x *CollectionDef_NodeList) GetValue() []string
- func (*CollectionDef_NodeList) ProtoMessage()
- func (x *CollectionDef_NodeList) ProtoReflect() protoreflect.Message
- func (x *CollectionDef_NodeList) Reset()
- func (x *CollectionDef_NodeList) String() string
- type CollectionDef_NodeList_
- type CompleteGroupRequest
- func (*CompleteGroupRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CompleteGroupRequest) GetDeviceName() []string
- func (x *CompleteGroupRequest) GetDeviceType() string
- func (x *CompleteGroupRequest) GetGroupKey() int32
- func (x *CompleteGroupRequest) GetGroupSize() int32
- func (*CompleteGroupRequest) ProtoMessage()
- func (x *CompleteGroupRequest) ProtoReflect() protoreflect.Message
- func (x *CompleteGroupRequest) Reset()
- func (x *CompleteGroupRequest) String() string
- type CompleteGroupResponse
- func (*CompleteGroupResponse) Descriptor() ([]byte, []int)deprecated
- func (x *CompleteGroupResponse) GetDeviceName() []string
- func (x *CompleteGroupResponse) GetDeviceType() string
- func (x *CompleteGroupResponse) GetGroupKey() int32
- func (x *CompleteGroupResponse) GetGroupSize() int32
- func (x *CompleteGroupResponse) GetNumTasks() int32
- func (x *CompleteGroupResponse) GetTaskName() []string
- func (*CompleteGroupResponse) ProtoMessage()
- func (x *CompleteGroupResponse) ProtoReflect() protoreflect.Message
- func (x *CompleteGroupResponse) Reset()
- func (x *CompleteGroupResponse) String() string
- type CompleteInstanceRequest
- func (*CompleteInstanceRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CompleteInstanceRequest) GetDataType() framework.DataType
- func (x *CompleteInstanceRequest) GetDevice() string
- func (x *CompleteInstanceRequest) GetDeviceType() string
- func (x *CompleteInstanceRequest) GetGroupKey() int32
- func (x *CompleteInstanceRequest) GetGroupSize() int32
- func (x *CompleteInstanceRequest) GetInstanceKey() int32
- func (x *CompleteInstanceRequest) GetIsSource() bool
- func (x *CompleteInstanceRequest) GetName() string
- func (x *CompleteInstanceRequest) GetShape() *framework.TensorShapeProto
- func (x *CompleteInstanceRequest) GetSubdivOffset() []int32
- func (x *CompleteInstanceRequest) GetType() int32
- func (*CompleteInstanceRequest) ProtoMessage()
- func (x *CompleteInstanceRequest) ProtoReflect() protoreflect.Message
- func (x *CompleteInstanceRequest) Reset()
- func (x *CompleteInstanceRequest) String() string
- type CompleteInstanceResponse
- func (*CompleteInstanceResponse) Descriptor() ([]byte, []int)deprecated
- func (x *CompleteInstanceResponse) GetInstanceKey() int32
- func (x *CompleteInstanceResponse) GetSourceRank() int32
- func (*CompleteInstanceResponse) ProtoMessage()
- func (x *CompleteInstanceResponse) ProtoReflect() protoreflect.Message
- func (x *CompleteInstanceResponse) Reset()
- func (x *CompleteInstanceResponse) String() string
- type CondContextDef
- func (*CondContextDef) Descriptor() ([]byte, []int)deprecated
- func (x *CondContextDef) GetBranch() int32
- func (x *CondContextDef) GetContextName() string
- func (x *CondContextDef) GetNestedContexts() []*ControlFlowContextDef
- func (x *CondContextDef) GetPivotName() string
- func (x *CondContextDef) GetPredName() string
- func (x *CondContextDef) GetValuesDef() *ValuesDef
- func (*CondContextDef) ProtoMessage()
- func (x *CondContextDef) ProtoReflect() protoreflect.Message
- func (x *CondContextDef) Reset()
- func (x *CondContextDef) String() string
- type ConfigProto
- func (*ConfigProto) Descriptor() ([]byte, []int)deprecated
- func (x *ConfigProto) GetAllowSoftPlacement() bool
- func (x *ConfigProto) GetClusterDef() *ClusterDef
- func (x *ConfigProto) GetDeviceCount() map[string]int32
- func (x *ConfigProto) GetDeviceFilters() []string
- func (x *ConfigProto) GetExperimental() *ConfigProto_Experimental
- func (x *ConfigProto) GetGpuOptions() *GPUOptions
- func (x *ConfigProto) GetGraphOptions() *GraphOptions
- func (x *ConfigProto) GetInterOpParallelismThreads() int32
- func (x *ConfigProto) GetIntraOpParallelismThreads() int32
- func (x *ConfigProto) GetIsolateSessionState() bool
- func (x *ConfigProto) GetLogDevicePlacement() bool
- func (x *ConfigProto) GetOperationTimeoutInMs() int64
- func (x *ConfigProto) GetPlacementPeriod() int32
- func (x *ConfigProto) GetRpcOptions() *RPCOptions
- func (x *ConfigProto) GetSessionInterOpThreadPool() []*ThreadPoolOptionProto
- func (x *ConfigProto) GetUsePerSessionThreads() bool
- func (*ConfigProto) ProtoMessage()
- func (x *ConfigProto) ProtoReflect() protoreflect.Message
- func (x *ConfigProto) Reset()
- func (x *ConfigProto) String() string
- type ConfigProto_Experimental
- func (*ConfigProto_Experimental) Descriptor() ([]byte, []int)deprecated
- func (x *ConfigProto_Experimental) GetCollectiveGroupLeader() string
- func (x *ConfigProto_Experimental) GetExecutorType() string
- func (x *ConfigProto_Experimental) GetRecvBufMaxChunk() int32
- func (x *ConfigProto_Experimental) GetUseNumaAffinity() bool
- func (*ConfigProto_Experimental) ProtoMessage()
- func (x *ConfigProto_Experimental) ProtoReflect() protoreflect.Message
- func (x *ConfigProto_Experimental) Reset()
- func (x *ConfigProto_Experimental) String() string
- type ControlFlowContextDef
- func (*ControlFlowContextDef) Descriptor() ([]byte, []int)deprecated
- func (x *ControlFlowContextDef) GetCondCtxt() *CondContextDef
- func (m *ControlFlowContextDef) GetCtxt() isControlFlowContextDef_Ctxt
- func (x *ControlFlowContextDef) GetWhileCtxt() *WhileContextDef
- func (*ControlFlowContextDef) ProtoMessage()
- func (x *ControlFlowContextDef) ProtoReflect() protoreflect.Message
- func (x *ControlFlowContextDef) Reset()
- func (x *ControlFlowContextDef) String() string
- type ControlFlowContextDef_CondCtxt
- type ControlFlowContextDef_WhileCtxt
- type CreateContextRequest
- func (*CreateContextRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CreateContextRequest) GetAsync() bool
- func (x *CreateContextRequest) GetKeepAliveSecs() int64
- func (x *CreateContextRequest) GetRendezvousId() int64
- func (x *CreateContextRequest) GetServerDef() *ServerDef
- func (x *CreateContextRequest) GetVersionDef() *framework.VersionDef
- func (*CreateContextRequest) ProtoMessage()
- func (x *CreateContextRequest) ProtoReflect() protoreflect.Message
- func (x *CreateContextRequest) Reset()
- func (x *CreateContextRequest) String() string
- type CreateContextResponse
- func (*CreateContextResponse) Descriptor() ([]byte, []int)deprecated
- func (x *CreateContextResponse) GetContextId() uint64
- func (x *CreateContextResponse) GetDeviceAttributes() []*framework.DeviceAttributes
- func (*CreateContextResponse) ProtoMessage()
- func (x *CreateContextResponse) ProtoReflect() protoreflect.Message
- func (x *CreateContextResponse) Reset()
- func (x *CreateContextResponse) String() string
- type CreateSessionRequest
- func (*CreateSessionRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CreateSessionRequest) GetConfig() *ConfigProto
- func (x *CreateSessionRequest) GetGraphDef() *framework.GraphDef
- func (x *CreateSessionRequest) GetTarget() string
- func (*CreateSessionRequest) ProtoMessage()
- func (x *CreateSessionRequest) ProtoReflect() protoreflect.Message
- func (x *CreateSessionRequest) Reset()
- func (x *CreateSessionRequest) String() string
- type CreateSessionResponse
- func (*CreateSessionResponse) Descriptor() ([]byte, []int)deprecated
- func (x *CreateSessionResponse) GetGraphVersion() int64
- func (x *CreateSessionResponse) GetSessionHandle() string
- func (*CreateSessionResponse) ProtoMessage()
- func (x *CreateSessionResponse) ProtoReflect() protoreflect.Message
- func (x *CreateSessionResponse) Reset()
- func (x *CreateSessionResponse) String() string
- type CreateWorkerSessionRequest
- func (*CreateWorkerSessionRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CreateWorkerSessionRequest) GetIsolateSessionState() bool
- func (x *CreateWorkerSessionRequest) GetServerDef() *ServerDef
- func (x *CreateWorkerSessionRequest) GetSessionHandle() string
- func (*CreateWorkerSessionRequest) ProtoMessage()
- func (x *CreateWorkerSessionRequest) ProtoReflect() protoreflect.Message
- func (x *CreateWorkerSessionRequest) Reset()
- func (x *CreateWorkerSessionRequest) String() string
- type CreateWorkerSessionResponse
- type CriticalSectionDef
- func (*CriticalSectionDef) Descriptor() ([]byte, []int)deprecated
- func (x *CriticalSectionDef) GetCriticalSectionName() string
- func (*CriticalSectionDef) ProtoMessage()
- func (x *CriticalSectionDef) ProtoReflect() protoreflect.Message
- func (x *CriticalSectionDef) Reset()
- func (x *CriticalSectionDef) String() string
- type CriticalSectionExecutionDef
- func (*CriticalSectionExecutionDef) Descriptor() ([]byte, []int)deprecated
- func (x *CriticalSectionExecutionDef) GetExclusiveResourceAccess() bool
- func (x *CriticalSectionExecutionDef) GetExecuteInCriticalSectionName() string
- func (*CriticalSectionExecutionDef) ProtoMessage()
- func (x *CriticalSectionExecutionDef) ProtoReflect() protoreflect.Message
- func (x *CriticalSectionExecutionDef) Reset()
- func (x *CriticalSectionExecutionDef) String() string
- type DebugOptions
- func (*DebugOptions) Descriptor() ([]byte, []int)deprecated
- func (x *DebugOptions) GetDebugTensorWatchOpts() []*DebugTensorWatch
- func (x *DebugOptions) GetGlobalStep() int64
- func (x *DebugOptions) GetResetDiskByteUsage() bool
- func (*DebugOptions) ProtoMessage()
- func (x *DebugOptions) ProtoReflect() protoreflect.Message
- func (x *DebugOptions) Reset()
- func (x *DebugOptions) String() string
- type DebugTensorWatch
- func (*DebugTensorWatch) Descriptor() ([]byte, []int)deprecated
- func (x *DebugTensorWatch) GetDebugOps() []string
- func (x *DebugTensorWatch) GetDebugUrls() []string
- func (x *DebugTensorWatch) GetNodeName() string
- func (x *DebugTensorWatch) GetOutputSlot() int32
- func (x *DebugTensorWatch) GetTolerateDebugOpCreationFailures() bool
- func (*DebugTensorWatch) ProtoMessage()
- func (x *DebugTensorWatch) ProtoReflect() protoreflect.Message
- func (x *DebugTensorWatch) Reset()
- func (x *DebugTensorWatch) String() string
- type DebuggedSourceFile
- func (*DebuggedSourceFile) Descriptor() ([]byte, []int)deprecated
- func (x *DebuggedSourceFile) GetBytes() int64
- func (x *DebuggedSourceFile) GetFilePath() string
- func (x *DebuggedSourceFile) GetHost() string
- func (x *DebuggedSourceFile) GetLastModified() int64
- func (x *DebuggedSourceFile) GetLines() []string
- func (*DebuggedSourceFile) ProtoMessage()
- func (x *DebuggedSourceFile) ProtoReflect() protoreflect.Message
- func (x *DebuggedSourceFile) Reset()
- func (x *DebuggedSourceFile) String() string
- type DebuggedSourceFiles
- func (*DebuggedSourceFiles) Descriptor() ([]byte, []int)deprecated
- func (x *DebuggedSourceFiles) GetSourceFiles() []*DebuggedSourceFile
- func (*DebuggedSourceFiles) ProtoMessage()
- func (x *DebuggedSourceFiles) ProtoReflect() protoreflect.Message
- func (x *DebuggedSourceFiles) Reset()
- func (x *DebuggedSourceFiles) String() string
- type DeleteWorkerSessionRequest
- func (*DeleteWorkerSessionRequest) Descriptor() ([]byte, []int)deprecated
- func (x *DeleteWorkerSessionRequest) GetSessionHandle() string
- func (*DeleteWorkerSessionRequest) ProtoMessage()
- func (x *DeleteWorkerSessionRequest) ProtoReflect() protoreflect.Message
- func (x *DeleteWorkerSessionRequest) Reset()
- func (x *DeleteWorkerSessionRequest) String() string
- type DeleteWorkerSessionResponse
- type DeregisterGraphRequest
- func (*DeregisterGraphRequest) Descriptor() ([]byte, []int)deprecated
- func (x *DeregisterGraphRequest) GetCreateWorkerSessionCalled() bool
- func (x *DeregisterGraphRequest) GetGraphHandle() string
- func (x *DeregisterGraphRequest) GetSessionHandle() string
- func (*DeregisterGraphRequest) ProtoMessage()
- func (x *DeregisterGraphRequest) ProtoReflect() protoreflect.Message
- func (x *DeregisterGraphRequest) Reset()
- func (x *DeregisterGraphRequest) String() string
- type DeregisterGraphResponse
- type DeviceProperties
- func (*DeviceProperties) Descriptor() ([]byte, []int)deprecated
- func (x *DeviceProperties) GetBandwidth() int64
- func (x *DeviceProperties) GetEnvironment() map[string]string
- func (x *DeviceProperties) GetFrequency() int64
- func (x *DeviceProperties) GetL1CacheSize() int64
- func (x *DeviceProperties) GetL2CacheSize() int64
- func (x *DeviceProperties) GetL3CacheSize() int64
- func (x *DeviceProperties) GetMemorySize() int64
- func (x *DeviceProperties) GetModel() string
- func (x *DeviceProperties) GetNumCores() int64
- func (x *DeviceProperties) GetNumRegisters() int64
- func (x *DeviceProperties) GetSharedMemorySizePerMultiprocessor() int64
- func (x *DeviceProperties) GetType() string
- func (x *DeviceProperties) GetVendor() string
- func (*DeviceProperties) ProtoMessage()
- func (x *DeviceProperties) ProtoReflect() protoreflect.Message
- func (x *DeviceProperties) Reset()
- func (x *DeviceProperties) String() string
- type EnqueueRequest
- func (*EnqueueRequest) Descriptor() ([]byte, []int)deprecated
- func (x *EnqueueRequest) GetContextId() uint64
- func (x *EnqueueRequest) GetQueue() []*QueueItem
- func (*EnqueueRequest) ProtoMessage()
- func (x *EnqueueRequest) ProtoReflect() protoreflect.Message
- func (x *EnqueueRequest) Reset()
- func (x *EnqueueRequest) String() string
- type EnqueueResponse
- func (*EnqueueResponse) Descriptor() ([]byte, []int)deprecated
- func (x *EnqueueResponse) GetQueueResponse() []*QueueResponse
- func (*EnqueueResponse) ProtoMessage()
- func (x *EnqueueResponse) ProtoReflect() protoreflect.Message
- func (x *EnqueueResponse) Reset()
- func (x *EnqueueResponse) String() string
- type ExecutorOpts
- func (*ExecutorOpts) Descriptor() ([]byte, []int)deprecated
- func (x *ExecutorOpts) GetRecordCosts() bool
- func (x *ExecutorOpts) GetRecordPartitionGraphs() bool
- func (x *ExecutorOpts) GetRecordTimeline() bool
- func (x *ExecutorOpts) GetReportTensorAllocationsUponOom() bool
- func (*ExecutorOpts) ProtoMessage()
- func (x *ExecutorOpts) ProtoReflect() protoreflect.Message
- func (x *ExecutorOpts) Reset()
- func (x *ExecutorOpts) String() string
- type ExtendSessionRequest
- func (*ExtendSessionRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ExtendSessionRequest) GetCurrentGraphVersion() int64
- func (x *ExtendSessionRequest) GetGraphDef() *framework.GraphDef
- func (x *ExtendSessionRequest) GetSessionHandle() string
- func (*ExtendSessionRequest) ProtoMessage()
- func (x *ExtendSessionRequest) ProtoReflect() protoreflect.Message
- func (x *ExtendSessionRequest) Reset()
- func (x *ExtendSessionRequest) String() string
- type ExtendSessionResponse
- func (*ExtendSessionResponse) Descriptor() ([]byte, []int)deprecated
- func (x *ExtendSessionResponse) GetNewGraphVersion() int64
- func (*ExtendSessionResponse) ProtoMessage()
- func (x *ExtendSessionResponse) ProtoReflect() protoreflect.Message
- func (x *ExtendSessionResponse) Reset()
- func (x *ExtendSessionResponse) String() string
- type GPUOptions
- func (*GPUOptions) Descriptor() ([]byte, []int)deprecated
- func (x *GPUOptions) GetAllocatorType() string
- func (x *GPUOptions) GetAllowGrowth() bool
- func (x *GPUOptions) GetDeferredDeletionBytes() int64
- func (x *GPUOptions) GetExperimental() *GPUOptions_Experimental
- func (x *GPUOptions) GetForceGpuCompatible() bool
- func (x *GPUOptions) GetPerProcessGpuMemoryFraction() float64
- func (x *GPUOptions) GetPollingActiveDelayUsecs() int32
- func (x *GPUOptions) GetPollingInactiveDelayMsecs() int32
- func (x *GPUOptions) GetVisibleDeviceList() string
- func (*GPUOptions) ProtoMessage()
- func (x *GPUOptions) ProtoReflect() protoreflect.Message
- func (x *GPUOptions) Reset()
- func (x *GPUOptions) String() string
- type GPUOptions_Experimental
- func (*GPUOptions_Experimental) Descriptor() ([]byte, []int)deprecated
- func (x *GPUOptions_Experimental) GetCollectiveRingOrder() string
- func (x *GPUOptions_Experimental) GetNumDevToDevCopyStreams() int32
- func (x *GPUOptions_Experimental) GetUseUnifiedMemory() bool
- func (x *GPUOptions_Experimental) GetVirtualDevices() []*GPUOptions_Experimental_VirtualDevices
- func (*GPUOptions_Experimental) ProtoMessage()
- func (x *GPUOptions_Experimental) ProtoReflect() protoreflect.Message
- func (x *GPUOptions_Experimental) Reset()
- func (x *GPUOptions_Experimental) String() string
- type GPUOptions_Experimental_VirtualDevices
- func (*GPUOptions_Experimental_VirtualDevices) Descriptor() ([]byte, []int)deprecated
- func (x *GPUOptions_Experimental_VirtualDevices) GetMemoryLimitMb() []float32
- func (*GPUOptions_Experimental_VirtualDevices) ProtoMessage()
- func (x *GPUOptions_Experimental_VirtualDevices) ProtoReflect() protoreflect.Message
- func (x *GPUOptions_Experimental_VirtualDevices) Reset()
- func (x *GPUOptions_Experimental_VirtualDevices) String() string
- type GetStatusRequest
- type GetStatusResponse
- func (*GetStatusResponse) Descriptor() ([]byte, []int)deprecated
- func (x *GetStatusResponse) GetDeviceAttributes() []*framework.DeviceAttributes
- func (*GetStatusResponse) ProtoMessage()
- func (x *GetStatusResponse) ProtoReflect() protoreflect.Message
- func (x *GetStatusResponse) Reset()
- func (x *GetStatusResponse) String() string
- type GetStepSequenceRequest
- func (*GetStepSequenceRequest) Descriptor() ([]byte, []int)deprecated
- func (x *GetStepSequenceRequest) GetGraphKey() []int64
- func (*GetStepSequenceRequest) ProtoMessage()
- func (x *GetStepSequenceRequest) ProtoReflect() protoreflect.Message
- func (x *GetStepSequenceRequest) Reset()
- func (x *GetStepSequenceRequest) String() string
- type GetStepSequenceResponse
- func (*GetStepSequenceResponse) Descriptor() ([]byte, []int)deprecated
- func (x *GetStepSequenceResponse) GetStepSequence() []*StepSequence
- func (*GetStepSequenceResponse) ProtoMessage()
- func (x *GetStepSequenceResponse) ProtoReflect() protoreflect.Message
- func (x *GetStepSequenceResponse) Reset()
- func (x *GetStepSequenceResponse) String() string
- type GraphOptions
- func (*GraphOptions) Descriptor() ([]byte, []int)deprecated
- func (x *GraphOptions) GetBuildCostModel() int64
- func (x *GraphOptions) GetBuildCostModelAfter() int64
- func (x *GraphOptions) GetEnableBfloat16Sendrecv() bool
- func (x *GraphOptions) GetEnableRecvScheduling() bool
- func (x *GraphOptions) GetInferShapes() bool
- func (x *GraphOptions) GetOptimizerOptions() *OptimizerOptions
- func (x *GraphOptions) GetPlacePrunedGraph() bool
- func (x *GraphOptions) GetRewriteOptions() *RewriterConfig
- func (x *GraphOptions) GetTimelineStep() int32
- func (*GraphOptions) ProtoMessage()
- func (x *GraphOptions) ProtoReflect() protoreflect.Message
- func (x *GraphOptions) Reset()
- func (x *GraphOptions) String() string
- type JobDef
- type KeepAliveRequest
- type KeepAliveResponse
- type LabeledStepStats
- func (*LabeledStepStats) Descriptor() ([]byte, []int)deprecated
- func (x *LabeledStepStats) GetStepId() int64
- func (x *LabeledStepStats) GetStepStats() *framework.StepStats
- func (*LabeledStepStats) ProtoMessage()
- func (x *LabeledStepStats) ProtoReflect() protoreflect.Message
- func (x *LabeledStepStats) Reset()
- func (x *LabeledStepStats) String() string
- type ListDevicesRequest
- func (*ListDevicesRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ListDevicesRequest) GetSessionHandle() string
- func (*ListDevicesRequest) ProtoMessage()
- func (x *ListDevicesRequest) ProtoReflect() protoreflect.Message
- func (x *ListDevicesRequest) Reset()
- func (x *ListDevicesRequest) String() string
- type ListDevicesResponse
- func (*ListDevicesResponse) Descriptor() ([]byte, []int)deprecated
- func (x *ListDevicesResponse) GetLocalDevice() []*framework.DeviceAttributes
- func (x *ListDevicesResponse) GetRemoteDevice() []*framework.DeviceAttributes
- func (*ListDevicesResponse) ProtoMessage()
- func (x *ListDevicesResponse) ProtoReflect() protoreflect.Message
- func (x *ListDevicesResponse) Reset()
- func (x *ListDevicesResponse) String() string
- type LoggingRequest
- func (*LoggingRequest) Descriptor() ([]byte, []int)deprecated
- func (x *LoggingRequest) GetClear() bool
- func (x *LoggingRequest) GetDisableRpcLogging() bool
- func (x *LoggingRequest) GetEnableRpcLogging() bool
- func (x *LoggingRequest) GetFetchStepId() []int64
- func (*LoggingRequest) ProtoMessage()
- func (x *LoggingRequest) ProtoReflect() protoreflect.Message
- func (x *LoggingRequest) Reset()
- func (x *LoggingRequest) String() string
- type LoggingResponse
- type MakeCallableRequest
- func (*MakeCallableRequest) Descriptor() ([]byte, []int)deprecated
- func (x *MakeCallableRequest) GetOptions() *CallableOptions
- func (x *MakeCallableRequest) GetSessionHandle() string
- func (*MakeCallableRequest) ProtoMessage()
- func (x *MakeCallableRequest) ProtoReflect() protoreflect.Message
- func (x *MakeCallableRequest) Reset()
- func (x *MakeCallableRequest) String() string
- type MakeCallableResponse
- func (*MakeCallableResponse) Descriptor() ([]byte, []int)deprecated
- func (x *MakeCallableResponse) GetHandle() int64
- func (*MakeCallableResponse) ProtoMessage()
- func (x *MakeCallableResponse) ProtoReflect() protoreflect.Message
- func (x *MakeCallableResponse) Reset()
- func (x *MakeCallableResponse) String() string
- type MetaGraphDef
- func (*MetaGraphDef) Descriptor() ([]byte, []int)deprecated
- func (x *MetaGraphDef) GetAssetFileDef() []*AssetFileDef
- func (x *MetaGraphDef) GetCollectionDef() map[string]*CollectionDef
- func (x *MetaGraphDef) GetGraphDef() *framework.GraphDef
- func (x *MetaGraphDef) GetMetaInfoDef() *MetaGraphDef_MetaInfoDef
- func (x *MetaGraphDef) GetSaverDef() *SaverDef
- func (x *MetaGraphDef) GetSignatureDef() map[string]*SignatureDef
- func (*MetaGraphDef) ProtoMessage()
- func (x *MetaGraphDef) ProtoReflect() protoreflect.Message
- func (x *MetaGraphDef) Reset()
- func (x *MetaGraphDef) String() string
- type MetaGraphDef_MetaInfoDef
- func (*MetaGraphDef_MetaInfoDef) Descriptor() ([]byte, []int)deprecated
- func (x *MetaGraphDef_MetaInfoDef) GetAnyInfo() *any.Any
- func (x *MetaGraphDef_MetaInfoDef) GetMetaGraphVersion() string
- func (x *MetaGraphDef_MetaInfoDef) GetStrippedDefaultAttrs() bool
- func (x *MetaGraphDef_MetaInfoDef) GetStrippedOpList() *framework.OpList
- func (x *MetaGraphDef_MetaInfoDef) GetTags() []string
- func (x *MetaGraphDef_MetaInfoDef) GetTensorflowGitVersion() string
- func (x *MetaGraphDef_MetaInfoDef) GetTensorflowVersion() string
- func (*MetaGraphDef_MetaInfoDef) ProtoMessage()
- func (x *MetaGraphDef_MetaInfoDef) ProtoReflect() protoreflect.Message
- func (x *MetaGraphDef_MetaInfoDef) Reset()
- func (x *MetaGraphDef_MetaInfoDef) String() string
- type NamedDevice
- func (*NamedDevice) Descriptor() ([]byte, []int)deprecated
- func (x *NamedDevice) GetName() string
- func (x *NamedDevice) GetProperties() *DeviceProperties
- func (*NamedDevice) ProtoMessage()
- func (x *NamedDevice) ProtoReflect() protoreflect.Message
- func (x *NamedDevice) Reset()
- func (x *NamedDevice) String() string
- type NamedTensorProto
- func (*NamedTensorProto) Descriptor() ([]byte, []int)deprecated
- func (x *NamedTensorProto) GetName() string
- func (x *NamedTensorProto) GetTensor() *framework.TensorProto
- func (*NamedTensorProto) ProtoMessage()
- func (x *NamedTensorProto) ProtoReflect() protoreflect.Message
- func (x *NamedTensorProto) Reset()
- func (x *NamedTensorProto) String() string
- type NewReplaySession
- func (*NewReplaySession) Descriptor() ([]byte, []int)deprecated
- func (x *NewReplaySession) GetDevices() *ListDevicesResponse
- func (x *NewReplaySession) GetSessionHandle() string
- func (*NewReplaySession) ProtoMessage()
- func (x *NewReplaySession) ProtoReflect() protoreflect.Message
- func (x *NewReplaySession) Reset()
- func (x *NewReplaySession) String() string
- type Operation
- func (*Operation) Descriptor() ([]byte, []int)deprecated
- func (x *Operation) GetAttrs() map[string]*framework.AttrValue
- func (x *Operation) GetControlOpIds() []int64
- func (x *Operation) GetDevice() string
- func (x *Operation) GetId() int64
- func (x *Operation) GetInputs() []*RemoteTensorHandle
- func (x *Operation) GetName() string
- func (*Operation) ProtoMessage()
- func (x *Operation) ProtoReflect() protoreflect.Message
- func (x *Operation) Reset()
- func (x *Operation) String() string
- type OptimizerOptions
- func (*OptimizerOptions) Descriptor() ([]byte, []int)deprecated
- func (x *OptimizerOptions) GetDoCommonSubexpressionElimination() bool
- func (x *OptimizerOptions) GetDoConstantFolding() bool
- func (x *OptimizerOptions) GetDoFunctionInlining() bool
- func (x *OptimizerOptions) GetGlobalJitLevel() OptimizerOptions_GlobalJitLevel
- func (x *OptimizerOptions) GetMaxFoldedConstantInBytes() int64
- func (x *OptimizerOptions) GetOptLevel() OptimizerOptions_Level
- func (*OptimizerOptions) ProtoMessage()
- func (x *OptimizerOptions) ProtoReflect() protoreflect.Message
- func (x *OptimizerOptions) Reset()
- func (x *OptimizerOptions) String() string
- type OptimizerOptions_GlobalJitLevel
- func (OptimizerOptions_GlobalJitLevel) Descriptor() protoreflect.EnumDescriptor
- func (x OptimizerOptions_GlobalJitLevel) Enum() *OptimizerOptions_GlobalJitLevel
- func (OptimizerOptions_GlobalJitLevel) EnumDescriptor() ([]byte, []int)deprecated
- func (x OptimizerOptions_GlobalJitLevel) Number() protoreflect.EnumNumber
- func (x OptimizerOptions_GlobalJitLevel) String() string
- func (OptimizerOptions_GlobalJitLevel) Type() protoreflect.EnumType
- type OptimizerOptions_Level
- func (OptimizerOptions_Level) Descriptor() protoreflect.EnumDescriptor
- func (x OptimizerOptions_Level) Enum() *OptimizerOptions_Level
- func (OptimizerOptions_Level) EnumDescriptor() ([]byte, []int)deprecated
- func (x OptimizerOptions_Level) Number() protoreflect.EnumNumber
- func (x OptimizerOptions_Level) String() string
- func (OptimizerOptions_Level) Type() protoreflect.EnumType
- type PartialRunSetupRequest
- func (*PartialRunSetupRequest) Descriptor() ([]byte, []int)deprecated
- func (x *PartialRunSetupRequest) GetFeed() []string
- func (x *PartialRunSetupRequest) GetFetch() []string
- func (x *PartialRunSetupRequest) GetSessionHandle() string
- func (x *PartialRunSetupRequest) GetTarget() []string
- func (*PartialRunSetupRequest) ProtoMessage()
- func (x *PartialRunSetupRequest) ProtoReflect() protoreflect.Message
- func (x *PartialRunSetupRequest) Reset()
- func (x *PartialRunSetupRequest) String() string
- type PartialRunSetupResponse
- func (*PartialRunSetupResponse) Descriptor() ([]byte, []int)deprecated
- func (x *PartialRunSetupResponse) GetPartialRunHandle() string
- func (*PartialRunSetupResponse) ProtoMessage()
- func (x *PartialRunSetupResponse) ProtoReflect() protoreflect.Message
- func (x *PartialRunSetupResponse) Reset()
- func (x *PartialRunSetupResponse) String() string
- type QueueItem
- func (*QueueItem) Descriptor() ([]byte, []int)deprecated
- func (x *QueueItem) GetHandleToDecref() *RemoteTensorHandle
- func (m *QueueItem) GetItem() isQueueItem_Item
- func (x *QueueItem) GetOperation() *Operation
- func (*QueueItem) ProtoMessage()
- func (x *QueueItem) ProtoReflect() protoreflect.Message
- func (x *QueueItem) Reset()
- func (x *QueueItem) String() string
- type QueueItem_HandleToDecref
- type QueueItem_Operation
- type QueueResponse
- type QueueRunnerDef
- func (*QueueRunnerDef) Descriptor() ([]byte, []int)deprecated
- func (x *QueueRunnerDef) GetCancelOpName() string
- func (x *QueueRunnerDef) GetCloseOpName() string
- func (x *QueueRunnerDef) GetEnqueueOpName() []string
- func (x *QueueRunnerDef) GetQueueClosedExceptionTypes() []core.Code
- func (x *QueueRunnerDef) GetQueueName() string
- func (*QueueRunnerDef) ProtoMessage()
- func (x *QueueRunnerDef) ProtoReflect() protoreflect.Message
- func (x *QueueRunnerDef) Reset()
- func (x *QueueRunnerDef) String() string
- type RPCOptions
- func (*RPCOptions) Descriptor() ([]byte, []int)deprecated
- func (x *RPCOptions) GetCompressionAlgorithm() string
- func (x *RPCOptions) GetCompressionLevel() int32
- func (x *RPCOptions) GetUseRpcForInprocessMaster() bool
- func (*RPCOptions) ProtoMessage()
- func (x *RPCOptions) ProtoReflect() protoreflect.Message
- func (x *RPCOptions) Reset()
- func (x *RPCOptions) String() string
- type RecvBufRequest
- func (*RecvBufRequest) Descriptor() ([]byte, []int)deprecated
- func (x *RecvBufRequest) GetBufPtr() uint64
- func (x *RecvBufRequest) GetBufRendezvousKey() string
- func (x *RecvBufRequest) GetClientLocality() *framework.DeviceLocality
- func (x *RecvBufRequest) GetDstDevice() string
- func (x *RecvBufRequest) GetNumBytes() int64
- func (x *RecvBufRequest) GetRequestId() int64
- func (x *RecvBufRequest) GetServerLocality() *framework.DeviceLocality
- func (x *RecvBufRequest) GetSrcDevice() string
- func (x *RecvBufRequest) GetStepId() int64
- func (x *RecvBufRequest) GetTransportOptions() *any.Any
- func (*RecvBufRequest) ProtoMessage()
- func (x *RecvBufRequest) ProtoReflect() protoreflect.Message
- func (x *RecvBufRequest) Reset()
- func (x *RecvBufRequest) String() string
- type RecvBufRespExtra
- type RecvBufResponse
- func (*RecvBufResponse) Descriptor() ([]byte, []int)deprecated
- func (x *RecvBufResponse) GetBufPtr() uint64
- func (x *RecvBufResponse) GetIsDead() bool
- func (x *RecvBufResponse) GetNumBytes() int64
- func (x *RecvBufResponse) GetSendStartMicros() int64
- func (x *RecvBufResponse) GetTransportOptions() *any.Any
- func (*RecvBufResponse) ProtoMessage()
- func (x *RecvBufResponse) ProtoReflect() protoreflect.Message
- func (x *RecvBufResponse) Reset()
- func (x *RecvBufResponse) String() string
- type RecvTensorRequest
- func (*RecvTensorRequest) Descriptor() ([]byte, []int)deprecated
- func (x *RecvTensorRequest) GetClientLocality() *framework.DeviceLocality
- func (x *RecvTensorRequest) GetDmaOk() bool
- func (x *RecvTensorRequest) GetRendezvousKey() string
- func (x *RecvTensorRequest) GetRequestId() int64
- func (x *RecvTensorRequest) GetServerLocality() *framework.DeviceLocality
- func (x *RecvTensorRequest) GetStepId() int64
- func (x *RecvTensorRequest) GetTransportOptions() *any.Any
- func (*RecvTensorRequest) ProtoMessage()
- func (x *RecvTensorRequest) ProtoReflect() protoreflect.Message
- func (x *RecvTensorRequest) Reset()
- func (x *RecvTensorRequest) String() string
- type RecvTensorResponse
- func (*RecvTensorResponse) Descriptor() ([]byte, []int)deprecated
- func (x *RecvTensorResponse) GetIsDead() bool
- func (x *RecvTensorResponse) GetSendStartMicros() int64
- func (x *RecvTensorResponse) GetTensor() *framework.TensorProto
- func (x *RecvTensorResponse) GetTransportOptions() *any.Any
- func (*RecvTensorResponse) ProtoMessage()
- func (x *RecvTensorResponse) ProtoReflect() protoreflect.Message
- func (x *RecvTensorResponse) Reset()
- func (x *RecvTensorResponse) String() string
- type RegisterFunctionRequest
- func (*RegisterFunctionRequest) Descriptor() ([]byte, []int)deprecated
- func (x *RegisterFunctionRequest) GetContextId() uint64
- func (x *RegisterFunctionRequest) GetFunctionDef() *framework.FunctionDef
- func (*RegisterFunctionRequest) ProtoMessage()
- func (x *RegisterFunctionRequest) ProtoReflect() protoreflect.Message
- func (x *RegisterFunctionRequest) Reset()
- func (x *RegisterFunctionRequest) String() string
- type RegisterFunctionResponse
- type RegisterGraphRequest
- func (*RegisterGraphRequest) Descriptor() ([]byte, []int)deprecated
- func (x *RegisterGraphRequest) GetCollectiveGraphKey() int64
- func (x *RegisterGraphRequest) GetCreateWorkerSessionCalled() bool
- func (x *RegisterGraphRequest) GetDebugOptions() *DebugOptions
- func (x *RegisterGraphRequest) GetGraphDef() *framework.GraphDef
- func (x *RegisterGraphRequest) GetGraphOptions() *GraphOptions
- func (x *RegisterGraphRequest) GetHasControlFlow() booldeprecated
- func (x *RegisterGraphRequest) GetSessionHandle() string
- func (*RegisterGraphRequest) ProtoMessage()
- func (x *RegisterGraphRequest) ProtoReflect() protoreflect.Message
- func (x *RegisterGraphRequest) Reset()
- func (x *RegisterGraphRequest) String() string
- type RegisterGraphResponse
- func (*RegisterGraphResponse) Descriptor() ([]byte, []int)deprecated
- func (x *RegisterGraphResponse) GetGraphHandle() string
- func (*RegisterGraphResponse) ProtoMessage()
- func (x *RegisterGraphResponse) ProtoReflect() protoreflect.Message
- func (x *RegisterGraphResponse) Reset()
- func (x *RegisterGraphResponse) String() string
- type ReleaseCallableRequest
- func (*ReleaseCallableRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ReleaseCallableRequest) GetHandle() int64
- func (x *ReleaseCallableRequest) GetSessionHandle() string
- func (*ReleaseCallableRequest) ProtoMessage()
- func (x *ReleaseCallableRequest) ProtoReflect() protoreflect.Message
- func (x *ReleaseCallableRequest) Reset()
- func (x *ReleaseCallableRequest) String() string
- type ReleaseCallableResponse
- type RemoteTensorHandle
- func (*RemoteTensorHandle) Descriptor() ([]byte, []int)deprecated
- func (x *RemoteTensorHandle) GetOpId() int64
- func (x *RemoteTensorHandle) GetOutputNum() int32
- func (*RemoteTensorHandle) ProtoMessage()
- func (x *RemoteTensorHandle) ProtoReflect() protoreflect.Message
- func (x *RemoteTensorHandle) Reset()
- func (x *RemoteTensorHandle) String() string
- type ReplayOp
- func (*ReplayOp) Descriptor() ([]byte, []int)deprecated
- func (x *ReplayOp) GetCloseSession() *CloseSessionRequest
- func (x *ReplayOp) GetCloseSessionResponse() *CloseSessionResponse
- func (x *ReplayOp) GetCreateSession() *CreateSessionRequest
- func (x *ReplayOp) GetCreateSessionResponse() *CreateSessionResponse
- func (x *ReplayOp) GetEndTimeUs() float64
- func (x *ReplayOp) GetExtendSession() *ExtendSessionRequest
- func (x *ReplayOp) GetExtendSessionResponse() *ExtendSessionResponse
- func (x *ReplayOp) GetListDevices() *ListDevicesRequest
- func (x *ReplayOp) GetListDevicesResponse() *ListDevicesResponse
- func (x *ReplayOp) GetMakeCallable() *MakeCallableRequest
- func (x *ReplayOp) GetMakeCallableResponse() *MakeCallableResponse
- func (x *ReplayOp) GetNewReplaySession() *NewReplaySession
- func (m *ReplayOp) GetOp() isReplayOp_Op
- func (x *ReplayOp) GetPartialRunSetup() *PartialRunSetupRequest
- func (x *ReplayOp) GetPartialRunSetupResponse() *PartialRunSetupResponse
- func (x *ReplayOp) GetReleaseCallable() *ReleaseCallableRequest
- func (x *ReplayOp) GetReleaseCallableResponse() *ReleaseCallableResponse
- func (x *ReplayOp) GetResetRequest() *ResetRequest
- func (x *ReplayOp) GetResetRequestResponse() *ResetResponse
- func (m *ReplayOp) GetResponse() isReplayOp_Response
- func (x *ReplayOp) GetRunCallable() *RunCallableRequest
- func (x *ReplayOp) GetRunCallableResponse() *RunCallableResponse
- func (x *ReplayOp) GetRunStep() *RunStepRequest
- func (x *ReplayOp) GetRunStepResponse() *RunStepResponse
- func (x *ReplayOp) GetStartTimeUs() float64
- func (*ReplayOp) ProtoMessage()
- func (x *ReplayOp) ProtoReflect() protoreflect.Message
- func (x *ReplayOp) Reset()
- func (x *ReplayOp) String() string
- type ReplayOp_CloseSession
- type ReplayOp_CloseSessionResponse
- type ReplayOp_CreateSession
- type ReplayOp_CreateSessionResponse
- type ReplayOp_ExtendSession
- type ReplayOp_ExtendSessionResponse
- type ReplayOp_ListDevices
- type ReplayOp_ListDevicesResponse
- type ReplayOp_MakeCallable
- type ReplayOp_MakeCallableResponse
- type ReplayOp_NewReplaySession
- type ReplayOp_PartialRunSetup
- type ReplayOp_PartialRunSetupResponse
- type ReplayOp_ReleaseCallable
- type ReplayOp_ReleaseCallableResponse
- type ReplayOp_ResetRequest
- type ReplayOp_ResetRequestResponse
- type ReplayOp_RunCallable
- type ReplayOp_RunCallableResponse
- type ReplayOp_RunStep
- type ReplayOp_RunStepResponse
- type ResetRequest
- func (*ResetRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ResetRequest) GetContainer() []string
- func (x *ResetRequest) GetDeviceFilters() []string
- func (*ResetRequest) ProtoMessage()
- func (x *ResetRequest) ProtoReflect() protoreflect.Message
- func (x *ResetRequest) Reset()
- func (x *ResetRequest) String() string
- type ResetResponse
- type RewriterConfig
- func (*RewriterConfig) Descriptor() ([]byte, []int)deprecated
- func (x *RewriterConfig) GetArithmeticOptimization() RewriterConfig_Toggle
- func (x *RewriterConfig) GetAutoParallel() *AutoParallelOptions
- func (x *RewriterConfig) GetConstantFolding() RewriterConfig_Toggle
- func (x *RewriterConfig) GetCustomOptimizers() []*RewriterConfig_CustomGraphOptimizer
- func (x *RewriterConfig) GetDebugStripper() RewriterConfig_Toggle
- func (x *RewriterConfig) GetDependencyOptimization() RewriterConfig_Toggle
- func (x *RewriterConfig) GetDisableMetaOptimizer() bool
- func (x *RewriterConfig) GetDisableModelPruning() bool
- func (x *RewriterConfig) GetFailOnOptimizerErrors() bool
- func (x *RewriterConfig) GetFunctionOptimization() RewriterConfig_Toggle
- func (x *RewriterConfig) GetLayoutOptimizer() RewriterConfig_Toggle
- func (x *RewriterConfig) GetLoopOptimization() RewriterConfig_Toggle
- func (x *RewriterConfig) GetMemoryOptimization() RewriterConfig_MemOptType
- func (x *RewriterConfig) GetMemoryOptimizerTargetNodeNameScope() string
- func (x *RewriterConfig) GetMetaOptimizerIterations() RewriterConfig_NumIterationsType
- func (x *RewriterConfig) GetMetaOptimizerTimeoutMs() int64
- func (x *RewriterConfig) GetMinGraphNodes() int32
- func (x *RewriterConfig) GetOptimizers() []string
- func (x *RewriterConfig) GetPinToHostOptimization() RewriterConfig_Toggle
- func (x *RewriterConfig) GetRemapping() RewriterConfig_Toggle
- func (x *RewriterConfig) GetScopedAllocatorOptimization() RewriterConfig_Toggle
- func (x *RewriterConfig) GetScopedAllocatorOpts() *ScopedAllocatorOptions
- func (x *RewriterConfig) GetShapeOptimization() RewriterConfig_Toggle
- func (*RewriterConfig) ProtoMessage()
- func (x *RewriterConfig) ProtoReflect() protoreflect.Message
- func (x *RewriterConfig) Reset()
- func (x *RewriterConfig) String() string
- type RewriterConfig_CustomGraphOptimizer
- func (*RewriterConfig_CustomGraphOptimizer) Descriptor() ([]byte, []int)deprecated
- func (x *RewriterConfig_CustomGraphOptimizer) GetName() string
- func (x *RewriterConfig_CustomGraphOptimizer) GetParameterMap() map[string]*framework.AttrValue
- func (*RewriterConfig_CustomGraphOptimizer) ProtoMessage()
- func (x *RewriterConfig_CustomGraphOptimizer) ProtoReflect() protoreflect.Message
- func (x *RewriterConfig_CustomGraphOptimizer) Reset()
- func (x *RewriterConfig_CustomGraphOptimizer) String() string
- type RewriterConfig_MemOptType
- func (RewriterConfig_MemOptType) Descriptor() protoreflect.EnumDescriptor
- func (x RewriterConfig_MemOptType) Enum() *RewriterConfig_MemOptType
- func (RewriterConfig_MemOptType) EnumDescriptor() ([]byte, []int)deprecated
- func (x RewriterConfig_MemOptType) Number() protoreflect.EnumNumber
- func (x RewriterConfig_MemOptType) String() string
- func (RewriterConfig_MemOptType) Type() protoreflect.EnumType
- type RewriterConfig_NumIterationsType
- func (RewriterConfig_NumIterationsType) Descriptor() protoreflect.EnumDescriptor
- func (x RewriterConfig_NumIterationsType) Enum() *RewriterConfig_NumIterationsType
- func (RewriterConfig_NumIterationsType) EnumDescriptor() ([]byte, []int)deprecated
- func (x RewriterConfig_NumIterationsType) Number() protoreflect.EnumNumber
- func (x RewriterConfig_NumIterationsType) String() string
- func (RewriterConfig_NumIterationsType) Type() protoreflect.EnumType
- type RewriterConfig_Toggle
- func (RewriterConfig_Toggle) Descriptor() protoreflect.EnumDescriptor
- func (x RewriterConfig_Toggle) Enum() *RewriterConfig_Toggle
- func (RewriterConfig_Toggle) EnumDescriptor() ([]byte, []int)deprecated
- func (x RewriterConfig_Toggle) Number() protoreflect.EnumNumber
- func (x RewriterConfig_Toggle) String() string
- func (RewriterConfig_Toggle) Type() protoreflect.EnumType
- type RunCallableRequest
- func (*RunCallableRequest) Descriptor() ([]byte, []int)deprecated
- func (x *RunCallableRequest) GetFeed() []*framework.TensorProto
- func (x *RunCallableRequest) GetHandle() int64
- func (x *RunCallableRequest) GetSessionHandle() string
- func (*RunCallableRequest) ProtoMessage()
- func (x *RunCallableRequest) ProtoReflect() protoreflect.Message
- func (x *RunCallableRequest) Reset()
- func (x *RunCallableRequest) String() string
- type RunCallableResponse
- func (*RunCallableResponse) Descriptor() ([]byte, []int)deprecated
- func (x *RunCallableResponse) GetFetch() []*framework.TensorProto
- func (x *RunCallableResponse) GetMetadata() *RunMetadata
- func (*RunCallableResponse) ProtoMessage()
- func (x *RunCallableResponse) ProtoReflect() protoreflect.Message
- func (x *RunCallableResponse) Reset()
- func (x *RunCallableResponse) String() string
- type RunGraphRequest
- func (*RunGraphRequest) Descriptor() ([]byte, []int)deprecated
- func (x *RunGraphRequest) GetCreateWorkerSessionCalled() bool
- func (x *RunGraphRequest) GetExecOpts() *ExecutorOpts
- func (x *RunGraphRequest) GetGraphHandle() string
- func (x *RunGraphRequest) GetIsLastPartialRun() bool
- func (x *RunGraphRequest) GetIsPartial() bool
- func (x *RunGraphRequest) GetRecvKey() []string
- func (x *RunGraphRequest) GetSend() []*NamedTensorProto
- func (x *RunGraphRequest) GetSessionHandle() string
- func (x *RunGraphRequest) GetStepId() int64
- func (x *RunGraphRequest) GetStoreErrorsInResponseBody() bool
- func (*RunGraphRequest) ProtoMessage()
- func (x *RunGraphRequest) ProtoReflect() protoreflect.Message
- func (x *RunGraphRequest) Reset()
- func (x *RunGraphRequest) String() string
- type RunGraphResponse
- func (*RunGraphResponse) Descriptor() ([]byte, []int)deprecated
- func (x *RunGraphResponse) GetCostGraph() *framework.CostGraphDef
- func (x *RunGraphResponse) GetPartitionGraph() []*framework.GraphDef
- func (x *RunGraphResponse) GetRecv() []*NamedTensorProto
- func (x *RunGraphResponse) GetStatusCode() core.Code
- func (x *RunGraphResponse) GetStatusErrorMessage() string
- func (x *RunGraphResponse) GetStepStats() *framework.StepStats
- func (*RunGraphResponse) ProtoMessage()
- func (x *RunGraphResponse) ProtoReflect() protoreflect.Message
- func (x *RunGraphResponse) Reset()
- func (x *RunGraphResponse) String() string
- type RunMetadata
- func (*RunMetadata) Descriptor() ([]byte, []int)deprecated
- func (x *RunMetadata) GetCostGraph() *framework.CostGraphDef
- func (x *RunMetadata) GetPartitionGraphs() []*framework.GraphDef
- func (x *RunMetadata) GetStepStats() *framework.StepStats
- func (*RunMetadata) ProtoMessage()
- func (x *RunMetadata) ProtoReflect() protoreflect.Message
- func (x *RunMetadata) Reset()
- func (x *RunMetadata) String() string
- type RunOptions
- func (*RunOptions) Descriptor() ([]byte, []int)deprecated
- func (x *RunOptions) GetDebugOptions() *DebugOptions
- func (x *RunOptions) GetExperimental() *RunOptions_Experimental
- func (x *RunOptions) GetInterOpThreadPool() int32
- func (x *RunOptions) GetOutputPartitionGraphs() bool
- func (x *RunOptions) GetReportTensorAllocationsUponOom() bool
- func (x *RunOptions) GetTimeoutInMs() int64
- func (x *RunOptions) GetTraceLevel() RunOptions_TraceLevel
- func (*RunOptions) ProtoMessage()
- func (x *RunOptions) ProtoReflect() protoreflect.Message
- func (x *RunOptions) Reset()
- func (x *RunOptions) String() string
- type RunOptions_Experimental
- func (*RunOptions_Experimental) Descriptor() ([]byte, []int)deprecated
- func (x *RunOptions_Experimental) GetCollectiveGraphKey() int64
- func (x *RunOptions_Experimental) GetUseRunHandlerPool() bool
- func (*RunOptions_Experimental) ProtoMessage()
- func (x *RunOptions_Experimental) ProtoReflect() protoreflect.Message
- func (x *RunOptions_Experimental) Reset()
- func (x *RunOptions_Experimental) String() string
- type RunOptions_TraceLevel
- func (RunOptions_TraceLevel) Descriptor() protoreflect.EnumDescriptor
- func (x RunOptions_TraceLevel) Enum() *RunOptions_TraceLevel
- func (RunOptions_TraceLevel) EnumDescriptor() ([]byte, []int)deprecated
- func (x RunOptions_TraceLevel) Number() protoreflect.EnumNumber
- func (x RunOptions_TraceLevel) String() string
- func (RunOptions_TraceLevel) Type() protoreflect.EnumType
- type RunStepRequest
- func (*RunStepRequest) Descriptor() ([]byte, []int)deprecated
- func (x *RunStepRequest) GetFeed() []*NamedTensorProto
- func (x *RunStepRequest) GetFetch() []string
- func (x *RunStepRequest) GetOptions() *RunOptions
- func (x *RunStepRequest) GetPartialRunHandle() string
- func (x *RunStepRequest) GetSessionHandle() string
- func (x *RunStepRequest) GetStoreErrorsInResponseBody() bool
- func (x *RunStepRequest) GetTarget() []string
- func (*RunStepRequest) ProtoMessage()
- func (x *RunStepRequest) ProtoReflect() protoreflect.Message
- func (x *RunStepRequest) Reset()
- func (x *RunStepRequest) String() string
- type RunStepResponse
- func (*RunStepResponse) Descriptor() ([]byte, []int)deprecated
- func (x *RunStepResponse) GetMetadata() *RunMetadata
- func (x *RunStepResponse) GetStatusCode() core.Code
- func (x *RunStepResponse) GetStatusErrorMessage() string
- func (x *RunStepResponse) GetTensor() []*NamedTensorProto
- func (*RunStepResponse) ProtoMessage()
- func (x *RunStepResponse) ProtoReflect() protoreflect.Message
- func (x *RunStepResponse) Reset()
- func (x *RunStepResponse) String() string
- type SavedModel
- func (*SavedModel) Descriptor() ([]byte, []int)deprecated
- func (x *SavedModel) GetMetaGraphs() []*MetaGraphDef
- func (x *SavedModel) GetSavedModelSchemaVersion() int64
- func (*SavedModel) ProtoMessage()
- func (x *SavedModel) ProtoReflect() protoreflect.Message
- func (x *SavedModel) Reset()
- func (x *SavedModel) String() string
- type SaverDef
- func (*SaverDef) Descriptor() ([]byte, []int)deprecated
- func (x *SaverDef) GetFilenameTensorName() string
- func (x *SaverDef) GetKeepCheckpointEveryNHours() float32
- func (x *SaverDef) GetMaxToKeep() int32
- func (x *SaverDef) GetRestoreOpName() string
- func (x *SaverDef) GetSaveTensorName() string
- func (x *SaverDef) GetSharded() bool
- func (x *SaverDef) GetVersion() SaverDef_CheckpointFormatVersion
- func (*SaverDef) ProtoMessage()
- func (x *SaverDef) ProtoReflect() protoreflect.Message
- func (x *SaverDef) Reset()
- func (x *SaverDef) String() string
- type SaverDef_CheckpointFormatVersion
- func (SaverDef_CheckpointFormatVersion) Descriptor() protoreflect.EnumDescriptor
- func (x SaverDef_CheckpointFormatVersion) Enum() *SaverDef_CheckpointFormatVersion
- func (SaverDef_CheckpointFormatVersion) EnumDescriptor() ([]byte, []int)deprecated
- func (x SaverDef_CheckpointFormatVersion) Number() protoreflect.EnumNumber
- func (x SaverDef_CheckpointFormatVersion) String() string
- func (SaverDef_CheckpointFormatVersion) Type() protoreflect.EnumType
- type ScopedAllocatorOptions
- func (*ScopedAllocatorOptions) Descriptor() ([]byte, []int)deprecated
- func (x *ScopedAllocatorOptions) GetEnableOp() []string
- func (*ScopedAllocatorOptions) ProtoMessage()
- func (x *ScopedAllocatorOptions) ProtoReflect() protoreflect.Message
- func (x *ScopedAllocatorOptions) Reset()
- func (x *ScopedAllocatorOptions) String() string
- type SendTensorRequest
- func (*SendTensorRequest) Descriptor() ([]byte, []int)deprecated
- func (x *SendTensorRequest) GetContextId() uint64
- func (x *SendTensorRequest) GetDeviceName() string
- func (x *SendTensorRequest) GetOpId() int64
- func (x *SendTensorRequest) GetTensors() []*framework.TensorProto
- func (*SendTensorRequest) ProtoMessage()
- func (x *SendTensorRequest) ProtoReflect() protoreflect.Message
- func (x *SendTensorRequest) Reset()
- func (x *SendTensorRequest) String() string
- type SendTensorResponse
- type ServerDef
- func (*ServerDef) Descriptor() ([]byte, []int)deprecated
- func (x *ServerDef) GetCluster() *ClusterDef
- func (x *ServerDef) GetDefaultSessionConfig() *ConfigProto
- func (x *ServerDef) GetJobName() string
- func (x *ServerDef) GetProtocol() string
- func (x *ServerDef) GetTaskIndex() int32
- func (*ServerDef) ProtoMessage()
- func (x *ServerDef) ProtoReflect() protoreflect.Message
- func (x *ServerDef) Reset()
- func (x *ServerDef) String() string
- type SignatureDef
- func (*SignatureDef) Descriptor() ([]byte, []int)deprecated
- func (x *SignatureDef) GetInputs() map[string]*TensorInfo
- func (x *SignatureDef) GetMethodName() string
- func (x *SignatureDef) GetOutputs() map[string]*TensorInfo
- func (*SignatureDef) ProtoMessage()
- func (x *SignatureDef) ProtoReflect() protoreflect.Message
- func (x *SignatureDef) Reset()
- func (x *SignatureDef) String() string
- type StepSequence
- func (*StepSequence) Descriptor() ([]byte, []int)deprecated
- func (x *StepSequence) GetGraphKey() int64
- func (x *StepSequence) GetNextStepId() int64
- func (*StepSequence) ProtoMessage()
- func (x *StepSequence) ProtoReflect() protoreflect.Message
- func (x *StepSequence) Reset()
- func (x *StepSequence) String() string
- type TensorConnection
- func (*TensorConnection) Descriptor() ([]byte, []int)deprecated
- func (x *TensorConnection) GetFromTensor() string
- func (x *TensorConnection) GetToTensor() string
- func (*TensorConnection) ProtoMessage()
- func (x *TensorConnection) ProtoReflect() protoreflect.Message
- func (x *TensorConnection) Reset()
- func (x *TensorConnection) String() string
- type TensorInfo
- func (*TensorInfo) Descriptor() ([]byte, []int)deprecated
- func (x *TensorInfo) GetCooSparse() *TensorInfo_CooSparse
- func (x *TensorInfo) GetDtype() framework.DataType
- func (m *TensorInfo) GetEncoding() isTensorInfo_Encoding
- func (x *TensorInfo) GetName() string
- func (x *TensorInfo) GetTensorShape() *framework.TensorShapeProto
- func (*TensorInfo) ProtoMessage()
- func (x *TensorInfo) ProtoReflect() protoreflect.Message
- func (x *TensorInfo) Reset()
- func (x *TensorInfo) String() string
- type TensorInfo_CooSparse
- func (*TensorInfo_CooSparse) Descriptor() ([]byte, []int)deprecated
- func (x *TensorInfo_CooSparse) GetDenseShapeTensorName() string
- func (x *TensorInfo_CooSparse) GetIndicesTensorName() string
- func (x *TensorInfo_CooSparse) GetValuesTensorName() string
- func (*TensorInfo_CooSparse) ProtoMessage()
- func (x *TensorInfo_CooSparse) ProtoReflect() protoreflect.Message
- func (x *TensorInfo_CooSparse) Reset()
- func (x *TensorInfo_CooSparse) String() string
- type TensorInfo_CooSparse_
- type TensorInfo_Name
- type ThreadPoolOptionProto
- func (*ThreadPoolOptionProto) Descriptor() ([]byte, []int)deprecated
- func (x *ThreadPoolOptionProto) GetGlobalName() string
- func (x *ThreadPoolOptionProto) GetNumThreads() int32
- func (*ThreadPoolOptionProto) ProtoMessage()
- func (x *ThreadPoolOptionProto) ProtoReflect() protoreflect.Message
- func (x *ThreadPoolOptionProto) Reset()
- func (x *ThreadPoolOptionProto) String() string
- type TraceOpts
- func (*TraceOpts) Descriptor() ([]byte, []int)deprecated
- func (x *TraceOpts) GetDuration() float64
- func (x *TraceOpts) GetUseExtendedProfiler() bool
- func (x *TraceOpts) GetUseGpuProfiler() bool
- func (x *TraceOpts) GetUseKernelProfiler() bool
- func (x *TraceOpts) GetUseSampleProfiler() bool
- func (x *TraceOpts) GetUseStepProfiler() bool
- func (*TraceOpts) ProtoMessage()
- func (x *TraceOpts) ProtoReflect() protoreflect.Message
- func (x *TraceOpts) Reset()
- func (x *TraceOpts) String() string
- type TracingRequest
- type TracingResponse
- type ValuesDef
- func (*ValuesDef) Descriptor() ([]byte, []int)deprecated
- func (x *ValuesDef) GetExternalValues() map[string]string
- func (x *ValuesDef) GetValues() []string
- func (*ValuesDef) ProtoMessage()
- func (x *ValuesDef) ProtoReflect() protoreflect.Message
- func (x *ValuesDef) Reset()
- func (x *ValuesDef) String() string
- type WaitQueueDoneRequest
- func (*WaitQueueDoneRequest) Descriptor() ([]byte, []int)deprecated
- func (x *WaitQueueDoneRequest) GetContextId() uint64
- func (x *WaitQueueDoneRequest) GetOpId() []int64
- func (*WaitQueueDoneRequest) ProtoMessage()
- func (x *WaitQueueDoneRequest) ProtoReflect() protoreflect.Message
- func (x *WaitQueueDoneRequest) Reset()
- func (x *WaitQueueDoneRequest) String() string
- type WaitQueueDoneResponse
- type WhileContextDef
- func (*WhileContextDef) Descriptor() ([]byte, []int)deprecated
- func (x *WhileContextDef) GetBackProp() bool
- func (x *WhileContextDef) GetContextName() string
- func (x *WhileContextDef) GetLoopEnterNames() []string
- func (x *WhileContextDef) GetLoopExitNames() []string
- func (x *WhileContextDef) GetMaximumIterationsName() string
- func (x *WhileContextDef) GetNestedContexts() []*ControlFlowContextDef
- func (x *WhileContextDef) GetParallelIterations() int32
- func (x *WhileContextDef) GetPivotForBodyName() string
- func (x *WhileContextDef) GetPivotForPredName() string
- func (x *WhileContextDef) GetPivotName() string
- func (x *WhileContextDef) GetSwapMemory() bool
- func (x *WhileContextDef) GetValuesDef() *ValuesDef
- func (*WhileContextDef) ProtoMessage()
- func (x *WhileContextDef) ProtoReflect() protoreflect.Message
- func (x *WhileContextDef) Reset()
- func (x *WhileContextDef) String() string
Constants ¶
This section is empty.
Variables ¶
var ( OptimizerOptions_Level_name = map[int32]string{ 0: "L1", -1: "L0", } OptimizerOptions_Level_value = map[string]int32{ "L1": 0, "L0": -1, } )
Enum value maps for OptimizerOptions_Level.
var ( OptimizerOptions_GlobalJitLevel_name = map[int32]string{ 0: "DEFAULT", -1: "OFF", 1: "ON_1", 2: "ON_2", } OptimizerOptions_GlobalJitLevel_value = map[string]int32{ "DEFAULT": 0, "OFF": -1, "ON_1": 1, "ON_2": 2, } )
Enum value maps for OptimizerOptions_GlobalJitLevel.
var ( RunOptions_TraceLevel_name = map[int32]string{ 0: "NO_TRACE", 1: "SOFTWARE_TRACE", 2: "HARDWARE_TRACE", 3: "FULL_TRACE", } RunOptions_TraceLevel_value = map[string]int32{ "NO_TRACE": 0, "SOFTWARE_TRACE": 1, "HARDWARE_TRACE": 2, "FULL_TRACE": 3, } )
Enum value maps for RunOptions_TraceLevel.
var ( RewriterConfig_Toggle_name = map[int32]string{ 0: "DEFAULT", 1: "ON", 2: "OFF", 3: "AGGRESSIVE", } RewriterConfig_Toggle_value = map[string]int32{ "DEFAULT": 0, "ON": 1, "OFF": 2, "AGGRESSIVE": 3, } )
Enum value maps for RewriterConfig_Toggle.
var ( RewriterConfig_NumIterationsType_name = map[int32]string{ 0: "DEFAULT_NUM_ITERS", 1: "ONE", 2: "TWO", } RewriterConfig_NumIterationsType_value = map[string]int32{ "DEFAULT_NUM_ITERS": 0, "ONE": 1, "TWO": 2, } )
Enum value maps for RewriterConfig_NumIterationsType.
var ( RewriterConfig_MemOptType_name = map[int32]string{ 0: "DEFAULT_MEM_OPT", 1: "NO_MEM_OPT", 2: "MANUAL", 4: "SWAPPING_HEURISTICS", 5: "RECOMPUTATION_HEURISTICS", 6: "SCHEDULING_HEURISTICS", 3: "HEURISTICS", } RewriterConfig_MemOptType_value = map[string]int32{ "DEFAULT_MEM_OPT": 0, "NO_MEM_OPT": 1, "MANUAL": 2, "SWAPPING_HEURISTICS": 4, "RECOMPUTATION_HEURISTICS": 5, "SCHEDULING_HEURISTICS": 6, "HEURISTICS": 3, } )
Enum value maps for RewriterConfig_MemOptType.
var ( SaverDef_CheckpointFormatVersion_name = map[int32]string{ 0: "LEGACY", 1: "V1", 2: "V2", } SaverDef_CheckpointFormatVersion_value = map[string]int32{ "LEGACY": 0, "V1": 1, "V2": 2, } )
Enum value maps for SaverDef_CheckpointFormatVersion.
var ( BundleHeaderProto_Endianness_name = map[int32]string{ 0: "LITTLE", 1: "BIG", } BundleHeaderProto_Endianness_value = map[string]int32{ "LITTLE": 0, "BIG": 1, } )
Enum value maps for BundleHeaderProto_Endianness.
var File_tensorflow_core_protobuf_checkpointable_object_graph_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_cluster_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_config_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_control_flow_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_critical_section_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_debug_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_device_properties_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_eager_service_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_master_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_master_service_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_meta_graph_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_named_tensor_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_queue_runner_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_replay_log_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_rewriter_config_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_saved_model_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_saver_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_tensor_bundle_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_tensorflow_server_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_transport_options_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_worker_proto protoreflect.FileDescriptor
var File_tensorflow_core_protobuf_worker_service_proto protoreflect.FileDescriptor
Functions ¶
This section is empty.
Types ¶
type AssetFileDef ¶
type AssetFileDef struct { // The tensor to bind the asset filename to. TensorInfo *TensorInfo `protobuf:"bytes,1,opt,name=tensor_info,json=tensorInfo,proto3" json:"tensor_info,omitempty"` // The filename within an assets directory. Note: does not include the path // prefix, i.e. directories. For an asset at /tmp/path/vocab.txt, the filename // would be "vocab.txt". Filename string `protobuf:"bytes,2,opt,name=filename,proto3" json:"filename,omitempty"` // contains filtered or unexported fields }
An asset file def for a single file or a set of sharded files with the same name.
func (*AssetFileDef) Descriptor
deprecated
func (*AssetFileDef) Descriptor() ([]byte, []int)
Deprecated: Use AssetFileDef.ProtoReflect.Descriptor instead.
func (*AssetFileDef) GetFilename ¶
func (x *AssetFileDef) GetFilename() string
func (*AssetFileDef) GetTensorInfo ¶
func (x *AssetFileDef) GetTensorInfo() *TensorInfo
func (*AssetFileDef) ProtoMessage ¶
func (*AssetFileDef) ProtoMessage()
func (*AssetFileDef) ProtoReflect ¶
func (x *AssetFileDef) ProtoReflect() protoreflect.Message
func (*AssetFileDef) Reset ¶
func (x *AssetFileDef) Reset()
func (*AssetFileDef) String ¶
func (x *AssetFileDef) String() string
type AutoParallelOptions ¶
type AutoParallelOptions struct { Enable bool `protobuf:"varint,1,opt,name=enable,proto3" json:"enable,omitempty"` NumReplicas int32 `protobuf:"varint,2,opt,name=num_replicas,json=numReplicas,proto3" json:"num_replicas,omitempty"` // contains filtered or unexported fields }
func (*AutoParallelOptions) Descriptor
deprecated
func (*AutoParallelOptions) Descriptor() ([]byte, []int)
Deprecated: Use AutoParallelOptions.ProtoReflect.Descriptor instead.
func (*AutoParallelOptions) GetEnable ¶
func (x *AutoParallelOptions) GetEnable() bool
func (*AutoParallelOptions) GetNumReplicas ¶
func (x *AutoParallelOptions) GetNumReplicas() int32
func (*AutoParallelOptions) ProtoMessage ¶
func (*AutoParallelOptions) ProtoMessage()
func (*AutoParallelOptions) ProtoReflect ¶
func (x *AutoParallelOptions) ProtoReflect() protoreflect.Message
func (*AutoParallelOptions) Reset ¶
func (x *AutoParallelOptions) Reset()
func (*AutoParallelOptions) String ¶
func (x *AutoParallelOptions) String() string
type BundleEntryProto ¶
type BundleEntryProto struct { // The tensor dtype and shape. Dtype framework.DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` Shape *framework.TensorShapeProto `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"` // The binary content of the tensor lies in: // File "shard_id": bytes [offset, offset + size). ShardId int32 `protobuf:"varint,3,opt,name=shard_id,json=shardId,proto3" json:"shard_id,omitempty"` Offset int64 `protobuf:"varint,4,opt,name=offset,proto3" json:"offset,omitempty"` Size int64 `protobuf:"varint,5,opt,name=size,proto3" json:"size,omitempty"` // The CRC32C checksum of the tensor bytes. Crc32C uint32 `protobuf:"fixed32,6,opt,name=crc32c,proto3" json:"crc32c,omitempty"` // Iff present, this entry represents a partitioned tensor. The previous // fields are interpreted as follows: // // "dtype", "shape": describe the full tensor. // "shard_id", "offset", "size", "crc32c": all IGNORED. // These information for each slice can be looked up in their own // BundleEntryProto, keyed by each "slice_name". Slices []*framework.TensorSliceProto `protobuf:"bytes,7,rep,name=slices,proto3" json:"slices,omitempty"` // contains filtered or unexported fields }
Describes the metadata related to a checkpointed tensor.
func (*BundleEntryProto) Descriptor
deprecated
func (*BundleEntryProto) Descriptor() ([]byte, []int)
Deprecated: Use BundleEntryProto.ProtoReflect.Descriptor instead.
func (*BundleEntryProto) GetCrc32C ¶
func (x *BundleEntryProto) GetCrc32C() uint32
func (*BundleEntryProto) GetDtype ¶
func (x *BundleEntryProto) GetDtype() framework.DataType
func (*BundleEntryProto) GetOffset ¶
func (x *BundleEntryProto) GetOffset() int64
func (*BundleEntryProto) GetShape ¶
func (x *BundleEntryProto) GetShape() *framework.TensorShapeProto
func (*BundleEntryProto) GetShardId ¶
func (x *BundleEntryProto) GetShardId() int32
func (*BundleEntryProto) GetSize ¶
func (x *BundleEntryProto) GetSize() int64
func (*BundleEntryProto) GetSlices ¶
func (x *BundleEntryProto) GetSlices() []*framework.TensorSliceProto
func (*BundleEntryProto) ProtoMessage ¶
func (*BundleEntryProto) ProtoMessage()
func (*BundleEntryProto) ProtoReflect ¶
func (x *BundleEntryProto) ProtoReflect() protoreflect.Message
func (*BundleEntryProto) Reset ¶
func (x *BundleEntryProto) Reset()
func (*BundleEntryProto) String ¶
func (x *BundleEntryProto) String() string
type BundleHeaderProto ¶
type BundleHeaderProto struct { // Number of data files in the bundle. NumShards int32 `protobuf:"varint,1,opt,name=num_shards,json=numShards,proto3" json:"num_shards,omitempty"` Endianness BundleHeaderProto_Endianness `protobuf:"varint,2,opt,name=endianness,proto3,enum=tensorflow.BundleHeaderProto_Endianness" json:"endianness,omitempty"` // Versioning of the tensor bundle format. Version *framework.VersionDef `protobuf:"bytes,3,opt,name=version,proto3" json:"version,omitempty"` // contains filtered or unexported fields }
Special header that is associated with a bundle.
TODO(zongheng,zhifengc): maybe in the future, we can add information about which binary produced this checkpoint, timestamp, etc. Sometime, these can be valuable debugging information. And if needed, these can be used as defensive information ensuring reader (binary version) of the checkpoint and the writer (binary version) must match within certain range, etc.
func (*BundleHeaderProto) Descriptor
deprecated
func (*BundleHeaderProto) Descriptor() ([]byte, []int)
Deprecated: Use BundleHeaderProto.ProtoReflect.Descriptor instead.
func (*BundleHeaderProto) GetEndianness ¶
func (x *BundleHeaderProto) GetEndianness() BundleHeaderProto_Endianness
func (*BundleHeaderProto) GetNumShards ¶
func (x *BundleHeaderProto) GetNumShards() int32
func (*BundleHeaderProto) GetVersion ¶
func (x *BundleHeaderProto) GetVersion() *framework.VersionDef
func (*BundleHeaderProto) ProtoMessage ¶
func (*BundleHeaderProto) ProtoMessage()
func (*BundleHeaderProto) ProtoReflect ¶
func (x *BundleHeaderProto) ProtoReflect() protoreflect.Message
func (*BundleHeaderProto) Reset ¶
func (x *BundleHeaderProto) Reset()
func (*BundleHeaderProto) String ¶
func (x *BundleHeaderProto) String() string
type BundleHeaderProto_Endianness ¶
type BundleHeaderProto_Endianness int32
An enum indicating the endianness of the platform that produced this bundle. A bundle can only be read by a platform with matching endianness. Defaults to LITTLE, as most modern platforms are little-endian.
Affects the binary tensor data bytes only, not the metadata in protobufs.
const ( BundleHeaderProto_LITTLE BundleHeaderProto_Endianness = 0 BundleHeaderProto_BIG BundleHeaderProto_Endianness = 1 )
func (BundleHeaderProto_Endianness) Descriptor ¶
func (BundleHeaderProto_Endianness) Descriptor() protoreflect.EnumDescriptor
func (BundleHeaderProto_Endianness) Enum ¶
func (x BundleHeaderProto_Endianness) Enum() *BundleHeaderProto_Endianness
func (BundleHeaderProto_Endianness) EnumDescriptor
deprecated
func (BundleHeaderProto_Endianness) EnumDescriptor() ([]byte, []int)
Deprecated: Use BundleHeaderProto_Endianness.Descriptor instead.
func (BundleHeaderProto_Endianness) Number ¶
func (x BundleHeaderProto_Endianness) Number() protoreflect.EnumNumber
func (BundleHeaderProto_Endianness) String ¶
func (x BundleHeaderProto_Endianness) String() string
func (BundleHeaderProto_Endianness) Type ¶
func (BundleHeaderProto_Endianness) Type() protoreflect.EnumType
type CallableOptions ¶
type CallableOptions struct { // Tensors to be fed in the callable. Each feed is the name of a tensor. Feed []string `protobuf:"bytes,1,rep,name=feed,proto3" json:"feed,omitempty"` // Fetches. A list of tensor names. The caller of the callable expects a // tensor to be returned for each fetch[i] (see RunStepResponse.tensor). The // order of specified fetches does not change the execution order. Fetch []string `protobuf:"bytes,2,rep,name=fetch,proto3" json:"fetch,omitempty"` // Target Nodes. A list of node names. The named nodes will be run by the // callable but their outputs will not be returned. Target []string `protobuf:"bytes,3,rep,name=target,proto3" json:"target,omitempty"` // Options that will be applied to each run. RunOptions *RunOptions `protobuf:"bytes,4,opt,name=run_options,json=runOptions,proto3" json:"run_options,omitempty"` // Tensors to be connected in the callable. Each TensorConnection denotes // a pair of tensors in the graph, between which an edge will be created // in the callable. TensorConnection []*TensorConnection `protobuf:"bytes,5,rep,name=tensor_connection,json=tensorConnection,proto3" json:"tensor_connection,omitempty"` // The Tensor objects fed in the callable and fetched from the callable // are expected to be backed by host (CPU) memory by default. // // The options below allow changing that - feeding tensors backed by // device memory, or returning tensors that are backed by device memory. // // The maps below map the name of a feed/fetch tensor (which appears in // 'feed' or 'fetch' fields above), to the fully qualified name of the device // owning the memory backing the contents of the tensor. // // For example, creating a callable with the following options: // // CallableOptions { // feed: "a:0" // feed: "b:0" // // fetch: "x:0" // fetch: "y:0" // // feed_devices: { // "a:0": "/job:localhost/replica:0/task:0/device:GPU:0" // } // // fetch_devices: { // "y:0": "/job:localhost/replica:0/task:0/device:GPU:0" // } // } // // means that the Callable expects: // - The first argument ("a:0") is a Tensor backed by GPU memory. // - The second argument ("b:0") is a Tensor backed by host memory. // and of its return values: // - The first output ("x:0") will be backed by host memory. // - The second output ("y:0") will be backed by GPU memory. // // FEEDS: // It is the responsibility of the caller to ensure that the memory of the fed // tensors will be correctly initialized and synchronized before it is // accessed by operations executed during the call to Session::RunCallable(). // // This is typically ensured by using the TensorFlow memory allocators // (Device::GetAllocator()) to create the Tensor to be fed. // // Alternatively, for CUDA-enabled GPU devices, this typically means that the // operation that produced the contents of the tensor has completed, i.e., the // CUDA stream has been synchronized (e.g., via cuCtxSynchronize() or // cuStreamSynchronize()). FeedDevices map[string]string `` /* 182-byte string literal not displayed */ FetchDevices map[string]string `` /* 185-byte string literal not displayed */ // By default, RunCallable() will synchronize the GPU stream before returning // fetched tensors on a GPU device, to ensure that the values in those tensors // have been produced. This simplifies interacting with the tensors, but // potentially incurs a performance hit. // // If this options is set to true, the caller is responsible for ensuring // that the values in the fetched tensors have been produced before they are // used. The caller can do this by invoking `Device::Sync()` on the underlying // device(s), or by feeding the tensors back to the same Session using // `feed_devices` with the same corresponding device name. FetchSkipSync bool `protobuf:"varint,8,opt,name=fetch_skip_sync,json=fetchSkipSync,proto3" json:"fetch_skip_sync,omitempty"` // contains filtered or unexported fields }
Defines a subgraph in another `GraphDef` as a set of feed points and nodes to be fetched or executed.
Compare with the arguments to `Session::Run()`.
func (*CallableOptions) Descriptor
deprecated
func (*CallableOptions) Descriptor() ([]byte, []int)
Deprecated: Use CallableOptions.ProtoReflect.Descriptor instead.
func (*CallableOptions) GetFeed ¶
func (x *CallableOptions) GetFeed() []string
func (*CallableOptions) GetFeedDevices ¶
func (x *CallableOptions) GetFeedDevices() map[string]string
func (*CallableOptions) GetFetch ¶
func (x *CallableOptions) GetFetch() []string
func (*CallableOptions) GetFetchDevices ¶
func (x *CallableOptions) GetFetchDevices() map[string]string
func (*CallableOptions) GetFetchSkipSync ¶
func (x *CallableOptions) GetFetchSkipSync() bool
func (*CallableOptions) GetRunOptions ¶
func (x *CallableOptions) GetRunOptions() *RunOptions
func (*CallableOptions) GetTarget ¶
func (x *CallableOptions) GetTarget() []string
func (*CallableOptions) GetTensorConnection ¶
func (x *CallableOptions) GetTensorConnection() []*TensorConnection
func (*CallableOptions) ProtoMessage ¶
func (*CallableOptions) ProtoMessage()
func (*CallableOptions) ProtoReflect ¶
func (x *CallableOptions) ProtoReflect() protoreflect.Message
func (*CallableOptions) Reset ¶
func (x *CallableOptions) Reset()
func (*CallableOptions) String ¶
func (x *CallableOptions) String() string
type CheckpointableObjectGraph ¶
type CheckpointableObjectGraph struct { Nodes []*CheckpointableObjectGraph_CheckpointableObject `protobuf:"bytes,1,rep,name=nodes,proto3" json:"nodes,omitempty"` // contains filtered or unexported fields }
func (*CheckpointableObjectGraph) Descriptor
deprecated
func (*CheckpointableObjectGraph) Descriptor() ([]byte, []int)
Deprecated: Use CheckpointableObjectGraph.ProtoReflect.Descriptor instead.
func (*CheckpointableObjectGraph) GetNodes ¶
func (x *CheckpointableObjectGraph) GetNodes() []*CheckpointableObjectGraph_CheckpointableObject
func (*CheckpointableObjectGraph) ProtoMessage ¶
func (*CheckpointableObjectGraph) ProtoMessage()
func (*CheckpointableObjectGraph) ProtoReflect ¶
func (x *CheckpointableObjectGraph) ProtoReflect() protoreflect.Message
func (*CheckpointableObjectGraph) Reset ¶
func (x *CheckpointableObjectGraph) Reset()
func (*CheckpointableObjectGraph) String ¶
func (x *CheckpointableObjectGraph) String() string
type CheckpointableObjectGraph_CheckpointableObject ¶
type CheckpointableObjectGraph_CheckpointableObject struct { // Objects which this object depends on. Children []*CheckpointableObjectGraph_CheckpointableObject_ObjectReference `protobuf:"bytes,1,rep,name=children,proto3" json:"children,omitempty"` // Serialized data specific to this object. Attributes []*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor `protobuf:"bytes,2,rep,name=attributes,proto3" json:"attributes,omitempty"` // Slot variables owned by this object. SlotVariables []*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference `protobuf:"bytes,3,rep,name=slot_variables,json=slotVariables,proto3" json:"slot_variables,omitempty"` // contains filtered or unexported fields }
func (*CheckpointableObjectGraph_CheckpointableObject) Descriptor
deprecated
func (*CheckpointableObjectGraph_CheckpointableObject) Descriptor() ([]byte, []int)
Deprecated: Use CheckpointableObjectGraph_CheckpointableObject.ProtoReflect.Descriptor instead.
func (*CheckpointableObjectGraph_CheckpointableObject) GetAttributes ¶
func (x *CheckpointableObjectGraph_CheckpointableObject) GetAttributes() []*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor
func (*CheckpointableObjectGraph_CheckpointableObject) GetChildren ¶
func (x *CheckpointableObjectGraph_CheckpointableObject) GetChildren() []*CheckpointableObjectGraph_CheckpointableObject_ObjectReference
func (*CheckpointableObjectGraph_CheckpointableObject) GetSlotVariables ¶
func (x *CheckpointableObjectGraph_CheckpointableObject) GetSlotVariables() []*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference
func (*CheckpointableObjectGraph_CheckpointableObject) ProtoMessage ¶
func (*CheckpointableObjectGraph_CheckpointableObject) ProtoMessage()
func (*CheckpointableObjectGraph_CheckpointableObject) ProtoReflect ¶
func (x *CheckpointableObjectGraph_CheckpointableObject) ProtoReflect() protoreflect.Message
func (*CheckpointableObjectGraph_CheckpointableObject) Reset ¶
func (x *CheckpointableObjectGraph_CheckpointableObject) Reset()
func (*CheckpointableObjectGraph_CheckpointableObject) String ¶
func (x *CheckpointableObjectGraph_CheckpointableObject) String() string
type CheckpointableObjectGraph_CheckpointableObject_ObjectReference ¶
type CheckpointableObjectGraph_CheckpointableObject_ObjectReference struct { // An index into `CheckpointableObjectGraph.nodes`, indicating the object // being referenced. NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"` // A user-provided name for the edge. LocalName string `protobuf:"bytes,2,opt,name=local_name,json=localName,proto3" json:"local_name,omitempty"` // contains filtered or unexported fields }
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) Descriptor
deprecated
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) Descriptor() ([]byte, []int)
Deprecated: Use CheckpointableObjectGraph_CheckpointableObject_ObjectReference.ProtoReflect.Descriptor instead.
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) GetLocalName ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) GetLocalName() string
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) GetNodeId ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) GetNodeId() int32
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) ProtoMessage ¶
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) ProtoMessage()
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) ProtoReflect ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) ProtoReflect() protoreflect.Message
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) Reset ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) Reset()
func (*CheckpointableObjectGraph_CheckpointableObject_ObjectReference) String ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_ObjectReference) String() string
type CheckpointableObjectGraph_CheckpointableObject_SerializedTensor ¶
type CheckpointableObjectGraph_CheckpointableObject_SerializedTensor struct { // A name for the Tensor. Simple variables have only one // `SerializedTensor` named "VARIABLE_VALUE" by convention. This value may // be restored on object creation as an optimization. Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // The full name of the variable/tensor, if applicable. Used to allow // name-based loading of checkpoints which were saved using an // object-based API. Should match the checkpoint key which would have been // assigned by tf.train.Saver. FullName string `protobuf:"bytes,2,opt,name=full_name,json=fullName,proto3" json:"full_name,omitempty"` // The generated name of the Tensor in the checkpoint. CheckpointKey string `protobuf:"bytes,3,opt,name=checkpoint_key,json=checkpointKey,proto3" json:"checkpoint_key,omitempty"` // contains filtered or unexported fields }
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) Descriptor
deprecated
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) Descriptor() ([]byte, []int)
Deprecated: Use CheckpointableObjectGraph_CheckpointableObject_SerializedTensor.ProtoReflect.Descriptor instead.
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetCheckpointKey ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetCheckpointKey() string
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetFullName ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetFullName() string
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetName ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) GetName() string
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) ProtoMessage ¶
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) ProtoMessage()
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) ProtoReflect ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) ProtoReflect() protoreflect.Message
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) Reset ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) Reset()
func (*CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) String ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SerializedTensor) String() string
type CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference ¶
type CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference struct { // An index into `CheckpointableObjectGraph.nodes`, indicating the // variable object this slot was created for. OriginalVariableNodeId int32 `` /* 132-byte string literal not displayed */ // The name of the slot (e.g. "m"/"v"). SlotName string `protobuf:"bytes,2,opt,name=slot_name,json=slotName,proto3" json:"slot_name,omitempty"` // An index into `CheckpointableObjectGraph.nodes`, indicating the // `Object` with the value of the slot variable. SlotVariableNodeId int32 `protobuf:"varint,3,opt,name=slot_variable_node_id,json=slotVariableNodeId,proto3" json:"slot_variable_node_id,omitempty"` // contains filtered or unexported fields }
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) Descriptor
deprecated
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) Descriptor() ([]byte, []int)
Deprecated: Use CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference.ProtoReflect.Descriptor instead.
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetOriginalVariableNodeId ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetOriginalVariableNodeId() int32
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetSlotName ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetSlotName() string
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetSlotVariableNodeId ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) GetSlotVariableNodeId() int32
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) ProtoMessage ¶
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) ProtoMessage()
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) ProtoReflect ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) ProtoReflect() protoreflect.Message
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) Reset ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) Reset()
func (*CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) String ¶
func (x *CheckpointableObjectGraph_CheckpointableObject_SlotVariableReference) String() string
type CleanupAllRequest ¶
type CleanupAllRequest struct { // A list of container names. // // If 'container' is not empty, releases resources in the given // containers in all devices. // // If 'container' is empty, releases resources in the default // container in all devices. Container []string `protobuf:"bytes,1,rep,name=container,proto3" json:"container,omitempty"` // contains filtered or unexported fields }
func (*CleanupAllRequest) Descriptor
deprecated
func (*CleanupAllRequest) Descriptor() ([]byte, []int)
Deprecated: Use CleanupAllRequest.ProtoReflect.Descriptor instead.
func (*CleanupAllRequest) GetContainer ¶
func (x *CleanupAllRequest) GetContainer() []string
func (*CleanupAllRequest) ProtoMessage ¶
func (*CleanupAllRequest) ProtoMessage()
func (*CleanupAllRequest) ProtoReflect ¶
func (x *CleanupAllRequest) ProtoReflect() protoreflect.Message
func (*CleanupAllRequest) Reset ¶
func (x *CleanupAllRequest) Reset()
func (*CleanupAllRequest) String ¶
func (x *CleanupAllRequest) String() string
type CleanupAllResponse ¶
type CleanupAllResponse struct {
// contains filtered or unexported fields
}
func (*CleanupAllResponse) Descriptor
deprecated
func (*CleanupAllResponse) Descriptor() ([]byte, []int)
Deprecated: Use CleanupAllResponse.ProtoReflect.Descriptor instead.
func (*CleanupAllResponse) ProtoMessage ¶
func (*CleanupAllResponse) ProtoMessage()
func (*CleanupAllResponse) ProtoReflect ¶
func (x *CleanupAllResponse) ProtoReflect() protoreflect.Message
func (*CleanupAllResponse) Reset ¶
func (x *CleanupAllResponse) Reset()
func (*CleanupAllResponse) String ¶
func (x *CleanupAllResponse) String() string
type CleanupGraphRequest ¶
type CleanupGraphRequest struct { StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // contains filtered or unexported fields }
func (*CleanupGraphRequest) Descriptor
deprecated
func (*CleanupGraphRequest) Descriptor() ([]byte, []int)
Deprecated: Use CleanupGraphRequest.ProtoReflect.Descriptor instead.
func (*CleanupGraphRequest) GetStepId ¶
func (x *CleanupGraphRequest) GetStepId() int64
func (*CleanupGraphRequest) ProtoMessage ¶
func (*CleanupGraphRequest) ProtoMessage()
func (*CleanupGraphRequest) ProtoReflect ¶
func (x *CleanupGraphRequest) ProtoReflect() protoreflect.Message
func (*CleanupGraphRequest) Reset ¶
func (x *CleanupGraphRequest) Reset()
func (*CleanupGraphRequest) String ¶
func (x *CleanupGraphRequest) String() string
type CleanupGraphResponse ¶
type CleanupGraphResponse struct {
// contains filtered or unexported fields
}
func (*CleanupGraphResponse) Descriptor
deprecated
func (*CleanupGraphResponse) Descriptor() ([]byte, []int)
Deprecated: Use CleanupGraphResponse.ProtoReflect.Descriptor instead.
func (*CleanupGraphResponse) ProtoMessage ¶
func (*CleanupGraphResponse) ProtoMessage()
func (*CleanupGraphResponse) ProtoReflect ¶
func (x *CleanupGraphResponse) ProtoReflect() protoreflect.Message
func (*CleanupGraphResponse) Reset ¶
func (x *CleanupGraphResponse) Reset()
func (*CleanupGraphResponse) String ¶
func (x *CleanupGraphResponse) String() string
type CloseContextRequest ¶
type CloseContextRequest struct { ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"` // contains filtered or unexported fields }
func (*CloseContextRequest) Descriptor
deprecated
func (*CloseContextRequest) Descriptor() ([]byte, []int)
Deprecated: Use CloseContextRequest.ProtoReflect.Descriptor instead.
func (*CloseContextRequest) GetContextId ¶
func (x *CloseContextRequest) GetContextId() uint64
func (*CloseContextRequest) ProtoMessage ¶
func (*CloseContextRequest) ProtoMessage()
func (*CloseContextRequest) ProtoReflect ¶
func (x *CloseContextRequest) ProtoReflect() protoreflect.Message
func (*CloseContextRequest) Reset ¶
func (x *CloseContextRequest) Reset()
func (*CloseContextRequest) String ¶
func (x *CloseContextRequest) String() string
type CloseContextResponse ¶
type CloseContextResponse struct {
// contains filtered or unexported fields
}
func (*CloseContextResponse) Descriptor
deprecated
func (*CloseContextResponse) Descriptor() ([]byte, []int)
Deprecated: Use CloseContextResponse.ProtoReflect.Descriptor instead.
func (*CloseContextResponse) ProtoMessage ¶
func (*CloseContextResponse) ProtoMessage()
func (*CloseContextResponse) ProtoReflect ¶
func (x *CloseContextResponse) ProtoReflect() protoreflect.Message
func (*CloseContextResponse) Reset ¶
func (x *CloseContextResponse) Reset()
func (*CloseContextResponse) String ¶
func (x *CloseContextResponse) String() string
type CloseSessionRequest ¶
type CloseSessionRequest struct { // REQUIRED: session_handle must be returned by a CreateSession call // to the same master service. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // contains filtered or unexported fields }
func (*CloseSessionRequest) Descriptor
deprecated
func (*CloseSessionRequest) Descriptor() ([]byte, []int)
Deprecated: Use CloseSessionRequest.ProtoReflect.Descriptor instead.
func (*CloseSessionRequest) GetSessionHandle ¶
func (x *CloseSessionRequest) GetSessionHandle() string
func (*CloseSessionRequest) ProtoMessage ¶
func (*CloseSessionRequest) ProtoMessage()
func (*CloseSessionRequest) ProtoReflect ¶
func (x *CloseSessionRequest) ProtoReflect() protoreflect.Message
func (*CloseSessionRequest) Reset ¶
func (x *CloseSessionRequest) Reset()
func (*CloseSessionRequest) String ¶
func (x *CloseSessionRequest) String() string
type CloseSessionResponse ¶
type CloseSessionResponse struct {
// contains filtered or unexported fields
}
func (*CloseSessionResponse) Descriptor
deprecated
func (*CloseSessionResponse) Descriptor() ([]byte, []int)
Deprecated: Use CloseSessionResponse.ProtoReflect.Descriptor instead.
func (*CloseSessionResponse) ProtoMessage ¶
func (*CloseSessionResponse) ProtoMessage()
func (*CloseSessionResponse) ProtoReflect ¶
func (x *CloseSessionResponse) ProtoReflect() protoreflect.Message
func (*CloseSessionResponse) Reset ¶
func (x *CloseSessionResponse) Reset()
func (*CloseSessionResponse) String ¶
func (x *CloseSessionResponse) String() string
type ClusterDef ¶
type ClusterDef struct { // The jobs that comprise the cluster. Job []*JobDef `protobuf:"bytes,1,rep,name=job,proto3" json:"job,omitempty"` // contains filtered or unexported fields }
Defines a TensorFlow cluster as a set of jobs.
func (*ClusterDef) Descriptor
deprecated
func (*ClusterDef) Descriptor() ([]byte, []int)
Deprecated: Use ClusterDef.ProtoReflect.Descriptor instead.
func (*ClusterDef) GetJob ¶
func (x *ClusterDef) GetJob() []*JobDef
func (*ClusterDef) ProtoMessage ¶
func (*ClusterDef) ProtoMessage()
func (*ClusterDef) ProtoReflect ¶
func (x *ClusterDef) ProtoReflect() protoreflect.Message
func (*ClusterDef) Reset ¶
func (x *ClusterDef) Reset()
func (*ClusterDef) String ¶
func (x *ClusterDef) String() string
type CollectionDef ¶
type CollectionDef struct { // Types that are assignable to Kind: // *CollectionDef_NodeList_ // *CollectionDef_BytesList_ // *CollectionDef_Int64List_ // *CollectionDef_FloatList_ // *CollectionDef_AnyList_ Kind isCollectionDef_Kind `protobuf_oneof:"kind"` // contains filtered or unexported fields }
CollectionDef should cover most collections. To add a user-defined collection, do one of the following:
- For simple data types, such as string, int, float: tf.add_to_collection("your_collection_name", your_simple_value) strings will be stored as bytes_list.
2. For Protobuf types, there are three ways to add them:
tf.add_to_collection("your_collection_name", your_proto.SerializeToString())
collection_def { key: "user_defined_bytes_collection" value { bytes_list { value: "queue_name: \"test_queue\"\n" } } }
or
tf.add_to_collection("your_collection_name", str(your_proto))
collection_def { key: "user_defined_string_collection" value { bytes_list { value: "\n\ntest_queue" } } }
or
any_buf = any_pb2.Any() tf.add_to_collection("your_collection_name", any_buf.Pack(your_proto))
collection_def { key: "user_defined_any_collection" value { any_list { value { type_url: "type.googleapis.com/tensorflow.QueueRunnerDef" value: "\n\ntest_queue" } } } }
- For Python objects, implement to_proto() and from_proto(), and register them in the following manner: ops.register_proto_function("your_collection_name", proto_type, to_proto=YourPythonObject.to_proto, from_proto=YourPythonObject.from_proto) These functions will be invoked to serialize and de-serialize the collection. For example, ops.register_proto_function(ops.GraphKeys.GLOBAL_VARIABLES, proto_type=variable_pb2.VariableDef, to_proto=Variable.to_proto, from_proto=Variable.from_proto)
func (*CollectionDef) Descriptor
deprecated
func (*CollectionDef) Descriptor() ([]byte, []int)
Deprecated: Use CollectionDef.ProtoReflect.Descriptor instead.
func (*CollectionDef) GetAnyList ¶
func (x *CollectionDef) GetAnyList() *CollectionDef_AnyList
func (*CollectionDef) GetBytesList ¶
func (x *CollectionDef) GetBytesList() *CollectionDef_BytesList
func (*CollectionDef) GetFloatList ¶
func (x *CollectionDef) GetFloatList() *CollectionDef_FloatList
func (*CollectionDef) GetInt64List ¶
func (x *CollectionDef) GetInt64List() *CollectionDef_Int64List
func (*CollectionDef) GetKind ¶
func (m *CollectionDef) GetKind() isCollectionDef_Kind
func (*CollectionDef) GetNodeList ¶
func (x *CollectionDef) GetNodeList() *CollectionDef_NodeList
func (*CollectionDef) ProtoMessage ¶
func (*CollectionDef) ProtoMessage()
func (*CollectionDef) ProtoReflect ¶
func (x *CollectionDef) ProtoReflect() protoreflect.Message
func (*CollectionDef) Reset ¶
func (x *CollectionDef) Reset()
func (*CollectionDef) String ¶
func (x *CollectionDef) String() string
type CollectionDef_AnyList ¶
type CollectionDef_AnyList struct { Value []*any.Any `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"` // contains filtered or unexported fields }
AnyList is used for collecting Any protos.
func (*CollectionDef_AnyList) Descriptor
deprecated
func (*CollectionDef_AnyList) Descriptor() ([]byte, []int)
Deprecated: Use CollectionDef_AnyList.ProtoReflect.Descriptor instead.
func (*CollectionDef_AnyList) GetValue ¶
func (x *CollectionDef_AnyList) GetValue() []*any.Any
func (*CollectionDef_AnyList) ProtoMessage ¶
func (*CollectionDef_AnyList) ProtoMessage()
func (*CollectionDef_AnyList) ProtoReflect ¶
func (x *CollectionDef_AnyList) ProtoReflect() protoreflect.Message
func (*CollectionDef_AnyList) Reset ¶
func (x *CollectionDef_AnyList) Reset()
func (*CollectionDef_AnyList) String ¶
func (x *CollectionDef_AnyList) String() string
type CollectionDef_AnyList_ ¶
type CollectionDef_AnyList_ struct {
AnyList *CollectionDef_AnyList `protobuf:"bytes,5,opt,name=any_list,json=anyList,proto3,oneof"`
}
type CollectionDef_BytesList ¶
type CollectionDef_BytesList struct { Value [][]byte `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"` // contains filtered or unexported fields }
BytesList is used for collecting strings and serialized protobufs. For example:
collection_def { key: "trainable_variables" value { bytes_list { value: "\n\017conv1/weights:0\022\024conv1/weights/Assign \032\024conv1/weights/read:0" value: "\n\016conv1/biases:0\022\023conv1/biases/Assign\032 \023conv1/biases/read:0" } } }
func (*CollectionDef_BytesList) Descriptor
deprecated
func (*CollectionDef_BytesList) Descriptor() ([]byte, []int)
Deprecated: Use CollectionDef_BytesList.ProtoReflect.Descriptor instead.
func (*CollectionDef_BytesList) GetValue ¶
func (x *CollectionDef_BytesList) GetValue() [][]byte
func (*CollectionDef_BytesList) ProtoMessage ¶
func (*CollectionDef_BytesList) ProtoMessage()
func (*CollectionDef_BytesList) ProtoReflect ¶
func (x *CollectionDef_BytesList) ProtoReflect() protoreflect.Message
func (*CollectionDef_BytesList) Reset ¶
func (x *CollectionDef_BytesList) Reset()
func (*CollectionDef_BytesList) String ¶
func (x *CollectionDef_BytesList) String() string
type CollectionDef_BytesList_ ¶
type CollectionDef_BytesList_ struct {
BytesList *CollectionDef_BytesList `protobuf:"bytes,2,opt,name=bytes_list,json=bytesList,proto3,oneof"`
}
type CollectionDef_FloatList ¶
type CollectionDef_FloatList struct { Value []float32 `protobuf:"fixed32,1,rep,packed,name=value,proto3" json:"value,omitempty"` // contains filtered or unexported fields }
FloatList is used for collecting float values.
func (*CollectionDef_FloatList) Descriptor
deprecated
func (*CollectionDef_FloatList) Descriptor() ([]byte, []int)
Deprecated: Use CollectionDef_FloatList.ProtoReflect.Descriptor instead.
func (*CollectionDef_FloatList) GetValue ¶
func (x *CollectionDef_FloatList) GetValue() []float32
func (*CollectionDef_FloatList) ProtoMessage ¶
func (*CollectionDef_FloatList) ProtoMessage()
func (*CollectionDef_FloatList) ProtoReflect ¶
func (x *CollectionDef_FloatList) ProtoReflect() protoreflect.Message
func (*CollectionDef_FloatList) Reset ¶
func (x *CollectionDef_FloatList) Reset()
func (*CollectionDef_FloatList) String ¶
func (x *CollectionDef_FloatList) String() string
type CollectionDef_FloatList_ ¶
type CollectionDef_FloatList_ struct {
FloatList *CollectionDef_FloatList `protobuf:"bytes,4,opt,name=float_list,json=floatList,proto3,oneof"`
}
type CollectionDef_Int64List ¶
type CollectionDef_Int64List struct { Value []int64 `protobuf:"varint,1,rep,packed,name=value,proto3" json:"value,omitempty"` // contains filtered or unexported fields }
Int64List is used for collecting int, int64 and long values.
func (*CollectionDef_Int64List) Descriptor
deprecated
func (*CollectionDef_Int64List) Descriptor() ([]byte, []int)
Deprecated: Use CollectionDef_Int64List.ProtoReflect.Descriptor instead.
func (*CollectionDef_Int64List) GetValue ¶
func (x *CollectionDef_Int64List) GetValue() []int64
func (*CollectionDef_Int64List) ProtoMessage ¶
func (*CollectionDef_Int64List) ProtoMessage()
func (*CollectionDef_Int64List) ProtoReflect ¶
func (x *CollectionDef_Int64List) ProtoReflect() protoreflect.Message
func (*CollectionDef_Int64List) Reset ¶
func (x *CollectionDef_Int64List) Reset()
func (*CollectionDef_Int64List) String ¶
func (x *CollectionDef_Int64List) String() string
type CollectionDef_Int64List_ ¶
type CollectionDef_Int64List_ struct {
Int64List *CollectionDef_Int64List `protobuf:"bytes,3,opt,name=int64_list,json=int64List,proto3,oneof"`
}
type CollectionDef_NodeList ¶
type CollectionDef_NodeList struct { Value []string `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"` // contains filtered or unexported fields }
NodeList is used for collecting nodes in graph. For example
collection_def { key: "summaries" value { node_list { value: "input_producer/ScalarSummary:0" value: "shuffle_batch/ScalarSummary:0" value: "ImageSummary:0" } }
func (*CollectionDef_NodeList) Descriptor
deprecated
func (*CollectionDef_NodeList) Descriptor() ([]byte, []int)
Deprecated: Use CollectionDef_NodeList.ProtoReflect.Descriptor instead.
func (*CollectionDef_NodeList) GetValue ¶
func (x *CollectionDef_NodeList) GetValue() []string
func (*CollectionDef_NodeList) ProtoMessage ¶
func (*CollectionDef_NodeList) ProtoMessage()
func (*CollectionDef_NodeList) ProtoReflect ¶
func (x *CollectionDef_NodeList) ProtoReflect() protoreflect.Message
func (*CollectionDef_NodeList) Reset ¶
func (x *CollectionDef_NodeList) Reset()
func (*CollectionDef_NodeList) String ¶
func (x *CollectionDef_NodeList) String() string
type CollectionDef_NodeList_ ¶
type CollectionDef_NodeList_ struct {
NodeList *CollectionDef_NodeList `protobuf:"bytes,1,opt,name=node_list,json=nodeList,proto3,oneof"`
}
type CompleteGroupRequest ¶
type CompleteGroupRequest struct { GroupKey int32 `protobuf:"varint,1,opt,name=group_key,json=groupKey,proto3" json:"group_key,omitempty"` GroupSize int32 `protobuf:"varint,2,opt,name=group_size,json=groupSize,proto3" json:"group_size,omitempty"` DeviceType string `protobuf:"bytes,3,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"` DeviceName []string `protobuf:"bytes,4,rep,name=device_name,json=deviceName,proto3" json:"device_name,omitempty"` // contains filtered or unexported fields }
Supplies one or more device names as members of the group identified by group_key. Service will respond when all group_size devices become known. All devices in group must have same type.
func (*CompleteGroupRequest) Descriptor
deprecated
func (*CompleteGroupRequest) Descriptor() ([]byte, []int)
Deprecated: Use CompleteGroupRequest.ProtoReflect.Descriptor instead.
func (*CompleteGroupRequest) GetDeviceName ¶
func (x *CompleteGroupRequest) GetDeviceName() []string
func (*CompleteGroupRequest) GetDeviceType ¶
func (x *CompleteGroupRequest) GetDeviceType() string
func (*CompleteGroupRequest) GetGroupKey ¶
func (x *CompleteGroupRequest) GetGroupKey() int32
func (*CompleteGroupRequest) GetGroupSize ¶
func (x *CompleteGroupRequest) GetGroupSize() int32
func (*CompleteGroupRequest) ProtoMessage ¶
func (*CompleteGroupRequest) ProtoMessage()
func (*CompleteGroupRequest) ProtoReflect ¶
func (x *CompleteGroupRequest) ProtoReflect() protoreflect.Message
func (*CompleteGroupRequest) Reset ¶
func (x *CompleteGroupRequest) Reset()
func (*CompleteGroupRequest) String ¶
func (x *CompleteGroupRequest) String() string
type CompleteGroupResponse ¶
type CompleteGroupResponse struct { GroupKey int32 `protobuf:"varint,1,opt,name=group_key,json=groupKey,proto3" json:"group_key,omitempty"` GroupSize int32 `protobuf:"varint,2,opt,name=group_size,json=groupSize,proto3" json:"group_size,omitempty"` DeviceType string `protobuf:"bytes,3,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"` NumTasks int32 `protobuf:"varint,4,opt,name=num_tasks,json=numTasks,proto3" json:"num_tasks,omitempty"` // number of distinct tasks hosting the devices DeviceName []string `protobuf:"bytes,5,rep,name=device_name,json=deviceName,proto3" json:"device_name,omitempty"` TaskName []string `protobuf:"bytes,6,rep,name=task_name,json=taskName,proto3" json:"task_name,omitempty"` // task name prefixes of device_names // contains filtered or unexported fields }
Gives the complete membership of the group identified by group_key.
func (*CompleteGroupResponse) Descriptor
deprecated
func (*CompleteGroupResponse) Descriptor() ([]byte, []int)
Deprecated: Use CompleteGroupResponse.ProtoReflect.Descriptor instead.
func (*CompleteGroupResponse) GetDeviceName ¶
func (x *CompleteGroupResponse) GetDeviceName() []string
func (*CompleteGroupResponse) GetDeviceType ¶
func (x *CompleteGroupResponse) GetDeviceType() string
func (*CompleteGroupResponse) GetGroupKey ¶
func (x *CompleteGroupResponse) GetGroupKey() int32
func (*CompleteGroupResponse) GetGroupSize ¶
func (x *CompleteGroupResponse) GetGroupSize() int32
func (*CompleteGroupResponse) GetNumTasks ¶
func (x *CompleteGroupResponse) GetNumTasks() int32
func (*CompleteGroupResponse) GetTaskName ¶
func (x *CompleteGroupResponse) GetTaskName() []string
func (*CompleteGroupResponse) ProtoMessage ¶
func (*CompleteGroupResponse) ProtoMessage()
func (*CompleteGroupResponse) ProtoReflect ¶
func (x *CompleteGroupResponse) ProtoReflect() protoreflect.Message
func (*CompleteGroupResponse) Reset ¶
func (x *CompleteGroupResponse) Reset()
func (*CompleteGroupResponse) String ¶
func (x *CompleteGroupResponse) String() string
type CompleteInstanceRequest ¶
type CompleteInstanceRequest struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` Type int32 `protobuf:"varint,2,opt,name=type,proto3" json:"type,omitempty"` DataType framework.DataType `protobuf:"varint,3,opt,name=data_type,json=dataType,proto3,enum=tensorflow.DataType" json:"data_type,omitempty"` Shape *framework.TensorShapeProto `protobuf:"bytes,4,opt,name=shape,proto3" json:"shape,omitempty"` GroupKey int32 `protobuf:"varint,5,opt,name=group_key,json=groupKey,proto3" json:"group_key,omitempty"` GroupSize int32 `protobuf:"varint,6,opt,name=group_size,json=groupSize,proto3" json:"group_size,omitempty"` InstanceKey int32 `protobuf:"varint,7,opt,name=instance_key,json=instanceKey,proto3" json:"instance_key,omitempty"` DeviceType string `protobuf:"bytes,8,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"` SubdivOffset []int32 `protobuf:"varint,9,rep,packed,name=subdiv_offset,json=subdivOffset,proto3" json:"subdiv_offset,omitempty"` Device string `protobuf:"bytes,10,opt,name=device,proto3" json:"device,omitempty"` IsSource bool `protobuf:"varint,11,opt,name=is_source,json=isSource,proto3" json:"is_source,omitempty"` // contains filtered or unexported fields }
Supplies data about one collective op belonging to the instance identified by instance_key. Service will respond when all group_size ops have become known. Most of the data being sent is for correctness checking, to ensure that all ops in the instance share common attributes.
func (*CompleteInstanceRequest) Descriptor
deprecated
func (*CompleteInstanceRequest) Descriptor() ([]byte, []int)
Deprecated: Use CompleteInstanceRequest.ProtoReflect.Descriptor instead.
func (*CompleteInstanceRequest) GetDataType ¶
func (x *CompleteInstanceRequest) GetDataType() framework.DataType
func (*CompleteInstanceRequest) GetDevice ¶
func (x *CompleteInstanceRequest) GetDevice() string
func (*CompleteInstanceRequest) GetDeviceType ¶
func (x *CompleteInstanceRequest) GetDeviceType() string
func (*CompleteInstanceRequest) GetGroupKey ¶
func (x *CompleteInstanceRequest) GetGroupKey() int32
func (*CompleteInstanceRequest) GetGroupSize ¶
func (x *CompleteInstanceRequest) GetGroupSize() int32
func (*CompleteInstanceRequest) GetInstanceKey ¶
func (x *CompleteInstanceRequest) GetInstanceKey() int32
func (*CompleteInstanceRequest) GetIsSource ¶
func (x *CompleteInstanceRequest) GetIsSource() bool
func (*CompleteInstanceRequest) GetName ¶
func (x *CompleteInstanceRequest) GetName() string
func (*CompleteInstanceRequest) GetShape ¶
func (x *CompleteInstanceRequest) GetShape() *framework.TensorShapeProto
func (*CompleteInstanceRequest) GetSubdivOffset ¶
func (x *CompleteInstanceRequest) GetSubdivOffset() []int32
func (*CompleteInstanceRequest) GetType ¶
func (x *CompleteInstanceRequest) GetType() int32
func (*CompleteInstanceRequest) ProtoMessage ¶
func (*CompleteInstanceRequest) ProtoMessage()
func (*CompleteInstanceRequest) ProtoReflect ¶
func (x *CompleteInstanceRequest) ProtoReflect() protoreflect.Message
func (*CompleteInstanceRequest) Reset ¶
func (x *CompleteInstanceRequest) Reset()
func (*CompleteInstanceRequest) String ¶
func (x *CompleteInstanceRequest) String() string
type CompleteInstanceResponse ¶
type CompleteInstanceResponse struct { InstanceKey int32 `protobuf:"varint,1,opt,name=instance_key,json=instanceKey,proto3" json:"instance_key,omitempty"` SourceRank int32 `protobuf:"varint,2,opt,name=source_rank,json=sourceRank,proto3" json:"source_rank,omitempty"` // contains filtered or unexported fields }
Confirms that every op in the instance has consistently declared itself. Also gives the source_rank in case of broadcast.
func (*CompleteInstanceResponse) Descriptor
deprecated
func (*CompleteInstanceResponse) Descriptor() ([]byte, []int)
Deprecated: Use CompleteInstanceResponse.ProtoReflect.Descriptor instead.
func (*CompleteInstanceResponse) GetInstanceKey ¶
func (x *CompleteInstanceResponse) GetInstanceKey() int32
func (*CompleteInstanceResponse) GetSourceRank ¶
func (x *CompleteInstanceResponse) GetSourceRank() int32
func (*CompleteInstanceResponse) ProtoMessage ¶
func (*CompleteInstanceResponse) ProtoMessage()
func (*CompleteInstanceResponse) ProtoReflect ¶
func (x *CompleteInstanceResponse) ProtoReflect() protoreflect.Message
func (*CompleteInstanceResponse) Reset ¶
func (x *CompleteInstanceResponse) Reset()
func (*CompleteInstanceResponse) String ¶
func (x *CompleteInstanceResponse) String() string
type CondContextDef ¶
type CondContextDef struct { // Name of the context. ContextName string `protobuf:"bytes,1,opt,name=context_name,json=contextName,proto3" json:"context_name,omitempty"` // Name of the pred tensor. PredName string `protobuf:"bytes,2,opt,name=pred_name,json=predName,proto3" json:"pred_name,omitempty"` // Name of the pivot tensor. PivotName string `protobuf:"bytes,3,opt,name=pivot_name,json=pivotName,proto3" json:"pivot_name,omitempty"` // Branch prediction. 0 or 1. Branch int32 `protobuf:"varint,4,opt,name=branch,proto3" json:"branch,omitempty"` // Values and external values in control flow context. ValuesDef *ValuesDef `protobuf:"bytes,5,opt,name=values_def,json=valuesDef,proto3" json:"values_def,omitempty"` // Contexts contained inside this context (e.g. nested conds). NestedContexts []*ControlFlowContextDef `protobuf:"bytes,6,rep,name=nested_contexts,json=nestedContexts,proto3" json:"nested_contexts,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing a CondContext object.
func (*CondContextDef) Descriptor
deprecated
func (*CondContextDef) Descriptor() ([]byte, []int)
Deprecated: Use CondContextDef.ProtoReflect.Descriptor instead.
func (*CondContextDef) GetBranch ¶
func (x *CondContextDef) GetBranch() int32
func (*CondContextDef) GetContextName ¶
func (x *CondContextDef) GetContextName() string
func (*CondContextDef) GetNestedContexts ¶
func (x *CondContextDef) GetNestedContexts() []*ControlFlowContextDef
func (*CondContextDef) GetPivotName ¶
func (x *CondContextDef) GetPivotName() string
func (*CondContextDef) GetPredName ¶
func (x *CondContextDef) GetPredName() string
func (*CondContextDef) GetValuesDef ¶
func (x *CondContextDef) GetValuesDef() *ValuesDef
func (*CondContextDef) ProtoMessage ¶
func (*CondContextDef) ProtoMessage()
func (*CondContextDef) ProtoReflect ¶
func (x *CondContextDef) ProtoReflect() protoreflect.Message
func (*CondContextDef) Reset ¶
func (x *CondContextDef) Reset()
func (*CondContextDef) String ¶
func (x *CondContextDef) String() string
type ConfigProto ¶
type ConfigProto struct { // Map from device type name (e.g., "CPU" or "GPU" ) to maximum // number of devices of that type to use. If a particular device // type is not found in the map, the system picks an appropriate // number. DeviceCount map[string]int32 `` /* 183-byte string literal not displayed */ // The execution of an individual op (for some op types) can be // parallelized on a pool of intra_op_parallelism_threads. // 0 means the system picks an appropriate number. IntraOpParallelismThreads int32 `` /* 141-byte string literal not displayed */ // Nodes that perform blocking operations are enqueued on a pool of // inter_op_parallelism_threads available in each process. // // 0 means the system picks an appropriate number. // // Note that the first Session created in the process sets the // number of threads for all future sessions unless use_per_session_threads is // true or session_inter_op_thread_pool is configured. InterOpParallelismThreads int32 `` /* 141-byte string literal not displayed */ // If true, use a new set of threads for this session rather than the global // pool of threads. Only supported by direct sessions. // // If false, use the global threads created by the first session, or the // per-session thread pools configured by session_inter_op_thread_pool. // // This option is deprecated. The same effect can be achieved by setting // session_inter_op_thread_pool to have one element, whose num_threads equals // inter_op_parallelism_threads. UsePerSessionThreads bool `` /* 126-byte string literal not displayed */ // This option is experimental - it may be replaced with a different mechanism // in the future. // // Configures session thread pools. If this is configured, then RunOptions for // a Run call can select the thread pool to use. // // The intended use is for when some session invocations need to run in a // background pool limited to a small number of threads: // - For example, a session may be configured to have one large pool (for // regular compute) and one small pool (for periodic, low priority work); // using the small pool is currently the mechanism for limiting the inter-op // parallelism of the low priority work. Note that it does not limit the // parallelism of work spawned by a single op kernel implementation. // - Using this setting is normally not needed in training, but may help some // serving use cases. // - It is also generally recommended to set the global_name field of this // proto, to avoid creating multiple large pools. It is typically better to // run the non-low-priority work, even across sessions, in a single large // pool. SessionInterOpThreadPool []*ThreadPoolOptionProto `` /* 140-byte string literal not displayed */ // Assignment of Nodes to Devices is recomputed every placement_period // steps until the system warms up (at which point the recomputation // typically slows down automatically). PlacementPeriod int32 `protobuf:"varint,3,opt,name=placement_period,json=placementPeriod,proto3" json:"placement_period,omitempty"` // When any filters are present sessions will ignore all devices which do not // match the filters. Each filter can be partially specified, e.g. "/job:ps" // "/job:worker/replica:3", etc. DeviceFilters []string `protobuf:"bytes,4,rep,name=device_filters,json=deviceFilters,proto3" json:"device_filters,omitempty"` // Options that apply to all GPUs. GpuOptions *GPUOptions `protobuf:"bytes,6,opt,name=gpu_options,json=gpuOptions,proto3" json:"gpu_options,omitempty"` // Whether soft placement is allowed. If allow_soft_placement is true, // an op will be placed on CPU if // 1. there's no GPU implementation for the OP // or // 2. no GPU devices are known or registered // or // 3. need to co-locate with reftype input(s) which are from CPU. AllowSoftPlacement bool `protobuf:"varint,7,opt,name=allow_soft_placement,json=allowSoftPlacement,proto3" json:"allow_soft_placement,omitempty"` // Whether device placements should be logged. LogDevicePlacement bool `protobuf:"varint,8,opt,name=log_device_placement,json=logDevicePlacement,proto3" json:"log_device_placement,omitempty"` // Options that apply to all graphs. GraphOptions *GraphOptions `protobuf:"bytes,10,opt,name=graph_options,json=graphOptions,proto3" json:"graph_options,omitempty"` // Global timeout for all blocking operations in this session. If non-zero, // and not overridden on a per-operation basis, this value will be used as the // deadline for all blocking operations. OperationTimeoutInMs int64 `` /* 127-byte string literal not displayed */ // Options that apply when this session uses the distributed runtime. RpcOptions *RPCOptions `protobuf:"bytes,13,opt,name=rpc_options,json=rpcOptions,proto3" json:"rpc_options,omitempty"` // Optional list of all workers to use in this session. ClusterDef *ClusterDef `protobuf:"bytes,14,opt,name=cluster_def,json=clusterDef,proto3" json:"cluster_def,omitempty"` // If true, any resources such as Variables used in the session will not be // shared with other sessions. IsolateSessionState bool `protobuf:"varint,15,opt,name=isolate_session_state,json=isolateSessionState,proto3" json:"isolate_session_state,omitempty"` Experimental *ConfigProto_Experimental `protobuf:"bytes,16,opt,name=experimental,proto3" json:"experimental,omitempty"` // contains filtered or unexported fields }
Session configuration parameters. The system picks appropriate values for fields that are not set.
func (*ConfigProto) Descriptor
deprecated
func (*ConfigProto) Descriptor() ([]byte, []int)
Deprecated: Use ConfigProto.ProtoReflect.Descriptor instead.
func (*ConfigProto) GetAllowSoftPlacement ¶
func (x *ConfigProto) GetAllowSoftPlacement() bool
func (*ConfigProto) GetClusterDef ¶
func (x *ConfigProto) GetClusterDef() *ClusterDef
func (*ConfigProto) GetDeviceCount ¶
func (x *ConfigProto) GetDeviceCount() map[string]int32
func (*ConfigProto) GetDeviceFilters ¶
func (x *ConfigProto) GetDeviceFilters() []string
func (*ConfigProto) GetExperimental ¶
func (x *ConfigProto) GetExperimental() *ConfigProto_Experimental
func (*ConfigProto) GetGpuOptions ¶
func (x *ConfigProto) GetGpuOptions() *GPUOptions
func (*ConfigProto) GetGraphOptions ¶
func (x *ConfigProto) GetGraphOptions() *GraphOptions
func (*ConfigProto) GetInterOpParallelismThreads ¶
func (x *ConfigProto) GetInterOpParallelismThreads() int32
func (*ConfigProto) GetIntraOpParallelismThreads ¶
func (x *ConfigProto) GetIntraOpParallelismThreads() int32
func (*ConfigProto) GetIsolateSessionState ¶
func (x *ConfigProto) GetIsolateSessionState() bool
func (*ConfigProto) GetLogDevicePlacement ¶
func (x *ConfigProto) GetLogDevicePlacement() bool
func (*ConfigProto) GetOperationTimeoutInMs ¶
func (x *ConfigProto) GetOperationTimeoutInMs() int64
func (*ConfigProto) GetPlacementPeriod ¶
func (x *ConfigProto) GetPlacementPeriod() int32
func (*ConfigProto) GetRpcOptions ¶
func (x *ConfigProto) GetRpcOptions() *RPCOptions
func (*ConfigProto) GetSessionInterOpThreadPool ¶
func (x *ConfigProto) GetSessionInterOpThreadPool() []*ThreadPoolOptionProto
func (*ConfigProto) GetUsePerSessionThreads ¶
func (x *ConfigProto) GetUsePerSessionThreads() bool
func (*ConfigProto) ProtoMessage ¶
func (*ConfigProto) ProtoMessage()
func (*ConfigProto) ProtoReflect ¶
func (x *ConfigProto) ProtoReflect() protoreflect.Message
func (*ConfigProto) Reset ¶
func (x *ConfigProto) Reset()
func (*ConfigProto) String ¶
func (x *ConfigProto) String() string
type ConfigProto_Experimental ¶
type ConfigProto_Experimental struct { // Task name for group resolution. CollectiveGroupLeader string `` /* 126-byte string literal not displayed */ // Which executor to use, the default executor will be used // if it is an empty string or "DEFAULT" ExecutorType string `protobuf:"bytes,3,opt,name=executor_type,json=executorType,proto3" json:"executor_type,omitempty"` // Guidance to formatting of large RecvBuf fields for transfer. // Any positive value sets the max chunk size. 0 defaults to 4096. // Any negative value indicates no max, i.e. one chunk only. RecvBufMaxChunk int32 `protobuf:"varint,4,opt,name=recv_buf_max_chunk,json=recvBufMaxChunk,proto3" json:"recv_buf_max_chunk,omitempty"` // If true, and supported by the platform, the runtime will attempt to // use NUMA affinity where applicable. One consequence will be the // existence of as many CPU devices as there are available NUMA nodes. UseNumaAffinity bool `protobuf:"varint,5,opt,name=use_numa_affinity,json=useNumaAffinity,proto3" json:"use_numa_affinity,omitempty"` // contains filtered or unexported fields }
Everything inside Experimental is subject to change and is not subject to API stability guarantees in https://www.tensorflow.org/guide/version_compat.
func (*ConfigProto_Experimental) Descriptor
deprecated
func (*ConfigProto_Experimental) Descriptor() ([]byte, []int)
Deprecated: Use ConfigProto_Experimental.ProtoReflect.Descriptor instead.
func (*ConfigProto_Experimental) GetCollectiveGroupLeader ¶
func (x *ConfigProto_Experimental) GetCollectiveGroupLeader() string
func (*ConfigProto_Experimental) GetExecutorType ¶
func (x *ConfigProto_Experimental) GetExecutorType() string
func (*ConfigProto_Experimental) GetRecvBufMaxChunk ¶
func (x *ConfigProto_Experimental) GetRecvBufMaxChunk() int32
func (*ConfigProto_Experimental) GetUseNumaAffinity ¶
func (x *ConfigProto_Experimental) GetUseNumaAffinity() bool
func (*ConfigProto_Experimental) ProtoMessage ¶
func (*ConfigProto_Experimental) ProtoMessage()
func (*ConfigProto_Experimental) ProtoReflect ¶
func (x *ConfigProto_Experimental) ProtoReflect() protoreflect.Message
func (*ConfigProto_Experimental) Reset ¶
func (x *ConfigProto_Experimental) Reset()
func (*ConfigProto_Experimental) String ¶
func (x *ConfigProto_Experimental) String() string
type ControlFlowContextDef ¶
type ControlFlowContextDef struct { // Types that are assignable to Ctxt: // *ControlFlowContextDef_CondCtxt // *ControlFlowContextDef_WhileCtxt Ctxt isControlFlowContextDef_Ctxt `protobuf_oneof:"ctxt"` // contains filtered or unexported fields }
Container for any kind of control flow context. Any other control flow contexts that are added below should also be added here.
func (*ControlFlowContextDef) Descriptor
deprecated
func (*ControlFlowContextDef) Descriptor() ([]byte, []int)
Deprecated: Use ControlFlowContextDef.ProtoReflect.Descriptor instead.
func (*ControlFlowContextDef) GetCondCtxt ¶
func (x *ControlFlowContextDef) GetCondCtxt() *CondContextDef
func (*ControlFlowContextDef) GetCtxt ¶
func (m *ControlFlowContextDef) GetCtxt() isControlFlowContextDef_Ctxt
func (*ControlFlowContextDef) GetWhileCtxt ¶
func (x *ControlFlowContextDef) GetWhileCtxt() *WhileContextDef
func (*ControlFlowContextDef) ProtoMessage ¶
func (*ControlFlowContextDef) ProtoMessage()
func (*ControlFlowContextDef) ProtoReflect ¶
func (x *ControlFlowContextDef) ProtoReflect() protoreflect.Message
func (*ControlFlowContextDef) Reset ¶
func (x *ControlFlowContextDef) Reset()
func (*ControlFlowContextDef) String ¶
func (x *ControlFlowContextDef) String() string
type ControlFlowContextDef_CondCtxt ¶
type ControlFlowContextDef_CondCtxt struct {
CondCtxt *CondContextDef `protobuf:"bytes,1,opt,name=cond_ctxt,json=condCtxt,proto3,oneof"`
}
type ControlFlowContextDef_WhileCtxt ¶
type ControlFlowContextDef_WhileCtxt struct {
WhileCtxt *WhileContextDef `protobuf:"bytes,2,opt,name=while_ctxt,json=whileCtxt,proto3,oneof"`
}
type CreateContextRequest ¶
type CreateContextRequest struct { // Identifies the full cluster, and this particular worker's position within. ServerDef *ServerDef `protobuf:"bytes,1,opt,name=server_def,json=serverDef,proto3" json:"server_def,omitempty"` // Whether the ops on the worker should be executed synchronously or // asynchronously. By default, ops are executed synchronously. Async bool `protobuf:"varint,2,opt,name=async,proto3" json:"async,omitempty"` // Number of seconds to keep the context alive. If more than keep_alive_secs // has passed since a particular context has been communicated with, it will // be garbage collected. KeepAliveSecs int64 `protobuf:"varint,3,opt,name=keep_alive_secs,json=keepAliveSecs,proto3" json:"keep_alive_secs,omitempty"` // This is the version for all the ops that will be enqueued by the client. VersionDef *framework.VersionDef `protobuf:"bytes,4,opt,name=version_def,json=versionDef,proto3" json:"version_def,omitempty"` // This ID will be used for all future communications. It is essential that // both ends use this ID for selecting a rendezvous to get everything to // match. RendezvousId int64 `protobuf:"varint,5,opt,name=rendezvous_id,json=rendezvousId,proto3" json:"rendezvous_id,omitempty"` // contains filtered or unexported fields }
func (*CreateContextRequest) Descriptor
deprecated
func (*CreateContextRequest) Descriptor() ([]byte, []int)
Deprecated: Use CreateContextRequest.ProtoReflect.Descriptor instead.
func (*CreateContextRequest) GetAsync ¶
func (x *CreateContextRequest) GetAsync() bool
func (*CreateContextRequest) GetKeepAliveSecs ¶
func (x *CreateContextRequest) GetKeepAliveSecs() int64
func (*CreateContextRequest) GetRendezvousId ¶
func (x *CreateContextRequest) GetRendezvousId() int64
func (*CreateContextRequest) GetServerDef ¶
func (x *CreateContextRequest) GetServerDef() *ServerDef
func (*CreateContextRequest) GetVersionDef ¶
func (x *CreateContextRequest) GetVersionDef() *framework.VersionDef
func (*CreateContextRequest) ProtoMessage ¶
func (*CreateContextRequest) ProtoMessage()
func (*CreateContextRequest) ProtoReflect ¶
func (x *CreateContextRequest) ProtoReflect() protoreflect.Message
func (*CreateContextRequest) Reset ¶
func (x *CreateContextRequest) Reset()
func (*CreateContextRequest) String ¶
func (x *CreateContextRequest) String() string
type CreateContextResponse ¶
type CreateContextResponse struct { // The ID of the created context. This is usually a randomly generated number, // that will be used to identify the context in future requests to the // service. Contexts are not persisted through server restarts. ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"` // List of devices that are locally accessible to the worker. DeviceAttributes []*framework.DeviceAttributes `protobuf:"bytes,2,rep,name=device_attributes,json=deviceAttributes,proto3" json:"device_attributes,omitempty"` // contains filtered or unexported fields }
func (*CreateContextResponse) Descriptor
deprecated
func (*CreateContextResponse) Descriptor() ([]byte, []int)
Deprecated: Use CreateContextResponse.ProtoReflect.Descriptor instead.
func (*CreateContextResponse) GetContextId ¶
func (x *CreateContextResponse) GetContextId() uint64
func (*CreateContextResponse) GetDeviceAttributes ¶
func (x *CreateContextResponse) GetDeviceAttributes() []*framework.DeviceAttributes
func (*CreateContextResponse) ProtoMessage ¶
func (*CreateContextResponse) ProtoMessage()
func (*CreateContextResponse) ProtoReflect ¶
func (x *CreateContextResponse) ProtoReflect() protoreflect.Message
func (*CreateContextResponse) Reset ¶
func (x *CreateContextResponse) Reset()
func (*CreateContextResponse) String ¶
func (x *CreateContextResponse) String() string
type CreateSessionRequest ¶
type CreateSessionRequest struct { // The initial graph definition. GraphDef *framework.GraphDef `protobuf:"bytes,1,opt,name=graph_def,json=graphDef,proto3" json:"graph_def,omitempty"` // Configuration options. Config *ConfigProto `protobuf:"bytes,2,opt,name=config,proto3" json:"config,omitempty"` // The target string used from the client's perspective. Target string `protobuf:"bytes,3,opt,name=target,proto3" json:"target,omitempty"` // contains filtered or unexported fields }
func (*CreateSessionRequest) Descriptor
deprecated
func (*CreateSessionRequest) Descriptor() ([]byte, []int)
Deprecated: Use CreateSessionRequest.ProtoReflect.Descriptor instead.
func (*CreateSessionRequest) GetConfig ¶
func (x *CreateSessionRequest) GetConfig() *ConfigProto
func (*CreateSessionRequest) GetGraphDef ¶
func (x *CreateSessionRequest) GetGraphDef() *framework.GraphDef
func (*CreateSessionRequest) GetTarget ¶
func (x *CreateSessionRequest) GetTarget() string
func (*CreateSessionRequest) ProtoMessage ¶
func (*CreateSessionRequest) ProtoMessage()
func (*CreateSessionRequest) ProtoReflect ¶
func (x *CreateSessionRequest) ProtoReflect() protoreflect.Message
func (*CreateSessionRequest) Reset ¶
func (x *CreateSessionRequest) Reset()
func (*CreateSessionRequest) String ¶
func (x *CreateSessionRequest) String() string
type CreateSessionResponse ¶
type CreateSessionResponse struct { // The session handle to be used in subsequent calls for the created session. // // The client must arrange to call CloseSession with this returned // session handle to close the session. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // The initial version number for the graph, to be used in the next call // to ExtendSession. GraphVersion int64 `protobuf:"varint,2,opt,name=graph_version,json=graphVersion,proto3" json:"graph_version,omitempty"` // contains filtered or unexported fields }
func (*CreateSessionResponse) Descriptor
deprecated
func (*CreateSessionResponse) Descriptor() ([]byte, []int)
Deprecated: Use CreateSessionResponse.ProtoReflect.Descriptor instead.
func (*CreateSessionResponse) GetGraphVersion ¶
func (x *CreateSessionResponse) GetGraphVersion() int64
func (*CreateSessionResponse) GetSessionHandle ¶
func (x *CreateSessionResponse) GetSessionHandle() string
func (*CreateSessionResponse) ProtoMessage ¶
func (*CreateSessionResponse) ProtoMessage()
func (*CreateSessionResponse) ProtoReflect ¶
func (x *CreateSessionResponse) ProtoReflect() protoreflect.Message
func (*CreateSessionResponse) Reset ¶
func (x *CreateSessionResponse) Reset()
func (*CreateSessionResponse) String ¶
func (x *CreateSessionResponse) String() string
type CreateWorkerSessionRequest ¶
type CreateWorkerSessionRequest struct { // Sessions are identified by a given handle. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // Defines the configuration of a TensorFlow worker. ServerDef *ServerDef `protobuf:"bytes,2,opt,name=server_def,json=serverDef,proto3" json:"server_def,omitempty"` // If true, any resources such as Variables used in the session will not be // shared with other sessions. IsolateSessionState bool `protobuf:"varint,3,opt,name=isolate_session_state,json=isolateSessionState,proto3" json:"isolate_session_state,omitempty"` // contains filtered or unexported fields }
func (*CreateWorkerSessionRequest) Descriptor
deprecated
func (*CreateWorkerSessionRequest) Descriptor() ([]byte, []int)
Deprecated: Use CreateWorkerSessionRequest.ProtoReflect.Descriptor instead.
func (*CreateWorkerSessionRequest) GetIsolateSessionState ¶
func (x *CreateWorkerSessionRequest) GetIsolateSessionState() bool
func (*CreateWorkerSessionRequest) GetServerDef ¶
func (x *CreateWorkerSessionRequest) GetServerDef() *ServerDef
func (*CreateWorkerSessionRequest) GetSessionHandle ¶
func (x *CreateWorkerSessionRequest) GetSessionHandle() string
func (*CreateWorkerSessionRequest) ProtoMessage ¶
func (*CreateWorkerSessionRequest) ProtoMessage()
func (*CreateWorkerSessionRequest) ProtoReflect ¶
func (x *CreateWorkerSessionRequest) ProtoReflect() protoreflect.Message
func (*CreateWorkerSessionRequest) Reset ¶
func (x *CreateWorkerSessionRequest) Reset()
func (*CreateWorkerSessionRequest) String ¶
func (x *CreateWorkerSessionRequest) String() string
type CreateWorkerSessionResponse ¶
type CreateWorkerSessionResponse struct {
// contains filtered or unexported fields
}
func (*CreateWorkerSessionResponse) Descriptor
deprecated
func (*CreateWorkerSessionResponse) Descriptor() ([]byte, []int)
Deprecated: Use CreateWorkerSessionResponse.ProtoReflect.Descriptor instead.
func (*CreateWorkerSessionResponse) ProtoMessage ¶
func (*CreateWorkerSessionResponse) ProtoMessage()
func (*CreateWorkerSessionResponse) ProtoReflect ¶
func (x *CreateWorkerSessionResponse) ProtoReflect() protoreflect.Message
func (*CreateWorkerSessionResponse) Reset ¶
func (x *CreateWorkerSessionResponse) Reset()
func (*CreateWorkerSessionResponse) String ¶
func (x *CreateWorkerSessionResponse) String() string
type CriticalSectionDef ¶
type CriticalSectionDef struct { // Name of the critical section handle. CriticalSectionName string `protobuf:"bytes,1,opt,name=critical_section_name,json=criticalSectionName,proto3" json:"critical_section_name,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing a CriticalSection.
func (*CriticalSectionDef) Descriptor
deprecated
func (*CriticalSectionDef) Descriptor() ([]byte, []int)
Deprecated: Use CriticalSectionDef.ProtoReflect.Descriptor instead.
func (*CriticalSectionDef) GetCriticalSectionName ¶
func (x *CriticalSectionDef) GetCriticalSectionName() string
func (*CriticalSectionDef) ProtoMessage ¶
func (*CriticalSectionDef) ProtoMessage()
func (*CriticalSectionDef) ProtoReflect ¶
func (x *CriticalSectionDef) ProtoReflect() protoreflect.Message
func (*CriticalSectionDef) Reset ¶
func (x *CriticalSectionDef) Reset()
func (*CriticalSectionDef) String ¶
func (x *CriticalSectionDef) String() string
type CriticalSectionExecutionDef ¶
type CriticalSectionExecutionDef struct { // Name of the critical section handle. ExecuteInCriticalSectionName string `` /* 151-byte string literal not displayed */ // Whether this operation requires exclusive access to its resources, // (i.e., no other CriticalSections may request the same resources). ExclusiveResourceAccess bool `` /* 133-byte string literal not displayed */ // contains filtered or unexported fields }
Protocol buffer representing a CriticalSection execution.
func (*CriticalSectionExecutionDef) Descriptor
deprecated
func (*CriticalSectionExecutionDef) Descriptor() ([]byte, []int)
Deprecated: Use CriticalSectionExecutionDef.ProtoReflect.Descriptor instead.
func (*CriticalSectionExecutionDef) GetExclusiveResourceAccess ¶
func (x *CriticalSectionExecutionDef) GetExclusiveResourceAccess() bool
func (*CriticalSectionExecutionDef) GetExecuteInCriticalSectionName ¶
func (x *CriticalSectionExecutionDef) GetExecuteInCriticalSectionName() string
func (*CriticalSectionExecutionDef) ProtoMessage ¶
func (*CriticalSectionExecutionDef) ProtoMessage()
func (*CriticalSectionExecutionDef) ProtoReflect ¶
func (x *CriticalSectionExecutionDef) ProtoReflect() protoreflect.Message
func (*CriticalSectionExecutionDef) Reset ¶
func (x *CriticalSectionExecutionDef) Reset()
func (*CriticalSectionExecutionDef) String ¶
func (x *CriticalSectionExecutionDef) String() string
type DebugOptions ¶
type DebugOptions struct { // Debugging options DebugTensorWatchOpts []*DebugTensorWatch `protobuf:"bytes,4,rep,name=debug_tensor_watch_opts,json=debugTensorWatchOpts,proto3" json:"debug_tensor_watch_opts,omitempty"` // Caller-specified global step count. // Note that this is distinct from the session run count and the executor // step count. GlobalStep int64 `protobuf:"varint,10,opt,name=global_step,json=globalStep,proto3" json:"global_step,omitempty"` // Whether the total disk usage of tfdbg is to be reset to zero // in this Session.run call. This is used by wrappers and hooks // such as the local CLI ones to indicate that the dumped tensors // are cleaned up from the disk after each Session.run. ResetDiskByteUsage bool `protobuf:"varint,11,opt,name=reset_disk_byte_usage,json=resetDiskByteUsage,proto3" json:"reset_disk_byte_usage,omitempty"` // contains filtered or unexported fields }
Options for initializing DebuggerState in TensorFlow Debugger (tfdbg).
func (*DebugOptions) Descriptor
deprecated
func (*DebugOptions) Descriptor() ([]byte, []int)
Deprecated: Use DebugOptions.ProtoReflect.Descriptor instead.
func (*DebugOptions) GetDebugTensorWatchOpts ¶
func (x *DebugOptions) GetDebugTensorWatchOpts() []*DebugTensorWatch
func (*DebugOptions) GetGlobalStep ¶
func (x *DebugOptions) GetGlobalStep() int64
func (*DebugOptions) GetResetDiskByteUsage ¶
func (x *DebugOptions) GetResetDiskByteUsage() bool
func (*DebugOptions) ProtoMessage ¶
func (*DebugOptions) ProtoMessage()
func (*DebugOptions) ProtoReflect ¶
func (x *DebugOptions) ProtoReflect() protoreflect.Message
func (*DebugOptions) Reset ¶
func (x *DebugOptions) Reset()
func (*DebugOptions) String ¶
func (x *DebugOptions) String() string
type DebugTensorWatch ¶
type DebugTensorWatch struct { // Name of the node to watch. NodeName string `protobuf:"bytes,1,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"` // Output slot to watch. // The semantics of output_slot == -1 is that the node is only watched for // completion, but not for any output tensors. See NodeCompletionCallback // in debug_gateway.h. // TODO(cais): Implement this semantics. OutputSlot int32 `protobuf:"varint,2,opt,name=output_slot,json=outputSlot,proto3" json:"output_slot,omitempty"` // Name(s) of the debugging op(s). // One or more than one probes on a tensor. // e.g., {"DebugIdentity", "DebugNanCount"} DebugOps []string `protobuf:"bytes,3,rep,name=debug_ops,json=debugOps,proto3" json:"debug_ops,omitempty"` // URL(s) for debug targets(s). // // Supported URL formats are: // - file:///foo/tfdbg_dump: Writes out Event content to file // /foo/tfdbg_dump. Assumes all directories can be created if they don't // already exist. // - grpc://localhost:11011: Sends an RPC request to an EventListener // service running at localhost:11011 with the event. // - memcbk:///event_key: Routes tensors to clients using the // callback registered with the DebugCallbackRegistry for event_key. // // Each debug op listed in debug_ops will publish its output tensor (debug // signal) to all URLs in debug_urls. // // N.B. Session::Run() supports concurrent invocations of the same inputs // (feed keys), outputs and target nodes. If such concurrent invocations // are to be debugged, the callers of Session::Run() must use distinct // debug_urls to make sure that the streamed or dumped events do not overlap // among the invocations. // TODO(cais): More visible documentation of this in g3docs. DebugUrls []string `protobuf:"bytes,4,rep,name=debug_urls,json=debugUrls,proto3" json:"debug_urls,omitempty"` // Do not error out if debug op creation fails (e.g., due to dtype // incompatibility). Instead, just log the failure. TolerateDebugOpCreationFailures bool `` /* 161-byte string literal not displayed */ // contains filtered or unexported fields }
Option for watching a node in TensorFlow Debugger (tfdbg).
func (*DebugTensorWatch) Descriptor
deprecated
func (*DebugTensorWatch) Descriptor() ([]byte, []int)
Deprecated: Use DebugTensorWatch.ProtoReflect.Descriptor instead.
func (*DebugTensorWatch) GetDebugOps ¶
func (x *DebugTensorWatch) GetDebugOps() []string
func (*DebugTensorWatch) GetDebugUrls ¶
func (x *DebugTensorWatch) GetDebugUrls() []string
func (*DebugTensorWatch) GetNodeName ¶
func (x *DebugTensorWatch) GetNodeName() string
func (*DebugTensorWatch) GetOutputSlot ¶
func (x *DebugTensorWatch) GetOutputSlot() int32
func (*DebugTensorWatch) GetTolerateDebugOpCreationFailures ¶
func (x *DebugTensorWatch) GetTolerateDebugOpCreationFailures() bool
func (*DebugTensorWatch) ProtoMessage ¶
func (*DebugTensorWatch) ProtoMessage()
func (*DebugTensorWatch) ProtoReflect ¶
func (x *DebugTensorWatch) ProtoReflect() protoreflect.Message
func (*DebugTensorWatch) Reset ¶
func (x *DebugTensorWatch) Reset()
func (*DebugTensorWatch) String ¶
func (x *DebugTensorWatch) String() string
type DebuggedSourceFile ¶
type DebuggedSourceFile struct { // The host name on which a source code file is located. Host string `protobuf:"bytes,1,opt,name=host,proto3" json:"host,omitempty"` // Path to the source code file. FilePath string `protobuf:"bytes,2,opt,name=file_path,json=filePath,proto3" json:"file_path,omitempty"` // The timestamp at which the source code file is last modified. LastModified int64 `protobuf:"varint,3,opt,name=last_modified,json=lastModified,proto3" json:"last_modified,omitempty"` // Byte size of the file. Bytes int64 `protobuf:"varint,4,opt,name=bytes,proto3" json:"bytes,omitempty"` // Line-by-line content of the source code file. Lines []string `protobuf:"bytes,5,rep,name=lines,proto3" json:"lines,omitempty"` // contains filtered or unexported fields }
func (*DebuggedSourceFile) Descriptor
deprecated
func (*DebuggedSourceFile) Descriptor() ([]byte, []int)
Deprecated: Use DebuggedSourceFile.ProtoReflect.Descriptor instead.
func (*DebuggedSourceFile) GetBytes ¶
func (x *DebuggedSourceFile) GetBytes() int64
func (*DebuggedSourceFile) GetFilePath ¶
func (x *DebuggedSourceFile) GetFilePath() string
func (*DebuggedSourceFile) GetHost ¶
func (x *DebuggedSourceFile) GetHost() string
func (*DebuggedSourceFile) GetLastModified ¶
func (x *DebuggedSourceFile) GetLastModified() int64
func (*DebuggedSourceFile) GetLines ¶
func (x *DebuggedSourceFile) GetLines() []string
func (*DebuggedSourceFile) ProtoMessage ¶
func (*DebuggedSourceFile) ProtoMessage()
func (*DebuggedSourceFile) ProtoReflect ¶
func (x *DebuggedSourceFile) ProtoReflect() protoreflect.Message
func (*DebuggedSourceFile) Reset ¶
func (x *DebuggedSourceFile) Reset()
func (*DebuggedSourceFile) String ¶
func (x *DebuggedSourceFile) String() string
type DebuggedSourceFiles ¶
type DebuggedSourceFiles struct { // A collection of source code files. SourceFiles []*DebuggedSourceFile `protobuf:"bytes,1,rep,name=source_files,json=sourceFiles,proto3" json:"source_files,omitempty"` // contains filtered or unexported fields }
func (*DebuggedSourceFiles) Descriptor
deprecated
func (*DebuggedSourceFiles) Descriptor() ([]byte, []int)
Deprecated: Use DebuggedSourceFiles.ProtoReflect.Descriptor instead.
func (*DebuggedSourceFiles) GetSourceFiles ¶
func (x *DebuggedSourceFiles) GetSourceFiles() []*DebuggedSourceFile
func (*DebuggedSourceFiles) ProtoMessage ¶
func (*DebuggedSourceFiles) ProtoMessage()
func (*DebuggedSourceFiles) ProtoReflect ¶
func (x *DebuggedSourceFiles) ProtoReflect() protoreflect.Message
func (*DebuggedSourceFiles) Reset ¶
func (x *DebuggedSourceFiles) Reset()
func (*DebuggedSourceFiles) String ¶
func (x *DebuggedSourceFiles) String() string
type DeleteWorkerSessionRequest ¶
type DeleteWorkerSessionRequest struct { // Sessions are identified by a given handle. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // contains filtered or unexported fields }
func (*DeleteWorkerSessionRequest) Descriptor
deprecated
func (*DeleteWorkerSessionRequest) Descriptor() ([]byte, []int)
Deprecated: Use DeleteWorkerSessionRequest.ProtoReflect.Descriptor instead.
func (*DeleteWorkerSessionRequest) GetSessionHandle ¶
func (x *DeleteWorkerSessionRequest) GetSessionHandle() string
func (*DeleteWorkerSessionRequest) ProtoMessage ¶
func (*DeleteWorkerSessionRequest) ProtoMessage()
func (*DeleteWorkerSessionRequest) ProtoReflect ¶
func (x *DeleteWorkerSessionRequest) ProtoReflect() protoreflect.Message
func (*DeleteWorkerSessionRequest) Reset ¶
func (x *DeleteWorkerSessionRequest) Reset()
func (*DeleteWorkerSessionRequest) String ¶
func (x *DeleteWorkerSessionRequest) String() string
type DeleteWorkerSessionResponse ¶
type DeleteWorkerSessionResponse struct {
// contains filtered or unexported fields
}
func (*DeleteWorkerSessionResponse) Descriptor
deprecated
func (*DeleteWorkerSessionResponse) Descriptor() ([]byte, []int)
Deprecated: Use DeleteWorkerSessionResponse.ProtoReflect.Descriptor instead.
func (*DeleteWorkerSessionResponse) ProtoMessage ¶
func (*DeleteWorkerSessionResponse) ProtoMessage()
func (*DeleteWorkerSessionResponse) ProtoReflect ¶
func (x *DeleteWorkerSessionResponse) ProtoReflect() protoreflect.Message
func (*DeleteWorkerSessionResponse) Reset ¶
func (x *DeleteWorkerSessionResponse) Reset()
func (*DeleteWorkerSessionResponse) String ¶
func (x *DeleteWorkerSessionResponse) String() string
type DeregisterGraphRequest ¶
type DeregisterGraphRequest struct { // The session_handle used when registering the graph. If session_handle is // empty, a single global namespace is used. SessionHandle string `protobuf:"bytes,2,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // Set to true if `CreateWorkerSession` was called for `session_handle`. CreateWorkerSessionCalled bool `` /* 141-byte string literal not displayed */ // REQUIRED: graph_handle must be returned by a RegisterGraph call // to the same WorkerService. GraphHandle string `protobuf:"bytes,1,opt,name=graph_handle,json=graphHandle,proto3" json:"graph_handle,omitempty"` // contains filtered or unexported fields }
func (*DeregisterGraphRequest) Descriptor
deprecated
func (*DeregisterGraphRequest) Descriptor() ([]byte, []int)
Deprecated: Use DeregisterGraphRequest.ProtoReflect.Descriptor instead.
func (*DeregisterGraphRequest) GetCreateWorkerSessionCalled ¶
func (x *DeregisterGraphRequest) GetCreateWorkerSessionCalled() bool
func (*DeregisterGraphRequest) GetGraphHandle ¶
func (x *DeregisterGraphRequest) GetGraphHandle() string
func (*DeregisterGraphRequest) GetSessionHandle ¶
func (x *DeregisterGraphRequest) GetSessionHandle() string
func (*DeregisterGraphRequest) ProtoMessage ¶
func (*DeregisterGraphRequest) ProtoMessage()
func (*DeregisterGraphRequest) ProtoReflect ¶
func (x *DeregisterGraphRequest) ProtoReflect() protoreflect.Message
func (*DeregisterGraphRequest) Reset ¶
func (x *DeregisterGraphRequest) Reset()
func (*DeregisterGraphRequest) String ¶
func (x *DeregisterGraphRequest) String() string
type DeregisterGraphResponse ¶
type DeregisterGraphResponse struct {
// contains filtered or unexported fields
}
func (*DeregisterGraphResponse) Descriptor
deprecated
func (*DeregisterGraphResponse) Descriptor() ([]byte, []int)
Deprecated: Use DeregisterGraphResponse.ProtoReflect.Descriptor instead.
func (*DeregisterGraphResponse) ProtoMessage ¶
func (*DeregisterGraphResponse) ProtoMessage()
func (*DeregisterGraphResponse) ProtoReflect ¶
func (x *DeregisterGraphResponse) ProtoReflect() protoreflect.Message
func (*DeregisterGraphResponse) Reset ¶
func (x *DeregisterGraphResponse) Reset()
func (*DeregisterGraphResponse) String ¶
func (x *DeregisterGraphResponse) String() string
type DeviceProperties ¶
type DeviceProperties struct { // Device type (CPU, GPU, ...) Type string `protobuf:"bytes,1,opt,name=type,proto3" json:"type,omitempty"` // Vendor (Intel, nvidia, ...) Vendor string `protobuf:"bytes,2,opt,name=vendor,proto3" json:"vendor,omitempty"` // Model (Haswell, K40, ...) Model string `protobuf:"bytes,3,opt,name=model,proto3" json:"model,omitempty"` // Core Frequency in Mhz Frequency int64 `protobuf:"varint,4,opt,name=frequency,proto3" json:"frequency,omitempty"` // Number of cores NumCores int64 `protobuf:"varint,5,opt,name=num_cores,json=numCores,proto3" json:"num_cores,omitempty"` // Version of the tools and libraries used with this device (e.g. gcc 4.9, // cudnn 5.1) Environment map[string]string `` /* 163-byte string literal not displayed */ // Number of registers per core. NumRegisters int64 `protobuf:"varint,7,opt,name=num_registers,json=numRegisters,proto3" json:"num_registers,omitempty"` // L1 cache size in bytes L1CacheSize int64 `protobuf:"varint,8,opt,name=l1_cache_size,json=l1CacheSize,proto3" json:"l1_cache_size,omitempty"` // L2 cache size in bytes L2CacheSize int64 `protobuf:"varint,9,opt,name=l2_cache_size,json=l2CacheSize,proto3" json:"l2_cache_size,omitempty"` // L3 cache size in bytes L3CacheSize int64 `protobuf:"varint,10,opt,name=l3_cache_size,json=l3CacheSize,proto3" json:"l3_cache_size,omitempty"` // applicable to GPUs only. SharedMemorySizePerMultiprocessor int64 `` /* 168-byte string literal not displayed */ // Memory size in bytes MemorySize int64 `protobuf:"varint,12,opt,name=memory_size,json=memorySize,proto3" json:"memory_size,omitempty"` // Memory bandwidth in KB/s Bandwidth int64 `protobuf:"varint,13,opt,name=bandwidth,proto3" json:"bandwidth,omitempty"` // contains filtered or unexported fields }
func (*DeviceProperties) Descriptor
deprecated
func (*DeviceProperties) Descriptor() ([]byte, []int)
Deprecated: Use DeviceProperties.ProtoReflect.Descriptor instead.
func (*DeviceProperties) GetBandwidth ¶
func (x *DeviceProperties) GetBandwidth() int64
func (*DeviceProperties) GetEnvironment ¶
func (x *DeviceProperties) GetEnvironment() map[string]string
func (*DeviceProperties) GetFrequency ¶
func (x *DeviceProperties) GetFrequency() int64
func (*DeviceProperties) GetL1CacheSize ¶
func (x *DeviceProperties) GetL1CacheSize() int64
func (*DeviceProperties) GetL2CacheSize ¶
func (x *DeviceProperties) GetL2CacheSize() int64
func (*DeviceProperties) GetL3CacheSize ¶
func (x *DeviceProperties) GetL3CacheSize() int64
func (*DeviceProperties) GetMemorySize ¶
func (x *DeviceProperties) GetMemorySize() int64
func (*DeviceProperties) GetModel ¶
func (x *DeviceProperties) GetModel() string
func (*DeviceProperties) GetNumCores ¶
func (x *DeviceProperties) GetNumCores() int64
func (*DeviceProperties) GetNumRegisters ¶
func (x *DeviceProperties) GetNumRegisters() int64
func (*DeviceProperties) GetSharedMemorySizePerMultiprocessor ¶
func (x *DeviceProperties) GetSharedMemorySizePerMultiprocessor() int64
func (*DeviceProperties) GetType ¶
func (x *DeviceProperties) GetType() string
func (*DeviceProperties) GetVendor ¶
func (x *DeviceProperties) GetVendor() string
func (*DeviceProperties) ProtoMessage ¶
func (*DeviceProperties) ProtoMessage()
func (*DeviceProperties) ProtoReflect ¶
func (x *DeviceProperties) ProtoReflect() protoreflect.Message
func (*DeviceProperties) Reset ¶
func (x *DeviceProperties) Reset()
func (*DeviceProperties) String ¶
func (x *DeviceProperties) String() string
type EnqueueRequest ¶
type EnqueueRequest struct { ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"` Queue []*QueueItem `protobuf:"bytes,3,rep,name=queue,proto3" json:"queue,omitempty"` // contains filtered or unexported fields }
func (*EnqueueRequest) Descriptor
deprecated
func (*EnqueueRequest) Descriptor() ([]byte, []int)
Deprecated: Use EnqueueRequest.ProtoReflect.Descriptor instead.
func (*EnqueueRequest) GetContextId ¶
func (x *EnqueueRequest) GetContextId() uint64
func (*EnqueueRequest) GetQueue ¶
func (x *EnqueueRequest) GetQueue() []*QueueItem
func (*EnqueueRequest) ProtoMessage ¶
func (*EnqueueRequest) ProtoMessage()
func (*EnqueueRequest) ProtoReflect ¶
func (x *EnqueueRequest) ProtoReflect() protoreflect.Message
func (*EnqueueRequest) Reset ¶
func (x *EnqueueRequest) Reset()
func (*EnqueueRequest) String ¶
func (x *EnqueueRequest) String() string
type EnqueueResponse ¶
type EnqueueResponse struct { // A single operation response for every item in the request. QueueResponse []*QueueResponse `protobuf:"bytes,1,rep,name=queue_response,json=queueResponse,proto3" json:"queue_response,omitempty"` // contains filtered or unexported fields }
func (*EnqueueResponse) Descriptor
deprecated
func (*EnqueueResponse) Descriptor() ([]byte, []int)
Deprecated: Use EnqueueResponse.ProtoReflect.Descriptor instead.
func (*EnqueueResponse) GetQueueResponse ¶
func (x *EnqueueResponse) GetQueueResponse() []*QueueResponse
func (*EnqueueResponse) ProtoMessage ¶
func (*EnqueueResponse) ProtoMessage()
func (*EnqueueResponse) ProtoReflect ¶
func (x *EnqueueResponse) ProtoReflect() protoreflect.Message
func (*EnqueueResponse) Reset ¶
func (x *EnqueueResponse) Reset()
func (*EnqueueResponse) String ¶
func (x *EnqueueResponse) String() string
type ExecutorOpts ¶
type ExecutorOpts struct { RecordCosts bool `protobuf:"varint,1,opt,name=record_costs,json=recordCosts,proto3" json:"record_costs,omitempty"` RecordTimeline bool `protobuf:"varint,3,opt,name=record_timeline,json=recordTimeline,proto3" json:"record_timeline,omitempty"` RecordPartitionGraphs bool `` /* 127-byte string literal not displayed */ ReportTensorAllocationsUponOom bool `` /* 158-byte string literal not displayed */ // contains filtered or unexported fields }
Options specific to the execution of a single step.
func (*ExecutorOpts) Descriptor
deprecated
func (*ExecutorOpts) Descriptor() ([]byte, []int)
Deprecated: Use ExecutorOpts.ProtoReflect.Descriptor instead.
func (*ExecutorOpts) GetRecordCosts ¶
func (x *ExecutorOpts) GetRecordCosts() bool
func (*ExecutorOpts) GetRecordPartitionGraphs ¶
func (x *ExecutorOpts) GetRecordPartitionGraphs() bool
func (*ExecutorOpts) GetRecordTimeline ¶
func (x *ExecutorOpts) GetRecordTimeline() bool
func (*ExecutorOpts) GetReportTensorAllocationsUponOom ¶
func (x *ExecutorOpts) GetReportTensorAllocationsUponOom() bool
func (*ExecutorOpts) ProtoMessage ¶
func (*ExecutorOpts) ProtoMessage()
func (*ExecutorOpts) ProtoReflect ¶
func (x *ExecutorOpts) ProtoReflect() protoreflect.Message
func (*ExecutorOpts) Reset ¶
func (x *ExecutorOpts) Reset()
func (*ExecutorOpts) String ¶
func (x *ExecutorOpts) String() string
type ExtendSessionRequest ¶
type ExtendSessionRequest struct { // REQUIRED: session_handle must be returned by a CreateSession call // to the same master service. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // REQUIRED: The nodes to be added to the session's graph. If any node has // the same name as an existing node, the operation will fail with // ILLEGAL_ARGUMENT. GraphDef *framework.GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef,proto3" json:"graph_def,omitempty"` // REQUIRED: The version number of the graph to be extended. This will be // tested against the current server-side version number, and the operation // will fail with FAILED_PRECONDITION if they do not match. CurrentGraphVersion int64 `protobuf:"varint,3,opt,name=current_graph_version,json=currentGraphVersion,proto3" json:"current_graph_version,omitempty"` // contains filtered or unexported fields }
func (*ExtendSessionRequest) Descriptor
deprecated
func (*ExtendSessionRequest) Descriptor() ([]byte, []int)
Deprecated: Use ExtendSessionRequest.ProtoReflect.Descriptor instead.
func (*ExtendSessionRequest) GetCurrentGraphVersion ¶
func (x *ExtendSessionRequest) GetCurrentGraphVersion() int64
func (*ExtendSessionRequest) GetGraphDef ¶
func (x *ExtendSessionRequest) GetGraphDef() *framework.GraphDef
func (*ExtendSessionRequest) GetSessionHandle ¶
func (x *ExtendSessionRequest) GetSessionHandle() string
func (*ExtendSessionRequest) ProtoMessage ¶
func (*ExtendSessionRequest) ProtoMessage()
func (*ExtendSessionRequest) ProtoReflect ¶
func (x *ExtendSessionRequest) ProtoReflect() protoreflect.Message
func (*ExtendSessionRequest) Reset ¶
func (x *ExtendSessionRequest) Reset()
func (*ExtendSessionRequest) String ¶
func (x *ExtendSessionRequest) String() string
type ExtendSessionResponse ¶
type ExtendSessionResponse struct { // The new version number for the extended graph, to be used in the next call // to ExtendSession. NewGraphVersion int64 `protobuf:"varint,4,opt,name=new_graph_version,json=newGraphVersion,proto3" json:"new_graph_version,omitempty"` // contains filtered or unexported fields }
func (*ExtendSessionResponse) Descriptor
deprecated
func (*ExtendSessionResponse) Descriptor() ([]byte, []int)
Deprecated: Use ExtendSessionResponse.ProtoReflect.Descriptor instead.
func (*ExtendSessionResponse) GetNewGraphVersion ¶
func (x *ExtendSessionResponse) GetNewGraphVersion() int64
func (*ExtendSessionResponse) ProtoMessage ¶
func (*ExtendSessionResponse) ProtoMessage()
func (*ExtendSessionResponse) ProtoReflect ¶
func (x *ExtendSessionResponse) ProtoReflect() protoreflect.Message
func (*ExtendSessionResponse) Reset ¶
func (x *ExtendSessionResponse) Reset()
func (*ExtendSessionResponse) String ¶
func (x *ExtendSessionResponse) String() string
type GPUOptions ¶
type GPUOptions struct { // Fraction of the available GPU memory to allocate for each process. // 1 means to allocate all of the GPU memory, 0.5 means the process // allocates up to ~50% of the available GPU memory. // // GPU memory is pre-allocated unless the allow_growth option is enabled. // // If greater than 1.0, uses CUDA unified memory to potentially oversubscribe // the amount of memory available on the GPU device by using host memory as a // swap space. Accessing memory not available on the device will be // significantly slower as that would require memory transfer between the host // and the device. Options to reduce the memory requirement should be // considered before enabling this option as this may come with a negative // performance impact. Oversubscription using the unified memory requires // Pascal class or newer GPUs and it is currently only supported on the Linux // operating system. See // https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#um-requirements // for the detailed requirements. PerProcessGpuMemoryFraction float64 `` /* 150-byte string literal not displayed */ // If true, the allocator does not pre-allocate the entire specified // GPU memory region, instead starting small and growing as needed. AllowGrowth bool `protobuf:"varint,4,opt,name=allow_growth,json=allowGrowth,proto3" json:"allow_growth,omitempty"` // The type of GPU allocation strategy to use. // // Allowed values: // "": The empty string (default) uses a system-chosen default // which may change over time. // // "BFC": A "Best-fit with coalescing" algorithm, simplified from a // version of dlmalloc. AllocatorType string `protobuf:"bytes,2,opt,name=allocator_type,json=allocatorType,proto3" json:"allocator_type,omitempty"` // Delay deletion of up to this many bytes to reduce the number of // interactions with gpu driver code. If 0, the system chooses // a reasonable default (several MBs). DeferredDeletionBytes int64 `` /* 127-byte string literal not displayed */ // A comma-separated list of GPU ids that determines the 'visible' // to 'virtual' mapping of GPU devices. For example, if TensorFlow // can see 8 GPU devices in the process, and one wanted to map // visible GPU devices 5 and 3 as "/device:GPU:0", and "/device:GPU:1", // then one would specify this field as "5,3". This field is similar in // spirit to the CUDA_VISIBLE_DEVICES environment variable, except // it applies to the visible GPU devices in the process. // // NOTE: // 1. The GPU driver provides the process with the visible GPUs // in an order which is not guaranteed to have any correlation to // the *physical* GPU id in the machine. This field is used for // remapping "visible" to "virtual", which means this operates only // after the process starts. Users are required to use vendor // specific mechanisms (e.g., CUDA_VISIBLE_DEVICES) to control the // physical to visible device mapping prior to invoking TensorFlow. // 2. In the code, the ids in this list are also called "platform GPU id"s, // and the 'virtual' ids of GPU devices (i.e. the ids in the device // name "/device:GPU:<id>") are also called "TF GPU id"s. Please // refer to third_party/tensorflow/core/common_runtime/gpu/gpu_id.h // for more information. VisibleDeviceList string `protobuf:"bytes,5,opt,name=visible_device_list,json=visibleDeviceList,proto3" json:"visible_device_list,omitempty"` // In the event polling loop sleep this many microseconds between // PollEvents calls, when the queue is not empty. If value is not // set or set to 0, gets set to a non-zero default. PollingActiveDelayUsecs int32 `` /* 135-byte string literal not displayed */ // This field is deprecated and ignored. PollingInactiveDelayMsecs int32 `` /* 141-byte string literal not displayed */ // Force all tensors to be gpu_compatible. On a GPU-enabled TensorFlow, // enabling this option forces all CPU tensors to be allocated with Cuda // pinned memory. Normally, TensorFlow will infer which tensors should be // allocated as the pinned memory. But in case where the inference is // incomplete, this option can significantly speed up the cross-device memory // copy performance as long as it fits the memory. // Note that this option is not something that should be // enabled by default for unknown or very large models, since all Cuda pinned // memory is unpageable, having too much pinned memory might negatively impact // the overall host system performance. ForceGpuCompatible bool `protobuf:"varint,8,opt,name=force_gpu_compatible,json=forceGpuCompatible,proto3" json:"force_gpu_compatible,omitempty"` // Everything inside experimental is subject to change and is not subject // to API stability guarantees in // https://www.tensorflow.org/guide/version_compat. Experimental *GPUOptions_Experimental `protobuf:"bytes,9,opt,name=experimental,proto3" json:"experimental,omitempty"` // contains filtered or unexported fields }
func (*GPUOptions) Descriptor
deprecated
func (*GPUOptions) Descriptor() ([]byte, []int)
Deprecated: Use GPUOptions.ProtoReflect.Descriptor instead.
func (*GPUOptions) GetAllocatorType ¶
func (x *GPUOptions) GetAllocatorType() string
func (*GPUOptions) GetAllowGrowth ¶
func (x *GPUOptions) GetAllowGrowth() bool
func (*GPUOptions) GetDeferredDeletionBytes ¶
func (x *GPUOptions) GetDeferredDeletionBytes() int64
func (*GPUOptions) GetExperimental ¶
func (x *GPUOptions) GetExperimental() *GPUOptions_Experimental
func (*GPUOptions) GetForceGpuCompatible ¶
func (x *GPUOptions) GetForceGpuCompatible() bool
func (*GPUOptions) GetPerProcessGpuMemoryFraction ¶
func (x *GPUOptions) GetPerProcessGpuMemoryFraction() float64
func (*GPUOptions) GetPollingActiveDelayUsecs ¶
func (x *GPUOptions) GetPollingActiveDelayUsecs() int32
func (*GPUOptions) GetPollingInactiveDelayMsecs ¶
func (x *GPUOptions) GetPollingInactiveDelayMsecs() int32
func (*GPUOptions) GetVisibleDeviceList ¶
func (x *GPUOptions) GetVisibleDeviceList() string
func (*GPUOptions) ProtoMessage ¶
func (*GPUOptions) ProtoMessage()
func (*GPUOptions) ProtoReflect ¶
func (x *GPUOptions) ProtoReflect() protoreflect.Message
func (*GPUOptions) Reset ¶
func (x *GPUOptions) Reset()
func (*GPUOptions) String ¶
func (x *GPUOptions) String() string
type GPUOptions_Experimental ¶
type GPUOptions_Experimental struct { // The multi virtual device settings. If empty (not set), it will create // single virtual device on each visible GPU, according to the settings // in "visible_device_list" above. Otherwise, the number of elements in the // list must be the same as the number of visible GPUs (after // "visible_device_list" filtering if it is set), and the string represented // device names (e.g. /device:GPU:<id>) will refer to the virtual // devices and have the <id> field assigned sequentially starting from 0, // according to the order they appear in this list and the "memory_limit" // list inside each element. For example, // visible_device_list = "1,0" // virtual_devices { memory_limit: 1GB memory_limit: 2GB } // virtual_devices {} // will create three virtual devices as: // /device:GPU:0 -> visible GPU 1 with 1GB memory // /device:GPU:1 -> visible GPU 1 with 2GB memory // /device:GPU:2 -> visible GPU 0 with all available memory // // NOTE: // 1. It's invalid to set both this and "per_process_gpu_memory_fraction" // at the same time. // 2. Currently this setting is per-process, not per-session. Using // different settings in different sessions within same process will // result in undefined behavior. VirtualDevices []*GPUOptions_Experimental_VirtualDevices `protobuf:"bytes,1,rep,name=virtual_devices,json=virtualDevices,proto3" json:"virtual_devices,omitempty"` // If true, uses CUDA unified memory for memory allocations. If // per_process_gpu_memory_fraction option is greater than 1.0, then unified // memory is used regardless of the value for this field. See comments for // per_process_gpu_memory_fraction field for more details and requirements // of the unified memory. This option is useful to oversubscribe memory if // multiple processes are sharing a single GPU while individually using less // than 1.0 per process memory fraction. UseUnifiedMemory bool `protobuf:"varint,2,opt,name=use_unified_memory,json=useUnifiedMemory,proto3" json:"use_unified_memory,omitempty"` // If > 1, the number of device-to-device copy streams to create // for each GPUDevice. Default value is 0, which is automatically // converted to 1. NumDevToDevCopyStreams int32 `` /* 136-byte string literal not displayed */ // If non-empty, defines a good GPU ring order on a single worker based on // device interconnect. This assumes that all workers have the same GPU // topology. Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4". // This ring order is used by the RingReducer implementation of // CollectiveReduce, and serves as an override to automatic ring order // generation in OrderTaskDeviceMap() during CollectiveParam resolution. CollectiveRingOrder string `protobuf:"bytes,4,opt,name=collective_ring_order,json=collectiveRingOrder,proto3" json:"collective_ring_order,omitempty"` // contains filtered or unexported fields }
func (*GPUOptions_Experimental) Descriptor
deprecated
func (*GPUOptions_Experimental) Descriptor() ([]byte, []int)
Deprecated: Use GPUOptions_Experimental.ProtoReflect.Descriptor instead.
func (*GPUOptions_Experimental) GetCollectiveRingOrder ¶
func (x *GPUOptions_Experimental) GetCollectiveRingOrder() string
func (*GPUOptions_Experimental) GetNumDevToDevCopyStreams ¶
func (x *GPUOptions_Experimental) GetNumDevToDevCopyStreams() int32
func (*GPUOptions_Experimental) GetUseUnifiedMemory ¶
func (x *GPUOptions_Experimental) GetUseUnifiedMemory() bool
func (*GPUOptions_Experimental) GetVirtualDevices ¶
func (x *GPUOptions_Experimental) GetVirtualDevices() []*GPUOptions_Experimental_VirtualDevices
func (*GPUOptions_Experimental) ProtoMessage ¶
func (*GPUOptions_Experimental) ProtoMessage()
func (*GPUOptions_Experimental) ProtoReflect ¶
func (x *GPUOptions_Experimental) ProtoReflect() protoreflect.Message
func (*GPUOptions_Experimental) Reset ¶
func (x *GPUOptions_Experimental) Reset()
func (*GPUOptions_Experimental) String ¶
func (x *GPUOptions_Experimental) String() string
type GPUOptions_Experimental_VirtualDevices ¶
type GPUOptions_Experimental_VirtualDevices struct { // Per "virtual" device memory limit, in MB. The number of elements in // the list is the number of virtual devices to create on the // corresponding visible GPU (see "virtual_devices" below). // If empty, it will create single virtual device taking all available // memory from the device. // // For the concept of "visible" and "virtual" GPU, see the comments for // "visible_device_list" above for more information. MemoryLimitMb []float32 `protobuf:"fixed32,1,rep,packed,name=memory_limit_mb,json=memoryLimitMb,proto3" json:"memory_limit_mb,omitempty"` // contains filtered or unexported fields }
Configuration for breaking down a visible GPU into multiple "virtual" devices.
func (*GPUOptions_Experimental_VirtualDevices) Descriptor
deprecated
func (*GPUOptions_Experimental_VirtualDevices) Descriptor() ([]byte, []int)
Deprecated: Use GPUOptions_Experimental_VirtualDevices.ProtoReflect.Descriptor instead.
func (*GPUOptions_Experimental_VirtualDevices) GetMemoryLimitMb ¶
func (x *GPUOptions_Experimental_VirtualDevices) GetMemoryLimitMb() []float32
func (*GPUOptions_Experimental_VirtualDevices) ProtoMessage ¶
func (*GPUOptions_Experimental_VirtualDevices) ProtoMessage()
func (*GPUOptions_Experimental_VirtualDevices) ProtoReflect ¶
func (x *GPUOptions_Experimental_VirtualDevices) ProtoReflect() protoreflect.Message
func (*GPUOptions_Experimental_VirtualDevices) Reset ¶
func (x *GPUOptions_Experimental_VirtualDevices) Reset()
func (*GPUOptions_Experimental_VirtualDevices) String ¶
func (x *GPUOptions_Experimental_VirtualDevices) String() string
type GetStatusRequest ¶
type GetStatusRequest struct {
// contains filtered or unexported fields
}
func (*GetStatusRequest) Descriptor
deprecated
func (*GetStatusRequest) Descriptor() ([]byte, []int)
Deprecated: Use GetStatusRequest.ProtoReflect.Descriptor instead.
func (*GetStatusRequest) ProtoMessage ¶
func (*GetStatusRequest) ProtoMessage()
func (*GetStatusRequest) ProtoReflect ¶
func (x *GetStatusRequest) ProtoReflect() protoreflect.Message
func (*GetStatusRequest) Reset ¶
func (x *GetStatusRequest) Reset()
func (*GetStatusRequest) String ¶
func (x *GetStatusRequest) String() string
type GetStatusResponse ¶
type GetStatusResponse struct { DeviceAttributes []*framework.DeviceAttributes `protobuf:"bytes,1,rep,name=device_attributes,json=deviceAttributes,proto3" json:"device_attributes,omitempty"` // contains filtered or unexported fields }
func (*GetStatusResponse) Descriptor
deprecated
func (*GetStatusResponse) Descriptor() ([]byte, []int)
Deprecated: Use GetStatusResponse.ProtoReflect.Descriptor instead.
func (*GetStatusResponse) GetDeviceAttributes ¶
func (x *GetStatusResponse) GetDeviceAttributes() []*framework.DeviceAttributes
func (*GetStatusResponse) ProtoMessage ¶
func (*GetStatusResponse) ProtoMessage()
func (*GetStatusResponse) ProtoReflect ¶
func (x *GetStatusResponse) ProtoReflect() protoreflect.Message
func (*GetStatusResponse) Reset ¶
func (x *GetStatusResponse) Reset()
func (*GetStatusResponse) String ¶
func (x *GetStatusResponse) String() string
type GetStepSequenceRequest ¶
type GetStepSequenceRequest struct { GraphKey []int64 `protobuf:"varint,1,rep,packed,name=graph_key,json=graphKey,proto3" json:"graph_key,omitempty"` // contains filtered or unexported fields }
Request for next agreed-upon step_id for the specified graph_keys. This is used to enable multiple graphs containing nodes from a common collective instance to coordinate using the same step_ids.
func (*GetStepSequenceRequest) Descriptor
deprecated
func (*GetStepSequenceRequest) Descriptor() ([]byte, []int)
Deprecated: Use GetStepSequenceRequest.ProtoReflect.Descriptor instead.
func (*GetStepSequenceRequest) GetGraphKey ¶
func (x *GetStepSequenceRequest) GetGraphKey() []int64
func (*GetStepSequenceRequest) ProtoMessage ¶
func (*GetStepSequenceRequest) ProtoMessage()
func (*GetStepSequenceRequest) ProtoReflect ¶
func (x *GetStepSequenceRequest) ProtoReflect() protoreflect.Message
func (*GetStepSequenceRequest) Reset ¶
func (x *GetStepSequenceRequest) Reset()
func (*GetStepSequenceRequest) String ¶
func (x *GetStepSequenceRequest) String() string
type GetStepSequenceResponse ¶
type GetStepSequenceResponse struct { StepSequence []*StepSequence `protobuf:"bytes,1,rep,name=step_sequence,json=stepSequence,proto3" json:"step_sequence,omitempty"` // contains filtered or unexported fields }
Next valid step_ids for one or more graph_keys.
func (*GetStepSequenceResponse) Descriptor
deprecated
func (*GetStepSequenceResponse) Descriptor() ([]byte, []int)
Deprecated: Use GetStepSequenceResponse.ProtoReflect.Descriptor instead.
func (*GetStepSequenceResponse) GetStepSequence ¶
func (x *GetStepSequenceResponse) GetStepSequence() []*StepSequence
func (*GetStepSequenceResponse) ProtoMessage ¶
func (*GetStepSequenceResponse) ProtoMessage()
func (*GetStepSequenceResponse) ProtoReflect ¶
func (x *GetStepSequenceResponse) ProtoReflect() protoreflect.Message
func (*GetStepSequenceResponse) Reset ¶
func (x *GetStepSequenceResponse) Reset()
func (*GetStepSequenceResponse) String ¶
func (x *GetStepSequenceResponse) String() string
type GraphOptions ¶
type GraphOptions struct { // If true, use control flow to schedule the activation of Recv nodes. // (Currently ignored.) EnableRecvScheduling bool `protobuf:"varint,2,opt,name=enable_recv_scheduling,json=enableRecvScheduling,proto3" json:"enable_recv_scheduling,omitempty"` // Options controlling how graph is optimized. OptimizerOptions *OptimizerOptions `protobuf:"bytes,3,opt,name=optimizer_options,json=optimizerOptions,proto3" json:"optimizer_options,omitempty"` // The number of steps to run before returning a cost model detailing // the memory usage and performance of each node of the graph. 0 means // no cost model. BuildCostModel int64 `protobuf:"varint,4,opt,name=build_cost_model,json=buildCostModel,proto3" json:"build_cost_model,omitempty"` // The number of steps to skip before collecting statistics for the // cost model. BuildCostModelAfter int64 `protobuf:"varint,9,opt,name=build_cost_model_after,json=buildCostModelAfter,proto3" json:"build_cost_model_after,omitempty"` // Annotate each Node with Op output shape data, to the extent it can // be statically inferred. InferShapes bool `protobuf:"varint,5,opt,name=infer_shapes,json=inferShapes,proto3" json:"infer_shapes,omitempty"` // Only place the subgraphs that are run, rather than the entire graph. // // This is useful for interactive graph building, where one might // produce graphs that cannot be placed during the debugging // process. In particular, it allows the client to continue work in // a session after adding a node to a graph whose placement // constraints are unsatisfiable. PlacePrunedGraph bool `protobuf:"varint,6,opt,name=place_pruned_graph,json=placePrunedGraph,proto3" json:"place_pruned_graph,omitempty"` // If true, transfer float values between processes as bfloat16. EnableBfloat16Sendrecv bool `` /* 130-byte string literal not displayed */ // If > 0, record a timeline every this many steps. // EXPERIMENTAL: This currently has no effect in MasterSession. TimelineStep int32 `protobuf:"varint,8,opt,name=timeline_step,json=timelineStep,proto3" json:"timeline_step,omitempty"` // Options that control the type and amount of graph rewriting. // Not currently configurable via the public Python API (i.e. there is no API // stability guarantee if you import RewriterConfig explicitly). RewriteOptions *RewriterConfig `protobuf:"bytes,10,opt,name=rewrite_options,json=rewriteOptions,proto3" json:"rewrite_options,omitempty"` // contains filtered or unexported fields }
func (*GraphOptions) Descriptor
deprecated
func (*GraphOptions) Descriptor() ([]byte, []int)
Deprecated: Use GraphOptions.ProtoReflect.Descriptor instead.
func (*GraphOptions) GetBuildCostModel ¶
func (x *GraphOptions) GetBuildCostModel() int64
func (*GraphOptions) GetBuildCostModelAfter ¶
func (x *GraphOptions) GetBuildCostModelAfter() int64
func (*GraphOptions) GetEnableBfloat16Sendrecv ¶
func (x *GraphOptions) GetEnableBfloat16Sendrecv() bool
func (*GraphOptions) GetEnableRecvScheduling ¶
func (x *GraphOptions) GetEnableRecvScheduling() bool
func (*GraphOptions) GetInferShapes ¶
func (x *GraphOptions) GetInferShapes() bool
func (*GraphOptions) GetOptimizerOptions ¶
func (x *GraphOptions) GetOptimizerOptions() *OptimizerOptions
func (*GraphOptions) GetPlacePrunedGraph ¶
func (x *GraphOptions) GetPlacePrunedGraph() bool
func (*GraphOptions) GetRewriteOptions ¶
func (x *GraphOptions) GetRewriteOptions() *RewriterConfig
func (*GraphOptions) GetTimelineStep ¶
func (x *GraphOptions) GetTimelineStep() int32
func (*GraphOptions) ProtoMessage ¶
func (*GraphOptions) ProtoMessage()
func (*GraphOptions) ProtoReflect ¶
func (x *GraphOptions) ProtoReflect() protoreflect.Message
func (*GraphOptions) Reset ¶
func (x *GraphOptions) Reset()
func (*GraphOptions) String ¶
func (x *GraphOptions) String() string
type JobDef ¶
type JobDef struct { // The name of this job. Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // Mapping from task ID to "hostname:port" string. // // If the `name` field contains "worker", and the `tasks` map contains a // mapping from 7 to "example.org:2222", then the device prefix // "/job:worker/task:7" will be assigned to "example.org:2222". Tasks map[int32]string `` /* 152-byte string literal not displayed */ // contains filtered or unexported fields }
Defines a single job in a TensorFlow cluster.
func (*JobDef) Descriptor
deprecated
func (*JobDef) ProtoMessage ¶
func (*JobDef) ProtoMessage()
func (*JobDef) ProtoReflect ¶
func (x *JobDef) ProtoReflect() protoreflect.Message
type KeepAliveRequest ¶
type KeepAliveRequest struct { ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"` // contains filtered or unexported fields }
func (*KeepAliveRequest) Descriptor
deprecated
func (*KeepAliveRequest) Descriptor() ([]byte, []int)
Deprecated: Use KeepAliveRequest.ProtoReflect.Descriptor instead.
func (*KeepAliveRequest) GetContextId ¶
func (x *KeepAliveRequest) GetContextId() uint64
func (*KeepAliveRequest) ProtoMessage ¶
func (*KeepAliveRequest) ProtoMessage()
func (*KeepAliveRequest) ProtoReflect ¶
func (x *KeepAliveRequest) ProtoReflect() protoreflect.Message
func (*KeepAliveRequest) Reset ¶
func (x *KeepAliveRequest) Reset()
func (*KeepAliveRequest) String ¶
func (x *KeepAliveRequest) String() string
type KeepAliveResponse ¶
type KeepAliveResponse struct {
// contains filtered or unexported fields
}
func (*KeepAliveResponse) Descriptor
deprecated
func (*KeepAliveResponse) Descriptor() ([]byte, []int)
Deprecated: Use KeepAliveResponse.ProtoReflect.Descriptor instead.
func (*KeepAliveResponse) ProtoMessage ¶
func (*KeepAliveResponse) ProtoMessage()
func (*KeepAliveResponse) ProtoReflect ¶
func (x *KeepAliveResponse) ProtoReflect() protoreflect.Message
func (*KeepAliveResponse) Reset ¶
func (x *KeepAliveResponse) Reset()
func (*KeepAliveResponse) String ¶
func (x *KeepAliveResponse) String() string
type LabeledStepStats ¶
type LabeledStepStats struct { StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` StepStats *framework.StepStats `protobuf:"bytes,2,opt,name=step_stats,json=stepStats,proto3" json:"step_stats,omitempty"` // contains filtered or unexported fields }
func (*LabeledStepStats) Descriptor
deprecated
func (*LabeledStepStats) Descriptor() ([]byte, []int)
Deprecated: Use LabeledStepStats.ProtoReflect.Descriptor instead.
func (*LabeledStepStats) GetStepId ¶
func (x *LabeledStepStats) GetStepId() int64
func (*LabeledStepStats) GetStepStats ¶
func (x *LabeledStepStats) GetStepStats() *framework.StepStats
func (*LabeledStepStats) ProtoMessage ¶
func (*LabeledStepStats) ProtoMessage()
func (*LabeledStepStats) ProtoReflect ¶
func (x *LabeledStepStats) ProtoReflect() protoreflect.Message
func (*LabeledStepStats) Reset ¶
func (x *LabeledStepStats) Reset()
func (*LabeledStepStats) String ¶
func (x *LabeledStepStats) String() string
type ListDevicesRequest ¶
type ListDevicesRequest struct { // Optional: session_handle must be returned by a CreateSession call to the // same master service. // // When session_handle is empty, the ClusterSpec provided when the master was // started is used to compute the available devices. If the session_handle is // provided but not recognized, an error is returned. Finally, if a valid // session_handle is provided, the cluster configuration for that session is // used when computing the response. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // contains filtered or unexported fields }
func (*ListDevicesRequest) Descriptor
deprecated
func (*ListDevicesRequest) Descriptor() ([]byte, []int)
Deprecated: Use ListDevicesRequest.ProtoReflect.Descriptor instead.
func (*ListDevicesRequest) GetSessionHandle ¶
func (x *ListDevicesRequest) GetSessionHandle() string
func (*ListDevicesRequest) ProtoMessage ¶
func (*ListDevicesRequest) ProtoMessage()
func (*ListDevicesRequest) ProtoReflect ¶
func (x *ListDevicesRequest) ProtoReflect() protoreflect.Message
func (*ListDevicesRequest) Reset ¶
func (x *ListDevicesRequest) Reset()
func (*ListDevicesRequest) String ¶
func (x *ListDevicesRequest) String() string
type ListDevicesResponse ¶
type ListDevicesResponse struct { LocalDevice []*framework.DeviceAttributes `protobuf:"bytes,1,rep,name=local_device,json=localDevice,proto3" json:"local_device,omitempty"` RemoteDevice []*framework.DeviceAttributes `protobuf:"bytes,2,rep,name=remote_device,json=remoteDevice,proto3" json:"remote_device,omitempty"` // contains filtered or unexported fields }
func (*ListDevicesResponse) Descriptor
deprecated
func (*ListDevicesResponse) Descriptor() ([]byte, []int)
Deprecated: Use ListDevicesResponse.ProtoReflect.Descriptor instead.
func (*ListDevicesResponse) GetLocalDevice ¶
func (x *ListDevicesResponse) GetLocalDevice() []*framework.DeviceAttributes
func (*ListDevicesResponse) GetRemoteDevice ¶
func (x *ListDevicesResponse) GetRemoteDevice() []*framework.DeviceAttributes
func (*ListDevicesResponse) ProtoMessage ¶
func (*ListDevicesResponse) ProtoMessage()
func (*ListDevicesResponse) ProtoReflect ¶
func (x *ListDevicesResponse) ProtoReflect() protoreflect.Message
func (*ListDevicesResponse) Reset ¶
func (x *ListDevicesResponse) Reset()
func (*ListDevicesResponse) String ¶
func (x *ListDevicesResponse) String() string
type LoggingRequest ¶
type LoggingRequest struct { // If true, RPC logging will be enabled. EnableRpcLogging bool `protobuf:"varint,1,opt,name=enable_rpc_logging,json=enableRpcLogging,proto3" json:"enable_rpc_logging,omitempty"` // If true, RPC logging will be disabled. DisableRpcLogging bool `protobuf:"varint,4,opt,name=disable_rpc_logging,json=disableRpcLogging,proto3" json:"disable_rpc_logging,omitempty"` // If true, discard any saved logging data (for all steps). Clear bool `protobuf:"varint,2,opt,name=clear,proto3" json:"clear,omitempty"` // When set, requests all saved log data pertaining to the step. // Any log data retrieved is eliminated from the store and cannot be // retrieved again. FetchStepId []int64 `protobuf:"varint,3,rep,packed,name=fetch_step_id,json=fetchStepId,proto3" json:"fetch_step_id,omitempty"` // contains filtered or unexported fields }
Out-of-band request to begin or end logging, or to retrieve logs for particular steps.
func (*LoggingRequest) Descriptor
deprecated
func (*LoggingRequest) Descriptor() ([]byte, []int)
Deprecated: Use LoggingRequest.ProtoReflect.Descriptor instead.
func (*LoggingRequest) GetClear ¶
func (x *LoggingRequest) GetClear() bool
func (*LoggingRequest) GetDisableRpcLogging ¶
func (x *LoggingRequest) GetDisableRpcLogging() bool
func (*LoggingRequest) GetEnableRpcLogging ¶
func (x *LoggingRequest) GetEnableRpcLogging() bool
func (*LoggingRequest) GetFetchStepId ¶
func (x *LoggingRequest) GetFetchStepId() []int64
func (*LoggingRequest) ProtoMessage ¶
func (*LoggingRequest) ProtoMessage()
func (*LoggingRequest) ProtoReflect ¶
func (x *LoggingRequest) ProtoReflect() protoreflect.Message
func (*LoggingRequest) Reset ¶
func (x *LoggingRequest) Reset()
func (*LoggingRequest) String ¶
func (x *LoggingRequest) String() string
type LoggingResponse ¶
type LoggingResponse struct { Step []*LabeledStepStats `protobuf:"bytes,1,rep,name=step,proto3" json:"step,omitempty"` // contains filtered or unexported fields }
func (*LoggingResponse) Descriptor
deprecated
func (*LoggingResponse) Descriptor() ([]byte, []int)
Deprecated: Use LoggingResponse.ProtoReflect.Descriptor instead.
func (*LoggingResponse) GetStep ¶
func (x *LoggingResponse) GetStep() []*LabeledStepStats
func (*LoggingResponse) ProtoMessage ¶
func (*LoggingResponse) ProtoMessage()
func (*LoggingResponse) ProtoReflect ¶
func (x *LoggingResponse) ProtoReflect() protoreflect.Message
func (*LoggingResponse) Reset ¶
func (x *LoggingResponse) Reset()
func (*LoggingResponse) String ¶
func (x *LoggingResponse) String() string
type MakeCallableRequest ¶
type MakeCallableRequest struct { // REQUIRED: session_handle must be returned by a CreateSession call // to the same master service. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // Options that define the behavior of the created callable. Options *CallableOptions `protobuf:"bytes,2,opt,name=options,proto3" json:"options,omitempty"` // contains filtered or unexported fields }
func (*MakeCallableRequest) Descriptor
deprecated
func (*MakeCallableRequest) Descriptor() ([]byte, []int)
Deprecated: Use MakeCallableRequest.ProtoReflect.Descriptor instead.
func (*MakeCallableRequest) GetOptions ¶
func (x *MakeCallableRequest) GetOptions() *CallableOptions
func (*MakeCallableRequest) GetSessionHandle ¶
func (x *MakeCallableRequest) GetSessionHandle() string
func (*MakeCallableRequest) ProtoMessage ¶
func (*MakeCallableRequest) ProtoMessage()
func (*MakeCallableRequest) ProtoReflect ¶
func (x *MakeCallableRequest) ProtoReflect() protoreflect.Message
func (*MakeCallableRequest) Reset ¶
func (x *MakeCallableRequest) Reset()
func (*MakeCallableRequest) String ¶
func (x *MakeCallableRequest) String() string
type MakeCallableResponse ¶
type MakeCallableResponse struct { // A handle to the created callable. Handle int64 `protobuf:"varint,1,opt,name=handle,proto3" json:"handle,omitempty"` // contains filtered or unexported fields }
func (*MakeCallableResponse) Descriptor
deprecated
func (*MakeCallableResponse) Descriptor() ([]byte, []int)
Deprecated: Use MakeCallableResponse.ProtoReflect.Descriptor instead.
func (*MakeCallableResponse) GetHandle ¶
func (x *MakeCallableResponse) GetHandle() int64
func (*MakeCallableResponse) ProtoMessage ¶
func (*MakeCallableResponse) ProtoMessage()
func (*MakeCallableResponse) ProtoReflect ¶
func (x *MakeCallableResponse) ProtoReflect() protoreflect.Message
func (*MakeCallableResponse) Reset ¶
func (x *MakeCallableResponse) Reset()
func (*MakeCallableResponse) String ¶
func (x *MakeCallableResponse) String() string
type MetaGraphDef ¶
type MetaGraphDef struct { MetaInfoDef *MetaGraphDef_MetaInfoDef `protobuf:"bytes,1,opt,name=meta_info_def,json=metaInfoDef,proto3" json:"meta_info_def,omitempty"` // GraphDef. GraphDef *framework.GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef,proto3" json:"graph_def,omitempty"` // SaverDef. SaverDef *SaverDef `protobuf:"bytes,3,opt,name=saver_def,json=saverDef,proto3" json:"saver_def,omitempty"` // collection_def: Map from collection name to collections. // See CollectionDef section for details. CollectionDef map[string]*CollectionDef `` /* 188-byte string literal not displayed */ // signature_def: Map from user supplied key for a signature to a single // SignatureDef. SignatureDef map[string]*SignatureDef `` /* 185-byte string literal not displayed */ // Asset file def to be used with the defined graph. AssetFileDef []*AssetFileDef `protobuf:"bytes,6,rep,name=asset_file_def,json=assetFileDef,proto3" json:"asset_file_def,omitempty"` // contains filtered or unexported fields }
NOTE: This protocol buffer is evolving, and will go through revisions in the coming months.
Protocol buffer containing the following which are necessary to restart training, run inference. It can be used to serialize/de-serialize memory objects necessary for running computation in a graph when crossing the process boundary. It can be used for long term storage of graphs, cross-language execution of graphs, etc.
MetaInfoDef GraphDef SaverDef CollectionDef TensorInfo SignatureDef
func (*MetaGraphDef) Descriptor
deprecated
func (*MetaGraphDef) Descriptor() ([]byte, []int)
Deprecated: Use MetaGraphDef.ProtoReflect.Descriptor instead.
func (*MetaGraphDef) GetAssetFileDef ¶
func (x *MetaGraphDef) GetAssetFileDef() []*AssetFileDef
func (*MetaGraphDef) GetCollectionDef ¶
func (x *MetaGraphDef) GetCollectionDef() map[string]*CollectionDef
func (*MetaGraphDef) GetGraphDef ¶
func (x *MetaGraphDef) GetGraphDef() *framework.GraphDef
func (*MetaGraphDef) GetMetaInfoDef ¶
func (x *MetaGraphDef) GetMetaInfoDef() *MetaGraphDef_MetaInfoDef
func (*MetaGraphDef) GetSaverDef ¶
func (x *MetaGraphDef) GetSaverDef() *SaverDef
func (*MetaGraphDef) GetSignatureDef ¶
func (x *MetaGraphDef) GetSignatureDef() map[string]*SignatureDef
func (*MetaGraphDef) ProtoMessage ¶
func (*MetaGraphDef) ProtoMessage()
func (*MetaGraphDef) ProtoReflect ¶
func (x *MetaGraphDef) ProtoReflect() protoreflect.Message
func (*MetaGraphDef) Reset ¶
func (x *MetaGraphDef) Reset()
func (*MetaGraphDef) String ¶
func (x *MetaGraphDef) String() string
type MetaGraphDef_MetaInfoDef ¶
type MetaGraphDef_MetaInfoDef struct { // User specified Version string. Can be the name of the model and revision, // steps this model has been trained to, etc. MetaGraphVersion string `protobuf:"bytes,1,opt,name=meta_graph_version,json=metaGraphVersion,proto3" json:"meta_graph_version,omitempty"` // A copy of the OpDefs used by the producer of this graph_def. // Descriptions and Ops not used in graph_def are stripped out. StrippedOpList *framework.OpList `protobuf:"bytes,2,opt,name=stripped_op_list,json=strippedOpList,proto3" json:"stripped_op_list,omitempty"` // A serialized protobuf. Can be the time this meta graph is created, or // modified, or name of the model. AnyInfo *any.Any `protobuf:"bytes,3,opt,name=any_info,json=anyInfo,proto3" json:"any_info,omitempty"` // User supplied tag(s) on the meta_graph and included graph_def. // // MetaGraphDefs should be tagged with their capabilities or use-cases. // Examples: "train", "serve", "gpu", "tpu", etc. // These tags enable loaders to access the MetaGraph(s) appropriate for a // specific use-case or runtime environment. Tags []string `protobuf:"bytes,4,rep,name=tags,proto3" json:"tags,omitempty"` // The __version__ string of the tensorflow build used to write this graph. // This will be populated by the framework, which will overwrite any user // supplied value. TensorflowVersion string `protobuf:"bytes,5,opt,name=tensorflow_version,json=tensorflowVersion,proto3" json:"tensorflow_version,omitempty"` // The __git_version__ string of the tensorflow build used to write this // graph. This will be populated by the framework, which will overwrite any // user supplied value. TensorflowGitVersion string `protobuf:"bytes,6,opt,name=tensorflow_git_version,json=tensorflowGitVersion,proto3" json:"tensorflow_git_version,omitempty"` // A flag to denote whether default-valued attrs have been stripped from // the nodes in this graph_def. StrippedDefaultAttrs bool `protobuf:"varint,7,opt,name=stripped_default_attrs,json=strippedDefaultAttrs,proto3" json:"stripped_default_attrs,omitempty"` // contains filtered or unexported fields }
Meta information regarding the graph to be exported. To be used by users of this protocol buffer to encode information regarding their meta graph.
func (*MetaGraphDef_MetaInfoDef) Descriptor
deprecated
func (*MetaGraphDef_MetaInfoDef) Descriptor() ([]byte, []int)
Deprecated: Use MetaGraphDef_MetaInfoDef.ProtoReflect.Descriptor instead.
func (*MetaGraphDef_MetaInfoDef) GetAnyInfo ¶
func (x *MetaGraphDef_MetaInfoDef) GetAnyInfo() *any.Any
func (*MetaGraphDef_MetaInfoDef) GetMetaGraphVersion ¶
func (x *MetaGraphDef_MetaInfoDef) GetMetaGraphVersion() string
func (*MetaGraphDef_MetaInfoDef) GetStrippedDefaultAttrs ¶
func (x *MetaGraphDef_MetaInfoDef) GetStrippedDefaultAttrs() bool
func (*MetaGraphDef_MetaInfoDef) GetStrippedOpList ¶
func (x *MetaGraphDef_MetaInfoDef) GetStrippedOpList() *framework.OpList
func (*MetaGraphDef_MetaInfoDef) GetTags ¶
func (x *MetaGraphDef_MetaInfoDef) GetTags() []string
func (*MetaGraphDef_MetaInfoDef) GetTensorflowGitVersion ¶
func (x *MetaGraphDef_MetaInfoDef) GetTensorflowGitVersion() string
func (*MetaGraphDef_MetaInfoDef) GetTensorflowVersion ¶
func (x *MetaGraphDef_MetaInfoDef) GetTensorflowVersion() string
func (*MetaGraphDef_MetaInfoDef) ProtoMessage ¶
func (*MetaGraphDef_MetaInfoDef) ProtoMessage()
func (*MetaGraphDef_MetaInfoDef) ProtoReflect ¶
func (x *MetaGraphDef_MetaInfoDef) ProtoReflect() protoreflect.Message
func (*MetaGraphDef_MetaInfoDef) Reset ¶
func (x *MetaGraphDef_MetaInfoDef) Reset()
func (*MetaGraphDef_MetaInfoDef) String ¶
func (x *MetaGraphDef_MetaInfoDef) String() string
type NamedDevice ¶
type NamedDevice struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` Properties *DeviceProperties `protobuf:"bytes,2,opt,name=properties,proto3" json:"properties,omitempty"` // contains filtered or unexported fields }
func (*NamedDevice) Descriptor
deprecated
func (*NamedDevice) Descriptor() ([]byte, []int)
Deprecated: Use NamedDevice.ProtoReflect.Descriptor instead.
func (*NamedDevice) GetName ¶
func (x *NamedDevice) GetName() string
func (*NamedDevice) GetProperties ¶
func (x *NamedDevice) GetProperties() *DeviceProperties
func (*NamedDevice) ProtoMessage ¶
func (*NamedDevice) ProtoMessage()
func (*NamedDevice) ProtoReflect ¶
func (x *NamedDevice) ProtoReflect() protoreflect.Message
func (*NamedDevice) Reset ¶
func (x *NamedDevice) Reset()
func (*NamedDevice) String ¶
func (x *NamedDevice) String() string
type NamedTensorProto ¶
type NamedTensorProto struct { // Name of the tensor. Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // The client can populate a TensorProto using a tensorflow::Tensor`, or // directly using the protobuf field accessors. // // The client specifies whether the returned tensor values should be // filled tensor fields (float_val, int_val, etc.) or encoded in a // compact form in tensor.tensor_content. Tensor *framework.TensorProto `protobuf:"bytes,2,opt,name=tensor,proto3" json:"tensor,omitempty"` // contains filtered or unexported fields }
A pair of tensor name and tensor values.
func (*NamedTensorProto) Descriptor
deprecated
func (*NamedTensorProto) Descriptor() ([]byte, []int)
Deprecated: Use NamedTensorProto.ProtoReflect.Descriptor instead.
func (*NamedTensorProto) GetName ¶
func (x *NamedTensorProto) GetName() string
func (*NamedTensorProto) GetTensor ¶
func (x *NamedTensorProto) GetTensor() *framework.TensorProto
func (*NamedTensorProto) ProtoMessage ¶
func (*NamedTensorProto) ProtoMessage()
func (*NamedTensorProto) ProtoReflect ¶
func (x *NamedTensorProto) ProtoReflect() protoreflect.Message
func (*NamedTensorProto) Reset ¶
func (x *NamedTensorProto) Reset()
func (*NamedTensorProto) String ¶
func (x *NamedTensorProto) String() string
type NewReplaySession ¶
type NewReplaySession struct { Devices *ListDevicesResponse `protobuf:"bytes,1,opt,name=devices,proto3" json:"devices,omitempty"` SessionHandle string `protobuf:"bytes,2,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // contains filtered or unexported fields }
Records the creation of a new replay session. We record the device listing here to capture the state of the cluster.
func (*NewReplaySession) Descriptor
deprecated
func (*NewReplaySession) Descriptor() ([]byte, []int)
Deprecated: Use NewReplaySession.ProtoReflect.Descriptor instead.
func (*NewReplaySession) GetDevices ¶
func (x *NewReplaySession) GetDevices() *ListDevicesResponse
func (*NewReplaySession) GetSessionHandle ¶
func (x *NewReplaySession) GetSessionHandle() string
func (*NewReplaySession) ProtoMessage ¶
func (*NewReplaySession) ProtoMessage()
func (*NewReplaySession) ProtoReflect ¶
func (x *NewReplaySession) ProtoReflect() protoreflect.Message
func (*NewReplaySession) Reset ¶
func (x *NewReplaySession) Reset()
func (*NewReplaySession) String ¶
func (x *NewReplaySession) String() string
type Operation ¶
type Operation struct { // A unique identifier for the operation. Set by the client so that the client // can uniquely identify the outputs of the scheduled operation. // // In the initial implementation, sending duplicate IDs has undefined // behaviour, but additional constraints may be placed upon this in the // future. Id int64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"` Name string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"` Inputs []*RemoteTensorHandle `protobuf:"bytes,3,rep,name=inputs,proto3" json:"inputs,omitempty"` // Control Operation IDs that will be respected when ops are re-ordered by // async execution. If async execution (+ op re-ordering) is not enabled, this // should have no effect. ControlOpIds []int64 `protobuf:"varint,4,rep,packed,name=control_op_ids,json=controlOpIds,proto3" json:"control_op_ids,omitempty"` Attrs map[string]*framework.AttrValue `` /* 151-byte string literal not displayed */ Device string `protobuf:"bytes,6,opt,name=device,proto3" json:"device,omitempty"` // contains filtered or unexported fields }
A proto representation of an eager operation.
func (*Operation) Descriptor
deprecated
func (*Operation) GetControlOpIds ¶
func (*Operation) GetInputs ¶
func (x *Operation) GetInputs() []*RemoteTensorHandle
func (*Operation) ProtoMessage ¶
func (*Operation) ProtoMessage()
func (*Operation) ProtoReflect ¶
func (x *Operation) ProtoReflect() protoreflect.Message
type OptimizerOptions ¶
type OptimizerOptions struct { // If true, optimize the graph using common subexpression elimination. DoCommonSubexpressionElimination bool `` /* 162-byte string literal not displayed */ // If true, perform constant folding optimization on the graph. DoConstantFolding bool `protobuf:"varint,2,opt,name=do_constant_folding,json=doConstantFolding,proto3" json:"do_constant_folding,omitempty"` // Constant folding optimization replaces tensors whose values can be // predetermined, with constant nodes. To avoid inserting too large constants, // the size of each constant created can be limited. If this value is zero, a // default limit of 10 MiB will be applied. If constant folding optimization // is disabled, this value is ignored. MaxFoldedConstantInBytes int64 `` /* 140-byte string literal not displayed */ // If true, perform function inlining on the graph. DoFunctionInlining bool `protobuf:"varint,4,opt,name=do_function_inlining,json=doFunctionInlining,proto3" json:"do_function_inlining,omitempty"` // Overall optimization level. The actual optimizations applied will be the // logical OR of the flags that this level implies and any flags already set. OptLevel OptimizerOptions_Level `protobuf:"varint,3,opt,name=opt_level,json=optLevel,proto3,enum=tensorflow.OptimizerOptions_Level" json:"opt_level,omitempty"` GlobalJitLevel OptimizerOptions_GlobalJitLevel `` /* 154-byte string literal not displayed */ // contains filtered or unexported fields }
Options passed to the graph optimizer
func (*OptimizerOptions) Descriptor
deprecated
func (*OptimizerOptions) Descriptor() ([]byte, []int)
Deprecated: Use OptimizerOptions.ProtoReflect.Descriptor instead.
func (*OptimizerOptions) GetDoCommonSubexpressionElimination ¶
func (x *OptimizerOptions) GetDoCommonSubexpressionElimination() bool
func (*OptimizerOptions) GetDoConstantFolding ¶
func (x *OptimizerOptions) GetDoConstantFolding() bool
func (*OptimizerOptions) GetDoFunctionInlining ¶
func (x *OptimizerOptions) GetDoFunctionInlining() bool
func (*OptimizerOptions) GetGlobalJitLevel ¶
func (x *OptimizerOptions) GetGlobalJitLevel() OptimizerOptions_GlobalJitLevel
func (*OptimizerOptions) GetMaxFoldedConstantInBytes ¶
func (x *OptimizerOptions) GetMaxFoldedConstantInBytes() int64
func (*OptimizerOptions) GetOptLevel ¶
func (x *OptimizerOptions) GetOptLevel() OptimizerOptions_Level
func (*OptimizerOptions) ProtoMessage ¶
func (*OptimizerOptions) ProtoMessage()
func (*OptimizerOptions) ProtoReflect ¶
func (x *OptimizerOptions) ProtoReflect() protoreflect.Message
func (*OptimizerOptions) Reset ¶
func (x *OptimizerOptions) Reset()
func (*OptimizerOptions) String ¶
func (x *OptimizerOptions) String() string
type OptimizerOptions_GlobalJitLevel ¶
type OptimizerOptions_GlobalJitLevel int32
Control the use of the compiler/jit. Experimental.
const ( OptimizerOptions_DEFAULT OptimizerOptions_GlobalJitLevel = 0 // Default setting ("off" now, but later expected to be "on") OptimizerOptions_OFF OptimizerOptions_GlobalJitLevel = -1 // The following settings turn on compilation, with higher values being // more aggressive. Higher values may reduce opportunities for parallelism // and may use more memory. (At present, there is no distinction, but this // is expected to change.) OptimizerOptions_ON_1 OptimizerOptions_GlobalJitLevel = 1 OptimizerOptions_ON_2 OptimizerOptions_GlobalJitLevel = 2 )
func (OptimizerOptions_GlobalJitLevel) Descriptor ¶
func (OptimizerOptions_GlobalJitLevel) Descriptor() protoreflect.EnumDescriptor
func (OptimizerOptions_GlobalJitLevel) Enum ¶
func (x OptimizerOptions_GlobalJitLevel) Enum() *OptimizerOptions_GlobalJitLevel
func (OptimizerOptions_GlobalJitLevel) EnumDescriptor
deprecated
func (OptimizerOptions_GlobalJitLevel) EnumDescriptor() ([]byte, []int)
Deprecated: Use OptimizerOptions_GlobalJitLevel.Descriptor instead.
func (OptimizerOptions_GlobalJitLevel) Number ¶
func (x OptimizerOptions_GlobalJitLevel) Number() protoreflect.EnumNumber
func (OptimizerOptions_GlobalJitLevel) String ¶
func (x OptimizerOptions_GlobalJitLevel) String() string
func (OptimizerOptions_GlobalJitLevel) Type ¶
func (OptimizerOptions_GlobalJitLevel) Type() protoreflect.EnumType
type OptimizerOptions_Level ¶
type OptimizerOptions_Level int32
Optimization level
const ( // L1 is the default level. // Optimization performed at L1 : // 1. Common subexpression elimination // 2. Constant folding OptimizerOptions_L1 OptimizerOptions_Level = 0 // No optimizations OptimizerOptions_L0 OptimizerOptions_Level = -1 )
func (OptimizerOptions_Level) Descriptor ¶
func (OptimizerOptions_Level) Descriptor() protoreflect.EnumDescriptor
func (OptimizerOptions_Level) Enum ¶
func (x OptimizerOptions_Level) Enum() *OptimizerOptions_Level
func (OptimizerOptions_Level) EnumDescriptor
deprecated
func (OptimizerOptions_Level) EnumDescriptor() ([]byte, []int)
Deprecated: Use OptimizerOptions_Level.Descriptor instead.
func (OptimizerOptions_Level) Number ¶
func (x OptimizerOptions_Level) Number() protoreflect.EnumNumber
func (OptimizerOptions_Level) String ¶
func (x OptimizerOptions_Level) String() string
func (OptimizerOptions_Level) Type ¶
func (OptimizerOptions_Level) Type() protoreflect.EnumType
type PartialRunSetupRequest ¶
type PartialRunSetupRequest struct { // REQUIRED: session_handle must be returned by a CreateSession call // to the same master service. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // Tensors to be fed in future steps. Feed []string `protobuf:"bytes,2,rep,name=feed,proto3" json:"feed,omitempty"` // Fetches. A list of tensor names. The caller expects a tensor to be returned // for each fetch[i] (see RunStepResponse.tensor), for corresponding partial // RunStepRequests. The order of specified fetches does not change the // execution order. Fetch []string `protobuf:"bytes,3,rep,name=fetch,proto3" json:"fetch,omitempty"` // Target Nodes. A list of node names. The named nodes will be run in future // steps, but their outputs will not be fetched. Target []string `protobuf:"bytes,4,rep,name=target,proto3" json:"target,omitempty"` // contains filtered or unexported fields }
func (*PartialRunSetupRequest) Descriptor
deprecated
func (*PartialRunSetupRequest) Descriptor() ([]byte, []int)
Deprecated: Use PartialRunSetupRequest.ProtoReflect.Descriptor instead.
func (*PartialRunSetupRequest) GetFeed ¶
func (x *PartialRunSetupRequest) GetFeed() []string
func (*PartialRunSetupRequest) GetFetch ¶
func (x *PartialRunSetupRequest) GetFetch() []string
func (*PartialRunSetupRequest) GetSessionHandle ¶
func (x *PartialRunSetupRequest) GetSessionHandle() string
func (*PartialRunSetupRequest) GetTarget ¶
func (x *PartialRunSetupRequest) GetTarget() []string
func (*PartialRunSetupRequest) ProtoMessage ¶
func (*PartialRunSetupRequest) ProtoMessage()
func (*PartialRunSetupRequest) ProtoReflect ¶
func (x *PartialRunSetupRequest) ProtoReflect() protoreflect.Message
func (*PartialRunSetupRequest) Reset ¶
func (x *PartialRunSetupRequest) Reset()
func (*PartialRunSetupRequest) String ¶
func (x *PartialRunSetupRequest) String() string
type PartialRunSetupResponse ¶
type PartialRunSetupResponse struct { // The unique handle corresponding to the ongoing partial run call setup by // the invocation to PartialRunSetup. This handle may be passed to // RunStepRequest to send and receive tensors for this partial run. PartialRunHandle string `protobuf:"bytes,1,opt,name=partial_run_handle,json=partialRunHandle,proto3" json:"partial_run_handle,omitempty"` // contains filtered or unexported fields }
func (*PartialRunSetupResponse) Descriptor
deprecated
func (*PartialRunSetupResponse) Descriptor() ([]byte, []int)
Deprecated: Use PartialRunSetupResponse.ProtoReflect.Descriptor instead.
func (*PartialRunSetupResponse) GetPartialRunHandle ¶
func (x *PartialRunSetupResponse) GetPartialRunHandle() string
func (*PartialRunSetupResponse) ProtoMessage ¶
func (*PartialRunSetupResponse) ProtoMessage()
func (*PartialRunSetupResponse) ProtoReflect ¶
func (x *PartialRunSetupResponse) ProtoReflect() protoreflect.Message
func (*PartialRunSetupResponse) Reset ¶
func (x *PartialRunSetupResponse) Reset()
func (*PartialRunSetupResponse) String ¶
func (x *PartialRunSetupResponse) String() string
type QueueItem ¶
type QueueItem struct { // The remote executor should be able to handle either executing ops directly, // or releasing any unused tensor handles, since the tensor lifetime is // maintained by the client. // // Types that are assignable to Item: // *QueueItem_HandleToDecref // *QueueItem_Operation Item isQueueItem_Item `protobuf_oneof:"item"` // contains filtered or unexported fields }
func (*QueueItem) Descriptor
deprecated
func (*QueueItem) GetHandleToDecref ¶
func (x *QueueItem) GetHandleToDecref() *RemoteTensorHandle
func (*QueueItem) GetOperation ¶
func (*QueueItem) ProtoMessage ¶
func (*QueueItem) ProtoMessage()
func (*QueueItem) ProtoReflect ¶
func (x *QueueItem) ProtoReflect() protoreflect.Message
type QueueItem_HandleToDecref ¶
type QueueItem_HandleToDecref struct {
HandleToDecref *RemoteTensorHandle `protobuf:"bytes,1,opt,name=handle_to_decref,json=handleToDecref,proto3,oneof"`
}
type QueueItem_Operation ¶
type QueueItem_Operation struct {
Operation *Operation `protobuf:"bytes,2,opt,name=operation,proto3,oneof"`
}
type QueueResponse ¶
type QueueResponse struct { Shape []*framework.TensorShapeProto `protobuf:"bytes,1,rep,name=shape,proto3" json:"shape,omitempty"` // contains filtered or unexported fields }
func (*QueueResponse) Descriptor
deprecated
func (*QueueResponse) Descriptor() ([]byte, []int)
Deprecated: Use QueueResponse.ProtoReflect.Descriptor instead.
func (*QueueResponse) GetShape ¶
func (x *QueueResponse) GetShape() []*framework.TensorShapeProto
func (*QueueResponse) ProtoMessage ¶
func (*QueueResponse) ProtoMessage()
func (*QueueResponse) ProtoReflect ¶
func (x *QueueResponse) ProtoReflect() protoreflect.Message
func (*QueueResponse) Reset ¶
func (x *QueueResponse) Reset()
func (*QueueResponse) String ¶
func (x *QueueResponse) String() string
type QueueRunnerDef ¶
type QueueRunnerDef struct { // Queue name. QueueName string `protobuf:"bytes,1,opt,name=queue_name,json=queueName,proto3" json:"queue_name,omitempty"` // A list of enqueue operations. EnqueueOpName []string `protobuf:"bytes,2,rep,name=enqueue_op_name,json=enqueueOpName,proto3" json:"enqueue_op_name,omitempty"` // The operation to run to close the queue. CloseOpName string `protobuf:"bytes,3,opt,name=close_op_name,json=closeOpName,proto3" json:"close_op_name,omitempty"` // The operation to run to cancel the queue. CancelOpName string `protobuf:"bytes,4,opt,name=cancel_op_name,json=cancelOpName,proto3" json:"cancel_op_name,omitempty"` // A list of exception types considered to signal a safely closed queue // if raised during enqueue operations. QueueClosedExceptionTypes []core.Code `` /* 175-byte string literal not displayed */ // contains filtered or unexported fields }
Protocol buffer representing a QueueRunner.
func (*QueueRunnerDef) Descriptor
deprecated
func (*QueueRunnerDef) Descriptor() ([]byte, []int)
Deprecated: Use QueueRunnerDef.ProtoReflect.Descriptor instead.
func (*QueueRunnerDef) GetCancelOpName ¶
func (x *QueueRunnerDef) GetCancelOpName() string
func (*QueueRunnerDef) GetCloseOpName ¶
func (x *QueueRunnerDef) GetCloseOpName() string
func (*QueueRunnerDef) GetEnqueueOpName ¶
func (x *QueueRunnerDef) GetEnqueueOpName() []string
func (*QueueRunnerDef) GetQueueClosedExceptionTypes ¶
func (x *QueueRunnerDef) GetQueueClosedExceptionTypes() []core.Code
func (*QueueRunnerDef) GetQueueName ¶
func (x *QueueRunnerDef) GetQueueName() string
func (*QueueRunnerDef) ProtoMessage ¶
func (*QueueRunnerDef) ProtoMessage()
func (*QueueRunnerDef) ProtoReflect ¶
func (x *QueueRunnerDef) ProtoReflect() protoreflect.Message
func (*QueueRunnerDef) Reset ¶
func (x *QueueRunnerDef) Reset()
func (*QueueRunnerDef) String ¶
func (x *QueueRunnerDef) String() string
type RPCOptions ¶
type RPCOptions struct { // If true, always use RPC to contact the session target. // // If false (the default option), TensorFlow may use an optimized // transport for client-master communication that avoids the RPC // stack. This option is primarily for used testing the RPC stack. UseRpcForInprocessMaster bool `` /* 140-byte string literal not displayed */ // The compression algorithm to be used. One of "deflate", "gzip". CompressionAlgorithm string `protobuf:"bytes,2,opt,name=compression_algorithm,json=compressionAlgorithm,proto3" json:"compression_algorithm,omitempty"` // If compression_algorithm is set, the compression level to be used. // From 0 (no compression), up to 3. CompressionLevel int32 `protobuf:"varint,3,opt,name=compression_level,json=compressionLevel,proto3" json:"compression_level,omitempty"` // contains filtered or unexported fields }
func (*RPCOptions) Descriptor
deprecated
func (*RPCOptions) Descriptor() ([]byte, []int)
Deprecated: Use RPCOptions.ProtoReflect.Descriptor instead.
func (*RPCOptions) GetCompressionAlgorithm ¶
func (x *RPCOptions) GetCompressionAlgorithm() string
func (*RPCOptions) GetCompressionLevel ¶
func (x *RPCOptions) GetCompressionLevel() int32
func (*RPCOptions) GetUseRpcForInprocessMaster ¶
func (x *RPCOptions) GetUseRpcForInprocessMaster() bool
func (*RPCOptions) ProtoMessage ¶
func (*RPCOptions) ProtoMessage()
func (*RPCOptions) ProtoReflect ¶
func (x *RPCOptions) ProtoReflect() protoreflect.Message
func (*RPCOptions) Reset ¶
func (x *RPCOptions) Reset()
func (*RPCOptions) String ¶
func (x *RPCOptions) String() string
type RecvBufRequest ¶
type RecvBufRequest struct { // Used at server side to find the correct BufRendezvous. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // Arbitrary string identifying a BufRendezvous entry. BufRendezvousKey string `protobuf:"bytes,2,opt,name=buf_rendezvous_key,json=bufRendezvousKey,proto3" json:"buf_rendezvous_key,omitempty"` // Size of value expected, must agree with BufRendezvous entry. NumBytes int64 `protobuf:"varint,3,opt,name=num_bytes,json=numBytes,proto3" json:"num_bytes,omitempty"` // When RDMA is in use, address of destination field on client. BufPtr uint64 `protobuf:"fixed64,4,opt,name=buf_ptr,json=bufPtr,proto3" json:"buf_ptr,omitempty"` // Optional information on client-side device locality. ClientLocality *framework.DeviceLocality `protobuf:"bytes,5,opt,name=client_locality,json=clientLocality,proto3" json:"client_locality,omitempty"` // Optional information on server-side device locality. ServerLocality *framework.DeviceLocality `protobuf:"bytes,6,opt,name=server_locality,json=serverLocality,proto3" json:"server_locality,omitempty"` // Optional, implementation-specific data. TransportOptions *any.Any `protobuf:"bytes,7,opt,name=transport_options,json=transportOptions,proto3" json:"transport_options,omitempty"` // Optional, for annotating the timeline. SrcDevice string `protobuf:"bytes,8,opt,name=src_device,json=srcDevice,proto3" json:"src_device,omitempty"` DstDevice string `protobuf:"bytes,9,opt,name=dst_device,json=dstDevice,proto3" json:"dst_device,omitempty"` // Depending on the RPC system in use, it may be necessary to set this // id to detect resends of RPCs where the server is not aware that // the prior RPC failed. RequestId int64 `protobuf:"varint,10,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"` // contains filtered or unexported fields }
func (*RecvBufRequest) Descriptor
deprecated
func (*RecvBufRequest) Descriptor() ([]byte, []int)
Deprecated: Use RecvBufRequest.ProtoReflect.Descriptor instead.
func (*RecvBufRequest) GetBufPtr ¶
func (x *RecvBufRequest) GetBufPtr() uint64
func (*RecvBufRequest) GetBufRendezvousKey ¶
func (x *RecvBufRequest) GetBufRendezvousKey() string
func (*RecvBufRequest) GetClientLocality ¶
func (x *RecvBufRequest) GetClientLocality() *framework.DeviceLocality
func (*RecvBufRequest) GetDstDevice ¶
func (x *RecvBufRequest) GetDstDevice() string
func (*RecvBufRequest) GetNumBytes ¶
func (x *RecvBufRequest) GetNumBytes() int64
func (*RecvBufRequest) GetRequestId ¶
func (x *RecvBufRequest) GetRequestId() int64
func (*RecvBufRequest) GetServerLocality ¶
func (x *RecvBufRequest) GetServerLocality() *framework.DeviceLocality
func (*RecvBufRequest) GetSrcDevice ¶
func (x *RecvBufRequest) GetSrcDevice() string
func (*RecvBufRequest) GetStepId ¶
func (x *RecvBufRequest) GetStepId() int64
func (*RecvBufRequest) GetTransportOptions ¶
func (x *RecvBufRequest) GetTransportOptions() *any.Any
func (*RecvBufRequest) ProtoMessage ¶
func (*RecvBufRequest) ProtoMessage()
func (*RecvBufRequest) ProtoReflect ¶
func (x *RecvBufRequest) ProtoReflect() protoreflect.Message
func (*RecvBufRequest) Reset ¶
func (x *RecvBufRequest) Reset()
func (*RecvBufRequest) String ¶
func (x *RecvBufRequest) String() string
type RecvBufRespExtra ¶
type RecvBufRespExtra struct { TensorContent [][]byte `protobuf:"bytes,1,rep,name=tensor_content,json=tensorContent,proto3" json:"tensor_content,omitempty"` // contains filtered or unexported fields }
Extra data needed on a non-RDMA RecvBufResponse.
func (*RecvBufRespExtra) Descriptor
deprecated
func (*RecvBufRespExtra) Descriptor() ([]byte, []int)
Deprecated: Use RecvBufRespExtra.ProtoReflect.Descriptor instead.
func (*RecvBufRespExtra) GetTensorContent ¶
func (x *RecvBufRespExtra) GetTensorContent() [][]byte
func (*RecvBufRespExtra) ProtoMessage ¶
func (*RecvBufRespExtra) ProtoMessage()
func (*RecvBufRespExtra) ProtoReflect ¶
func (x *RecvBufRespExtra) ProtoReflect() protoreflect.Message
func (*RecvBufRespExtra) Reset ¶
func (x *RecvBufRespExtra) Reset()
func (*RecvBufRespExtra) String ¶
func (x *RecvBufRespExtra) String() string
type RecvBufResponse ¶
type RecvBufResponse struct { BufPtr uint64 `protobuf:"fixed64,1,opt,name=buf_ptr,json=bufPtr,proto3" json:"buf_ptr,omitempty"` // Address of source field on server. NumBytes int64 `protobuf:"varint,2,opt,name=num_bytes,json=numBytes,proto3" json:"num_bytes,omitempty"` // Byte length of buf_ptr field, if set. IsDead bool `protobuf:"varint,3,opt,name=is_dead,json=isDead,proto3" json:"is_dead,omitempty"` // True if value is 'dead' like a tensor. // Optional, implementation-specific data. TransportOptions *any.Any `protobuf:"bytes,4,opt,name=transport_options,json=transportOptions,proto3" json:"transport_options,omitempty"` // Optional, for timeline. SendStartMicros int64 `protobuf:"varint,5,opt,name=send_start_micros,json=sendStartMicros,proto3" json:"send_start_micros,omitempty"` // contains filtered or unexported fields }
func (*RecvBufResponse) Descriptor
deprecated
func (*RecvBufResponse) Descriptor() ([]byte, []int)
Deprecated: Use RecvBufResponse.ProtoReflect.Descriptor instead.
func (*RecvBufResponse) GetBufPtr ¶
func (x *RecvBufResponse) GetBufPtr() uint64
func (*RecvBufResponse) GetIsDead ¶
func (x *RecvBufResponse) GetIsDead() bool
func (*RecvBufResponse) GetNumBytes ¶
func (x *RecvBufResponse) GetNumBytes() int64
func (*RecvBufResponse) GetSendStartMicros ¶
func (x *RecvBufResponse) GetSendStartMicros() int64
func (*RecvBufResponse) GetTransportOptions ¶
func (x *RecvBufResponse) GetTransportOptions() *any.Any
func (*RecvBufResponse) ProtoMessage ¶
func (*RecvBufResponse) ProtoMessage()
func (*RecvBufResponse) ProtoReflect ¶
func (x *RecvBufResponse) ProtoReflect() protoreflect.Message
func (*RecvBufResponse) Reset ¶
func (x *RecvBufResponse) Reset()
func (*RecvBufResponse) String ¶
func (x *RecvBufResponse) String() string
type RecvTensorRequest ¶
type RecvTensorRequest struct { // The step in which the tensor will be produced. // // REQUIRED: This must eventually correspond to the `step_id` passed // into a RunGraph call on the same WorkerService. StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // A key identifying the channel to receive tensors from. A RecvTensor request // retrieves one tensor from the channel, but multiple tensors can be sent and // received over the same channel with multiple RecvTensor requests. See // rendezvous.h for details. RendezvousKey string `protobuf:"bytes,2,opt,name=rendezvous_key,json=rendezvousKey,proto3" json:"rendezvous_key,omitempty"` // If true, use an out-of-band DMA mechanism to transfer the // received tensor. DmaOk bool `protobuf:"varint,3,opt,name=dma_ok,json=dmaOk,proto3" json:"dma_ok,omitempty"` // Optional information on client-side device locality. ClientLocality *framework.DeviceLocality `protobuf:"bytes,4,opt,name=client_locality,json=clientLocality,proto3" json:"client_locality,omitempty"` // Optional information on server-side device locality. ServerLocality *framework.DeviceLocality `protobuf:"bytes,5,opt,name=server_locality,json=serverLocality,proto3" json:"server_locality,omitempty"` // Optional information needed by the RPC subsystem. TransportOptions *any.Any `protobuf:"bytes,6,opt,name=transport_options,json=transportOptions,proto3" json:"transport_options,omitempty"` // Unique identifier for this request. Every RecvTensorRequest must have a // unique request_id, and retried RecvTensorRequests must have the same // request_id. If request_id is zero, retry detection is disabled. // // Retried RecvTensorRequests are problematic because a RecvTensor with no // corresponding sender will wait forever, and the tensor may have been // delivered to a previous retry. Workers use request_ids to reject retried // RecvTensor requests instead of waiting forever. RequestId int64 `protobuf:"varint,7,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"` // contains filtered or unexported fields }
func (*RecvTensorRequest) Descriptor
deprecated
func (*RecvTensorRequest) Descriptor() ([]byte, []int)
Deprecated: Use RecvTensorRequest.ProtoReflect.Descriptor instead.
func (*RecvTensorRequest) GetClientLocality ¶
func (x *RecvTensorRequest) GetClientLocality() *framework.DeviceLocality
func (*RecvTensorRequest) GetDmaOk ¶
func (x *RecvTensorRequest) GetDmaOk() bool
func (*RecvTensorRequest) GetRendezvousKey ¶
func (x *RecvTensorRequest) GetRendezvousKey() string
func (*RecvTensorRequest) GetRequestId ¶
func (x *RecvTensorRequest) GetRequestId() int64
func (*RecvTensorRequest) GetServerLocality ¶
func (x *RecvTensorRequest) GetServerLocality() *framework.DeviceLocality
func (*RecvTensorRequest) GetStepId ¶
func (x *RecvTensorRequest) GetStepId() int64
func (*RecvTensorRequest) GetTransportOptions ¶
func (x *RecvTensorRequest) GetTransportOptions() *any.Any
func (*RecvTensorRequest) ProtoMessage ¶
func (*RecvTensorRequest) ProtoMessage()
func (*RecvTensorRequest) ProtoReflect ¶
func (x *RecvTensorRequest) ProtoReflect() protoreflect.Message
func (*RecvTensorRequest) Reset ¶
func (x *RecvTensorRequest) Reset()
func (*RecvTensorRequest) String ¶
func (x *RecvTensorRequest) String() string
type RecvTensorResponse ¶
type RecvTensorResponse struct { // The tensor as a proto. Tensor *framework.TensorProto `protobuf:"bytes,1,opt,name=tensor,proto3" json:"tensor,omitempty"` // If true, this tensor was the output of a dead node, and the // content is invalid. IsDead bool `protobuf:"varint,2,opt,name=is_dead,json=isDead,proto3" json:"is_dead,omitempty"` // The time at which tensor was available and started to be returned. SendStartMicros int64 `protobuf:"varint,3,opt,name=send_start_micros,json=sendStartMicros,proto3" json:"send_start_micros,omitempty"` // Optional additional information about how to receive the tensor, // e.g. in the event that `RecvTensorRequest.dma_ok` was true. TransportOptions *any.Any `protobuf:"bytes,4,opt,name=transport_options,json=transportOptions,proto3" json:"transport_options,omitempty"` // contains filtered or unexported fields }
func (*RecvTensorResponse) Descriptor
deprecated
func (*RecvTensorResponse) Descriptor() ([]byte, []int)
Deprecated: Use RecvTensorResponse.ProtoReflect.Descriptor instead.
func (*RecvTensorResponse) GetIsDead ¶
func (x *RecvTensorResponse) GetIsDead() bool
func (*RecvTensorResponse) GetSendStartMicros ¶
func (x *RecvTensorResponse) GetSendStartMicros() int64
func (*RecvTensorResponse) GetTensor ¶
func (x *RecvTensorResponse) GetTensor() *framework.TensorProto
func (*RecvTensorResponse) GetTransportOptions ¶
func (x *RecvTensorResponse) GetTransportOptions() *any.Any
func (*RecvTensorResponse) ProtoMessage ¶
func (*RecvTensorResponse) ProtoMessage()
func (*RecvTensorResponse) ProtoReflect ¶
func (x *RecvTensorResponse) ProtoReflect() protoreflect.Message
func (*RecvTensorResponse) Reset ¶
func (x *RecvTensorResponse) Reset()
func (*RecvTensorResponse) String ¶
func (x *RecvTensorResponse) String() string
type RegisterFunctionRequest ¶
type RegisterFunctionRequest struct { ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"` FunctionDef *framework.FunctionDef `protobuf:"bytes,2,opt,name=function_def,json=functionDef,proto3" json:"function_def,omitempty"` // contains filtered or unexported fields }
func (*RegisterFunctionRequest) Descriptor
deprecated
func (*RegisterFunctionRequest) Descriptor() ([]byte, []int)
Deprecated: Use RegisterFunctionRequest.ProtoReflect.Descriptor instead.
func (*RegisterFunctionRequest) GetContextId ¶
func (x *RegisterFunctionRequest) GetContextId() uint64
func (*RegisterFunctionRequest) GetFunctionDef ¶
func (x *RegisterFunctionRequest) GetFunctionDef() *framework.FunctionDef
func (*RegisterFunctionRequest) ProtoMessage ¶
func (*RegisterFunctionRequest) ProtoMessage()
func (*RegisterFunctionRequest) ProtoReflect ¶
func (x *RegisterFunctionRequest) ProtoReflect() protoreflect.Message
func (*RegisterFunctionRequest) Reset ¶
func (x *RegisterFunctionRequest) Reset()
func (*RegisterFunctionRequest) String ¶
func (x *RegisterFunctionRequest) String() string
type RegisterFunctionResponse ¶
type RegisterFunctionResponse struct {
// contains filtered or unexported fields
}
func (*RegisterFunctionResponse) Descriptor
deprecated
func (*RegisterFunctionResponse) Descriptor() ([]byte, []int)
Deprecated: Use RegisterFunctionResponse.ProtoReflect.Descriptor instead.
func (*RegisterFunctionResponse) ProtoMessage ¶
func (*RegisterFunctionResponse) ProtoMessage()
func (*RegisterFunctionResponse) ProtoReflect ¶
func (x *RegisterFunctionResponse) ProtoReflect() protoreflect.Message
func (*RegisterFunctionResponse) Reset ¶
func (x *RegisterFunctionResponse) Reset()
func (*RegisterFunctionResponse) String ¶
func (x *RegisterFunctionResponse) String() string
type RegisterGraphRequest ¶
type RegisterGraphRequest struct { // Subgraphs are scoped within one session. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // Set to true if `CreateWorkerSession` was called for `session_handle`. CreateWorkerSessionCalled bool `` /* 141-byte string literal not displayed */ // "graph_def" has the subgraph of nodes for this worker, with each node // having its device_name filled in. GraphDef *framework.GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef,proto3" json:"graph_def,omitempty"` // True iff the graph (before partitioning) contains control flow nodes. // // As of 01/11/2015, this is no longer set by clients. // // Deprecated: Do not use. HasControlFlow bool `protobuf:"varint,3,opt,name=has_control_flow,json=hasControlFlow,proto3" json:"has_control_flow,omitempty"` // Configuration options for the session in which this graph was created. GraphOptions *GraphOptions `protobuf:"bytes,4,opt,name=graph_options,json=graphOptions,proto3" json:"graph_options,omitempty"` // Field(s) used by TensorFlow Debugger (tfdbg). DebugOptions *DebugOptions `protobuf:"bytes,5,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"` // If graph_def contains any collective ops this must be a positive // integer used to coordinate execution with other graphs. All // graphs in a distributed execution with the same // collective_graph_key will coordinate to use the same step_id // concurrently so that BufRendezvous entries will make the correct // values accessible. CollectiveGraphKey int64 `protobuf:"varint,7,opt,name=collective_graph_key,json=collectiveGraphKey,proto3" json:"collective_graph_key,omitempty"` // contains filtered or unexported fields }
func (*RegisterGraphRequest) Descriptor
deprecated
func (*RegisterGraphRequest) Descriptor() ([]byte, []int)
Deprecated: Use RegisterGraphRequest.ProtoReflect.Descriptor instead.
func (*RegisterGraphRequest) GetCollectiveGraphKey ¶
func (x *RegisterGraphRequest) GetCollectiveGraphKey() int64
func (*RegisterGraphRequest) GetCreateWorkerSessionCalled ¶
func (x *RegisterGraphRequest) GetCreateWorkerSessionCalled() bool
func (*RegisterGraphRequest) GetDebugOptions ¶
func (x *RegisterGraphRequest) GetDebugOptions() *DebugOptions
func (*RegisterGraphRequest) GetGraphDef ¶
func (x *RegisterGraphRequest) GetGraphDef() *framework.GraphDef
func (*RegisterGraphRequest) GetGraphOptions ¶
func (x *RegisterGraphRequest) GetGraphOptions() *GraphOptions
func (*RegisterGraphRequest) GetHasControlFlow
deprecated
func (x *RegisterGraphRequest) GetHasControlFlow() bool
Deprecated: Do not use.
func (*RegisterGraphRequest) GetSessionHandle ¶
func (x *RegisterGraphRequest) GetSessionHandle() string
func (*RegisterGraphRequest) ProtoMessage ¶
func (*RegisterGraphRequest) ProtoMessage()
func (*RegisterGraphRequest) ProtoReflect ¶
func (x *RegisterGraphRequest) ProtoReflect() protoreflect.Message
func (*RegisterGraphRequest) Reset ¶
func (x *RegisterGraphRequest) Reset()
func (*RegisterGraphRequest) String ¶
func (x *RegisterGraphRequest) String() string
type RegisterGraphResponse ¶
type RegisterGraphResponse struct { // If the registration succeeds, returns an opaque graph_handle to // the master. The master calls RunGraph with graph_handle to // compute different steps. GraphHandle string `protobuf:"bytes,1,opt,name=graph_handle,json=graphHandle,proto3" json:"graph_handle,omitempty"` // contains filtered or unexported fields }
func (*RegisterGraphResponse) Descriptor
deprecated
func (*RegisterGraphResponse) Descriptor() ([]byte, []int)
Deprecated: Use RegisterGraphResponse.ProtoReflect.Descriptor instead.
func (*RegisterGraphResponse) GetGraphHandle ¶
func (x *RegisterGraphResponse) GetGraphHandle() string
func (*RegisterGraphResponse) ProtoMessage ¶
func (*RegisterGraphResponse) ProtoMessage()
func (*RegisterGraphResponse) ProtoReflect ¶
func (x *RegisterGraphResponse) ProtoReflect() protoreflect.Message
func (*RegisterGraphResponse) Reset ¶
func (x *RegisterGraphResponse) Reset()
func (*RegisterGraphResponse) String ¶
func (x *RegisterGraphResponse) String() string
type ReleaseCallableRequest ¶
type ReleaseCallableRequest struct { // REQUIRED: session_handle must be returned by a CreateSession call // to the same master service. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // REQUIRED: handle must be returned by a MakeCallable call to the same // master service. Handle int64 `protobuf:"varint,2,opt,name=handle,proto3" json:"handle,omitempty"` // contains filtered or unexported fields }
func (*ReleaseCallableRequest) Descriptor
deprecated
func (*ReleaseCallableRequest) Descriptor() ([]byte, []int)
Deprecated: Use ReleaseCallableRequest.ProtoReflect.Descriptor instead.
func (*ReleaseCallableRequest) GetHandle ¶
func (x *ReleaseCallableRequest) GetHandle() int64
func (*ReleaseCallableRequest) GetSessionHandle ¶
func (x *ReleaseCallableRequest) GetSessionHandle() string
func (*ReleaseCallableRequest) ProtoMessage ¶
func (*ReleaseCallableRequest) ProtoMessage()
func (*ReleaseCallableRequest) ProtoReflect ¶
func (x *ReleaseCallableRequest) ProtoReflect() protoreflect.Message
func (*ReleaseCallableRequest) Reset ¶
func (x *ReleaseCallableRequest) Reset()
func (*ReleaseCallableRequest) String ¶
func (x *ReleaseCallableRequest) String() string
type ReleaseCallableResponse ¶
type ReleaseCallableResponse struct {
// contains filtered or unexported fields
}
func (*ReleaseCallableResponse) Descriptor
deprecated
func (*ReleaseCallableResponse) Descriptor() ([]byte, []int)
Deprecated: Use ReleaseCallableResponse.ProtoReflect.Descriptor instead.
func (*ReleaseCallableResponse) ProtoMessage ¶
func (*ReleaseCallableResponse) ProtoMessage()
func (*ReleaseCallableResponse) ProtoReflect ¶
func (x *ReleaseCallableResponse) ProtoReflect() protoreflect.Message
func (*ReleaseCallableResponse) Reset ¶
func (x *ReleaseCallableResponse) Reset()
func (*ReleaseCallableResponse) String ¶
func (x *ReleaseCallableResponse) String() string
type RemoteTensorHandle ¶
type RemoteTensorHandle struct { // The ID of the operation that produced this tensor. OpId int64 `protobuf:"varint,1,opt,name=op_id,json=opId,proto3" json:"op_id,omitempty"` // The index into the outputs of the operation that produced this tensor. OutputNum int32 `protobuf:"varint,2,opt,name=output_num,json=outputNum,proto3" json:"output_num,omitempty"` // contains filtered or unexported fields }
func (*RemoteTensorHandle) Descriptor
deprecated
func (*RemoteTensorHandle) Descriptor() ([]byte, []int)
Deprecated: Use RemoteTensorHandle.ProtoReflect.Descriptor instead.
func (*RemoteTensorHandle) GetOpId ¶
func (x *RemoteTensorHandle) GetOpId() int64
func (*RemoteTensorHandle) GetOutputNum ¶
func (x *RemoteTensorHandle) GetOutputNum() int32
func (*RemoteTensorHandle) ProtoMessage ¶
func (*RemoteTensorHandle) ProtoMessage()
func (*RemoteTensorHandle) ProtoReflect ¶
func (x *RemoteTensorHandle) ProtoReflect() protoreflect.Message
func (*RemoteTensorHandle) Reset ¶
func (x *RemoteTensorHandle) Reset()
func (*RemoteTensorHandle) String ¶
func (x *RemoteTensorHandle) String() string
type ReplayOp ¶
type ReplayOp struct { StartTimeUs float64 `protobuf:"fixed64,31,opt,name=start_time_us,json=startTimeUs,proto3" json:"start_time_us,omitempty"` EndTimeUs float64 `protobuf:"fixed64,32,opt,name=end_time_us,json=endTimeUs,proto3" json:"end_time_us,omitempty"` // Types that are assignable to Op: // *ReplayOp_CreateSession // *ReplayOp_ExtendSession // *ReplayOp_PartialRunSetup // *ReplayOp_RunStep // *ReplayOp_CloseSession // *ReplayOp_ListDevices // *ReplayOp_ResetRequest // *ReplayOp_MakeCallable // *ReplayOp_RunCallable // *ReplayOp_ReleaseCallable // *ReplayOp_NewReplaySession Op isReplayOp_Op `protobuf_oneof:"op"` // Types that are assignable to Response: // *ReplayOp_CreateSessionResponse // *ReplayOp_ExtendSessionResponse // *ReplayOp_PartialRunSetupResponse // *ReplayOp_RunStepResponse // *ReplayOp_CloseSessionResponse // *ReplayOp_ListDevicesResponse // *ReplayOp_ResetRequestResponse // *ReplayOp_MakeCallableResponse // *ReplayOp_RunCallableResponse // *ReplayOp_ReleaseCallableResponse Response isReplayOp_Response `protobuf_oneof:"response"` // contains filtered or unexported fields }
func (*ReplayOp) Descriptor
deprecated
func (*ReplayOp) GetCloseSession ¶
func (x *ReplayOp) GetCloseSession() *CloseSessionRequest
func (*ReplayOp) GetCloseSessionResponse ¶
func (x *ReplayOp) GetCloseSessionResponse() *CloseSessionResponse
func (*ReplayOp) GetCreateSession ¶
func (x *ReplayOp) GetCreateSession() *CreateSessionRequest
func (*ReplayOp) GetCreateSessionResponse ¶
func (x *ReplayOp) GetCreateSessionResponse() *CreateSessionResponse
func (*ReplayOp) GetEndTimeUs ¶
func (*ReplayOp) GetExtendSession ¶
func (x *ReplayOp) GetExtendSession() *ExtendSessionRequest
func (*ReplayOp) GetExtendSessionResponse ¶
func (x *ReplayOp) GetExtendSessionResponse() *ExtendSessionResponse
func (*ReplayOp) GetListDevices ¶
func (x *ReplayOp) GetListDevices() *ListDevicesRequest
func (*ReplayOp) GetListDevicesResponse ¶
func (x *ReplayOp) GetListDevicesResponse() *ListDevicesResponse
func (*ReplayOp) GetMakeCallable ¶
func (x *ReplayOp) GetMakeCallable() *MakeCallableRequest
func (*ReplayOp) GetMakeCallableResponse ¶
func (x *ReplayOp) GetMakeCallableResponse() *MakeCallableResponse
func (*ReplayOp) GetNewReplaySession ¶
func (x *ReplayOp) GetNewReplaySession() *NewReplaySession
func (*ReplayOp) GetPartialRunSetup ¶
func (x *ReplayOp) GetPartialRunSetup() *PartialRunSetupRequest
func (*ReplayOp) GetPartialRunSetupResponse ¶
func (x *ReplayOp) GetPartialRunSetupResponse() *PartialRunSetupResponse
func (*ReplayOp) GetReleaseCallable ¶
func (x *ReplayOp) GetReleaseCallable() *ReleaseCallableRequest
func (*ReplayOp) GetReleaseCallableResponse ¶
func (x *ReplayOp) GetReleaseCallableResponse() *ReleaseCallableResponse
func (*ReplayOp) GetResetRequest ¶
func (x *ReplayOp) GetResetRequest() *ResetRequest
func (*ReplayOp) GetResetRequestResponse ¶
func (x *ReplayOp) GetResetRequestResponse() *ResetResponse
func (*ReplayOp) GetResponse ¶
func (m *ReplayOp) GetResponse() isReplayOp_Response
func (*ReplayOp) GetRunCallable ¶
func (x *ReplayOp) GetRunCallable() *RunCallableRequest
func (*ReplayOp) GetRunCallableResponse ¶
func (x *ReplayOp) GetRunCallableResponse() *RunCallableResponse
func (*ReplayOp) GetRunStep ¶
func (x *ReplayOp) GetRunStep() *RunStepRequest
func (*ReplayOp) GetRunStepResponse ¶
func (x *ReplayOp) GetRunStepResponse() *RunStepResponse
func (*ReplayOp) GetStartTimeUs ¶
func (*ReplayOp) ProtoMessage ¶
func (*ReplayOp) ProtoMessage()
func (*ReplayOp) ProtoReflect ¶
func (x *ReplayOp) ProtoReflect() protoreflect.Message
type ReplayOp_CloseSession ¶
type ReplayOp_CloseSession struct {
CloseSession *CloseSessionRequest `protobuf:"bytes,5,opt,name=close_session,json=closeSession,proto3,oneof"`
}
type ReplayOp_CloseSessionResponse ¶
type ReplayOp_CloseSessionResponse struct {
CloseSessionResponse *CloseSessionResponse `protobuf:"bytes,25,opt,name=close_session_response,json=closeSessionResponse,proto3,oneof"`
}
type ReplayOp_CreateSession ¶
type ReplayOp_CreateSession struct {
CreateSession *CreateSessionRequest `protobuf:"bytes,1,opt,name=create_session,json=createSession,proto3,oneof"`
}
type ReplayOp_CreateSessionResponse ¶
type ReplayOp_CreateSessionResponse struct {
CreateSessionResponse *CreateSessionResponse `protobuf:"bytes,21,opt,name=create_session_response,json=createSessionResponse,proto3,oneof"`
}
type ReplayOp_ExtendSession ¶
type ReplayOp_ExtendSession struct {
ExtendSession *ExtendSessionRequest `protobuf:"bytes,2,opt,name=extend_session,json=extendSession,proto3,oneof"`
}
type ReplayOp_ExtendSessionResponse ¶
type ReplayOp_ExtendSessionResponse struct {
ExtendSessionResponse *ExtendSessionResponse `protobuf:"bytes,22,opt,name=extend_session_response,json=extendSessionResponse,proto3,oneof"`
}
type ReplayOp_ListDevices ¶
type ReplayOp_ListDevices struct {
ListDevices *ListDevicesRequest `protobuf:"bytes,6,opt,name=list_devices,json=listDevices,proto3,oneof"`
}
type ReplayOp_ListDevicesResponse ¶
type ReplayOp_ListDevicesResponse struct {
ListDevicesResponse *ListDevicesResponse `protobuf:"bytes,26,opt,name=list_devices_response,json=listDevicesResponse,proto3,oneof"`
}
type ReplayOp_MakeCallable ¶
type ReplayOp_MakeCallable struct {
MakeCallable *MakeCallableRequest `protobuf:"bytes,8,opt,name=make_callable,json=makeCallable,proto3,oneof"`
}
type ReplayOp_MakeCallableResponse ¶
type ReplayOp_MakeCallableResponse struct {
MakeCallableResponse *MakeCallableResponse `protobuf:"bytes,28,opt,name=make_callable_response,json=makeCallableResponse,proto3,oneof"`
}
type ReplayOp_NewReplaySession ¶
type ReplayOp_NewReplaySession struct {
NewReplaySession *NewReplaySession `protobuf:"bytes,11,opt,name=new_replay_session,json=newReplaySession,proto3,oneof"`
}
type ReplayOp_PartialRunSetup ¶
type ReplayOp_PartialRunSetup struct {
PartialRunSetup *PartialRunSetupRequest `protobuf:"bytes,3,opt,name=partial_run_setup,json=partialRunSetup,proto3,oneof"`
}
type ReplayOp_PartialRunSetupResponse ¶
type ReplayOp_PartialRunSetupResponse struct {
PartialRunSetupResponse *PartialRunSetupResponse `protobuf:"bytes,23,opt,name=partial_run_setup_response,json=partialRunSetupResponse,proto3,oneof"`
}
type ReplayOp_ReleaseCallable ¶
type ReplayOp_ReleaseCallable struct {
ReleaseCallable *ReleaseCallableRequest `protobuf:"bytes,10,opt,name=release_callable,json=releaseCallable,proto3,oneof"`
}
type ReplayOp_ReleaseCallableResponse ¶
type ReplayOp_ReleaseCallableResponse struct {
ReleaseCallableResponse *ReleaseCallableResponse `protobuf:"bytes,30,opt,name=release_callable_response,json=releaseCallableResponse,proto3,oneof"`
}
type ReplayOp_ResetRequest ¶
type ReplayOp_ResetRequest struct {
ResetRequest *ResetRequest `protobuf:"bytes,7,opt,name=reset_request,json=resetRequest,proto3,oneof"`
}
type ReplayOp_ResetRequestResponse ¶
type ReplayOp_ResetRequestResponse struct {
ResetRequestResponse *ResetResponse `protobuf:"bytes,27,opt,name=reset_request_response,json=resetRequestResponse,proto3,oneof"`
}
type ReplayOp_RunCallable ¶
type ReplayOp_RunCallable struct {
RunCallable *RunCallableRequest `protobuf:"bytes,9,opt,name=run_callable,json=runCallable,proto3,oneof"`
}
type ReplayOp_RunCallableResponse ¶
type ReplayOp_RunCallableResponse struct {
RunCallableResponse *RunCallableResponse `protobuf:"bytes,29,opt,name=run_callable_response,json=runCallableResponse,proto3,oneof"`
}
type ReplayOp_RunStep ¶
type ReplayOp_RunStep struct {
RunStep *RunStepRequest `protobuf:"bytes,4,opt,name=run_step,json=runStep,proto3,oneof"`
}
type ReplayOp_RunStepResponse ¶
type ReplayOp_RunStepResponse struct {
RunStepResponse *RunStepResponse `protobuf:"bytes,24,opt,name=run_step_response,json=runStepResponse,proto3,oneof"`
}
type ResetRequest ¶
type ResetRequest struct { // A list of container names, which may be empty. // // If 'container' is not empty, releases resources in the given // containers in all devices. // // If 'container' is empty, releases resources in the default // container in all devices. Container []string `protobuf:"bytes,1,rep,name=container,proto3" json:"container,omitempty"` // When any filters are present, only devices that match the filters // will be reset. Each filter can be partially specified, // e.g. "/job:ps" "/job:worker/replica:3", etc. DeviceFilters []string `protobuf:"bytes,2,rep,name=device_filters,json=deviceFilters,proto3" json:"device_filters,omitempty"` // contains filtered or unexported fields }
Reset() allows misbehaving or slow sessions to be aborted and closed, and causes their resources eventually to be released. Reset() does not wait for the computations in old sessions to cease; it merely starts the process of tearing them down. However, if a new session is started after a Reset(), the new session is isolated from changes that old sessions (started prior to the Reset()) may continue to make to resources, provided all those resources are in containers listed in "containers".
Old sessions may continue to have side-effects on resources not in containers listed in "containers", and thus may affect future sessions' results in ways that are hard to predict. Thus, if well-defined behavior is desired, is it recommended that all containers be listed in "containers". Similarly, if a device_filter is specified, results may be hard to predict.
func (*ResetRequest) Descriptor
deprecated
func (*ResetRequest) Descriptor() ([]byte, []int)
Deprecated: Use ResetRequest.ProtoReflect.Descriptor instead.
func (*ResetRequest) GetContainer ¶
func (x *ResetRequest) GetContainer() []string
func (*ResetRequest) GetDeviceFilters ¶
func (x *ResetRequest) GetDeviceFilters() []string
func (*ResetRequest) ProtoMessage ¶
func (*ResetRequest) ProtoMessage()
func (*ResetRequest) ProtoReflect ¶
func (x *ResetRequest) ProtoReflect() protoreflect.Message
func (*ResetRequest) Reset ¶
func (x *ResetRequest) Reset()
func (*ResetRequest) String ¶
func (x *ResetRequest) String() string
type ResetResponse ¶
type ResetResponse struct {
// contains filtered or unexported fields
}
func (*ResetResponse) Descriptor
deprecated
func (*ResetResponse) Descriptor() ([]byte, []int)
Deprecated: Use ResetResponse.ProtoReflect.Descriptor instead.
func (*ResetResponse) ProtoMessage ¶
func (*ResetResponse) ProtoMessage()
func (*ResetResponse) ProtoReflect ¶
func (x *ResetResponse) ProtoReflect() protoreflect.Message
func (*ResetResponse) Reset ¶
func (x *ResetResponse) Reset()
func (*ResetResponse) String ¶
func (x *ResetResponse) String() string
type RewriterConfig ¶
type RewriterConfig struct { // Optimize tensor layouts (default is ON) // e.g. This will try to use NCHW layout on GPU which is faster. LayoutOptimizer RewriterConfig_Toggle `` /* 145-byte string literal not displayed */ // Fold constants (default is ON) // Statically infer the value of tensors when possible, and materialize the // result using constants. ConstantFolding RewriterConfig_Toggle `` /* 145-byte string literal not displayed */ // Shape optimizations (default is ON) // Simplify computations made on shapes. ShapeOptimization RewriterConfig_Toggle `` /* 152-byte string literal not displayed */ // Remapping (default is ON) // Remap subgraphs onto more efficient implementations. Remapping RewriterConfig_Toggle `protobuf:"varint,14,opt,name=remapping,proto3,enum=tensorflow.RewriterConfig_Toggle" json:"remapping,omitempty"` // Arithmetic optimizations (default is ON) // e.g. Simplify arithmetic ops; merge ops with same value (like constants). ArithmeticOptimization RewriterConfig_Toggle `` /* 166-byte string literal not displayed */ // Control dependency optimizations (default is ON). // Remove redundant control dependencies, which may enable other optimization. DependencyOptimization RewriterConfig_Toggle `` /* 166-byte string literal not displayed */ // Loop optimizations (default is ON). LoopOptimization RewriterConfig_Toggle `` /* 148-byte string literal not displayed */ // Function optimizations (default is ON). FunctionOptimization RewriterConfig_Toggle `` /* 161-byte string literal not displayed */ // Strips debug-related nodes from the graph (off by default). DebugStripper RewriterConfig_Toggle `` /* 140-byte string literal not displayed */ // If true, don't remove unnecessary ops from the graph DisableModelPruning bool `protobuf:"varint,2,opt,name=disable_model_pruning,json=disableModelPruning,proto3" json:"disable_model_pruning,omitempty"` // Try to allocate some independent Op outputs contiguously in order to // merge or eliminate downstream Ops (off by default). ScopedAllocatorOptimization RewriterConfig_Toggle `` /* 184-byte string literal not displayed */ // Force small ops onto the CPU (default is OFF). PinToHostOptimization RewriterConfig_Toggle `` /* 168-byte string literal not displayed */ // Disable the entire meta optimizer (off by default). DisableMetaOptimizer bool `protobuf:"varint,19,opt,name=disable_meta_optimizer,json=disableMetaOptimizer,proto3" json:"disable_meta_optimizer,omitempty"` // Controls how many times we run the optimizers in meta optimizer (default // is once). MetaOptimizerIterations RewriterConfig_NumIterationsType `` /* 183-byte string literal not displayed */ // The minimum number of nodes in a graph to optimizer. For smaller graphs, // optimization is skipped. // 0 means the system picks an appropriate number. // < 0 means do not skip optimization. MinGraphNodes int32 `protobuf:"varint,17,opt,name=min_graph_nodes,json=minGraphNodes,proto3" json:"min_graph_nodes,omitempty"` // Configures memory optimization passes through the meta-optimizer. Has no // effect on manually requested memory optimization passes in the optimizers // field. MemoryOptimization RewriterConfig_MemOptType `` /* 158-byte string literal not displayed */ // A node name scope for node names which are valid outputs of recompuations. // Inputs to nodes that match this scope may be recomputed (subject either to // manual annotation of those input nodes or to manual annotation and // heuristics depending on memory_optimization), but the nodes themselves will // not be recomputed. This matches any sub-scopes as well, meaning the scope // can appear not just as a top-level scope. For example, if the value is // "gradients/", the default, it will match node name "gradients/foo", // "foo/gradients/bar", but not "foo_gradients/" MemoryOptimizerTargetNodeNameScope string `` /* 171-byte string literal not displayed */ // Maximum number of milliseconds to spend optimizing a single graph before // timing out. If equal to 0 the system picks a default (currently 5 minutes). // If less than 0 the optimizer will never time out. MetaOptimizerTimeoutMs int64 `` /* 133-byte string literal not displayed */ // Configures AutoParallel optimization passes either through the // meta-optimizer or when manually specified through the optimizers field. AutoParallel *AutoParallelOptions `protobuf:"bytes,5,opt,name=auto_parallel,json=autoParallel,proto3" json:"auto_parallel,omitempty"` // If true, any optimization pass failing will cause the MetaOptimizer to // stop with an error. By default - or when set to false, failing passes are // skipped silently. FailOnOptimizerErrors bool `` /* 130-byte string literal not displayed */ ScopedAllocatorOpts *ScopedAllocatorOptions `protobuf:"bytes,16,opt,name=scoped_allocator_opts,json=scopedAllocatorOpts,proto3" json:"scoped_allocator_opts,omitempty"` // If non-empty, will use this as an alternative way to specify a list of // optimizations to turn on and the order of the optimizations (replacing the // meta-optimizer). // // Of the RewriterConfig options, only the AutoParallel configuration options // (the auto_parallel field) apply to manually requested optimization passes // ("autoparallel"). Memory optimization passes ("memory") invoked here are // not configurable (in contrast to memory optimization passes through the // meta-optimizer) and act only on manual op annotations. // // Custom optimizers (see custom_optimizers) that are not part of this // schedule will be run after - in the order that they were specified. Optimizers []string `protobuf:"bytes,100,rep,name=optimizers,proto3" json:"optimizers,omitempty"` // list of CustomGraphOptimizers to apply. CustomOptimizers []*RewriterConfig_CustomGraphOptimizer `protobuf:"bytes,200,rep,name=custom_optimizers,json=customOptimizers,proto3" json:"custom_optimizers,omitempty"` // contains filtered or unexported fields }
func (*RewriterConfig) Descriptor
deprecated
func (*RewriterConfig) Descriptor() ([]byte, []int)
Deprecated: Use RewriterConfig.ProtoReflect.Descriptor instead.
func (*RewriterConfig) GetArithmeticOptimization ¶
func (x *RewriterConfig) GetArithmeticOptimization() RewriterConfig_Toggle
func (*RewriterConfig) GetAutoParallel ¶
func (x *RewriterConfig) GetAutoParallel() *AutoParallelOptions
func (*RewriterConfig) GetConstantFolding ¶
func (x *RewriterConfig) GetConstantFolding() RewriterConfig_Toggle
func (*RewriterConfig) GetCustomOptimizers ¶
func (x *RewriterConfig) GetCustomOptimizers() []*RewriterConfig_CustomGraphOptimizer
func (*RewriterConfig) GetDebugStripper ¶
func (x *RewriterConfig) GetDebugStripper() RewriterConfig_Toggle
func (*RewriterConfig) GetDependencyOptimization ¶
func (x *RewriterConfig) GetDependencyOptimization() RewriterConfig_Toggle
func (*RewriterConfig) GetDisableMetaOptimizer ¶
func (x *RewriterConfig) GetDisableMetaOptimizer() bool
func (*RewriterConfig) GetDisableModelPruning ¶
func (x *RewriterConfig) GetDisableModelPruning() bool
func (*RewriterConfig) GetFailOnOptimizerErrors ¶
func (x *RewriterConfig) GetFailOnOptimizerErrors() bool
func (*RewriterConfig) GetFunctionOptimization ¶
func (x *RewriterConfig) GetFunctionOptimization() RewriterConfig_Toggle
func (*RewriterConfig) GetLayoutOptimizer ¶
func (x *RewriterConfig) GetLayoutOptimizer() RewriterConfig_Toggle
func (*RewriterConfig) GetLoopOptimization ¶
func (x *RewriterConfig) GetLoopOptimization() RewriterConfig_Toggle
func (*RewriterConfig) GetMemoryOptimization ¶
func (x *RewriterConfig) GetMemoryOptimization() RewriterConfig_MemOptType
func (*RewriterConfig) GetMemoryOptimizerTargetNodeNameScope ¶
func (x *RewriterConfig) GetMemoryOptimizerTargetNodeNameScope() string
func (*RewriterConfig) GetMetaOptimizerIterations ¶
func (x *RewriterConfig) GetMetaOptimizerIterations() RewriterConfig_NumIterationsType
func (*RewriterConfig) GetMetaOptimizerTimeoutMs ¶
func (x *RewriterConfig) GetMetaOptimizerTimeoutMs() int64
func (*RewriterConfig) GetMinGraphNodes ¶
func (x *RewriterConfig) GetMinGraphNodes() int32
func (*RewriterConfig) GetOptimizers ¶
func (x *RewriterConfig) GetOptimizers() []string
func (*RewriterConfig) GetPinToHostOptimization ¶
func (x *RewriterConfig) GetPinToHostOptimization() RewriterConfig_Toggle
func (*RewriterConfig) GetRemapping ¶
func (x *RewriterConfig) GetRemapping() RewriterConfig_Toggle
func (*RewriterConfig) GetScopedAllocatorOptimization ¶
func (x *RewriterConfig) GetScopedAllocatorOptimization() RewriterConfig_Toggle
func (*RewriterConfig) GetScopedAllocatorOpts ¶
func (x *RewriterConfig) GetScopedAllocatorOpts() *ScopedAllocatorOptions
func (*RewriterConfig) GetShapeOptimization ¶
func (x *RewriterConfig) GetShapeOptimization() RewriterConfig_Toggle
func (*RewriterConfig) ProtoMessage ¶
func (*RewriterConfig) ProtoMessage()
func (*RewriterConfig) ProtoReflect ¶
func (x *RewriterConfig) ProtoReflect() protoreflect.Message
func (*RewriterConfig) Reset ¶
func (x *RewriterConfig) Reset()
func (*RewriterConfig) String ¶
func (x *RewriterConfig) String() string
type RewriterConfig_CustomGraphOptimizer ¶
type RewriterConfig_CustomGraphOptimizer struct { Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` ParameterMap map[string]*framework.AttrValue `` /* 185-byte string literal not displayed */ // contains filtered or unexported fields }
Message to describe custom graph optimizer and its parameters
func (*RewriterConfig_CustomGraphOptimizer) Descriptor
deprecated
func (*RewriterConfig_CustomGraphOptimizer) Descriptor() ([]byte, []int)
Deprecated: Use RewriterConfig_CustomGraphOptimizer.ProtoReflect.Descriptor instead.
func (*RewriterConfig_CustomGraphOptimizer) GetName ¶
func (x *RewriterConfig_CustomGraphOptimizer) GetName() string
func (*RewriterConfig_CustomGraphOptimizer) GetParameterMap ¶
func (x *RewriterConfig_CustomGraphOptimizer) GetParameterMap() map[string]*framework.AttrValue
func (*RewriterConfig_CustomGraphOptimizer) ProtoMessage ¶
func (*RewriterConfig_CustomGraphOptimizer) ProtoMessage()
func (*RewriterConfig_CustomGraphOptimizer) ProtoReflect ¶
func (x *RewriterConfig_CustomGraphOptimizer) ProtoReflect() protoreflect.Message
func (*RewriterConfig_CustomGraphOptimizer) Reset ¶
func (x *RewriterConfig_CustomGraphOptimizer) Reset()
func (*RewriterConfig_CustomGraphOptimizer) String ¶
func (x *RewriterConfig_CustomGraphOptimizer) String() string
type RewriterConfig_MemOptType ¶
type RewriterConfig_MemOptType int32
const ( // The default setting (SCHEDULING and SWAPPING HEURISTICS only) RewriterConfig_DEFAULT_MEM_OPT RewriterConfig_MemOptType = 0 // Disabled in the meta-optimizer. RewriterConfig_NO_MEM_OPT RewriterConfig_MemOptType = 1 // Driven by manual op-level annotations. RewriterConfig_MANUAL RewriterConfig_MemOptType = 2 // Swapping heuristic will move a tensor from the GPU to the CPU and move // it back when needed to reduce peak memory usage. RewriterConfig_SWAPPING_HEURISTICS RewriterConfig_MemOptType = 4 // Recomputation heuristics will recompute ops (such as Relu activation) // during backprop instead of storing them, reducing peak memory usage. RewriterConfig_RECOMPUTATION_HEURISTICS RewriterConfig_MemOptType = 5 // Scheduling will split big ops such as AddN and try to enforce a schedule // of the new computations that decreases peak memory usage. RewriterConfig_SCHEDULING_HEURISTICS RewriterConfig_MemOptType = 6 // Use any combination of swapping and recomputation heuristics. RewriterConfig_HEURISTICS RewriterConfig_MemOptType = 3 )
func (RewriterConfig_MemOptType) Descriptor ¶
func (RewriterConfig_MemOptType) Descriptor() protoreflect.EnumDescriptor
func (RewriterConfig_MemOptType) Enum ¶
func (x RewriterConfig_MemOptType) Enum() *RewriterConfig_MemOptType
func (RewriterConfig_MemOptType) EnumDescriptor
deprecated
func (RewriterConfig_MemOptType) EnumDescriptor() ([]byte, []int)
Deprecated: Use RewriterConfig_MemOptType.Descriptor instead.
func (RewriterConfig_MemOptType) Number ¶
func (x RewriterConfig_MemOptType) Number() protoreflect.EnumNumber
func (RewriterConfig_MemOptType) String ¶
func (x RewriterConfig_MemOptType) String() string
func (RewriterConfig_MemOptType) Type ¶
func (RewriterConfig_MemOptType) Type() protoreflect.EnumType
type RewriterConfig_NumIterationsType ¶
type RewriterConfig_NumIterationsType int32
Enum controlling the number of times to run optimizers. The default is to run them twice.
const ( RewriterConfig_DEFAULT_NUM_ITERS RewriterConfig_NumIterationsType = 0 RewriterConfig_ONE RewriterConfig_NumIterationsType = 1 RewriterConfig_TWO RewriterConfig_NumIterationsType = 2 )
func (RewriterConfig_NumIterationsType) Descriptor ¶
func (RewriterConfig_NumIterationsType) Descriptor() protoreflect.EnumDescriptor
func (RewriterConfig_NumIterationsType) Enum ¶
func (x RewriterConfig_NumIterationsType) Enum() *RewriterConfig_NumIterationsType
func (RewriterConfig_NumIterationsType) EnumDescriptor
deprecated
func (RewriterConfig_NumIterationsType) EnumDescriptor() ([]byte, []int)
Deprecated: Use RewriterConfig_NumIterationsType.Descriptor instead.
func (RewriterConfig_NumIterationsType) Number ¶
func (x RewriterConfig_NumIterationsType) Number() protoreflect.EnumNumber
func (RewriterConfig_NumIterationsType) String ¶
func (x RewriterConfig_NumIterationsType) String() string
func (RewriterConfig_NumIterationsType) Type ¶
func (RewriterConfig_NumIterationsType) Type() protoreflect.EnumType
type RewriterConfig_Toggle ¶
type RewriterConfig_Toggle int32
const ( RewriterConfig_DEFAULT RewriterConfig_Toggle = 0 RewriterConfig_ON RewriterConfig_Toggle = 1 RewriterConfig_OFF RewriterConfig_Toggle = 2 // Enable some aggressive optimizations that use assumptions that TF graphs // may break. For example, assume the shape of a placeholder matches its // actual feed. RewriterConfig_AGGRESSIVE RewriterConfig_Toggle = 3 )
func (RewriterConfig_Toggle) Descriptor ¶
func (RewriterConfig_Toggle) Descriptor() protoreflect.EnumDescriptor
func (RewriterConfig_Toggle) Enum ¶
func (x RewriterConfig_Toggle) Enum() *RewriterConfig_Toggle
func (RewriterConfig_Toggle) EnumDescriptor
deprecated
func (RewriterConfig_Toggle) EnumDescriptor() ([]byte, []int)
Deprecated: Use RewriterConfig_Toggle.Descriptor instead.
func (RewriterConfig_Toggle) Number ¶
func (x RewriterConfig_Toggle) Number() protoreflect.EnumNumber
func (RewriterConfig_Toggle) String ¶
func (x RewriterConfig_Toggle) String() string
func (RewriterConfig_Toggle) Type ¶
func (RewriterConfig_Toggle) Type() protoreflect.EnumType
type RunCallableRequest ¶
type RunCallableRequest struct { // REQUIRED: session_handle must be returned by a CreateSession call // to the same master service. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // REQUIRED: handle must be returned by a MakeCallable call to the same // master service. Handle int64 `protobuf:"varint,2,opt,name=handle,proto3" json:"handle,omitempty"` // Values of the tensors passed as arguments to the callable, in the order // defined in the CallableOptions.feed field passed to MakeCallable. Feed []*framework.TensorProto `protobuf:"bytes,3,rep,name=feed,proto3" json:"feed,omitempty"` // contains filtered or unexported fields }
func (*RunCallableRequest) Descriptor
deprecated
func (*RunCallableRequest) Descriptor() ([]byte, []int)
Deprecated: Use RunCallableRequest.ProtoReflect.Descriptor instead.
func (*RunCallableRequest) GetFeed ¶
func (x *RunCallableRequest) GetFeed() []*framework.TensorProto
func (*RunCallableRequest) GetHandle ¶
func (x *RunCallableRequest) GetHandle() int64
func (*RunCallableRequest) GetSessionHandle ¶
func (x *RunCallableRequest) GetSessionHandle() string
func (*RunCallableRequest) ProtoMessage ¶
func (*RunCallableRequest) ProtoMessage()
func (*RunCallableRequest) ProtoReflect ¶
func (x *RunCallableRequest) ProtoReflect() protoreflect.Message
func (*RunCallableRequest) Reset ¶
func (x *RunCallableRequest) Reset()
func (*RunCallableRequest) String ¶
func (x *RunCallableRequest) String() string
type RunCallableResponse ¶
type RunCallableResponse struct { // Values of the tensors returned by the callable, in the order defined in the // CallableOptions.fetch field passed to MakeCallable. Fetch []*framework.TensorProto `protobuf:"bytes,1,rep,name=fetch,proto3" json:"fetch,omitempty"` // Returned metadata if requested in the options. Metadata *RunMetadata `protobuf:"bytes,2,opt,name=metadata,proto3" json:"metadata,omitempty"` // contains filtered or unexported fields }
func (*RunCallableResponse) Descriptor
deprecated
func (*RunCallableResponse) Descriptor() ([]byte, []int)
Deprecated: Use RunCallableResponse.ProtoReflect.Descriptor instead.
func (*RunCallableResponse) GetFetch ¶
func (x *RunCallableResponse) GetFetch() []*framework.TensorProto
func (*RunCallableResponse) GetMetadata ¶
func (x *RunCallableResponse) GetMetadata() *RunMetadata
func (*RunCallableResponse) ProtoMessage ¶
func (*RunCallableResponse) ProtoMessage()
func (*RunCallableResponse) ProtoReflect ¶
func (x *RunCallableResponse) ProtoReflect() protoreflect.Message
func (*RunCallableResponse) Reset ¶
func (x *RunCallableResponse) Reset()
func (*RunCallableResponse) String ¶
func (x *RunCallableResponse) String() string
type RunGraphRequest ¶
type RunGraphRequest struct { // session_handle is the master-generated unique id for this session. // If session_handle is non-empty, it must be the same as used when // registering the graph. If it is empty, a single global namespace is used to // search for the graph_handle. SessionHandle string `protobuf:"bytes,8,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // Set to true if `CreateWorkerSession` was called for `session_handle`. CreateWorkerSessionCalled bool `` /* 142-byte string literal not displayed */ // REQUIRED: graph_handle must be returned by a RegisterGraph call // to the same WorkerService. GraphHandle string `protobuf:"bytes,1,opt,name=graph_handle,json=graphHandle,proto3" json:"graph_handle,omitempty"` // A unique ID to distinguish different runs of the same graph. // // The master generates a global unique `step_id` to distinguish // different runs of the graph computation. Subgraphs communicate // (e.g., send/recv ops) with each other using `step_id` to // distinguish tensors generated by different runs. StepId int64 `protobuf:"varint,2,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"` // Options for this step. ExecOpts *ExecutorOpts `protobuf:"bytes,5,opt,name=exec_opts,json=execOpts,proto3" json:"exec_opts,omitempty"` // Runs the graph. // // Sends the tensors in "send" into the graph before the run and // fetches the keys into `RunGraphResponse.recv` after the run. Send []*NamedTensorProto `protobuf:"bytes,3,rep,name=send,proto3" json:"send,omitempty"` RecvKey []string `protobuf:"bytes,4,rep,name=recv_key,json=recvKey,proto3" json:"recv_key,omitempty"` // True if the RunGraphRequest is a partial run request. IsPartial bool `protobuf:"varint,6,opt,name=is_partial,json=isPartial,proto3" json:"is_partial,omitempty"` // True if this is the last partial run request in a sequence of requests. IsLastPartialRun bool `protobuf:"varint,7,opt,name=is_last_partial_run,json=isLastPartialRun,proto3" json:"is_last_partial_run,omitempty"` // If true then some errors, e.g., execution errors that have long // error messages, may return an OK RunGraphResponse with the actual // error saved in the status_code/status_error_message fields of the // response body. This is a workaround since the RPC subsystem may // truncate long metadata messages. StoreErrorsInResponseBody bool `` /* 143-byte string literal not displayed */ // contains filtered or unexported fields }
func (*RunGraphRequest) Descriptor
deprecated
func (*RunGraphRequest) Descriptor() ([]byte, []int)
Deprecated: Use RunGraphRequest.ProtoReflect.Descriptor instead.
func (*RunGraphRequest) GetCreateWorkerSessionCalled ¶
func (x *RunGraphRequest) GetCreateWorkerSessionCalled() bool
func (*RunGraphRequest) GetExecOpts ¶
func (x *RunGraphRequest) GetExecOpts() *ExecutorOpts
func (*RunGraphRequest) GetGraphHandle ¶
func (x *RunGraphRequest) GetGraphHandle() string
func (*RunGraphRequest) GetIsLastPartialRun ¶
func (x *RunGraphRequest) GetIsLastPartialRun() bool
func (*RunGraphRequest) GetIsPartial ¶
func (x *RunGraphRequest) GetIsPartial() bool
func (*RunGraphRequest) GetRecvKey ¶
func (x *RunGraphRequest) GetRecvKey() []string
func (*RunGraphRequest) GetSend ¶
func (x *RunGraphRequest) GetSend() []*NamedTensorProto
func (*RunGraphRequest) GetSessionHandle ¶
func (x *RunGraphRequest) GetSessionHandle() string
func (*RunGraphRequest) GetStepId ¶
func (x *RunGraphRequest) GetStepId() int64
func (*RunGraphRequest) GetStoreErrorsInResponseBody ¶
func (x *RunGraphRequest) GetStoreErrorsInResponseBody() bool
func (*RunGraphRequest) ProtoMessage ¶
func (*RunGraphRequest) ProtoMessage()
func (*RunGraphRequest) ProtoReflect ¶
func (x *RunGraphRequest) ProtoReflect() protoreflect.Message
func (*RunGraphRequest) Reset ¶
func (x *RunGraphRequest) Reset()
func (*RunGraphRequest) String ¶
func (x *RunGraphRequest) String() string
type RunGraphResponse ¶
type RunGraphResponse struct { // A list of tensors corresponding to those requested by // `RunGraphRequest.recv_key`. Recv []*NamedTensorProto `protobuf:"bytes,1,rep,name=recv,proto3" json:"recv,omitempty"` // If the request asked for execution stats, the cost graph, or the partition // graphs, these are returned here. // TODO(suharshs): Package these in a RunMetadata instead. StepStats *framework.StepStats `protobuf:"bytes,2,opt,name=step_stats,json=stepStats,proto3" json:"step_stats,omitempty"` CostGraph *framework.CostGraphDef `protobuf:"bytes,3,opt,name=cost_graph,json=costGraph,proto3" json:"cost_graph,omitempty"` PartitionGraph []*framework.GraphDef `protobuf:"bytes,4,rep,name=partition_graph,json=partitionGraph,proto3" json:"partition_graph,omitempty"` // If store_errors_in_response_body is true in the request, then // optionally the server may return an OK status for the RPC and // fill the true status into the fields below, to allow for messages // that are too long to fit in metadata. StatusCode core.Code `protobuf:"varint,5,opt,name=status_code,json=statusCode,proto3,enum=tensorflow.error.Code" json:"status_code,omitempty"` StatusErrorMessage string `protobuf:"bytes,6,opt,name=status_error_message,json=statusErrorMessage,proto3" json:"status_error_message,omitempty"` // contains filtered or unexported fields }
func (*RunGraphResponse) Descriptor
deprecated
func (*RunGraphResponse) Descriptor() ([]byte, []int)
Deprecated: Use RunGraphResponse.ProtoReflect.Descriptor instead.
func (*RunGraphResponse) GetCostGraph ¶
func (x *RunGraphResponse) GetCostGraph() *framework.CostGraphDef
func (*RunGraphResponse) GetPartitionGraph ¶
func (x *RunGraphResponse) GetPartitionGraph() []*framework.GraphDef
func (*RunGraphResponse) GetRecv ¶
func (x *RunGraphResponse) GetRecv() []*NamedTensorProto
func (*RunGraphResponse) GetStatusCode ¶
func (x *RunGraphResponse) GetStatusCode() core.Code
func (*RunGraphResponse) GetStatusErrorMessage ¶
func (x *RunGraphResponse) GetStatusErrorMessage() string
func (*RunGraphResponse) GetStepStats ¶
func (x *RunGraphResponse) GetStepStats() *framework.StepStats
func (*RunGraphResponse) ProtoMessage ¶
func (*RunGraphResponse) ProtoMessage()
func (*RunGraphResponse) ProtoReflect ¶
func (x *RunGraphResponse) ProtoReflect() protoreflect.Message
func (*RunGraphResponse) Reset ¶
func (x *RunGraphResponse) Reset()
func (*RunGraphResponse) String ¶
func (x *RunGraphResponse) String() string
type RunMetadata ¶
type RunMetadata struct { // Statistics traced for this step. Populated if tracing is turned on via the // "RunOptions" proto. // EXPERIMENTAL: The format and set of events may change in future versions. StepStats *framework.StepStats `protobuf:"bytes,1,opt,name=step_stats,json=stepStats,proto3" json:"step_stats,omitempty"` // The cost graph for the computation defined by the run call. CostGraph *framework.CostGraphDef `protobuf:"bytes,2,opt,name=cost_graph,json=costGraph,proto3" json:"cost_graph,omitempty"` // Graphs of the partitions executed by executors. PartitionGraphs []*framework.GraphDef `protobuf:"bytes,3,rep,name=partition_graphs,json=partitionGraphs,proto3" json:"partition_graphs,omitempty"` // contains filtered or unexported fields }
Metadata output (i.e., non-Tensor) for a single Run() call.
func (*RunMetadata) Descriptor
deprecated
func (*RunMetadata) Descriptor() ([]byte, []int)
Deprecated: Use RunMetadata.ProtoReflect.Descriptor instead.
func (*RunMetadata) GetCostGraph ¶
func (x *RunMetadata) GetCostGraph() *framework.CostGraphDef
func (*RunMetadata) GetPartitionGraphs ¶
func (x *RunMetadata) GetPartitionGraphs() []*framework.GraphDef
func (*RunMetadata) GetStepStats ¶
func (x *RunMetadata) GetStepStats() *framework.StepStats
func (*RunMetadata) ProtoMessage ¶
func (*RunMetadata) ProtoMessage()
func (*RunMetadata) ProtoReflect ¶
func (x *RunMetadata) ProtoReflect() protoreflect.Message
func (*RunMetadata) Reset ¶
func (x *RunMetadata) Reset()
func (*RunMetadata) String ¶
func (x *RunMetadata) String() string
type RunOptions ¶
type RunOptions struct { TraceLevel RunOptions_TraceLevel `` /* 130-byte string literal not displayed */ // Time to wait for operation to complete in milliseconds. TimeoutInMs int64 `protobuf:"varint,2,opt,name=timeout_in_ms,json=timeoutInMs,proto3" json:"timeout_in_ms,omitempty"` // The thread pool to use, if session_inter_op_thread_pool is configured. // To use the caller thread set this to -1 - this uses the caller thread // to execute Session::Run() and thus avoids a context switch. Using the // caller thread to execute Session::Run() should be done ONLY for simple // graphs, where the overhead of an additional context switch is // comparable with the overhead of Session::Run(). InterOpThreadPool int32 `protobuf:"varint,3,opt,name=inter_op_thread_pool,json=interOpThreadPool,proto3" json:"inter_op_thread_pool,omitempty"` // Whether the partition graph(s) executed by the executor(s) should be // outputted via RunMetadata. OutputPartitionGraphs bool `` /* 127-byte string literal not displayed */ // EXPERIMENTAL. Options used to initialize DebuggerState, if enabled. DebugOptions *DebugOptions `protobuf:"bytes,6,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"` // When enabled, causes tensor allocation information to be included in // the error message when the Run() call fails because the allocator ran // out of memory (OOM). // // Enabling this option can slow down the Run() call. ReportTensorAllocationsUponOom bool `` /* 158-byte string literal not displayed */ Experimental *RunOptions_Experimental `protobuf:"bytes,8,opt,name=experimental,proto3" json:"experimental,omitempty"` // contains filtered or unexported fields }
Options for a single Run() call.
func (*RunOptions) Descriptor
deprecated
func (*RunOptions) Descriptor() ([]byte, []int)
Deprecated: Use RunOptions.ProtoReflect.Descriptor instead.
func (*RunOptions) GetDebugOptions ¶
func (x *RunOptions) GetDebugOptions() *DebugOptions
func (*RunOptions) GetExperimental ¶
func (x *RunOptions) GetExperimental() *RunOptions_Experimental
func (*RunOptions) GetInterOpThreadPool ¶
func (x *RunOptions) GetInterOpThreadPool() int32
func (*RunOptions) GetOutputPartitionGraphs ¶
func (x *RunOptions) GetOutputPartitionGraphs() bool
func (*RunOptions) GetReportTensorAllocationsUponOom ¶
func (x *RunOptions) GetReportTensorAllocationsUponOom() bool
func (*RunOptions) GetTimeoutInMs ¶
func (x *RunOptions) GetTimeoutInMs() int64
func (*RunOptions) GetTraceLevel ¶
func (x *RunOptions) GetTraceLevel() RunOptions_TraceLevel
func (*RunOptions) ProtoMessage ¶
func (*RunOptions) ProtoMessage()
func (*RunOptions) ProtoReflect ¶
func (x *RunOptions) ProtoReflect() protoreflect.Message
func (*RunOptions) Reset ¶
func (x *RunOptions) Reset()
func (*RunOptions) String ¶
func (x *RunOptions) String() string
type RunOptions_Experimental ¶
type RunOptions_Experimental struct { // If non-zero, declares that this graph is going to use collective // ops and must synchronize step_ids with any other graph with this // same group_key value (in a distributed computation where tasks // run disjoint graphs). CollectiveGraphKey int64 `protobuf:"varint,1,opt,name=collective_graph_key,json=collectiveGraphKey,proto3" json:"collective_graph_key,omitempty"` // If true, then operations (using the inter-op pool) across all // session::run() calls will be centrally scheduled, optimizing for (median // and tail) latency. // Consider using this option for CPU-bound workloads like inference. UseRunHandlerPool bool `protobuf:"varint,2,opt,name=use_run_handler_pool,json=useRunHandlerPool,proto3" json:"use_run_handler_pool,omitempty"` // contains filtered or unexported fields }
Everything inside Experimental is subject to change and is not subject to API stability guarantees in https://www.tensorflow.org/guide/version_compat.
func (*RunOptions_Experimental) Descriptor
deprecated
func (*RunOptions_Experimental) Descriptor() ([]byte, []int)
Deprecated: Use RunOptions_Experimental.ProtoReflect.Descriptor instead.
func (*RunOptions_Experimental) GetCollectiveGraphKey ¶
func (x *RunOptions_Experimental) GetCollectiveGraphKey() int64
func (*RunOptions_Experimental) GetUseRunHandlerPool ¶
func (x *RunOptions_Experimental) GetUseRunHandlerPool() bool
func (*RunOptions_Experimental) ProtoMessage ¶
func (*RunOptions_Experimental) ProtoMessage()
func (*RunOptions_Experimental) ProtoReflect ¶
func (x *RunOptions_Experimental) ProtoReflect() protoreflect.Message
func (*RunOptions_Experimental) Reset ¶
func (x *RunOptions_Experimental) Reset()
func (*RunOptions_Experimental) String ¶
func (x *RunOptions_Experimental) String() string
type RunOptions_TraceLevel ¶
type RunOptions_TraceLevel int32
TODO(pbar) Turn this into a TraceOptions proto which allows tracing to be controlled in a more orthogonal manner?
const ( RunOptions_NO_TRACE RunOptions_TraceLevel = 0 RunOptions_SOFTWARE_TRACE RunOptions_TraceLevel = 1 RunOptions_HARDWARE_TRACE RunOptions_TraceLevel = 2 RunOptions_FULL_TRACE RunOptions_TraceLevel = 3 )
func (RunOptions_TraceLevel) Descriptor ¶
func (RunOptions_TraceLevel) Descriptor() protoreflect.EnumDescriptor
func (RunOptions_TraceLevel) Enum ¶
func (x RunOptions_TraceLevel) Enum() *RunOptions_TraceLevel
func (RunOptions_TraceLevel) EnumDescriptor
deprecated
func (RunOptions_TraceLevel) EnumDescriptor() ([]byte, []int)
Deprecated: Use RunOptions_TraceLevel.Descriptor instead.
func (RunOptions_TraceLevel) Number ¶
func (x RunOptions_TraceLevel) Number() protoreflect.EnumNumber
func (RunOptions_TraceLevel) String ¶
func (x RunOptions_TraceLevel) String() string
func (RunOptions_TraceLevel) Type ¶
func (RunOptions_TraceLevel) Type() protoreflect.EnumType
type RunStepRequest ¶
type RunStepRequest struct { // REQUIRED: session_handle must be returned by a CreateSession call // to the same master service. SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"` // Tensors to be fed in the step. Each feed is a named tensor. Feed []*NamedTensorProto `protobuf:"bytes,2,rep,name=feed,proto3" json:"feed,omitempty"` // Fetches. A list of tensor names. The caller expects a tensor to // be returned for each fetch[i] (see RunStepResponse.tensor). The // order of specified fetches does not change the execution order. Fetch []string `protobuf:"bytes,3,rep,name=fetch,proto3" json:"fetch,omitempty"` // Target Nodes. A list of node names. The named nodes will be run // to but their outputs will not be fetched. Target []string `protobuf:"bytes,4,rep,name=target,proto3" json:"target,omitempty"` // Options for the run call. Options *RunOptions `protobuf:"bytes,5,opt,name=options,proto3" json:"options,omitempty"` // Partial run handle (optional). If specified, this will be a partial run // execution, run up to the specified fetches. PartialRunHandle string `protobuf:"bytes,6,opt,name=partial_run_handle,json=partialRunHandle,proto3" json:"partial_run_handle,omitempty"` // If true then some errors, e.g., execution errors that have long // error messages, may return an OK RunStepResponse with the actual // error saved in the status_code/status_error_message fields of the // response body. This is a workaround since the RPC subsystem may // truncate long metadata messages. StoreErrorsInResponseBody bool `` /* 143-byte string literal not displayed */ // contains filtered or unexported fields }
func (*RunStepRequest) Descriptor
deprecated
func (*RunStepRequest) Descriptor() ([]byte, []int)
Deprecated: Use RunStepRequest.ProtoReflect.Descriptor instead.
func (*RunStepRequest) GetFeed ¶
func (x *RunStepRequest) GetFeed() []*NamedTensorProto
func (*RunStepRequest) GetFetch ¶
func (x *RunStepRequest) GetFetch() []string
func (*RunStepRequest) GetOptions ¶
func (x *RunStepRequest) GetOptions() *RunOptions
func (*RunStepRequest) GetPartialRunHandle ¶
func (x *RunStepRequest) GetPartialRunHandle() string
func (*RunStepRequest) GetSessionHandle ¶
func (x *RunStepRequest) GetSessionHandle() string
func (*RunStepRequest) GetStoreErrorsInResponseBody ¶
func (x *RunStepRequest) GetStoreErrorsInResponseBody() bool
func (*RunStepRequest) GetTarget ¶
func (x *RunStepRequest) GetTarget() []string
func (*RunStepRequest) ProtoMessage ¶
func (*RunStepRequest) ProtoMessage()
func (*RunStepRequest) ProtoReflect ¶
func (x *RunStepRequest) ProtoReflect() protoreflect.Message
func (*RunStepRequest) Reset ¶
func (x *RunStepRequest) Reset()
func (*RunStepRequest) String ¶
func (x *RunStepRequest) String() string
type RunStepResponse ¶
type RunStepResponse struct { // NOTE: The order of the returned tensors may or may not match // the fetch order specified in RunStepRequest. Tensor []*NamedTensorProto `protobuf:"bytes,1,rep,name=tensor,proto3" json:"tensor,omitempty"` // Returned metadata if requested in the options. Metadata *RunMetadata `protobuf:"bytes,2,opt,name=metadata,proto3" json:"metadata,omitempty"` // If store_errors_in_response_body is true in the request, then // optionally the server may return an OK status for the RPC and // fill the true status into the fields below, to allow for messages // that are too long to fit in metadata. StatusCode core.Code `protobuf:"varint,3,opt,name=status_code,json=statusCode,proto3,enum=tensorflow.error.Code" json:"status_code,omitempty"` StatusErrorMessage string `protobuf:"bytes,4,opt,name=status_error_message,json=statusErrorMessage,proto3" json:"status_error_message,omitempty"` // contains filtered or unexported fields }
func (*RunStepResponse) Descriptor
deprecated
func (*RunStepResponse) Descriptor() ([]byte, []int)
Deprecated: Use RunStepResponse.ProtoReflect.Descriptor instead.
func (*RunStepResponse) GetMetadata ¶
func (x *RunStepResponse) GetMetadata() *RunMetadata
func (*RunStepResponse) GetStatusCode ¶
func (x *RunStepResponse) GetStatusCode() core.Code
func (*RunStepResponse) GetStatusErrorMessage ¶
func (x *RunStepResponse) GetStatusErrorMessage() string
func (*RunStepResponse) GetTensor ¶
func (x *RunStepResponse) GetTensor() []*NamedTensorProto
func (*RunStepResponse) ProtoMessage ¶
func (*RunStepResponse) ProtoMessage()
func (*RunStepResponse) ProtoReflect ¶
func (x *RunStepResponse) ProtoReflect() protoreflect.Message
func (*RunStepResponse) Reset ¶
func (x *RunStepResponse) Reset()
func (*RunStepResponse) String ¶
func (x *RunStepResponse) String() string
type SavedModel ¶
type SavedModel struct { // The schema version of the SavedModel instance. Used for versioning when // making future changes to the specification/implementation. Initial value // at release will be 1. SavedModelSchemaVersion int64 `` /* 135-byte string literal not displayed */ // One or more MetaGraphs. MetaGraphs []*MetaGraphDef `protobuf:"bytes,2,rep,name=meta_graphs,json=metaGraphs,proto3" json:"meta_graphs,omitempty"` // contains filtered or unexported fields }
SavedModel is the high level serialization format for TensorFlow Models. See [todo: doc links, similar to session_bundle] for more information.
func (*SavedModel) Descriptor
deprecated
func (*SavedModel) Descriptor() ([]byte, []int)
Deprecated: Use SavedModel.ProtoReflect.Descriptor instead.
func (*SavedModel) GetMetaGraphs ¶
func (x *SavedModel) GetMetaGraphs() []*MetaGraphDef
func (*SavedModel) GetSavedModelSchemaVersion ¶
func (x *SavedModel) GetSavedModelSchemaVersion() int64
func (*SavedModel) ProtoMessage ¶
func (*SavedModel) ProtoMessage()
func (*SavedModel) ProtoReflect ¶
func (x *SavedModel) ProtoReflect() protoreflect.Message
func (*SavedModel) Reset ¶
func (x *SavedModel) Reset()
func (*SavedModel) String ¶
func (x *SavedModel) String() string
type SaverDef ¶
type SaverDef struct { // The name of the tensor in which to specify the filename when saving or // restoring a model checkpoint. FilenameTensorName string `protobuf:"bytes,1,opt,name=filename_tensor_name,json=filenameTensorName,proto3" json:"filename_tensor_name,omitempty"` // The operation to run when saving a model checkpoint. SaveTensorName string `protobuf:"bytes,2,opt,name=save_tensor_name,json=saveTensorName,proto3" json:"save_tensor_name,omitempty"` // The operation to run when restoring a model checkpoint. RestoreOpName string `protobuf:"bytes,3,opt,name=restore_op_name,json=restoreOpName,proto3" json:"restore_op_name,omitempty"` // Maximum number of checkpoints to keep. If 0, no checkpoints are deleted. MaxToKeep int32 `protobuf:"varint,4,opt,name=max_to_keep,json=maxToKeep,proto3" json:"max_to_keep,omitempty"` // Shard the save files, one per device that has Variable nodes. Sharded bool `protobuf:"varint,5,opt,name=sharded,proto3" json:"sharded,omitempty"` // How often to keep an additional checkpoint. If not specified, only the last // "max_to_keep" checkpoints are kept; if specified, in addition to keeping // the last "max_to_keep" checkpoints, an additional checkpoint will be kept // for every n hours of training. KeepCheckpointEveryNHours float32 `` /* 144-byte string literal not displayed */ Version SaverDef_CheckpointFormatVersion `protobuf:"varint,7,opt,name=version,proto3,enum=tensorflow.SaverDef_CheckpointFormatVersion" json:"version,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing the configuration of a Saver.
func (*SaverDef) Descriptor
deprecated
func (*SaverDef) GetFilenameTensorName ¶
func (*SaverDef) GetKeepCheckpointEveryNHours ¶
func (*SaverDef) GetMaxToKeep ¶
func (*SaverDef) GetRestoreOpName ¶
func (*SaverDef) GetSaveTensorName ¶
func (*SaverDef) GetSharded ¶
func (*SaverDef) GetVersion ¶
func (x *SaverDef) GetVersion() SaverDef_CheckpointFormatVersion
func (*SaverDef) ProtoMessage ¶
func (*SaverDef) ProtoMessage()
func (*SaverDef) ProtoReflect ¶
func (x *SaverDef) ProtoReflect() protoreflect.Message
type SaverDef_CheckpointFormatVersion ¶
type SaverDef_CheckpointFormatVersion int32
A version number that identifies a different on-disk checkpoint format. Usually, each subclass of BaseSaverBuilder works with a particular version/format. However, it is possible that the same builder may be upgraded to support a newer checkpoint format in the future.
const ( // Internal legacy format. SaverDef_LEGACY SaverDef_CheckpointFormatVersion = 0 // Deprecated format: tf.Saver() which works with tensorflow::table::Table. SaverDef_V1 SaverDef_CheckpointFormatVersion = 1 // Current format: more efficient. SaverDef_V2 SaverDef_CheckpointFormatVersion = 2 )
func (SaverDef_CheckpointFormatVersion) Descriptor ¶
func (SaverDef_CheckpointFormatVersion) Descriptor() protoreflect.EnumDescriptor
func (SaverDef_CheckpointFormatVersion) Enum ¶
func (x SaverDef_CheckpointFormatVersion) Enum() *SaverDef_CheckpointFormatVersion
func (SaverDef_CheckpointFormatVersion) EnumDescriptor
deprecated
func (SaverDef_CheckpointFormatVersion) EnumDescriptor() ([]byte, []int)
Deprecated: Use SaverDef_CheckpointFormatVersion.Descriptor instead.
func (SaverDef_CheckpointFormatVersion) Number ¶
func (x SaverDef_CheckpointFormatVersion) Number() protoreflect.EnumNumber
func (SaverDef_CheckpointFormatVersion) String ¶
func (x SaverDef_CheckpointFormatVersion) String() string
func (SaverDef_CheckpointFormatVersion) Type ¶
func (SaverDef_CheckpointFormatVersion) Type() protoreflect.EnumType
type ScopedAllocatorOptions ¶
type ScopedAllocatorOptions struct { // If present, only perform optimization for these ops. EnableOp []string `protobuf:"bytes,1,rep,name=enable_op,json=enableOp,proto3" json:"enable_op,omitempty"` // contains filtered or unexported fields }
func (*ScopedAllocatorOptions) Descriptor
deprecated
func (*ScopedAllocatorOptions) Descriptor() ([]byte, []int)
Deprecated: Use ScopedAllocatorOptions.ProtoReflect.Descriptor instead.
func (*ScopedAllocatorOptions) GetEnableOp ¶
func (x *ScopedAllocatorOptions) GetEnableOp() []string
func (*ScopedAllocatorOptions) ProtoMessage ¶
func (*ScopedAllocatorOptions) ProtoMessage()
func (*ScopedAllocatorOptions) ProtoReflect ¶
func (x *ScopedAllocatorOptions) ProtoReflect() protoreflect.Message
func (*ScopedAllocatorOptions) Reset ¶
func (x *ScopedAllocatorOptions) Reset()
func (*ScopedAllocatorOptions) String ¶
func (x *ScopedAllocatorOptions) String() string
type SendTensorRequest ¶
type SendTensorRequest struct { ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"` // All remote tensors are identified by <Op ID, Output num>. To mimic this // situation when directly sending tensors, we include an "artificial" op ID // (which would have corresponded to the _Recv op when not using SendTensor). OpId int64 `protobuf:"varint,2,opt,name=op_id,json=opId,proto3" json:"op_id,omitempty"` // The index within the repeated field is the output number that will help // uniquely identify (along with the above op_id) the particular tensor. Tensors []*framework.TensorProto `protobuf:"bytes,3,rep,name=tensors,proto3" json:"tensors,omitempty"` // The device on which the tensors should be resident. DeviceName string `protobuf:"bytes,4,opt,name=device_name,json=deviceName,proto3" json:"device_name,omitempty"` // contains filtered or unexported fields }
func (*SendTensorRequest) Descriptor
deprecated
func (*SendTensorRequest) Descriptor() ([]byte, []int)
Deprecated: Use SendTensorRequest.ProtoReflect.Descriptor instead.
func (*SendTensorRequest) GetContextId ¶
func (x *SendTensorRequest) GetContextId() uint64
func (*SendTensorRequest) GetDeviceName ¶
func (x *SendTensorRequest) GetDeviceName() string
func (*SendTensorRequest) GetOpId ¶
func (x *SendTensorRequest) GetOpId() int64
func (*SendTensorRequest) GetTensors ¶
func (x *SendTensorRequest) GetTensors() []*framework.TensorProto
func (*SendTensorRequest) ProtoMessage ¶
func (*SendTensorRequest) ProtoMessage()
func (*SendTensorRequest) ProtoReflect ¶
func (x *SendTensorRequest) ProtoReflect() protoreflect.Message
func (*SendTensorRequest) Reset ¶
func (x *SendTensorRequest) Reset()
func (*SendTensorRequest) String ¶
func (x *SendTensorRequest) String() string
type SendTensorResponse ¶
type SendTensorResponse struct {
// contains filtered or unexported fields
}
func (*SendTensorResponse) Descriptor
deprecated
func (*SendTensorResponse) Descriptor() ([]byte, []int)
Deprecated: Use SendTensorResponse.ProtoReflect.Descriptor instead.
func (*SendTensorResponse) ProtoMessage ¶
func (*SendTensorResponse) ProtoMessage()
func (*SendTensorResponse) ProtoReflect ¶
func (x *SendTensorResponse) ProtoReflect() protoreflect.Message
func (*SendTensorResponse) Reset ¶
func (x *SendTensorResponse) Reset()
func (*SendTensorResponse) String ¶
func (x *SendTensorResponse) String() string
type ServerDef ¶
type ServerDef struct { // The cluster of which this server is a member. Cluster *ClusterDef `protobuf:"bytes,1,opt,name=cluster,proto3" json:"cluster,omitempty"` // The name of the job of which this server is a member. // // NOTE(mrry): The `cluster` field must contain a `JobDef` with a `name` field // that matches this name. JobName string `protobuf:"bytes,2,opt,name=job_name,json=jobName,proto3" json:"job_name,omitempty"` // The task index of this server in its job. // // NOTE: The `cluster` field must contain a `JobDef` with a matching `name` // and a mapping in its `tasks` field for this index. TaskIndex int32 `protobuf:"varint,3,opt,name=task_index,json=taskIndex,proto3" json:"task_index,omitempty"` // The default configuration for sessions that run on this server. DefaultSessionConfig *ConfigProto `protobuf:"bytes,4,opt,name=default_session_config,json=defaultSessionConfig,proto3" json:"default_session_config,omitempty"` // The protocol to be used by this server. // // Acceptable values include: "grpc", "grpc+verbs". Protocol string `protobuf:"bytes,5,opt,name=protocol,proto3" json:"protocol,omitempty"` // contains filtered or unexported fields }
Defines the configuration of a single TensorFlow server.
func (*ServerDef) Descriptor
deprecated
func (*ServerDef) GetCluster ¶
func (x *ServerDef) GetCluster() *ClusterDef
func (*ServerDef) GetDefaultSessionConfig ¶
func (x *ServerDef) GetDefaultSessionConfig() *ConfigProto
func (*ServerDef) GetJobName ¶
func (*ServerDef) GetProtocol ¶
func (*ServerDef) GetTaskIndex ¶
func (*ServerDef) ProtoMessage ¶
func (*ServerDef) ProtoMessage()
func (*ServerDef) ProtoReflect ¶
func (x *ServerDef) ProtoReflect() protoreflect.Message
type SignatureDef ¶
type SignatureDef struct { // Named input parameters. Inputs map[string]*TensorInfo `` /* 153-byte string literal not displayed */ // Named output parameters. Outputs map[string]*TensorInfo `` /* 155-byte string literal not displayed */ // Extensible method_name information enabling third-party users to mark a // SignatureDef as supporting a particular method. This enables producers and // consumers of SignatureDefs, e.g. a model definition library and a serving // library to have a clear hand-off regarding the semantics of a computation. // // Note that multiple SignatureDefs in a single MetaGraphDef may have the same // method_name. This is commonly used to support multi-headed computation, // where a single graph computation may return multiple results. MethodName string `protobuf:"bytes,3,opt,name=method_name,json=methodName,proto3" json:"method_name,omitempty"` // contains filtered or unexported fields }
SignatureDef defines the signature of a computation supported by a TensorFlow graph.
For example, a model with two loss computations, sharing a single input, might have the following signature_def map.
Note that across the two SignatureDefs "loss_A" and "loss_B", the input key, output key, and method_name are identical, and will be used by system(s) that implement or rely upon this particular loss method. The output tensor names differ, demonstrating how different outputs can exist for the same method.
signature_def { key: "loss_A" value { inputs { key: "input" value { name: "input:0" dtype: DT_STRING tensor_shape: ... } } outputs { key: "loss_output" value { name: "loss_output_A:0" dtype: DT_FLOAT tensor_shape: ... } } } ... method_name: "some/package/compute_loss" }
signature_def { key: "loss_B" value { inputs { key: "input" value { name: "input:0" dtype: DT_STRING tensor_shape: ... } } outputs { key: "loss_output" value { name: "loss_output_B:0" dtype: DT_FLOAT tensor_shape: ... } } } ... method_name: "some/package/compute_loss" }
func (*SignatureDef) Descriptor
deprecated
func (*SignatureDef) Descriptor() ([]byte, []int)
Deprecated: Use SignatureDef.ProtoReflect.Descriptor instead.
func (*SignatureDef) GetInputs ¶
func (x *SignatureDef) GetInputs() map[string]*TensorInfo
func (*SignatureDef) GetMethodName ¶
func (x *SignatureDef) GetMethodName() string
func (*SignatureDef) GetOutputs ¶
func (x *SignatureDef) GetOutputs() map[string]*TensorInfo
func (*SignatureDef) ProtoMessage ¶
func (*SignatureDef) ProtoMessage()
func (*SignatureDef) ProtoReflect ¶
func (x *SignatureDef) ProtoReflect() protoreflect.Message
func (*SignatureDef) Reset ¶
func (x *SignatureDef) Reset()
func (*SignatureDef) String ¶
func (x *SignatureDef) String() string
type StepSequence ¶
type StepSequence struct { GraphKey int64 `protobuf:"varint,1,opt,name=graph_key,json=graphKey,proto3" json:"graph_key,omitempty"` NextStepId int64 `protobuf:"varint,2,opt,name=next_step_id,json=nextStepId,proto3" json:"next_step_id,omitempty"` // contains filtered or unexported fields }
func (*StepSequence) Descriptor
deprecated
func (*StepSequence) Descriptor() ([]byte, []int)
Deprecated: Use StepSequence.ProtoReflect.Descriptor instead.
func (*StepSequence) GetGraphKey ¶
func (x *StepSequence) GetGraphKey() int64
func (*StepSequence) GetNextStepId ¶
func (x *StepSequence) GetNextStepId() int64
func (*StepSequence) ProtoMessage ¶
func (*StepSequence) ProtoMessage()
func (*StepSequence) ProtoReflect ¶
func (x *StepSequence) ProtoReflect() protoreflect.Message
func (*StepSequence) Reset ¶
func (x *StepSequence) Reset()
func (*StepSequence) String ¶
func (x *StepSequence) String() string
type TensorConnection ¶
type TensorConnection struct { // A tensor name. The value of this tensor will be substituted for // the tensor named in `to_tensor`. FromTensor string `protobuf:"bytes,1,opt,name=from_tensor,json=fromTensor,proto3" json:"from_tensor,omitempty"` // A tensor name. The value of this tensor will be bound to the // value of the tensor named in `from_tensor`. ToTensor string `protobuf:"bytes,2,opt,name=to_tensor,json=toTensor,proto3" json:"to_tensor,omitempty"` // contains filtered or unexported fields }
Defines a connection between two tensors in a `GraphDef`.
func (*TensorConnection) Descriptor
deprecated
func (*TensorConnection) Descriptor() ([]byte, []int)
Deprecated: Use TensorConnection.ProtoReflect.Descriptor instead.
func (*TensorConnection) GetFromTensor ¶
func (x *TensorConnection) GetFromTensor() string
func (*TensorConnection) GetToTensor ¶
func (x *TensorConnection) GetToTensor() string
func (*TensorConnection) ProtoMessage ¶
func (*TensorConnection) ProtoMessage()
func (*TensorConnection) ProtoReflect ¶
func (x *TensorConnection) ProtoReflect() protoreflect.Message
func (*TensorConnection) Reset ¶
func (x *TensorConnection) Reset()
func (*TensorConnection) String ¶
func (x *TensorConnection) String() string
type TensorInfo ¶
type TensorInfo struct { // Types that are assignable to Encoding: // *TensorInfo_Name // *TensorInfo_CooSparse_ Encoding isTensorInfo_Encoding `protobuf_oneof:"encoding"` Dtype framework.DataType `protobuf:"varint,2,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"` // The static shape should be recorded here, to the extent that it can // be known in advance. In the case of a SparseTensor, this field describes // the logical shape of the represented tensor (aka dense_shape). TensorShape *framework.TensorShapeProto `protobuf:"bytes,3,opt,name=tensor_shape,json=tensorShape,proto3" json:"tensor_shape,omitempty"` // contains filtered or unexported fields }
Information about a Tensor necessary for feeding or retrieval.
func (*TensorInfo) Descriptor
deprecated
func (*TensorInfo) Descriptor() ([]byte, []int)
Deprecated: Use TensorInfo.ProtoReflect.Descriptor instead.
func (*TensorInfo) GetCooSparse ¶
func (x *TensorInfo) GetCooSparse() *TensorInfo_CooSparse
func (*TensorInfo) GetDtype ¶
func (x *TensorInfo) GetDtype() framework.DataType
func (*TensorInfo) GetEncoding ¶
func (m *TensorInfo) GetEncoding() isTensorInfo_Encoding
func (*TensorInfo) GetName ¶
func (x *TensorInfo) GetName() string
func (*TensorInfo) GetTensorShape ¶
func (x *TensorInfo) GetTensorShape() *framework.TensorShapeProto
func (*TensorInfo) ProtoMessage ¶
func (*TensorInfo) ProtoMessage()
func (*TensorInfo) ProtoReflect ¶
func (x *TensorInfo) ProtoReflect() protoreflect.Message
func (*TensorInfo) Reset ¶
func (x *TensorInfo) Reset()
func (*TensorInfo) String ¶
func (x *TensorInfo) String() string
type TensorInfo_CooSparse ¶
type TensorInfo_CooSparse struct { // The shape of the values Tensor is [?]. Its dtype must be the dtype of // the SparseTensor as a whole, given in the enclosing TensorInfo. ValuesTensorName string `protobuf:"bytes,1,opt,name=values_tensor_name,json=valuesTensorName,proto3" json:"values_tensor_name,omitempty"` // The indices Tensor must have dtype int64 and shape [?, ?]. IndicesTensorName string `protobuf:"bytes,2,opt,name=indices_tensor_name,json=indicesTensorName,proto3" json:"indices_tensor_name,omitempty"` // The dynamic logical shape represented by the SparseTensor is recorded in // the Tensor referenced here. It must have dtype int64 and shape [?]. DenseShapeTensorName string `protobuf:"bytes,3,opt,name=dense_shape_tensor_name,json=denseShapeTensorName,proto3" json:"dense_shape_tensor_name,omitempty"` // contains filtered or unexported fields }
For sparse tensors, The COO encoding stores a triple of values, indices, and shape.
func (*TensorInfo_CooSparse) Descriptor
deprecated
func (*TensorInfo_CooSparse) Descriptor() ([]byte, []int)
Deprecated: Use TensorInfo_CooSparse.ProtoReflect.Descriptor instead.
func (*TensorInfo_CooSparse) GetDenseShapeTensorName ¶
func (x *TensorInfo_CooSparse) GetDenseShapeTensorName() string
func (*TensorInfo_CooSparse) GetIndicesTensorName ¶
func (x *TensorInfo_CooSparse) GetIndicesTensorName() string
func (*TensorInfo_CooSparse) GetValuesTensorName ¶
func (x *TensorInfo_CooSparse) GetValuesTensorName() string
func (*TensorInfo_CooSparse) ProtoMessage ¶
func (*TensorInfo_CooSparse) ProtoMessage()
func (*TensorInfo_CooSparse) ProtoReflect ¶
func (x *TensorInfo_CooSparse) ProtoReflect() protoreflect.Message
func (*TensorInfo_CooSparse) Reset ¶
func (x *TensorInfo_CooSparse) Reset()
func (*TensorInfo_CooSparse) String ¶
func (x *TensorInfo_CooSparse) String() string
type TensorInfo_CooSparse_ ¶
type TensorInfo_CooSparse_ struct { // There are many possible encodings of sparse matrices // (https://en.wikipedia.org/wiki/Sparse_matrix). Currently, TensorFlow // uses only the COO encoding. This is supported and documented in the // SparseTensor Python class. CooSparse *TensorInfo_CooSparse `protobuf:"bytes,4,opt,name=coo_sparse,json=cooSparse,proto3,oneof"` }
type TensorInfo_Name ¶
type TensorInfo_Name struct { // For dense `Tensor`s, the name of the tensor in the graph. Name string `protobuf:"bytes,1,opt,name=name,proto3,oneof"` }
type ThreadPoolOptionProto ¶
type ThreadPoolOptionProto struct { // The number of threads in the pool. // // 0 means the system picks a value based on where this option proto is used // (see the declaration of the specific field for more info). NumThreads int32 `protobuf:"varint,1,opt,name=num_threads,json=numThreads,proto3" json:"num_threads,omitempty"` // The global name of the threadpool. // // If empty, then the threadpool is made and used according to the scope it's // in - e.g., for a session threadpool, it is used by that session only. // // If non-empty, then: // - a global threadpool associated with this name is looked // up or created. This allows, for example, sharing one threadpool across // many sessions (e.g., like the default behavior, if // inter_op_parallelism_threads is not configured), but still partitioning // into a large and small pool. // - if the threadpool for this global_name already exists, then it is an // error if the existing pool was created using a different num_threads // value as is specified on this call. // - threadpools created this way are never garbage collected. GlobalName string `protobuf:"bytes,2,opt,name=global_name,json=globalName,proto3" json:"global_name,omitempty"` // contains filtered or unexported fields }
func (*ThreadPoolOptionProto) Descriptor
deprecated
func (*ThreadPoolOptionProto) Descriptor() ([]byte, []int)
Deprecated: Use ThreadPoolOptionProto.ProtoReflect.Descriptor instead.
func (*ThreadPoolOptionProto) GetGlobalName ¶
func (x *ThreadPoolOptionProto) GetGlobalName() string
func (*ThreadPoolOptionProto) GetNumThreads ¶
func (x *ThreadPoolOptionProto) GetNumThreads() int32
func (*ThreadPoolOptionProto) ProtoMessage ¶
func (*ThreadPoolOptionProto) ProtoMessage()
func (*ThreadPoolOptionProto) ProtoReflect ¶
func (x *ThreadPoolOptionProto) ProtoReflect() protoreflect.Message
func (*ThreadPoolOptionProto) Reset ¶
func (x *ThreadPoolOptionProto) Reset()
func (*ThreadPoolOptionProto) String ¶
func (x *ThreadPoolOptionProto) String() string
type TraceOpts ¶
type TraceOpts struct { // Length of the trace to be taken, in seconds. Duration float64 `protobuf:"fixed64,1,opt,name=duration,proto3" json:"duration,omitempty"` // If true, capture step profile locally in each worker. Currently // unimplemented. UseStepProfiler bool `protobuf:"varint,2,opt,name=use_step_profiler,json=useStepProfiler,proto3" json:"use_step_profiler,omitempty"` // If true, capture kernel events from each worker. UseKernelProfiler bool `protobuf:"varint,3,opt,name=use_kernel_profiler,json=useKernelProfiler,proto3" json:"use_kernel_profiler,omitempty"` // If true, capture extended profiling events from TensorFlow process. UseExtendedProfiler bool `protobuf:"varint,4,opt,name=use_extended_profiler,json=useExtendedProfiler,proto3" json:"use_extended_profiler,omitempty"` // If true, capture GPU profiling events locally on each // machine. Currently unimplemented. UseGpuProfiler bool `protobuf:"varint,5,opt,name=use_gpu_profiler,json=useGpuProfiler,proto3" json:"use_gpu_profiler,omitempty"` // If true, collect sampled profile events. Currently unimplemented. UseSampleProfiler bool `protobuf:"varint,6,opt,name=use_sample_profiler,json=useSampleProfiler,proto3" json:"use_sample_profiler,omitempty"` // contains filtered or unexported fields }
func (*TraceOpts) Descriptor
deprecated
func (*TraceOpts) GetDuration ¶
func (*TraceOpts) GetUseExtendedProfiler ¶
func (*TraceOpts) GetUseGpuProfiler ¶
func (*TraceOpts) GetUseKernelProfiler ¶
func (*TraceOpts) GetUseSampleProfiler ¶
func (*TraceOpts) GetUseStepProfiler ¶
func (*TraceOpts) ProtoMessage ¶
func (*TraceOpts) ProtoMessage()
func (*TraceOpts) ProtoReflect ¶
func (x *TraceOpts) ProtoReflect() protoreflect.Message
type TracingRequest ¶
type TracingRequest struct { Options *TraceOpts `protobuf:"bytes,1,opt,name=options,proto3" json:"options,omitempty"` // contains filtered or unexported fields }
Out-of-band request to configure distributed tracing.
func (*TracingRequest) Descriptor
deprecated
func (*TracingRequest) Descriptor() ([]byte, []int)
Deprecated: Use TracingRequest.ProtoReflect.Descriptor instead.
func (*TracingRequest) GetOptions ¶
func (x *TracingRequest) GetOptions() *TraceOpts
func (*TracingRequest) ProtoMessage ¶
func (*TracingRequest) ProtoMessage()
func (*TracingRequest) ProtoReflect ¶
func (x *TracingRequest) ProtoReflect() protoreflect.Message
func (*TracingRequest) Reset ¶
func (x *TracingRequest) Reset()
func (*TracingRequest) String ¶
func (x *TracingRequest) String() string
type TracingResponse ¶
type TracingResponse struct {
// contains filtered or unexported fields
}
func (*TracingResponse) Descriptor
deprecated
func (*TracingResponse) Descriptor() ([]byte, []int)
Deprecated: Use TracingResponse.ProtoReflect.Descriptor instead.
func (*TracingResponse) ProtoMessage ¶
func (*TracingResponse) ProtoMessage()
func (*TracingResponse) ProtoReflect ¶
func (x *TracingResponse) ProtoReflect() protoreflect.Message
func (*TracingResponse) Reset ¶
func (x *TracingResponse) Reset()
func (*TracingResponse) String ¶
func (x *TracingResponse) String() string
type ValuesDef ¶
type ValuesDef struct { // Value names that have been seen in this context. Values []string `protobuf:"bytes,1,rep,name=values,proto3" json:"values,omitempty"` // Value names referenced by but external to this context. ExternalValues map[string]string `` /* 191-byte string literal not displayed */ // contains filtered or unexported fields }
Protocol buffer representing the values in ControlFlowContext.
func (*ValuesDef) Descriptor
deprecated
func (*ValuesDef) GetExternalValues ¶
func (*ValuesDef) ProtoMessage ¶
func (*ValuesDef) ProtoMessage()
func (*ValuesDef) ProtoReflect ¶
func (x *ValuesDef) ProtoReflect() protoreflect.Message
type WaitQueueDoneRequest ¶
type WaitQueueDoneRequest struct { ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"` // Ids to wait on. If empty, wait on everything currently pending. OpId []int64 `protobuf:"varint,2,rep,packed,name=op_id,json=opId,proto3" json:"op_id,omitempty"` // contains filtered or unexported fields }
func (*WaitQueueDoneRequest) Descriptor
deprecated
func (*WaitQueueDoneRequest) Descriptor() ([]byte, []int)
Deprecated: Use WaitQueueDoneRequest.ProtoReflect.Descriptor instead.
func (*WaitQueueDoneRequest) GetContextId ¶
func (x *WaitQueueDoneRequest) GetContextId() uint64
func (*WaitQueueDoneRequest) GetOpId ¶
func (x *WaitQueueDoneRequest) GetOpId() []int64
func (*WaitQueueDoneRequest) ProtoMessage ¶
func (*WaitQueueDoneRequest) ProtoMessage()
func (*WaitQueueDoneRequest) ProtoReflect ¶
func (x *WaitQueueDoneRequest) ProtoReflect() protoreflect.Message
func (*WaitQueueDoneRequest) Reset ¶
func (x *WaitQueueDoneRequest) Reset()
func (*WaitQueueDoneRequest) String ¶
func (x *WaitQueueDoneRequest) String() string
type WaitQueueDoneResponse ¶
type WaitQueueDoneResponse struct {
// contains filtered or unexported fields
}
func (*WaitQueueDoneResponse) Descriptor
deprecated
func (*WaitQueueDoneResponse) Descriptor() ([]byte, []int)
Deprecated: Use WaitQueueDoneResponse.ProtoReflect.Descriptor instead.
func (*WaitQueueDoneResponse) ProtoMessage ¶
func (*WaitQueueDoneResponse) ProtoMessage()
func (*WaitQueueDoneResponse) ProtoReflect ¶
func (x *WaitQueueDoneResponse) ProtoReflect() protoreflect.Message
func (*WaitQueueDoneResponse) Reset ¶
func (x *WaitQueueDoneResponse) Reset()
func (*WaitQueueDoneResponse) String ¶
func (x *WaitQueueDoneResponse) String() string
type WhileContextDef ¶
type WhileContextDef struct { // Name of the context. ContextName string `protobuf:"bytes,1,opt,name=context_name,json=contextName,proto3" json:"context_name,omitempty"` // The number of iterations allowed to run in parallel. ParallelIterations int32 `protobuf:"varint,2,opt,name=parallel_iterations,json=parallelIterations,proto3" json:"parallel_iterations,omitempty"` // Whether backprop is enabled for this while loop. BackProp bool `protobuf:"varint,3,opt,name=back_prop,json=backProp,proto3" json:"back_prop,omitempty"` // Whether GPU-CPU memory swap is enabled for this loop. SwapMemory bool `protobuf:"varint,4,opt,name=swap_memory,json=swapMemory,proto3" json:"swap_memory,omitempty"` // Name of the pivot tensor. PivotName string `protobuf:"bytes,5,opt,name=pivot_name,json=pivotName,proto3" json:"pivot_name,omitempty"` // Name of the pivot_for_pred tensor. PivotForPredName string `protobuf:"bytes,6,opt,name=pivot_for_pred_name,json=pivotForPredName,proto3" json:"pivot_for_pred_name,omitempty"` // Name of the pivot_for_body tensor. PivotForBodyName string `protobuf:"bytes,7,opt,name=pivot_for_body_name,json=pivotForBodyName,proto3" json:"pivot_for_body_name,omitempty"` // List of names for exit tensors. LoopExitNames []string `protobuf:"bytes,8,rep,name=loop_exit_names,json=loopExitNames,proto3" json:"loop_exit_names,omitempty"` // List of names for enter tensors. LoopEnterNames []string `protobuf:"bytes,10,rep,name=loop_enter_names,json=loopEnterNames,proto3" json:"loop_enter_names,omitempty"` // Values and external values in control flow context. ValuesDef *ValuesDef `protobuf:"bytes,9,opt,name=values_def,json=valuesDef,proto3" json:"values_def,omitempty"` // Optional name of the maximum_iterations tensor. MaximumIterationsName string `` /* 127-byte string literal not displayed */ // Contexts contained inside this context (e.g. nested whiles). NestedContexts []*ControlFlowContextDef `protobuf:"bytes,12,rep,name=nested_contexts,json=nestedContexts,proto3" json:"nested_contexts,omitempty"` // contains filtered or unexported fields }
Protocol buffer representing a WhileContext object.
func (*WhileContextDef) Descriptor
deprecated
func (*WhileContextDef) Descriptor() ([]byte, []int)
Deprecated: Use WhileContextDef.ProtoReflect.Descriptor instead.
func (*WhileContextDef) GetBackProp ¶
func (x *WhileContextDef) GetBackProp() bool
func (*WhileContextDef) GetContextName ¶
func (x *WhileContextDef) GetContextName() string
func (*WhileContextDef) GetLoopEnterNames ¶
func (x *WhileContextDef) GetLoopEnterNames() []string
func (*WhileContextDef) GetLoopExitNames ¶
func (x *WhileContextDef) GetLoopExitNames() []string
func (*WhileContextDef) GetMaximumIterationsName ¶
func (x *WhileContextDef) GetMaximumIterationsName() string
func (*WhileContextDef) GetNestedContexts ¶
func (x *WhileContextDef) GetNestedContexts() []*ControlFlowContextDef
func (*WhileContextDef) GetParallelIterations ¶
func (x *WhileContextDef) GetParallelIterations() int32
func (*WhileContextDef) GetPivotForBodyName ¶
func (x *WhileContextDef) GetPivotForBodyName() string
func (*WhileContextDef) GetPivotForPredName ¶
func (x *WhileContextDef) GetPivotForPredName() string
func (*WhileContextDef) GetPivotName ¶
func (x *WhileContextDef) GetPivotName() string
func (*WhileContextDef) GetSwapMemory ¶
func (x *WhileContextDef) GetSwapMemory() bool
func (*WhileContextDef) GetValuesDef ¶
func (x *WhileContextDef) GetValuesDef() *ValuesDef
func (*WhileContextDef) ProtoMessage ¶
func (*WhileContextDef) ProtoMessage()
func (*WhileContextDef) ProtoReflect ¶
func (x *WhileContextDef) ProtoReflect() protoreflect.Message
func (*WhileContextDef) Reset ¶
func (x *WhileContextDef) Reset()
func (*WhileContextDef) String ¶
func (x *WhileContextDef) String() string
Source Files ¶
- checkpointable_object_graph.pb.go
- cluster.pb.go
- config.pb.go
- control_flow.pb.go
- critical_section.pb.go
- debug.pb.go
- device_properties.pb.go
- eager_service.pb.go
- master.pb.go
- master_service.pb.go
- meta_graph.pb.go
- named_tensor.pb.go
- queue_runner.pb.go
- replay_log.pb.go
- rewriter_config.pb.go
- saved_model.pb.go
- saver.pb.go
- tensor_bundle.pb.go
- tensorflow_server.pb.go
- transport_options.pb.go
- worker.pb.go
- worker_service.pb.go