Documentation
¶
Overview ¶
Package tensorflow is a generated protocol buffer package.
It is generated from these files:
allocation_description.proto attr_value.proto cluster.proto config.proto control_flow.proto cost_graph.proto debug.proto device_attributes.proto device_properties.proto error_codes.proto function.proto graph.proto graph_transfer_info.proto kernel_def.proto log_memory.proto master.proto master_service.proto meta_graph.proto named_tensor.proto node_def.proto op_def.proto op_gen_overrides.proto queue_runner.proto reader_base.proto remote_fused_graph_execute_info.proto resource_handle.proto rewriter_config.proto saved_model.proto saver.proto step_stats.proto summary.proto tensor_bundle.proto tensor_description.proto tensorflow_server.proto tensor.proto tensor_shape.proto tensor_slice.proto types.proto variable.proto versions.proto worker.proto worker_service.proto
It has these top-level messages:
AllocationDescription AttrValue NameAttrList JobDef ClusterDef GPUOptions OptimizerOptions GraphOptions ThreadPoolOptionProto RPCOptions ConfigProto RunOptions RunMetadata ValuesDef CondContextDef WhileContextDef CostGraphDef DebugTensorWatch DebugOptions DeviceLocality DeviceAttributes DeviceProperties Error FunctionDefLibrary FunctionDef GradientDef GraphDef GraphTransferInfo KernelDef MemoryLogStep MemoryLogTensorAllocation MemoryLogTensorDeallocation MemoryLogTensorOutput MemoryLogRawAllocation MemoryLogRawDeallocation CreateSessionRequest CreateSessionResponse ExtendSessionRequest ExtendSessionResponse RunStepRequest RunStepResponse PartialRunSetupRequest PartialRunSetupResponse CloseSessionRequest CloseSessionResponse ResetRequest ResetResponse ListDevicesRequest ListDevicesResponse MetaGraphDef CollectionDef TensorInfo SignatureDef AssetFileDef NamedTensorProto NodeDef OpDef OpDeprecation OpList OpGenOverride OpGenOverrides QueueRunnerDef ReaderBaseState RemoteFusedGraphExecuteInfo ResourceHandle AutoParallelOptions RewriterConfig SavedModel SaverDef AllocatorMemoryUsed NodeOutput MemoryStats NodeExecStats DeviceStepStats StepStats SummaryDescription HistogramProto Summary BundleHeaderProto BundleEntryProto TensorDescription ServerDef TensorProto TensorShapeProto TensorSliceProto VariableDef SaveSliceInfoDef VersionDef GetStatusRequest GetStatusResponse CreateWorkerSessionRequest CreateWorkerSessionResponse RegisterGraphRequest RegisterGraphResponse DeregisterGraphRequest DeregisterGraphResponse CleanupAllRequest CleanupAllResponse ExecutorOpts RunGraphRequest RunGraphResponse CleanupGraphRequest CleanupGraphResponse RecvTensorRequest RecvTensorResponse LoggingRequest LabeledStepStats LoggingResponse TraceOpts TracingRequest TracingResponse
Index ¶
- Variables
- func Asset(name string) ([]byte, error)
- func AssetDir(name string) ([]string, error)
- func AssetInfo(name string) (os.FileInfo, error)
- func AssetNames() []string
- func MustAsset(name string) []byte
- func RegisterMasterServiceServer(s *grpc.Server, srv MasterServiceServer)
- func RegisterWorkerServiceServer(s *grpc.Server, srv WorkerServiceServer)
- func RestoreAsset(dir, name string) error
- func RestoreAssets(dir, name string) error
- type AllocationDescription
- func (*AllocationDescription) Descriptor() ([]byte, []int)
- func (m *AllocationDescription) GetAllocatedBytes() int64
- func (m *AllocationDescription) GetAllocationId() int64
- func (m *AllocationDescription) GetAllocatorName() string
- func (m *AllocationDescription) GetHasSingleReference() bool
- func (m *AllocationDescription) GetPtr() uint64
- func (m *AllocationDescription) GetRequestedBytes() int64
- func (m *AllocationDescription) Marshal() (dAtA []byte, err error)
- func (m *AllocationDescription) MarshalTo(dAtA []byte) (int, error)
- func (*AllocationDescription) ProtoMessage()
- func (m *AllocationDescription) Reset()
- func (m *AllocationDescription) Size() (n int)
- func (m *AllocationDescription) String() string
- func (m *AllocationDescription) Unmarshal(dAtA []byte) error
- type AllocatorMemoryUsed
- func (*AllocatorMemoryUsed) Descriptor() ([]byte, []int)
- func (m *AllocatorMemoryUsed) GetAllocatorName() string
- func (m *AllocatorMemoryUsed) GetLiveBytes() int64
- func (m *AllocatorMemoryUsed) GetPeakBytes() int64
- func (m *AllocatorMemoryUsed) GetTotalBytes() int64
- func (m *AllocatorMemoryUsed) Marshal() (dAtA []byte, err error)
- func (m *AllocatorMemoryUsed) MarshalTo(dAtA []byte) (int, error)
- func (*AllocatorMemoryUsed) ProtoMessage()
- func (m *AllocatorMemoryUsed) Reset()
- func (m *AllocatorMemoryUsed) Size() (n int)
- func (m *AllocatorMemoryUsed) String() string
- func (m *AllocatorMemoryUsed) Unmarshal(dAtA []byte) error
- type AssetFileDef
- func (*AssetFileDef) Descriptor() ([]byte, []int)
- func (m *AssetFileDef) GetFilename() string
- func (m *AssetFileDef) GetTensorInfo() *TensorInfo
- func (m *AssetFileDef) Marshal() (dAtA []byte, err error)
- func (m *AssetFileDef) MarshalTo(dAtA []byte) (int, error)
- func (*AssetFileDef) ProtoMessage()
- func (m *AssetFileDef) Reset()
- func (m *AssetFileDef) Size() (n int)
- func (m *AssetFileDef) String() string
- func (m *AssetFileDef) Unmarshal(dAtA []byte) error
- type AttrValue
- func (*AttrValue) Descriptor() ([]byte, []int)
- func (m *AttrValue) GetB() bool
- func (m *AttrValue) GetF() float32
- func (m *AttrValue) GetFunc() *NameAttrList
- func (m *AttrValue) GetI() int64
- func (m *AttrValue) GetList() *AttrValue_ListValue
- func (m *AttrValue) GetPlaceholder() string
- func (m *AttrValue) GetS() []byte
- func (m *AttrValue) GetShape() *TensorShapeProto
- func (m *AttrValue) GetTensor() *TensorProto
- func (m *AttrValue) GetType() DataType
- func (m *AttrValue) GetValue() isAttrValue_Value
- func (m *AttrValue) Marshal() (dAtA []byte, err error)
- func (m *AttrValue) MarshalTo(dAtA []byte) (int, error)
- func (*AttrValue) ProtoMessage()
- func (m *AttrValue) Reset()
- func (m *AttrValue) Size() (n int)
- func (m *AttrValue) String() string
- func (m *AttrValue) Unmarshal(dAtA []byte) error
- func (*AttrValue) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, ...)
- type AttrValue_B
- type AttrValue_F
- type AttrValue_Func
- type AttrValue_I
- type AttrValue_List
- type AttrValue_ListValue
- func (*AttrValue_ListValue) Descriptor() ([]byte, []int)
- func (m *AttrValue_ListValue) GetB() []bool
- func (m *AttrValue_ListValue) GetF() []float32
- func (m *AttrValue_ListValue) GetFunc() []*NameAttrList
- func (m *AttrValue_ListValue) GetI() []int64
- func (m *AttrValue_ListValue) GetS() [][]byte
- func (m *AttrValue_ListValue) GetShape() []*TensorShapeProto
- func (m *AttrValue_ListValue) GetTensor() []*TensorProto
- func (m *AttrValue_ListValue) GetType() []DataType
- func (m *AttrValue_ListValue) Marshal() (dAtA []byte, err error)
- func (m *AttrValue_ListValue) MarshalTo(dAtA []byte) (int, error)
- func (*AttrValue_ListValue) ProtoMessage()
- func (m *AttrValue_ListValue) Reset()
- func (m *AttrValue_ListValue) Size() (n int)
- func (m *AttrValue_ListValue) String() string
- func (m *AttrValue_ListValue) Unmarshal(dAtA []byte) error
- type AttrValue_Placeholder
- type AttrValue_S
- type AttrValue_Shape
- type AttrValue_Tensor
- type AttrValue_Type
- type AutoParallelOptions
- func (*AutoParallelOptions) Descriptor() ([]byte, []int)
- func (m *AutoParallelOptions) GetEnable() bool
- func (m *AutoParallelOptions) GetNumReplicas() int32
- func (m *AutoParallelOptions) Marshal() (dAtA []byte, err error)
- func (m *AutoParallelOptions) MarshalTo(dAtA []byte) (int, error)
- func (*AutoParallelOptions) ProtoMessage()
- func (m *AutoParallelOptions) Reset()
- func (m *AutoParallelOptions) Size() (n int)
- func (m *AutoParallelOptions) String() string
- func (m *AutoParallelOptions) Unmarshal(dAtA []byte) error
- type BundleEntryProto
- func (*BundleEntryProto) Descriptor() ([]byte, []int)
- func (m *BundleEntryProto) GetCrc32C() uint32
- func (m *BundleEntryProto) GetDtype() DataType
- func (m *BundleEntryProto) GetOffset() int64
- func (m *BundleEntryProto) GetShape() *TensorShapeProto
- func (m *BundleEntryProto) GetShardId() int32
- func (m *BundleEntryProto) GetSize_() int64
- func (m *BundleEntryProto) GetSlices() []*TensorSliceProto
- func (m *BundleEntryProto) Marshal() (dAtA []byte, err error)
- func (m *BundleEntryProto) MarshalTo(dAtA []byte) (int, error)
- func (*BundleEntryProto) ProtoMessage()
- func (m *BundleEntryProto) Reset()
- func (m *BundleEntryProto) Size() (n int)
- func (m *BundleEntryProto) String() string
- func (m *BundleEntryProto) Unmarshal(dAtA []byte) error
- type BundleHeaderProto
- func (*BundleHeaderProto) Descriptor() ([]byte, []int)
- func (m *BundleHeaderProto) GetEndianness() BundleHeaderProto_Endianness
- func (m *BundleHeaderProto) GetNumShards() int32
- func (m *BundleHeaderProto) GetVersion() *VersionDef
- func (m *BundleHeaderProto) Marshal() (dAtA []byte, err error)
- func (m *BundleHeaderProto) MarshalTo(dAtA []byte) (int, error)
- func (*BundleHeaderProto) ProtoMessage()
- func (m *BundleHeaderProto) Reset()
- func (m *BundleHeaderProto) Size() (n int)
- func (m *BundleHeaderProto) String() string
- func (m *BundleHeaderProto) Unmarshal(dAtA []byte) error
- type BundleHeaderProto_Endianness
- type CleanupAllRequest
- func (*CleanupAllRequest) Descriptor() ([]byte, []int)
- func (m *CleanupAllRequest) GetContainer() []string
- func (m *CleanupAllRequest) Marshal() (dAtA []byte, err error)
- func (m *CleanupAllRequest) MarshalTo(dAtA []byte) (int, error)
- func (*CleanupAllRequest) ProtoMessage()
- func (m *CleanupAllRequest) Reset()
- func (m *CleanupAllRequest) Size() (n int)
- func (m *CleanupAllRequest) String() string
- func (m *CleanupAllRequest) Unmarshal(dAtA []byte) error
- type CleanupAllResponse
- func (*CleanupAllResponse) Descriptor() ([]byte, []int)
- func (m *CleanupAllResponse) Marshal() (dAtA []byte, err error)
- func (m *CleanupAllResponse) MarshalTo(dAtA []byte) (int, error)
- func (*CleanupAllResponse) ProtoMessage()
- func (m *CleanupAllResponse) Reset()
- func (m *CleanupAllResponse) Size() (n int)
- func (m *CleanupAllResponse) String() string
- func (m *CleanupAllResponse) Unmarshal(dAtA []byte) error
- type CleanupGraphRequest
- func (*CleanupGraphRequest) Descriptor() ([]byte, []int)
- func (m *CleanupGraphRequest) GetStepId() int64
- func (m *CleanupGraphRequest) Marshal() (dAtA []byte, err error)
- func (m *CleanupGraphRequest) MarshalTo(dAtA []byte) (int, error)
- func (*CleanupGraphRequest) ProtoMessage()
- func (m *CleanupGraphRequest) Reset()
- func (m *CleanupGraphRequest) Size() (n int)
- func (m *CleanupGraphRequest) String() string
- func (m *CleanupGraphRequest) Unmarshal(dAtA []byte) error
- type CleanupGraphResponse
- func (*CleanupGraphResponse) Descriptor() ([]byte, []int)
- func (m *CleanupGraphResponse) Marshal() (dAtA []byte, err error)
- func (m *CleanupGraphResponse) MarshalTo(dAtA []byte) (int, error)
- func (*CleanupGraphResponse) ProtoMessage()
- func (m *CleanupGraphResponse) Reset()
- func (m *CleanupGraphResponse) Size() (n int)
- func (m *CleanupGraphResponse) String() string
- func (m *CleanupGraphResponse) Unmarshal(dAtA []byte) error
- type CloseSessionRequest
- func (*CloseSessionRequest) Descriptor() ([]byte, []int)
- func (m *CloseSessionRequest) GetSessionHandle() string
- func (m *CloseSessionRequest) Marshal() (dAtA []byte, err error)
- func (m *CloseSessionRequest) MarshalTo(dAtA []byte) (int, error)
- func (*CloseSessionRequest) ProtoMessage()
- func (m *CloseSessionRequest) Reset()
- func (m *CloseSessionRequest) Size() (n int)
- func (m *CloseSessionRequest) String() string
- func (m *CloseSessionRequest) Unmarshal(dAtA []byte) error
- type CloseSessionResponse
- func (*CloseSessionResponse) Descriptor() ([]byte, []int)
- func (m *CloseSessionResponse) Marshal() (dAtA []byte, err error)
- func (m *CloseSessionResponse) MarshalTo(dAtA []byte) (int, error)
- func (*CloseSessionResponse) ProtoMessage()
- func (m *CloseSessionResponse) Reset()
- func (m *CloseSessionResponse) Size() (n int)
- func (m *CloseSessionResponse) String() string
- func (m *CloseSessionResponse) Unmarshal(dAtA []byte) error
- type ClusterDef
- func (*ClusterDef) Descriptor() ([]byte, []int)
- func (m *ClusterDef) GetJob() []*JobDef
- func (m *ClusterDef) Marshal() (dAtA []byte, err error)
- func (m *ClusterDef) MarshalTo(dAtA []byte) (int, error)
- func (*ClusterDef) ProtoMessage()
- func (m *ClusterDef) Reset()
- func (m *ClusterDef) Size() (n int)
- func (m *ClusterDef) String() string
- func (m *ClusterDef) Unmarshal(dAtA []byte) error
- type CollectionDef
- func (*CollectionDef) Descriptor() ([]byte, []int)
- func (m *CollectionDef) GetAnyList() *CollectionDef_AnyList
- func (m *CollectionDef) GetBytesList() *CollectionDef_BytesList
- func (m *CollectionDef) GetFloatList() *CollectionDef_FloatList
- func (m *CollectionDef) GetInt64List() *CollectionDef_Int64List
- func (m *CollectionDef) GetKind() isCollectionDef_Kind
- func (m *CollectionDef) GetNodeList() *CollectionDef_NodeList
- func (m *CollectionDef) Marshal() (dAtA []byte, err error)
- func (m *CollectionDef) MarshalTo(dAtA []byte) (int, error)
- func (*CollectionDef) ProtoMessage()
- func (m *CollectionDef) Reset()
- func (m *CollectionDef) Size() (n int)
- func (m *CollectionDef) String() string
- func (m *CollectionDef) Unmarshal(dAtA []byte) error
- func (*CollectionDef) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, ...)
- type CollectionDef_AnyList
- func (*CollectionDef_AnyList) Descriptor() ([]byte, []int)
- func (m *CollectionDef_AnyList) GetValue() []*google_protobuf.Any
- func (m *CollectionDef_AnyList) Marshal() (dAtA []byte, err error)
- func (m *CollectionDef_AnyList) MarshalTo(dAtA []byte) (int, error)
- func (*CollectionDef_AnyList) ProtoMessage()
- func (m *CollectionDef_AnyList) Reset()
- func (m *CollectionDef_AnyList) Size() (n int)
- func (m *CollectionDef_AnyList) String() string
- func (m *CollectionDef_AnyList) Unmarshal(dAtA []byte) error
- type CollectionDef_AnyList_
- type CollectionDef_BytesList
- func (*CollectionDef_BytesList) Descriptor() ([]byte, []int)
- func (m *CollectionDef_BytesList) GetValue() [][]byte
- func (m *CollectionDef_BytesList) Marshal() (dAtA []byte, err error)
- func (m *CollectionDef_BytesList) MarshalTo(dAtA []byte) (int, error)
- func (*CollectionDef_BytesList) ProtoMessage()
- func (m *CollectionDef_BytesList) Reset()
- func (m *CollectionDef_BytesList) Size() (n int)
- func (m *CollectionDef_BytesList) String() string
- func (m *CollectionDef_BytesList) Unmarshal(dAtA []byte) error
- type CollectionDef_BytesList_
- type CollectionDef_FloatList
- func (*CollectionDef_FloatList) Descriptor() ([]byte, []int)
- func (m *CollectionDef_FloatList) GetValue() []float32
- func (m *CollectionDef_FloatList) Marshal() (dAtA []byte, err error)
- func (m *CollectionDef_FloatList) MarshalTo(dAtA []byte) (int, error)
- func (*CollectionDef_FloatList) ProtoMessage()
- func (m *CollectionDef_FloatList) Reset()
- func (m *CollectionDef_FloatList) Size() (n int)
- func (m *CollectionDef_FloatList) String() string
- func (m *CollectionDef_FloatList) Unmarshal(dAtA []byte) error
- type CollectionDef_FloatList_
- type CollectionDef_Int64List
- func (*CollectionDef_Int64List) Descriptor() ([]byte, []int)
- func (m *CollectionDef_Int64List) GetValue() []int64
- func (m *CollectionDef_Int64List) Marshal() (dAtA []byte, err error)
- func (m *CollectionDef_Int64List) MarshalTo(dAtA []byte) (int, error)
- func (*CollectionDef_Int64List) ProtoMessage()
- func (m *CollectionDef_Int64List) Reset()
- func (m *CollectionDef_Int64List) Size() (n int)
- func (m *CollectionDef_Int64List) String() string
- func (m *CollectionDef_Int64List) Unmarshal(dAtA []byte) error
- type CollectionDef_Int64List_
- type CollectionDef_NodeList
- func (*CollectionDef_NodeList) Descriptor() ([]byte, []int)
- func (m *CollectionDef_NodeList) GetValue() []string
- func (m *CollectionDef_NodeList) Marshal() (dAtA []byte, err error)
- func (m *CollectionDef_NodeList) MarshalTo(dAtA []byte) (int, error)
- func (*CollectionDef_NodeList) ProtoMessage()
- func (m *CollectionDef_NodeList) Reset()
- func (m *CollectionDef_NodeList) Size() (n int)
- func (m *CollectionDef_NodeList) String() string
- func (m *CollectionDef_NodeList) Unmarshal(dAtA []byte) error
- type CollectionDef_NodeList_
- type CondContextDef
- func (*CondContextDef) Descriptor() ([]byte, []int)
- func (m *CondContextDef) GetBranch() int32
- func (m *CondContextDef) GetContextName() string
- func (m *CondContextDef) GetPivotName() string
- func (m *CondContextDef) GetPredName() string
- func (m *CondContextDef) GetValuesDef() *ValuesDef
- func (m *CondContextDef) Marshal() (dAtA []byte, err error)
- func (m *CondContextDef) MarshalTo(dAtA []byte) (int, error)
- func (*CondContextDef) ProtoMessage()
- func (m *CondContextDef) Reset()
- func (m *CondContextDef) Size() (n int)
- func (m *CondContextDef) String() string
- func (m *CondContextDef) Unmarshal(dAtA []byte) error
- type ConfigProto
- func (*ConfigProto) Descriptor() ([]byte, []int)
- func (m *ConfigProto) GetAllowSoftPlacement() bool
- func (m *ConfigProto) GetClusterDef() *ClusterDef
- func (m *ConfigProto) GetDeviceCount() map[string]int32
- func (m *ConfigProto) GetDeviceFilters() []string
- func (m *ConfigProto) GetGpuOptions() *GPUOptions
- func (m *ConfigProto) GetGraphOptions() *GraphOptions
- func (m *ConfigProto) GetInterOpParallelismThreads() int32
- func (m *ConfigProto) GetIntraOpParallelismThreads() int32
- func (m *ConfigProto) GetLogDevicePlacement() bool
- func (m *ConfigProto) GetOperationTimeoutInMs() int64
- func (m *ConfigProto) GetPlacementPeriod() int32
- func (m *ConfigProto) GetRpcOptions() *RPCOptions
- func (m *ConfigProto) GetSessionInterOpThreadPool() []*ThreadPoolOptionProto
- func (m *ConfigProto) GetUsePerSessionThreads() bool
- func (m *ConfigProto) Marshal() (dAtA []byte, err error)
- func (m *ConfigProto) MarshalTo(dAtA []byte) (int, error)
- func (*ConfigProto) ProtoMessage()
- func (m *ConfigProto) Reset()
- func (m *ConfigProto) Size() (n int)
- func (m *ConfigProto) String() string
- func (m *ConfigProto) Unmarshal(dAtA []byte) error
- type CostGraphDef
- func (*CostGraphDef) Descriptor() ([]byte, []int)
- func (m *CostGraphDef) GetNode() []*CostGraphDef_Node
- func (m *CostGraphDef) Marshal() (dAtA []byte, err error)
- func (m *CostGraphDef) MarshalTo(dAtA []byte) (int, error)
- func (*CostGraphDef) ProtoMessage()
- func (m *CostGraphDef) Reset()
- func (m *CostGraphDef) Size() (n int)
- func (m *CostGraphDef) String() string
- func (m *CostGraphDef) Unmarshal(dAtA []byte) error
- type CostGraphDef_Node
- func (*CostGraphDef_Node) Descriptor() ([]byte, []int)
- func (m *CostGraphDef_Node) GetComputeCost() int64
- func (m *CostGraphDef_Node) GetComputeTime() int64
- func (m *CostGraphDef_Node) GetControlInput() []int32
- func (m *CostGraphDef_Node) GetDevice() string
- func (m *CostGraphDef_Node) GetDevicePersistentMemorySize() int64
- func (m *CostGraphDef_Node) GetDeviceTempMemorySize() int64
- func (m *CostGraphDef_Node) GetHostPersistentMemorySize() int64
- func (m *CostGraphDef_Node) GetHostTempMemorySize() int64
- func (m *CostGraphDef_Node) GetId() int32
- func (m *CostGraphDef_Node) GetInputInfo() []*CostGraphDef_Node_InputInfo
- func (m *CostGraphDef_Node) GetIsFinal() bool
- func (m *CostGraphDef_Node) GetMemoryTime() int64
- func (m *CostGraphDef_Node) GetName() string
- func (m *CostGraphDef_Node) GetOutputInfo() []*CostGraphDef_Node_OutputInfo
- func (m *CostGraphDef_Node) GetTemporaryMemorySize() int64
- func (m *CostGraphDef_Node) Marshal() (dAtA []byte, err error)
- func (m *CostGraphDef_Node) MarshalTo(dAtA []byte) (int, error)
- func (*CostGraphDef_Node) ProtoMessage()
- func (m *CostGraphDef_Node) Reset()
- func (m *CostGraphDef_Node) Size() (n int)
- func (m *CostGraphDef_Node) String() string
- func (m *CostGraphDef_Node) Unmarshal(dAtA []byte) error
- type CostGraphDef_Node_InputInfo
- func (*CostGraphDef_Node_InputInfo) Descriptor() ([]byte, []int)
- func (m *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32
- func (m *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32
- func (m *CostGraphDef_Node_InputInfo) Marshal() (dAtA []byte, err error)
- func (m *CostGraphDef_Node_InputInfo) MarshalTo(dAtA []byte) (int, error)
- func (*CostGraphDef_Node_InputInfo) ProtoMessage()
- func (m *CostGraphDef_Node_InputInfo) Reset()
- func (m *CostGraphDef_Node_InputInfo) Size() (n int)
- func (m *CostGraphDef_Node_InputInfo) String() string
- func (m *CostGraphDef_Node_InputInfo) Unmarshal(dAtA []byte) error
- type CostGraphDef_Node_OutputInfo
- func (*CostGraphDef_Node_OutputInfo) Descriptor() ([]byte, []int)
- func (m *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64
- func (m *CostGraphDef_Node_OutputInfo) GetDtype() DataType
- func (m *CostGraphDef_Node_OutputInfo) GetShape() *TensorShapeProto
- func (m *CostGraphDef_Node_OutputInfo) GetSize_() int64
- func (m *CostGraphDef_Node_OutputInfo) Marshal() (dAtA []byte, err error)
- func (m *CostGraphDef_Node_OutputInfo) MarshalTo(dAtA []byte) (int, error)
- func (*CostGraphDef_Node_OutputInfo) ProtoMessage()
- func (m *CostGraphDef_Node_OutputInfo) Reset()
- func (m *CostGraphDef_Node_OutputInfo) Size() (n int)
- func (m *CostGraphDef_Node_OutputInfo) String() string
- func (m *CostGraphDef_Node_OutputInfo) Unmarshal(dAtA []byte) error
- type CreateSessionRequest
- func (*CreateSessionRequest) Descriptor() ([]byte, []int)
- func (m *CreateSessionRequest) GetConfig() *ConfigProto
- func (m *CreateSessionRequest) GetGraphDef() *GraphDef
- func (m *CreateSessionRequest) GetTarget() string
- func (m *CreateSessionRequest) Marshal() (dAtA []byte, err error)
- func (m *CreateSessionRequest) MarshalTo(dAtA []byte) (int, error)
- func (*CreateSessionRequest) ProtoMessage()
- func (m *CreateSessionRequest) Reset()
- func (m *CreateSessionRequest) Size() (n int)
- func (m *CreateSessionRequest) String() string
- func (m *CreateSessionRequest) Unmarshal(dAtA []byte) error
- type CreateSessionResponse
- func (*CreateSessionResponse) Descriptor() ([]byte, []int)
- func (m *CreateSessionResponse) GetGraphVersion() int64
- func (m *CreateSessionResponse) GetSessionHandle() string
- func (m *CreateSessionResponse) Marshal() (dAtA []byte, err error)
- func (m *CreateSessionResponse) MarshalTo(dAtA []byte) (int, error)
- func (*CreateSessionResponse) ProtoMessage()
- func (m *CreateSessionResponse) Reset()
- func (m *CreateSessionResponse) Size() (n int)
- func (m *CreateSessionResponse) String() string
- func (m *CreateSessionResponse) Unmarshal(dAtA []byte) error
- type CreateWorkerSessionRequest
- func (*CreateWorkerSessionRequest) Descriptor() ([]byte, []int)
- func (m *CreateWorkerSessionRequest) GetServerDef() *ServerDef
- func (m *CreateWorkerSessionRequest) GetSessionHandle() string
- func (m *CreateWorkerSessionRequest) Marshal() (dAtA []byte, err error)
- func (m *CreateWorkerSessionRequest) MarshalTo(dAtA []byte) (int, error)
- func (*CreateWorkerSessionRequest) ProtoMessage()
- func (m *CreateWorkerSessionRequest) Reset()
- func (m *CreateWorkerSessionRequest) Size() (n int)
- func (m *CreateWorkerSessionRequest) String() string
- func (m *CreateWorkerSessionRequest) Unmarshal(dAtA []byte) error
- type CreateWorkerSessionResponse
- func (*CreateWorkerSessionResponse) Descriptor() ([]byte, []int)
- func (m *CreateWorkerSessionResponse) Marshal() (dAtA []byte, err error)
- func (m *CreateWorkerSessionResponse) MarshalTo(dAtA []byte) (int, error)
- func (*CreateWorkerSessionResponse) ProtoMessage()
- func (m *CreateWorkerSessionResponse) Reset()
- func (m *CreateWorkerSessionResponse) Size() (n int)
- func (m *CreateWorkerSessionResponse) String() string
- func (m *CreateWorkerSessionResponse) Unmarshal(dAtA []byte) error
- type DataType
- type DebugOptions
- func (*DebugOptions) Descriptor() ([]byte, []int)
- func (m *DebugOptions) GetDebugTensorWatchOpts() []*DebugTensorWatch
- func (m *DebugOptions) GetGlobalStep() int64
- func (m *DebugOptions) Marshal() (dAtA []byte, err error)
- func (m *DebugOptions) MarshalTo(dAtA []byte) (int, error)
- func (*DebugOptions) ProtoMessage()
- func (m *DebugOptions) Reset()
- func (m *DebugOptions) Size() (n int)
- func (m *DebugOptions) String() string
- func (m *DebugOptions) Unmarshal(dAtA []byte) error
- type DebugTensorWatch
- func (*DebugTensorWatch) Descriptor() ([]byte, []int)
- func (m *DebugTensorWatch) GetDebugOps() []string
- func (m *DebugTensorWatch) GetDebugUrls() []string
- func (m *DebugTensorWatch) GetNodeName() string
- func (m *DebugTensorWatch) GetOutputSlot() int32
- func (m *DebugTensorWatch) GetTolerateDebugOpCreationFailures() bool
- func (m *DebugTensorWatch) Marshal() (dAtA []byte, err error)
- func (m *DebugTensorWatch) MarshalTo(dAtA []byte) (int, error)
- func (*DebugTensorWatch) ProtoMessage()
- func (m *DebugTensorWatch) Reset()
- func (m *DebugTensorWatch) Size() (n int)
- func (m *DebugTensorWatch) String() string
- func (m *DebugTensorWatch) Unmarshal(dAtA []byte) error
- type DeregisterGraphRequest
- func (*DeregisterGraphRequest) Descriptor() ([]byte, []int)
- func (m *DeregisterGraphRequest) GetGraphHandle() string
- func (m *DeregisterGraphRequest) GetSessionHandle() string
- func (m *DeregisterGraphRequest) Marshal() (dAtA []byte, err error)
- func (m *DeregisterGraphRequest) MarshalTo(dAtA []byte) (int, error)
- func (*DeregisterGraphRequest) ProtoMessage()
- func (m *DeregisterGraphRequest) Reset()
- func (m *DeregisterGraphRequest) Size() (n int)
- func (m *DeregisterGraphRequest) String() string
- func (m *DeregisterGraphRequest) Unmarshal(dAtA []byte) error
- type DeregisterGraphResponse
- func (*DeregisterGraphResponse) Descriptor() ([]byte, []int)
- func (m *DeregisterGraphResponse) Marshal() (dAtA []byte, err error)
- func (m *DeregisterGraphResponse) MarshalTo(dAtA []byte) (int, error)
- func (*DeregisterGraphResponse) ProtoMessage()
- func (m *DeregisterGraphResponse) Reset()
- func (m *DeregisterGraphResponse) Size() (n int)
- func (m *DeregisterGraphResponse) String() string
- func (m *DeregisterGraphResponse) Unmarshal(dAtA []byte) error
- type DeviceAttributes
- func (*DeviceAttributes) Descriptor() ([]byte, []int)
- func (m *DeviceAttributes) GetDeviceType() string
- func (m *DeviceAttributes) GetIncarnation() uint64
- func (m *DeviceAttributes) GetLocality() *DeviceLocality
- func (m *DeviceAttributes) GetMemoryLimit() int64
- func (m *DeviceAttributes) GetName() string
- func (m *DeviceAttributes) GetPhysicalDeviceDesc() string
- func (m *DeviceAttributes) Marshal() (dAtA []byte, err error)
- func (m *DeviceAttributes) MarshalTo(dAtA []byte) (int, error)
- func (*DeviceAttributes) ProtoMessage()
- func (m *DeviceAttributes) Reset()
- func (m *DeviceAttributes) Size() (n int)
- func (m *DeviceAttributes) String() string
- func (m *DeviceAttributes) Unmarshal(dAtA []byte) error
- type DeviceLocality
- func (*DeviceLocality) Descriptor() ([]byte, []int)
- func (m *DeviceLocality) GetBusId() int32
- func (m *DeviceLocality) Marshal() (dAtA []byte, err error)
- func (m *DeviceLocality) MarshalTo(dAtA []byte) (int, error)
- func (*DeviceLocality) ProtoMessage()
- func (m *DeviceLocality) Reset()
- func (m *DeviceLocality) Size() (n int)
- func (m *DeviceLocality) String() string
- func (m *DeviceLocality) Unmarshal(dAtA []byte) error
- type DeviceProperties
- func (*DeviceProperties) Descriptor() ([]byte, []int)
- func (m *DeviceProperties) GetBandwidth() int64
- func (m *DeviceProperties) GetEnvironment() map[string]string
- func (m *DeviceProperties) GetFrequency() int64
- func (m *DeviceProperties) GetL1CacheSize() int64
- func (m *DeviceProperties) GetL2CacheSize() int64
- func (m *DeviceProperties) GetL3CacheSize() int64
- func (m *DeviceProperties) GetMemorySize() int64
- func (m *DeviceProperties) GetModel() string
- func (m *DeviceProperties) GetNumCores() int64
- func (m *DeviceProperties) GetNumRegisters() int64
- func (m *DeviceProperties) GetSharedMemorySizePerMultiprocessor() int64
- func (m *DeviceProperties) GetType() string
- func (m *DeviceProperties) GetVendor() string
- func (m *DeviceProperties) Marshal() (dAtA []byte, err error)
- func (m *DeviceProperties) MarshalTo(dAtA []byte) (int, error)
- func (*DeviceProperties) ProtoMessage()
- func (m *DeviceProperties) Reset()
- func (m *DeviceProperties) Size() (n int)
- func (m *DeviceProperties) String() string
- func (m *DeviceProperties) Unmarshal(dAtA []byte) error
- type DeviceStepStats
- func (*DeviceStepStats) Descriptor() ([]byte, []int)
- func (m *DeviceStepStats) GetDevice() string
- func (m *DeviceStepStats) GetNodeStats() []*NodeExecStats
- func (m *DeviceStepStats) Marshal() (dAtA []byte, err error)
- func (m *DeviceStepStats) MarshalTo(dAtA []byte) (int, error)
- func (*DeviceStepStats) ProtoMessage()
- func (m *DeviceStepStats) Reset()
- func (m *DeviceStepStats) Size() (n int)
- func (m *DeviceStepStats) String() string
- func (m *DeviceStepStats) Unmarshal(dAtA []byte) error
- type Error
- func (*Error) Descriptor() ([]byte, []int)
- func (m *Error) Marshal() (dAtA []byte, err error)
- func (m *Error) MarshalTo(dAtA []byte) (int, error)
- func (*Error) ProtoMessage()
- func (m *Error) Reset()
- func (m *Error) Size() (n int)
- func (m *Error) String() string
- func (m *Error) Unmarshal(dAtA []byte) error
- type Error_Code
- type ExecutorOpts
- func (*ExecutorOpts) Descriptor() ([]byte, []int)
- func (m *ExecutorOpts) GetRecordCosts() bool
- func (m *ExecutorOpts) GetRecordTimeline() bool
- func (m *ExecutorOpts) Marshal() (dAtA []byte, err error)
- func (m *ExecutorOpts) MarshalTo(dAtA []byte) (int, error)
- func (*ExecutorOpts) ProtoMessage()
- func (m *ExecutorOpts) Reset()
- func (m *ExecutorOpts) Size() (n int)
- func (m *ExecutorOpts) String() string
- func (m *ExecutorOpts) Unmarshal(dAtA []byte) error
- type ExtendSessionRequest
- func (*ExtendSessionRequest) Descriptor() ([]byte, []int)
- func (m *ExtendSessionRequest) GetCurrentGraphVersion() int64
- func (m *ExtendSessionRequest) GetGraphDef() *GraphDef
- func (m *ExtendSessionRequest) GetSessionHandle() string
- func (m *ExtendSessionRequest) Marshal() (dAtA []byte, err error)
- func (m *ExtendSessionRequest) MarshalTo(dAtA []byte) (int, error)
- func (*ExtendSessionRequest) ProtoMessage()
- func (m *ExtendSessionRequest) Reset()
- func (m *ExtendSessionRequest) Size() (n int)
- func (m *ExtendSessionRequest) String() string
- func (m *ExtendSessionRequest) Unmarshal(dAtA []byte) error
- type ExtendSessionResponse
- func (*ExtendSessionResponse) Descriptor() ([]byte, []int)
- func (m *ExtendSessionResponse) GetNewGraphVersion() int64
- func (m *ExtendSessionResponse) Marshal() (dAtA []byte, err error)
- func (m *ExtendSessionResponse) MarshalTo(dAtA []byte) (int, error)
- func (*ExtendSessionResponse) ProtoMessage()
- func (m *ExtendSessionResponse) Reset()
- func (m *ExtendSessionResponse) Size() (n int)
- func (m *ExtendSessionResponse) String() string
- func (m *ExtendSessionResponse) Unmarshal(dAtA []byte) error
- type FunctionDef
- func (*FunctionDef) Descriptor() ([]byte, []int)
- func (m *FunctionDef) GetAttr() map[string]*AttrValue
- func (m *FunctionDef) GetNodeDef() []*NodeDef
- func (m *FunctionDef) GetRet() map[string]string
- func (m *FunctionDef) GetSignature() *OpDef
- func (m *FunctionDef) Marshal() (dAtA []byte, err error)
- func (m *FunctionDef) MarshalTo(dAtA []byte) (int, error)
- func (*FunctionDef) ProtoMessage()
- func (m *FunctionDef) Reset()
- func (m *FunctionDef) Size() (n int)
- func (m *FunctionDef) String() string
- func (m *FunctionDef) Unmarshal(dAtA []byte) error
- type FunctionDefLibrary
- func (*FunctionDefLibrary) Descriptor() ([]byte, []int)
- func (m *FunctionDefLibrary) GetFunction() []*FunctionDef
- func (m *FunctionDefLibrary) GetGradient() []*GradientDef
- func (m *FunctionDefLibrary) Marshal() (dAtA []byte, err error)
- func (m *FunctionDefLibrary) MarshalTo(dAtA []byte) (int, error)
- func (*FunctionDefLibrary) ProtoMessage()
- func (m *FunctionDefLibrary) Reset()
- func (m *FunctionDefLibrary) Size() (n int)
- func (m *FunctionDefLibrary) String() string
- func (m *FunctionDefLibrary) Unmarshal(dAtA []byte) error
- type GPUOptions
- func (*GPUOptions) Descriptor() ([]byte, []int)
- func (m *GPUOptions) GetAllocatorType() string
- func (m *GPUOptions) GetAllowGrowth() bool
- func (m *GPUOptions) GetDeferredDeletionBytes() int64
- func (m *GPUOptions) GetForceGpuCompatible() bool
- func (m *GPUOptions) GetPerProcessGpuMemoryFraction() float64
- func (m *GPUOptions) GetPollingActiveDelayUsecs() int32
- func (m *GPUOptions) GetPollingInactiveDelayMsecs() int32
- func (m *GPUOptions) GetVisibleDeviceList() string
- func (m *GPUOptions) Marshal() (dAtA []byte, err error)
- func (m *GPUOptions) MarshalTo(dAtA []byte) (int, error)
- func (*GPUOptions) ProtoMessage()
- func (m *GPUOptions) Reset()
- func (m *GPUOptions) Size() (n int)
- func (m *GPUOptions) String() string
- func (m *GPUOptions) Unmarshal(dAtA []byte) error
- type GetStatusRequest
- func (*GetStatusRequest) Descriptor() ([]byte, []int)
- func (m *GetStatusRequest) Marshal() (dAtA []byte, err error)
- func (m *GetStatusRequest) MarshalTo(dAtA []byte) (int, error)
- func (*GetStatusRequest) ProtoMessage()
- func (m *GetStatusRequest) Reset()
- func (m *GetStatusRequest) Size() (n int)
- func (m *GetStatusRequest) String() string
- func (m *GetStatusRequest) Unmarshal(dAtA []byte) error
- type GetStatusResponse
- func (*GetStatusResponse) Descriptor() ([]byte, []int)
- func (m *GetStatusResponse) GetDeviceAttributes() []*DeviceAttributes
- func (m *GetStatusResponse) Marshal() (dAtA []byte, err error)
- func (m *GetStatusResponse) MarshalTo(dAtA []byte) (int, error)
- func (*GetStatusResponse) ProtoMessage()
- func (m *GetStatusResponse) Reset()
- func (m *GetStatusResponse) Size() (n int)
- func (m *GetStatusResponse) String() string
- func (m *GetStatusResponse) Unmarshal(dAtA []byte) error
- type GradientDef
- func (*GradientDef) Descriptor() ([]byte, []int)
- func (m *GradientDef) GetFunctionName() string
- func (m *GradientDef) GetGradientFunc() string
- func (m *GradientDef) Marshal() (dAtA []byte, err error)
- func (m *GradientDef) MarshalTo(dAtA []byte) (int, error)
- func (*GradientDef) ProtoMessage()
- func (m *GradientDef) Reset()
- func (m *GradientDef) Size() (n int)
- func (m *GradientDef) String() string
- func (m *GradientDef) Unmarshal(dAtA []byte) error
- type GraphDef
- func (*GraphDef) Descriptor() ([]byte, []int)
- func (m *GraphDef) GetLibrary() *FunctionDefLibrary
- func (m *GraphDef) GetNode() []*NodeDef
- func (m *GraphDef) GetVersion() int32
- func (m *GraphDef) GetVersions() *VersionDef
- func (m *GraphDef) Marshal() (dAtA []byte, err error)
- func (m *GraphDef) MarshalTo(dAtA []byte) (int, error)
- func (*GraphDef) ProtoMessage()
- func (m *GraphDef) Reset()
- func (m *GraphDef) Size() (n int)
- func (m *GraphDef) String() string
- func (m *GraphDef) Unmarshal(dAtA []byte) error
- type GraphOptions
- func (*GraphOptions) Descriptor() ([]byte, []int)
- func (m *GraphOptions) GetBuildCostModel() int64
- func (m *GraphOptions) GetBuildCostModelAfter() int64
- func (m *GraphOptions) GetEnableBfloat16Sendrecv() bool
- func (m *GraphOptions) GetEnableRecvScheduling() bool
- func (m *GraphOptions) GetInferShapes() bool
- func (m *GraphOptions) GetOptimizerOptions() *OptimizerOptions
- func (m *GraphOptions) GetPlacePrunedGraph() bool
- func (m *GraphOptions) GetRewriteOptions() *RewriterConfig
- func (m *GraphOptions) GetTimelineStep() int32
- func (m *GraphOptions) Marshal() (dAtA []byte, err error)
- func (m *GraphOptions) MarshalTo(dAtA []byte) (int, error)
- func (*GraphOptions) ProtoMessage()
- func (m *GraphOptions) Reset()
- func (m *GraphOptions) Size() (n int)
- func (m *GraphOptions) String() string
- func (m *GraphOptions) Unmarshal(dAtA []byte) error
- type GraphTransferInfo
- func (*GraphTransferInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo) GetConstNodeInfo() []*GraphTransferInfo_ConstNodeInfo
- func (m *GraphTransferInfo) GetDestination() GraphTransferInfo_Destination
- func (m *GraphTransferInfo) GetGraphInputNodeInfo() []*GraphTransferInfo_GraphInputNodeInfo
- func (m *GraphTransferInfo) GetGraphOutputNodeInfo() []*GraphTransferInfo_GraphOutputNodeInfo
- func (m *GraphTransferInfo) GetNodeInfo() []*GraphTransferInfo_NodeInfo
- func (m *GraphTransferInfo) GetNodeInputInfo() []*GraphTransferInfo_NodeInputInfo
- func (m *GraphTransferInfo) GetNodeOutputInfo() []*GraphTransferInfo_NodeOutputInfo
- func (m *GraphTransferInfo) Marshal() (dAtA []byte, err error)
- func (m *GraphTransferInfo) MarshalTo(dAtA []byte) (int, error)
- func (*GraphTransferInfo) ProtoMessage()
- func (m *GraphTransferInfo) Reset()
- func (m *GraphTransferInfo) Size() (n int)
- func (m *GraphTransferInfo) String() string
- func (m *GraphTransferInfo) Unmarshal(dAtA []byte) error
- type GraphTransferInfo_ConstNodeInfo
- func (*GraphTransferInfo_ConstNodeInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_ConstNodeInfo) GetData() []byte
- func (m *GraphTransferInfo_ConstNodeInfo) GetDtype() DataType
- func (m *GraphTransferInfo_ConstNodeInfo) GetName() string
- func (m *GraphTransferInfo_ConstNodeInfo) GetNodeId() int32
- func (m *GraphTransferInfo_ConstNodeInfo) GetShape() []int64
- func (m *GraphTransferInfo_ConstNodeInfo) Marshal() (dAtA []byte, err error)
- func (m *GraphTransferInfo_ConstNodeInfo) MarshalTo(dAtA []byte) (int, error)
- func (*GraphTransferInfo_ConstNodeInfo) ProtoMessage()
- func (m *GraphTransferInfo_ConstNodeInfo) Reset()
- func (m *GraphTransferInfo_ConstNodeInfo) Size() (n int)
- func (m *GraphTransferInfo_ConstNodeInfo) String() string
- func (m *GraphTransferInfo_ConstNodeInfo) Unmarshal(dAtA []byte) error
- type GraphTransferInfo_Destination
- type GraphTransferInfo_GraphInputNodeInfo
- func (*GraphTransferInfo_GraphInputNodeInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_GraphInputNodeInfo) GetDtype() DataType
- func (m *GraphTransferInfo_GraphInputNodeInfo) GetName() string
- func (m *GraphTransferInfo_GraphInputNodeInfo) GetShape() []int64
- func (m *GraphTransferInfo_GraphInputNodeInfo) Marshal() (dAtA []byte, err error)
- func (m *GraphTransferInfo_GraphInputNodeInfo) MarshalTo(dAtA []byte) (int, error)
- func (*GraphTransferInfo_GraphInputNodeInfo) ProtoMessage()
- func (m *GraphTransferInfo_GraphInputNodeInfo) Reset()
- func (m *GraphTransferInfo_GraphInputNodeInfo) Size() (n int)
- func (m *GraphTransferInfo_GraphInputNodeInfo) String() string
- func (m *GraphTransferInfo_GraphInputNodeInfo) Unmarshal(dAtA []byte) error
- type GraphTransferInfo_GraphOutputNodeInfo
- func (*GraphTransferInfo_GraphOutputNodeInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_GraphOutputNodeInfo) GetDtype() DataType
- func (m *GraphTransferInfo_GraphOutputNodeInfo) GetName() string
- func (m *GraphTransferInfo_GraphOutputNodeInfo) GetShape() []int64
- func (m *GraphTransferInfo_GraphOutputNodeInfo) Marshal() (dAtA []byte, err error)
- func (m *GraphTransferInfo_GraphOutputNodeInfo) MarshalTo(dAtA []byte) (int, error)
- func (*GraphTransferInfo_GraphOutputNodeInfo) ProtoMessage()
- func (m *GraphTransferInfo_GraphOutputNodeInfo) Reset()
- func (m *GraphTransferInfo_GraphOutputNodeInfo) Size() (n int)
- func (m *GraphTransferInfo_GraphOutputNodeInfo) String() string
- func (m *GraphTransferInfo_GraphOutputNodeInfo) Unmarshal(dAtA []byte) error
- type GraphTransferInfo_NodeInfo
- func (*GraphTransferInfo_NodeInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_NodeInfo) GetInputCount() int32
- func (m *GraphTransferInfo_NodeInfo) GetName() string
- func (m *GraphTransferInfo_NodeInfo) GetNodeId() int32
- func (m *GraphTransferInfo_NodeInfo) GetOutputCount() int32
- func (m *GraphTransferInfo_NodeInfo) GetPaddingId() int32
- func (m *GraphTransferInfo_NodeInfo) GetSocOpId() int32
- func (m *GraphTransferInfo_NodeInfo) GetTypeName() string
- func (m *GraphTransferInfo_NodeInfo) Marshal() (dAtA []byte, err error)
- func (m *GraphTransferInfo_NodeInfo) MarshalTo(dAtA []byte) (int, error)
- func (*GraphTransferInfo_NodeInfo) ProtoMessage()
- func (m *GraphTransferInfo_NodeInfo) Reset()
- func (m *GraphTransferInfo_NodeInfo) Size() (n int)
- func (m *GraphTransferInfo_NodeInfo) String() string
- func (m *GraphTransferInfo_NodeInfo) Unmarshal(dAtA []byte) error
- type GraphTransferInfo_NodeInput
- func (*GraphTransferInfo_NodeInput) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_NodeInput) GetNodeId() int32
- func (m *GraphTransferInfo_NodeInput) GetOutputPort() int32
- func (m *GraphTransferInfo_NodeInput) Marshal() (dAtA []byte, err error)
- func (m *GraphTransferInfo_NodeInput) MarshalTo(dAtA []byte) (int, error)
- func (*GraphTransferInfo_NodeInput) ProtoMessage()
- func (m *GraphTransferInfo_NodeInput) Reset()
- func (m *GraphTransferInfo_NodeInput) Size() (n int)
- func (m *GraphTransferInfo_NodeInput) String() string
- func (m *GraphTransferInfo_NodeInput) Unmarshal(dAtA []byte) error
- type GraphTransferInfo_NodeInputInfo
- func (*GraphTransferInfo_NodeInputInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_NodeInputInfo) GetNodeId() int32
- func (m *GraphTransferInfo_NodeInputInfo) GetNodeInput() []*GraphTransferInfo_NodeInput
- func (m *GraphTransferInfo_NodeInputInfo) Marshal() (dAtA []byte, err error)
- func (m *GraphTransferInfo_NodeInputInfo) MarshalTo(dAtA []byte) (int, error)
- func (*GraphTransferInfo_NodeInputInfo) ProtoMessage()
- func (m *GraphTransferInfo_NodeInputInfo) Reset()
- func (m *GraphTransferInfo_NodeInputInfo) Size() (n int)
- func (m *GraphTransferInfo_NodeInputInfo) String() string
- func (m *GraphTransferInfo_NodeInputInfo) Unmarshal(dAtA []byte) error
- type GraphTransferInfo_NodeOutputInfo
- func (*GraphTransferInfo_NodeOutputInfo) Descriptor() ([]byte, []int)
- func (m *GraphTransferInfo_NodeOutputInfo) GetMaxByteSize() []int32
- func (m *GraphTransferInfo_NodeOutputInfo) GetNodeId() int32
- func (m *GraphTransferInfo_NodeOutputInfo) Marshal() (dAtA []byte, err error)
- func (m *GraphTransferInfo_NodeOutputInfo) MarshalTo(dAtA []byte) (int, error)
- func (*GraphTransferInfo_NodeOutputInfo) ProtoMessage()
- func (m *GraphTransferInfo_NodeOutputInfo) Reset()
- func (m *GraphTransferInfo_NodeOutputInfo) Size() (n int)
- func (m *GraphTransferInfo_NodeOutputInfo) String() string
- func (m *GraphTransferInfo_NodeOutputInfo) Unmarshal(dAtA []byte) error
- type HistogramProto
- func (*HistogramProto) Descriptor() ([]byte, []int)
- func (m *HistogramProto) GetBucket() []float64
- func (m *HistogramProto) GetBucketLimit() []float64
- func (m *HistogramProto) GetMax() float64
- func (m *HistogramProto) GetMin() float64
- func (m *HistogramProto) GetNum() float64
- func (m *HistogramProto) GetSum() float64
- func (m *HistogramProto) GetSumSquares() float64
- func (m *HistogramProto) Marshal() (dAtA []byte, err error)
- func (m *HistogramProto) MarshalTo(dAtA []byte) (int, error)
- func (*HistogramProto) ProtoMessage()
- func (m *HistogramProto) Reset()
- func (m *HistogramProto) Size() (n int)
- func (m *HistogramProto) String() string
- func (m *HistogramProto) Unmarshal(dAtA []byte) error
- type JobDef
- func (*JobDef) Descriptor() ([]byte, []int)
- func (m *JobDef) GetName() string
- func (m *JobDef) GetTasks() map[int32]string
- func (m *JobDef) Marshal() (dAtA []byte, err error)
- func (m *JobDef) MarshalTo(dAtA []byte) (int, error)
- func (*JobDef) ProtoMessage()
- func (m *JobDef) Reset()
- func (m *JobDef) Size() (n int)
- func (m *JobDef) String() string
- func (m *JobDef) Unmarshal(dAtA []byte) error
- type KernelDef
- func (*KernelDef) Descriptor() ([]byte, []int)
- func (m *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
- func (m *KernelDef) GetDeviceType() string
- func (m *KernelDef) GetHostMemoryArg() []string
- func (m *KernelDef) GetLabel() string
- func (m *KernelDef) GetOp() string
- func (m *KernelDef) Marshal() (dAtA []byte, err error)
- func (m *KernelDef) MarshalTo(dAtA []byte) (int, error)
- func (*KernelDef) ProtoMessage()
- func (m *KernelDef) Reset()
- func (m *KernelDef) Size() (n int)
- func (m *KernelDef) String() string
- func (m *KernelDef) Unmarshal(dAtA []byte) error
- type KernelDef_AttrConstraint
- func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)
- func (m *KernelDef_AttrConstraint) GetAllowedValues() *AttrValue
- func (m *KernelDef_AttrConstraint) GetName() string
- func (m *KernelDef_AttrConstraint) Marshal() (dAtA []byte, err error)
- func (m *KernelDef_AttrConstraint) MarshalTo(dAtA []byte) (int, error)
- func (*KernelDef_AttrConstraint) ProtoMessage()
- func (m *KernelDef_AttrConstraint) Reset()
- func (m *KernelDef_AttrConstraint) Size() (n int)
- func (m *KernelDef_AttrConstraint) String() string
- func (m *KernelDef_AttrConstraint) Unmarshal(dAtA []byte) error
- type LabeledStepStats
- func (*LabeledStepStats) Descriptor() ([]byte, []int)
- func (m *LabeledStepStats) GetStepId() int64
- func (m *LabeledStepStats) GetStepStats() *StepStats
- func (m *LabeledStepStats) Marshal() (dAtA []byte, err error)
- func (m *LabeledStepStats) MarshalTo(dAtA []byte) (int, error)
- func (*LabeledStepStats) ProtoMessage()
- func (m *LabeledStepStats) Reset()
- func (m *LabeledStepStats) Size() (n int)
- func (m *LabeledStepStats) String() string
- func (m *LabeledStepStats) Unmarshal(dAtA []byte) error
- type ListDevicesRequest
- func (*ListDevicesRequest) Descriptor() ([]byte, []int)
- func (m *ListDevicesRequest) Marshal() (dAtA []byte, err error)
- func (m *ListDevicesRequest) MarshalTo(dAtA []byte) (int, error)
- func (*ListDevicesRequest) ProtoMessage()
- func (m *ListDevicesRequest) Reset()
- func (m *ListDevicesRequest) Size() (n int)
- func (m *ListDevicesRequest) String() string
- func (m *ListDevicesRequest) Unmarshal(dAtA []byte) error
- type ListDevicesResponse
- func (*ListDevicesResponse) Descriptor() ([]byte, []int)
- func (m *ListDevicesResponse) GetLocalDevice() []*DeviceAttributes
- func (m *ListDevicesResponse) GetRemoteDevice() []*DeviceAttributes
- func (m *ListDevicesResponse) Marshal() (dAtA []byte, err error)
- func (m *ListDevicesResponse) MarshalTo(dAtA []byte) (int, error)
- func (*ListDevicesResponse) ProtoMessage()
- func (m *ListDevicesResponse) Reset()
- func (m *ListDevicesResponse) Size() (n int)
- func (m *ListDevicesResponse) String() string
- func (m *ListDevicesResponse) Unmarshal(dAtA []byte) error
- type LoggingRequest
- func (*LoggingRequest) Descriptor() ([]byte, []int)
- func (m *LoggingRequest) GetClear() bool
- func (m *LoggingRequest) GetFetchStepId() []int64
- func (m *LoggingRequest) GetRpcLogging() bool
- func (m *LoggingRequest) Marshal() (dAtA []byte, err error)
- func (m *LoggingRequest) MarshalTo(dAtA []byte) (int, error)
- func (*LoggingRequest) ProtoMessage()
- func (m *LoggingRequest) Reset()
- func (m *LoggingRequest) Size() (n int)
- func (m *LoggingRequest) String() string
- func (m *LoggingRequest) Unmarshal(dAtA []byte) error
- type LoggingResponse
- func (*LoggingResponse) Descriptor() ([]byte, []int)
- func (m *LoggingResponse) GetStep() []*LabeledStepStats
- func (m *LoggingResponse) Marshal() (dAtA []byte, err error)
- func (m *LoggingResponse) MarshalTo(dAtA []byte) (int, error)
- func (*LoggingResponse) ProtoMessage()
- func (m *LoggingResponse) Reset()
- func (m *LoggingResponse) Size() (n int)
- func (m *LoggingResponse) String() string
- func (m *LoggingResponse) Unmarshal(dAtA []byte) error
- type MasterServiceClient
- type MasterServiceServer
- type MemoryLogRawAllocation
- func (*MemoryLogRawAllocation) Descriptor() ([]byte, []int)
- func (m *MemoryLogRawAllocation) GetAllocationId() int64
- func (m *MemoryLogRawAllocation) GetAllocatorName() string
- func (m *MemoryLogRawAllocation) GetNumBytes() int64
- func (m *MemoryLogRawAllocation) GetOperation() string
- func (m *MemoryLogRawAllocation) GetPtr() uint64
- func (m *MemoryLogRawAllocation) GetStepId() int64
- func (m *MemoryLogRawAllocation) Marshal() (dAtA []byte, err error)
- func (m *MemoryLogRawAllocation) MarshalTo(dAtA []byte) (int, error)
- func (*MemoryLogRawAllocation) ProtoMessage()
- func (m *MemoryLogRawAllocation) Reset()
- func (m *MemoryLogRawAllocation) Size() (n int)
- func (m *MemoryLogRawAllocation) String() string
- func (m *MemoryLogRawAllocation) Unmarshal(dAtA []byte) error
- type MemoryLogRawDeallocation
- func (*MemoryLogRawDeallocation) Descriptor() ([]byte, []int)
- func (m *MemoryLogRawDeallocation) GetAllocationId() int64
- func (m *MemoryLogRawDeallocation) GetAllocatorName() string
- func (m *MemoryLogRawDeallocation) GetDeferred() bool
- func (m *MemoryLogRawDeallocation) GetOperation() string
- func (m *MemoryLogRawDeallocation) GetStepId() int64
- func (m *MemoryLogRawDeallocation) Marshal() (dAtA []byte, err error)
- func (m *MemoryLogRawDeallocation) MarshalTo(dAtA []byte) (int, error)
- func (*MemoryLogRawDeallocation) ProtoMessage()
- func (m *MemoryLogRawDeallocation) Reset()
- func (m *MemoryLogRawDeallocation) Size() (n int)
- func (m *MemoryLogRawDeallocation) String() string
- func (m *MemoryLogRawDeallocation) Unmarshal(dAtA []byte) error
- type MemoryLogStep
- func (*MemoryLogStep) Descriptor() ([]byte, []int)
- func (m *MemoryLogStep) GetHandle() string
- func (m *MemoryLogStep) GetStepId() int64
- func (m *MemoryLogStep) Marshal() (dAtA []byte, err error)
- func (m *MemoryLogStep) MarshalTo(dAtA []byte) (int, error)
- func (*MemoryLogStep) ProtoMessage()
- func (m *MemoryLogStep) Reset()
- func (m *MemoryLogStep) Size() (n int)
- func (m *MemoryLogStep) String() string
- func (m *MemoryLogStep) Unmarshal(dAtA []byte) error
- type MemoryLogTensorAllocation
- func (*MemoryLogTensorAllocation) Descriptor() ([]byte, []int)
- func (m *MemoryLogTensorAllocation) GetKernelName() string
- func (m *MemoryLogTensorAllocation) GetStepId() int64
- func (m *MemoryLogTensorAllocation) GetTensor() *TensorDescription
- func (m *MemoryLogTensorAllocation) Marshal() (dAtA []byte, err error)
- func (m *MemoryLogTensorAllocation) MarshalTo(dAtA []byte) (int, error)
- func (*MemoryLogTensorAllocation) ProtoMessage()
- func (m *MemoryLogTensorAllocation) Reset()
- func (m *MemoryLogTensorAllocation) Size() (n int)
- func (m *MemoryLogTensorAllocation) String() string
- func (m *MemoryLogTensorAllocation) Unmarshal(dAtA []byte) error
- type MemoryLogTensorDeallocation
- func (*MemoryLogTensorDeallocation) Descriptor() ([]byte, []int)
- func (m *MemoryLogTensorDeallocation) GetAllocationId() int64
- func (m *MemoryLogTensorDeallocation) GetAllocatorName() string
- func (m *MemoryLogTensorDeallocation) Marshal() (dAtA []byte, err error)
- func (m *MemoryLogTensorDeallocation) MarshalTo(dAtA []byte) (int, error)
- func (*MemoryLogTensorDeallocation) ProtoMessage()
- func (m *MemoryLogTensorDeallocation) Reset()
- func (m *MemoryLogTensorDeallocation) Size() (n int)
- func (m *MemoryLogTensorDeallocation) String() string
- func (m *MemoryLogTensorDeallocation) Unmarshal(dAtA []byte) error
- type MemoryLogTensorOutput
- func (*MemoryLogTensorOutput) Descriptor() ([]byte, []int)
- func (m *MemoryLogTensorOutput) GetIndex() int32
- func (m *MemoryLogTensorOutput) GetKernelName() string
- func (m *MemoryLogTensorOutput) GetStepId() int64
- func (m *MemoryLogTensorOutput) GetTensor() *TensorDescription
- func (m *MemoryLogTensorOutput) Marshal() (dAtA []byte, err error)
- func (m *MemoryLogTensorOutput) MarshalTo(dAtA []byte) (int, error)
- func (*MemoryLogTensorOutput) ProtoMessage()
- func (m *MemoryLogTensorOutput) Reset()
- func (m *MemoryLogTensorOutput) Size() (n int)
- func (m *MemoryLogTensorOutput) String() string
- func (m *MemoryLogTensorOutput) Unmarshal(dAtA []byte) error
- type MemoryStats
- func (*MemoryStats) Descriptor() ([]byte, []int)
- func (m *MemoryStats) GetDevicePersistentMemorySize() int64
- func (m *MemoryStats) GetDevicePersistentTensorAllocIds() []int64
- func (m *MemoryStats) GetDeviceTempMemorySize() int64
- func (m *MemoryStats) GetHostPersistentMemorySize() int64
- func (m *MemoryStats) GetHostPersistentTensorAllocIds() []int64
- func (m *MemoryStats) GetHostTempMemorySize() int64
- func (m *MemoryStats) Marshal() (dAtA []byte, err error)
- func (m *MemoryStats) MarshalTo(dAtA []byte) (int, error)
- func (*MemoryStats) ProtoMessage()
- func (m *MemoryStats) Reset()
- func (m *MemoryStats) Size() (n int)
- func (m *MemoryStats) String() string
- func (m *MemoryStats) Unmarshal(dAtA []byte) error
- type MetaGraphDef
- func (*MetaGraphDef) Descriptor() ([]byte, []int)
- func (m *MetaGraphDef) GetAssetFileDef() []*AssetFileDef
- func (m *MetaGraphDef) GetCollectionDef() map[string]*CollectionDef
- func (m *MetaGraphDef) GetGraphDef() *GraphDef
- func (m *MetaGraphDef) GetMetaInfoDef() *MetaGraphDef_MetaInfoDef
- func (m *MetaGraphDef) GetSaverDef() *SaverDef
- func (m *MetaGraphDef) GetSignatureDef() map[string]*SignatureDef
- func (m *MetaGraphDef) Marshal() (dAtA []byte, err error)
- func (m *MetaGraphDef) MarshalTo(dAtA []byte) (int, error)
- func (*MetaGraphDef) ProtoMessage()
- func (m *MetaGraphDef) Reset()
- func (m *MetaGraphDef) Size() (n int)
- func (m *MetaGraphDef) String() string
- func (m *MetaGraphDef) Unmarshal(dAtA []byte) error
- type MetaGraphDef_MetaInfoDef
- func (*MetaGraphDef_MetaInfoDef) Descriptor() ([]byte, []int)
- func (m *MetaGraphDef_MetaInfoDef) GetAnyInfo() *google_protobuf.Any
- func (m *MetaGraphDef_MetaInfoDef) GetMetaGraphVersion() string
- func (m *MetaGraphDef_MetaInfoDef) GetStrippedOpList() *OpList
- func (m *MetaGraphDef_MetaInfoDef) GetTags() []string
- func (m *MetaGraphDef_MetaInfoDef) GetTensorflowGitVersion() string
- func (m *MetaGraphDef_MetaInfoDef) GetTensorflowVersion() string
- func (m *MetaGraphDef_MetaInfoDef) Marshal() (dAtA []byte, err error)
- func (m *MetaGraphDef_MetaInfoDef) MarshalTo(dAtA []byte) (int, error)
- func (*MetaGraphDef_MetaInfoDef) ProtoMessage()
- func (m *MetaGraphDef_MetaInfoDef) Reset()
- func (m *MetaGraphDef_MetaInfoDef) Size() (n int)
- func (m *MetaGraphDef_MetaInfoDef) String() string
- func (m *MetaGraphDef_MetaInfoDef) Unmarshal(dAtA []byte) error
- type NameAttrList
- func (*NameAttrList) Descriptor() ([]byte, []int)
- func (m *NameAttrList) GetAttr() map[string]*AttrValue
- func (m *NameAttrList) GetName() string
- func (m *NameAttrList) Marshal() (dAtA []byte, err error)
- func (m *NameAttrList) MarshalTo(dAtA []byte) (int, error)
- func (*NameAttrList) ProtoMessage()
- func (m *NameAttrList) Reset()
- func (m *NameAttrList) Size() (n int)
- func (m *NameAttrList) String() string
- func (m *NameAttrList) Unmarshal(dAtA []byte) error
- type NamedTensorProto
- func (*NamedTensorProto) Descriptor() ([]byte, []int)
- func (m *NamedTensorProto) GetName() string
- func (m *NamedTensorProto) GetTensor() *TensorProto
- func (m *NamedTensorProto) Marshal() (dAtA []byte, err error)
- func (m *NamedTensorProto) MarshalTo(dAtA []byte) (int, error)
- func (*NamedTensorProto) ProtoMessage()
- func (m *NamedTensorProto) Reset()
- func (m *NamedTensorProto) Size() (n int)
- func (m *NamedTensorProto) String() string
- func (m *NamedTensorProto) Unmarshal(dAtA []byte) error
- type NodeDef
- func (*NodeDef) Descriptor() ([]byte, []int)
- func (m *NodeDef) GetAttr() map[string]*AttrValue
- func (m *NodeDef) GetDevice() string
- func (m *NodeDef) GetInput() []string
- func (m *NodeDef) GetName() string
- func (m *NodeDef) GetOp() string
- func (m *NodeDef) Marshal() (dAtA []byte, err error)
- func (m *NodeDef) MarshalTo(dAtA []byte) (int, error)
- func (*NodeDef) ProtoMessage()
- func (m *NodeDef) Reset()
- func (m *NodeDef) Size() (n int)
- func (m *NodeDef) String() string
- func (m *NodeDef) Unmarshal(dAtA []byte) error
- type NodeExecStats
- func (*NodeExecStats) Descriptor() ([]byte, []int)
- func (m *NodeExecStats) GetAllEndRelMicros() int64
- func (m *NodeExecStats) GetAllStartMicros() int64
- func (m *NodeExecStats) GetMemory() []*AllocatorMemoryUsed
- func (m *NodeExecStats) GetMemoryStats() *MemoryStats
- func (m *NodeExecStats) GetNodeName() string
- func (m *NodeExecStats) GetOpEndRelMicros() int64
- func (m *NodeExecStats) GetOpStartRelMicros() int64
- func (m *NodeExecStats) GetOutput() []*NodeOutput
- func (m *NodeExecStats) GetReferencedTensor() []*AllocationDescription
- func (m *NodeExecStats) GetScheduledMicros() int64
- func (m *NodeExecStats) GetThreadId() uint32
- func (m *NodeExecStats) GetTimelineLabel() string
- func (m *NodeExecStats) Marshal() (dAtA []byte, err error)
- func (m *NodeExecStats) MarshalTo(dAtA []byte) (int, error)
- func (*NodeExecStats) ProtoMessage()
- func (m *NodeExecStats) Reset()
- func (m *NodeExecStats) Size() (n int)
- func (m *NodeExecStats) String() string
- func (m *NodeExecStats) Unmarshal(dAtA []byte) error
- type NodeOutput
- func (*NodeOutput) Descriptor() ([]byte, []int)
- func (m *NodeOutput) GetSlot() int32
- func (m *NodeOutput) GetTensorDescription() *TensorDescription
- func (m *NodeOutput) Marshal() (dAtA []byte, err error)
- func (m *NodeOutput) MarshalTo(dAtA []byte) (int, error)
- func (*NodeOutput) ProtoMessage()
- func (m *NodeOutput) Reset()
- func (m *NodeOutput) Size() (n int)
- func (m *NodeOutput) String() string
- func (m *NodeOutput) Unmarshal(dAtA []byte) error
- type OpDef
- func (*OpDef) Descriptor() ([]byte, []int)
- func (m *OpDef) GetAllowsUninitializedInput() bool
- func (m *OpDef) GetAttr() []*OpDef_AttrDef
- func (m *OpDef) GetDeprecation() *OpDeprecation
- func (m *OpDef) GetDescription() string
- func (m *OpDef) GetInputArg() []*OpDef_ArgDef
- func (m *OpDef) GetIsAggregate() bool
- func (m *OpDef) GetIsCommutative() bool
- func (m *OpDef) GetIsStateful() bool
- func (m *OpDef) GetName() string
- func (m *OpDef) GetOutputArg() []*OpDef_ArgDef
- func (m *OpDef) GetSummary() string
- func (m *OpDef) Marshal() (dAtA []byte, err error)
- func (m *OpDef) MarshalTo(dAtA []byte) (int, error)
- func (*OpDef) ProtoMessage()
- func (m *OpDef) Reset()
- func (m *OpDef) Size() (n int)
- func (m *OpDef) String() string
- func (m *OpDef) Unmarshal(dAtA []byte) error
- type OpDef_ArgDef
- func (*OpDef_ArgDef) Descriptor() ([]byte, []int)
- func (m *OpDef_ArgDef) GetDescription() string
- func (m *OpDef_ArgDef) GetIsRef() bool
- func (m *OpDef_ArgDef) GetName() string
- func (m *OpDef_ArgDef) GetNumberAttr() string
- func (m *OpDef_ArgDef) GetType() DataType
- func (m *OpDef_ArgDef) GetTypeAttr() string
- func (m *OpDef_ArgDef) GetTypeListAttr() string
- func (m *OpDef_ArgDef) Marshal() (dAtA []byte, err error)
- func (m *OpDef_ArgDef) MarshalTo(dAtA []byte) (int, error)
- func (*OpDef_ArgDef) ProtoMessage()
- func (m *OpDef_ArgDef) Reset()
- func (m *OpDef_ArgDef) Size() (n int)
- func (m *OpDef_ArgDef) String() string
- func (m *OpDef_ArgDef) Unmarshal(dAtA []byte) error
- type OpDef_AttrDef
- func (*OpDef_AttrDef) Descriptor() ([]byte, []int)
- func (m *OpDef_AttrDef) GetAllowedValues() *AttrValue
- func (m *OpDef_AttrDef) GetDefaultValue() *AttrValue
- func (m *OpDef_AttrDef) GetDescription() string
- func (m *OpDef_AttrDef) GetHasMinimum() bool
- func (m *OpDef_AttrDef) GetMinimum() int64
- func (m *OpDef_AttrDef) GetName() string
- func (m *OpDef_AttrDef) GetType() string
- func (m *OpDef_AttrDef) Marshal() (dAtA []byte, err error)
- func (m *OpDef_AttrDef) MarshalTo(dAtA []byte) (int, error)
- func (*OpDef_AttrDef) ProtoMessage()
- func (m *OpDef_AttrDef) Reset()
- func (m *OpDef_AttrDef) Size() (n int)
- func (m *OpDef_AttrDef) String() string
- func (m *OpDef_AttrDef) Unmarshal(dAtA []byte) error
- type OpDeprecation
- func (*OpDeprecation) Descriptor() ([]byte, []int)
- func (m *OpDeprecation) GetExplanation() string
- func (m *OpDeprecation) GetVersion() int32
- func (m *OpDeprecation) Marshal() (dAtA []byte, err error)
- func (m *OpDeprecation) MarshalTo(dAtA []byte) (int, error)
- func (*OpDeprecation) ProtoMessage()
- func (m *OpDeprecation) Reset()
- func (m *OpDeprecation) Size() (n int)
- func (m *OpDeprecation) String() string
- func (m *OpDeprecation) Unmarshal(dAtA []byte) error
- type OpGenOverride
- func (*OpGenOverride) Descriptor() ([]byte, []int)
- func (m *OpGenOverride) GetAlias() []string
- func (m *OpGenOverride) GetAttrDefault() []*OpGenOverride_AttrDefault
- func (m *OpGenOverride) GetAttrRename() []*OpGenOverride_Rename
- func (m *OpGenOverride) GetHide() bool
- func (m *OpGenOverride) GetInputRename() []*OpGenOverride_Rename
- func (m *OpGenOverride) GetName() string
- func (m *OpGenOverride) GetOutputRename() []*OpGenOverride_Rename
- func (m *OpGenOverride) GetRenameTo() string
- func (m *OpGenOverride) GetSkip() bool
- func (m *OpGenOverride) Marshal() (dAtA []byte, err error)
- func (m *OpGenOverride) MarshalTo(dAtA []byte) (int, error)
- func (*OpGenOverride) ProtoMessage()
- func (m *OpGenOverride) Reset()
- func (m *OpGenOverride) Size() (n int)
- func (m *OpGenOverride) String() string
- func (m *OpGenOverride) Unmarshal(dAtA []byte) error
- type OpGenOverride_AttrDefault
- func (*OpGenOverride_AttrDefault) Descriptor() ([]byte, []int)
- func (m *OpGenOverride_AttrDefault) GetName() string
- func (m *OpGenOverride_AttrDefault) GetValue() *AttrValue
- func (m *OpGenOverride_AttrDefault) Marshal() (dAtA []byte, err error)
- func (m *OpGenOverride_AttrDefault) MarshalTo(dAtA []byte) (int, error)
- func (*OpGenOverride_AttrDefault) ProtoMessage()
- func (m *OpGenOverride_AttrDefault) Reset()
- func (m *OpGenOverride_AttrDefault) Size() (n int)
- func (m *OpGenOverride_AttrDefault) String() string
- func (m *OpGenOverride_AttrDefault) Unmarshal(dAtA []byte) error
- type OpGenOverride_Rename
- func (*OpGenOverride_Rename) Descriptor() ([]byte, []int)
- func (m *OpGenOverride_Rename) GetFrom() string
- func (m *OpGenOverride_Rename) GetTo() string
- func (m *OpGenOverride_Rename) Marshal() (dAtA []byte, err error)
- func (m *OpGenOverride_Rename) MarshalTo(dAtA []byte) (int, error)
- func (*OpGenOverride_Rename) ProtoMessage()
- func (m *OpGenOverride_Rename) Reset()
- func (m *OpGenOverride_Rename) Size() (n int)
- func (m *OpGenOverride_Rename) String() string
- func (m *OpGenOverride_Rename) Unmarshal(dAtA []byte) error
- type OpGenOverrides
- func (*OpGenOverrides) Descriptor() ([]byte, []int)
- func (m *OpGenOverrides) GetOp() []*OpGenOverride
- func (m *OpGenOverrides) Marshal() (dAtA []byte, err error)
- func (m *OpGenOverrides) MarshalTo(dAtA []byte) (int, error)
- func (*OpGenOverrides) ProtoMessage()
- func (m *OpGenOverrides) Reset()
- func (m *OpGenOverrides) Size() (n int)
- func (m *OpGenOverrides) String() string
- func (m *OpGenOverrides) Unmarshal(dAtA []byte) error
- type OpList
- func (*OpList) Descriptor() ([]byte, []int)
- func (m *OpList) GetOp() []*OpDef
- func (m *OpList) Marshal() (dAtA []byte, err error)
- func (m *OpList) MarshalTo(dAtA []byte) (int, error)
- func (*OpList) ProtoMessage()
- func (m *OpList) Reset()
- func (m *OpList) Size() (n int)
- func (m *OpList) String() string
- func (m *OpList) Unmarshal(dAtA []byte) error
- type OptimizerOptions
- func (*OptimizerOptions) Descriptor() ([]byte, []int)
- func (m *OptimizerOptions) GetDoCommonSubexpressionElimination() bool
- func (m *OptimizerOptions) GetDoConstantFolding() bool
- func (m *OptimizerOptions) GetDoFunctionInlining() bool
- func (m *OptimizerOptions) GetGlobalJitLevel() OptimizerOptions_GlobalJitLevel
- func (m *OptimizerOptions) GetOptLevel() OptimizerOptions_Level
- func (m *OptimizerOptions) Marshal() (dAtA []byte, err error)
- func (m *OptimizerOptions) MarshalTo(dAtA []byte) (int, error)
- func (*OptimizerOptions) ProtoMessage()
- func (m *OptimizerOptions) Reset()
- func (m *OptimizerOptions) Size() (n int)
- func (m *OptimizerOptions) String() string
- func (m *OptimizerOptions) Unmarshal(dAtA []byte) error
- type OptimizerOptions_GlobalJitLevel
- type OptimizerOptions_Level
- type PartialRunSetupRequest
- func (*PartialRunSetupRequest) Descriptor() ([]byte, []int)
- func (m *PartialRunSetupRequest) GetFeed() []string
- func (m *PartialRunSetupRequest) GetFetch() []string
- func (m *PartialRunSetupRequest) GetSessionHandle() string
- func (m *PartialRunSetupRequest) GetTarget() []string
- func (m *PartialRunSetupRequest) Marshal() (dAtA []byte, err error)
- func (m *PartialRunSetupRequest) MarshalTo(dAtA []byte) (int, error)
- func (*PartialRunSetupRequest) ProtoMessage()
- func (m *PartialRunSetupRequest) Reset()
- func (m *PartialRunSetupRequest) Size() (n int)
- func (m *PartialRunSetupRequest) String() string
- func (m *PartialRunSetupRequest) Unmarshal(dAtA []byte) error
- type PartialRunSetupResponse
- func (*PartialRunSetupResponse) Descriptor() ([]byte, []int)
- func (m *PartialRunSetupResponse) GetPartialRunHandle() string
- func (m *PartialRunSetupResponse) Marshal() (dAtA []byte, err error)
- func (m *PartialRunSetupResponse) MarshalTo(dAtA []byte) (int, error)
- func (*PartialRunSetupResponse) ProtoMessage()
- func (m *PartialRunSetupResponse) Reset()
- func (m *PartialRunSetupResponse) Size() (n int)
- func (m *PartialRunSetupResponse) String() string
- func (m *PartialRunSetupResponse) Unmarshal(dAtA []byte) error
- type QueueRunnerDef
- func (*QueueRunnerDef) Descriptor() ([]byte, []int)
- func (m *QueueRunnerDef) GetCancelOpName() string
- func (m *QueueRunnerDef) GetCloseOpName() string
- func (m *QueueRunnerDef) GetEnqueueOpName() []string
- func (m *QueueRunnerDef) GetQueueClosedExceptionTypes() []Error_Code
- func (m *QueueRunnerDef) GetQueueName() string
- func (m *QueueRunnerDef) Marshal() (dAtA []byte, err error)
- func (m *QueueRunnerDef) MarshalTo(dAtA []byte) (int, error)
- func (*QueueRunnerDef) ProtoMessage()
- func (m *QueueRunnerDef) Reset()
- func (m *QueueRunnerDef) Size() (n int)
- func (m *QueueRunnerDef) String() string
- func (m *QueueRunnerDef) Unmarshal(dAtA []byte) error
- type RPCOptions
- func (*RPCOptions) Descriptor() ([]byte, []int)
- func (m *RPCOptions) GetUseRpcForInprocessMaster() bool
- func (m *RPCOptions) Marshal() (dAtA []byte, err error)
- func (m *RPCOptions) MarshalTo(dAtA []byte) (int, error)
- func (*RPCOptions) ProtoMessage()
- func (m *RPCOptions) Reset()
- func (m *RPCOptions) Size() (n int)
- func (m *RPCOptions) String() string
- func (m *RPCOptions) Unmarshal(dAtA []byte) error
- type ReaderBaseState
- func (*ReaderBaseState) Descriptor() ([]byte, []int)
- func (m *ReaderBaseState) GetCurrentWork() []byte
- func (m *ReaderBaseState) GetNumRecordsProduced() int64
- func (m *ReaderBaseState) GetWorkFinished() int64
- func (m *ReaderBaseState) GetWorkStarted() int64
- func (m *ReaderBaseState) Marshal() (dAtA []byte, err error)
- func (m *ReaderBaseState) MarshalTo(dAtA []byte) (int, error)
- func (*ReaderBaseState) ProtoMessage()
- func (m *ReaderBaseState) Reset()
- func (m *ReaderBaseState) Size() (n int)
- func (m *ReaderBaseState) String() string
- func (m *ReaderBaseState) Unmarshal(dAtA []byte) error
- type RecvTensorRequest
- func (*RecvTensorRequest) Descriptor() ([]byte, []int)
- func (m *RecvTensorRequest) GetClientLocality() *DeviceLocality
- func (m *RecvTensorRequest) GetDmaOk() bool
- func (m *RecvTensorRequest) GetRendezvousKey() string
- func (m *RecvTensorRequest) GetServerLocality() *DeviceLocality
- func (m *RecvTensorRequest) GetStepId() int64
- func (m *RecvTensorRequest) GetTransportOptions() *google_protobuf.Any
- func (m *RecvTensorRequest) Marshal() (dAtA []byte, err error)
- func (m *RecvTensorRequest) MarshalTo(dAtA []byte) (int, error)
- func (*RecvTensorRequest) ProtoMessage()
- func (m *RecvTensorRequest) Reset()
- func (m *RecvTensorRequest) Size() (n int)
- func (m *RecvTensorRequest) String() string
- func (m *RecvTensorRequest) Unmarshal(dAtA []byte) error
- type RecvTensorResponse
- func (*RecvTensorResponse) Descriptor() ([]byte, []int)
- func (m *RecvTensorResponse) GetIsDead() bool
- func (m *RecvTensorResponse) GetSendStartMicros() int64
- func (m *RecvTensorResponse) GetTensor() *TensorProto
- func (m *RecvTensorResponse) GetTransportOptions() *google_protobuf.Any
- func (m *RecvTensorResponse) Marshal() (dAtA []byte, err error)
- func (m *RecvTensorResponse) MarshalTo(dAtA []byte) (int, error)
- func (*RecvTensorResponse) ProtoMessage()
- func (m *RecvTensorResponse) Reset()
- func (m *RecvTensorResponse) Size() (n int)
- func (m *RecvTensorResponse) String() string
- func (m *RecvTensorResponse) Unmarshal(dAtA []byte) error
- type RegisterGraphRequest
- func (*RegisterGraphRequest) Descriptor() ([]byte, []int)
- func (m *RegisterGraphRequest) GetDebugOptions() *DebugOptions
- func (m *RegisterGraphRequest) GetGraphDef() *GraphDef
- func (m *RegisterGraphRequest) GetGraphOptions() *GraphOptions
- func (m *RegisterGraphRequest) GetHasControlFlow() bool
- func (m *RegisterGraphRequest) GetSessionHandle() string
- func (m *RegisterGraphRequest) Marshal() (dAtA []byte, err error)
- func (m *RegisterGraphRequest) MarshalTo(dAtA []byte) (int, error)
- func (*RegisterGraphRequest) ProtoMessage()
- func (m *RegisterGraphRequest) Reset()
- func (m *RegisterGraphRequest) Size() (n int)
- func (m *RegisterGraphRequest) String() string
- func (m *RegisterGraphRequest) Unmarshal(dAtA []byte) error
- type RegisterGraphResponse
- func (*RegisterGraphResponse) Descriptor() ([]byte, []int)
- func (m *RegisterGraphResponse) GetGraphHandle() string
- func (m *RegisterGraphResponse) Marshal() (dAtA []byte, err error)
- func (m *RegisterGraphResponse) MarshalTo(dAtA []byte) (int, error)
- func (*RegisterGraphResponse) ProtoMessage()
- func (m *RegisterGraphResponse) Reset()
- func (m *RegisterGraphResponse) Size() (n int)
- func (m *RegisterGraphResponse) String() string
- func (m *RegisterGraphResponse) Unmarshal(dAtA []byte) error
- type RemoteFusedGraphExecuteInfo
- func (*RemoteFusedGraphExecuteInfo) Descriptor() ([]byte, []int)
- func (m *RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (m *RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (m *RemoteFusedGraphExecuteInfo) GetExecutorName() string
- func (m *RemoteFusedGraphExecuteInfo) GetGraphInputNodeName() []string
- func (m *RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName() []string
- func (m *RemoteFusedGraphExecuteInfo) GetRemoteGraph() *GraphDef
- func (m *RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters() []byte
- func (m *RemoteFusedGraphExecuteInfo) Marshal() (dAtA []byte, err error)
- func (m *RemoteFusedGraphExecuteInfo) MarshalTo(dAtA []byte) (int, error)
- func (*RemoteFusedGraphExecuteInfo) ProtoMessage()
- func (m *RemoteFusedGraphExecuteInfo) Reset()
- func (m *RemoteFusedGraphExecuteInfo) Size() (n int)
- func (m *RemoteFusedGraphExecuteInfo) String() string
- func (m *RemoteFusedGraphExecuteInfo) Unmarshal(dAtA []byte) error
- type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
- func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor() ([]byte, []int)
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype() DataType
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape() *TensorShapeProto
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Marshal() (dAtA []byte, err error)
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) MarshalTo(dAtA []byte) (int, error)
- func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage()
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset()
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Size() (n int)
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) String() string
- func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Unmarshal(dAtA []byte) error
- type ResetRequest
- func (*ResetRequest) Descriptor() ([]byte, []int)
- func (m *ResetRequest) GetContainer() []string
- func (m *ResetRequest) GetDeviceFilters() []string
- func (m *ResetRequest) Marshal() (dAtA []byte, err error)
- func (m *ResetRequest) MarshalTo(dAtA []byte) (int, error)
- func (*ResetRequest) ProtoMessage()
- func (m *ResetRequest) Reset()
- func (m *ResetRequest) Size() (n int)
- func (m *ResetRequest) String() string
- func (m *ResetRequest) Unmarshal(dAtA []byte) error
- type ResetResponse
- func (*ResetResponse) Descriptor() ([]byte, []int)
- func (m *ResetResponse) Marshal() (dAtA []byte, err error)
- func (m *ResetResponse) MarshalTo(dAtA []byte) (int, error)
- func (*ResetResponse) ProtoMessage()
- func (m *ResetResponse) Reset()
- func (m *ResetResponse) Size() (n int)
- func (m *ResetResponse) String() string
- func (m *ResetResponse) Unmarshal(dAtA []byte) error
- type ResourceHandle
- func (*ResourceHandle) Descriptor() ([]byte, []int)
- func (m *ResourceHandle) GetContainer() string
- func (m *ResourceHandle) GetDevice() string
- func (m *ResourceHandle) GetHashCode() uint64
- func (m *ResourceHandle) GetMaybeTypeName() string
- func (m *ResourceHandle) GetName() string
- func (m *ResourceHandle) Marshal() (dAtA []byte, err error)
- func (m *ResourceHandle) MarshalTo(dAtA []byte) (int, error)
- func (*ResourceHandle) ProtoMessage()
- func (m *ResourceHandle) Reset()
- func (m *ResourceHandle) Size() (n int)
- func (m *ResourceHandle) String() string
- func (m *ResourceHandle) Unmarshal(dAtA []byte) error
- type RewriterConfig
- func (*RewriterConfig) Descriptor() ([]byte, []int)
- func (m *RewriterConfig) GetAutoParallel() *AutoParallelOptions
- func (m *RewriterConfig) GetConstantFolding() bool
- func (m *RewriterConfig) GetDisableModelPruning() bool
- func (m *RewriterConfig) GetMemoryOptimization() RewriterConfig_MemOptType
- func (m *RewriterConfig) GetOptimizeTensorLayout() bool
- func (m *RewriterConfig) GetOptimizers() []string
- func (m *RewriterConfig) Marshal() (dAtA []byte, err error)
- func (m *RewriterConfig) MarshalTo(dAtA []byte) (int, error)
- func (*RewriterConfig) ProtoMessage()
- func (m *RewriterConfig) Reset()
- func (m *RewriterConfig) Size() (n int)
- func (m *RewriterConfig) String() string
- func (m *RewriterConfig) Unmarshal(dAtA []byte) error
- type RewriterConfig_MemOptType
- type RunGraphRequest
- func (*RunGraphRequest) Descriptor() ([]byte, []int)
- func (m *RunGraphRequest) GetExecOpts() *ExecutorOpts
- func (m *RunGraphRequest) GetGraphHandle() string
- func (m *RunGraphRequest) GetIsLastPartialRun() bool
- func (m *RunGraphRequest) GetIsPartial() bool
- func (m *RunGraphRequest) GetRecvKey() []string
- func (m *RunGraphRequest) GetSend() []*NamedTensorProto
- func (m *RunGraphRequest) GetSessionHandle() string
- func (m *RunGraphRequest) GetStepId() int64
- func (m *RunGraphRequest) Marshal() (dAtA []byte, err error)
- func (m *RunGraphRequest) MarshalTo(dAtA []byte) (int, error)
- func (*RunGraphRequest) ProtoMessage()
- func (m *RunGraphRequest) Reset()
- func (m *RunGraphRequest) Size() (n int)
- func (m *RunGraphRequest) String() string
- func (m *RunGraphRequest) Unmarshal(dAtA []byte) error
- type RunGraphResponse
- func (*RunGraphResponse) Descriptor() ([]byte, []int)
- func (m *RunGraphResponse) GetCostGraph() *CostGraphDef
- func (m *RunGraphResponse) GetRecv() []*NamedTensorProto
- func (m *RunGraphResponse) GetStepStats() *StepStats
- func (m *RunGraphResponse) Marshal() (dAtA []byte, err error)
- func (m *RunGraphResponse) MarshalTo(dAtA []byte) (int, error)
- func (*RunGraphResponse) ProtoMessage()
- func (m *RunGraphResponse) Reset()
- func (m *RunGraphResponse) Size() (n int)
- func (m *RunGraphResponse) String() string
- func (m *RunGraphResponse) Unmarshal(dAtA []byte) error
- type RunMetadata
- func (*RunMetadata) Descriptor() ([]byte, []int)
- func (m *RunMetadata) GetCostGraph() *CostGraphDef
- func (m *RunMetadata) GetPartitionGraphs() []*GraphDef
- func (m *RunMetadata) GetStepStats() *StepStats
- func (m *RunMetadata) Marshal() (dAtA []byte, err error)
- func (m *RunMetadata) MarshalTo(dAtA []byte) (int, error)
- func (*RunMetadata) ProtoMessage()
- func (m *RunMetadata) Reset()
- func (m *RunMetadata) Size() (n int)
- func (m *RunMetadata) String() string
- func (m *RunMetadata) Unmarshal(dAtA []byte) error
- type RunOptions
- func (*RunOptions) Descriptor() ([]byte, []int)
- func (m *RunOptions) GetDebugOptions() *DebugOptions
- func (m *RunOptions) GetInterOpThreadPool() int32
- func (m *RunOptions) GetOutputPartitionGraphs() bool
- func (m *RunOptions) GetTimeoutInMs() int64
- func (m *RunOptions) GetTraceLevel() RunOptions_TraceLevel
- func (m *RunOptions) Marshal() (dAtA []byte, err error)
- func (m *RunOptions) MarshalTo(dAtA []byte) (int, error)
- func (*RunOptions) ProtoMessage()
- func (m *RunOptions) Reset()
- func (m *RunOptions) Size() (n int)
- func (m *RunOptions) String() string
- func (m *RunOptions) Unmarshal(dAtA []byte) error
- type RunOptions_TraceLevel
- type RunStepRequest
- func (*RunStepRequest) Descriptor() ([]byte, []int)
- func (m *RunStepRequest) GetFeed() []*NamedTensorProto
- func (m *RunStepRequest) GetFetch() []string
- func (m *RunStepRequest) GetOptions() *RunOptions
- func (m *RunStepRequest) GetPartialRunHandle() string
- func (m *RunStepRequest) GetSessionHandle() string
- func (m *RunStepRequest) GetTarget() []string
- func (m *RunStepRequest) Marshal() (dAtA []byte, err error)
- func (m *RunStepRequest) MarshalTo(dAtA []byte) (int, error)
- func (*RunStepRequest) ProtoMessage()
- func (m *RunStepRequest) Reset()
- func (m *RunStepRequest) Size() (n int)
- func (m *RunStepRequest) String() string
- func (m *RunStepRequest) Unmarshal(dAtA []byte) error
- type RunStepResponse
- func (*RunStepResponse) Descriptor() ([]byte, []int)
- func (m *RunStepResponse) GetMetadata() *RunMetadata
- func (m *RunStepResponse) GetTensor() []*NamedTensorProto
- func (m *RunStepResponse) Marshal() (dAtA []byte, err error)
- func (m *RunStepResponse) MarshalTo(dAtA []byte) (int, error)
- func (*RunStepResponse) ProtoMessage()
- func (m *RunStepResponse) Reset()
- func (m *RunStepResponse) Size() (n int)
- func (m *RunStepResponse) String() string
- func (m *RunStepResponse) Unmarshal(dAtA []byte) error
- type SaveSliceInfoDef
- func (*SaveSliceInfoDef) Descriptor() ([]byte, []int)
- func (m *SaveSliceInfoDef) GetFullName() string
- func (m *SaveSliceInfoDef) GetFullShape() []int64
- func (m *SaveSliceInfoDef) GetVarOffset() []int64
- func (m *SaveSliceInfoDef) GetVarShape() []int64
- func (m *SaveSliceInfoDef) Marshal() (dAtA []byte, err error)
- func (m *SaveSliceInfoDef) MarshalTo(dAtA []byte) (int, error)
- func (*SaveSliceInfoDef) ProtoMessage()
- func (m *SaveSliceInfoDef) Reset()
- func (m *SaveSliceInfoDef) Size() (n int)
- func (m *SaveSliceInfoDef) String() string
- func (m *SaveSliceInfoDef) Unmarshal(dAtA []byte) error
- type SavedModel
- func (*SavedModel) Descriptor() ([]byte, []int)
- func (m *SavedModel) GetMetaGraphs() []*MetaGraphDef
- func (m *SavedModel) GetSavedModelSchemaVersion() int64
- func (m *SavedModel) Marshal() (dAtA []byte, err error)
- func (m *SavedModel) MarshalTo(dAtA []byte) (int, error)
- func (*SavedModel) ProtoMessage()
- func (m *SavedModel) Reset()
- func (m *SavedModel) Size() (n int)
- func (m *SavedModel) String() string
- func (m *SavedModel) Unmarshal(dAtA []byte) error
- type SaverDef
- func (*SaverDef) Descriptor() ([]byte, []int)
- func (m *SaverDef) GetFilenameTensorName() string
- func (m *SaverDef) GetKeepCheckpointEveryNHours() float32
- func (m *SaverDef) GetMaxToKeep() int32
- func (m *SaverDef) GetRestoreOpName() string
- func (m *SaverDef) GetSaveTensorName() string
- func (m *SaverDef) GetSharded() bool
- func (m *SaverDef) GetVersion() SaverDef_CheckpointFormatVersion
- func (m *SaverDef) Marshal() (dAtA []byte, err error)
- func (m *SaverDef) MarshalTo(dAtA []byte) (int, error)
- func (*SaverDef) ProtoMessage()
- func (m *SaverDef) Reset()
- func (m *SaverDef) Size() (n int)
- func (m *SaverDef) String() string
- func (m *SaverDef) Unmarshal(dAtA []byte) error
- type SaverDef_CheckpointFormatVersion
- type ServerDef
- func (*ServerDef) Descriptor() ([]byte, []int)
- func (m *ServerDef) GetCluster() *ClusterDef
- func (m *ServerDef) GetDefaultSessionConfig() *ConfigProto
- func (m *ServerDef) GetJobName() string
- func (m *ServerDef) GetProtocol() string
- func (m *ServerDef) GetTaskIndex() int32
- func (m *ServerDef) Marshal() (dAtA []byte, err error)
- func (m *ServerDef) MarshalTo(dAtA []byte) (int, error)
- func (*ServerDef) ProtoMessage()
- func (m *ServerDef) Reset()
- func (m *ServerDef) Size() (n int)
- func (m *ServerDef) String() string
- func (m *ServerDef) Unmarshal(dAtA []byte) error
- type SignatureDef
- func (*SignatureDef) Descriptor() ([]byte, []int)
- func (m *SignatureDef) GetInputs() map[string]*TensorInfo
- func (m *SignatureDef) GetMethodName() string
- func (m *SignatureDef) GetOutputs() map[string]*TensorInfo
- func (m *SignatureDef) Marshal() (dAtA []byte, err error)
- func (m *SignatureDef) MarshalTo(dAtA []byte) (int, error)
- func (*SignatureDef) ProtoMessage()
- func (m *SignatureDef) Reset()
- func (m *SignatureDef) Size() (n int)
- func (m *SignatureDef) String() string
- func (m *SignatureDef) Unmarshal(dAtA []byte) error
- type StepStats
- func (*StepStats) Descriptor() ([]byte, []int)
- func (m *StepStats) GetDevStats() []*DeviceStepStats
- func (m *StepStats) Marshal() (dAtA []byte, err error)
- func (m *StepStats) MarshalTo(dAtA []byte) (int, error)
- func (*StepStats) ProtoMessage()
- func (m *StepStats) Reset()
- func (m *StepStats) Size() (n int)
- func (m *StepStats) String() string
- func (m *StepStats) Unmarshal(dAtA []byte) error
- type Summary
- func (*Summary) Descriptor() ([]byte, []int)
- func (m *Summary) GetValue() []*Summary_Value
- func (m *Summary) Marshal() (dAtA []byte, err error)
- func (m *Summary) MarshalTo(dAtA []byte) (int, error)
- func (*Summary) ProtoMessage()
- func (m *Summary) Reset()
- func (m *Summary) Size() (n int)
- func (m *Summary) String() string
- func (m *Summary) Unmarshal(dAtA []byte) error
- type SummaryDescription
- func (*SummaryDescription) Descriptor() ([]byte, []int)
- func (m *SummaryDescription) GetTypeHint() string
- func (m *SummaryDescription) Marshal() (dAtA []byte, err error)
- func (m *SummaryDescription) MarshalTo(dAtA []byte) (int, error)
- func (*SummaryDescription) ProtoMessage()
- func (m *SummaryDescription) Reset()
- func (m *SummaryDescription) Size() (n int)
- func (m *SummaryDescription) String() string
- func (m *SummaryDescription) Unmarshal(dAtA []byte) error
- type Summary_Audio
- func (*Summary_Audio) Descriptor() ([]byte, []int)
- func (m *Summary_Audio) GetContentType() string
- func (m *Summary_Audio) GetEncodedAudioString() []byte
- func (m *Summary_Audio) GetLengthFrames() int64
- func (m *Summary_Audio) GetNumChannels() int64
- func (m *Summary_Audio) GetSampleRate() float32
- func (m *Summary_Audio) Marshal() (dAtA []byte, err error)
- func (m *Summary_Audio) MarshalTo(dAtA []byte) (int, error)
- func (*Summary_Audio) ProtoMessage()
- func (m *Summary_Audio) Reset()
- func (m *Summary_Audio) Size() (n int)
- func (m *Summary_Audio) String() string
- func (m *Summary_Audio) Unmarshal(dAtA []byte) error
- type Summary_Image
- func (*Summary_Image) Descriptor() ([]byte, []int)
- func (m *Summary_Image) GetColorspace() int32
- func (m *Summary_Image) GetEncodedImageString() []byte
- func (m *Summary_Image) GetHeight() int32
- func (m *Summary_Image) GetWidth() int32
- func (m *Summary_Image) Marshal() (dAtA []byte, err error)
- func (m *Summary_Image) MarshalTo(dAtA []byte) (int, error)
- func (*Summary_Image) ProtoMessage()
- func (m *Summary_Image) Reset()
- func (m *Summary_Image) Size() (n int)
- func (m *Summary_Image) String() string
- func (m *Summary_Image) Unmarshal(dAtA []byte) error
- type Summary_Value
- func (*Summary_Value) Descriptor() ([]byte, []int)
- func (m *Summary_Value) GetAudio() *Summary_Audio
- func (m *Summary_Value) GetHisto() *HistogramProto
- func (m *Summary_Value) GetImage() *Summary_Image
- func (m *Summary_Value) GetNodeName() string
- func (m *Summary_Value) GetObsoleteOldStyleHistogram() []byte
- func (m *Summary_Value) GetSimpleValue() float32
- func (m *Summary_Value) GetTag() string
- func (m *Summary_Value) GetTensor() *TensorProto
- func (m *Summary_Value) GetValue() isSummary_Value_Value
- func (m *Summary_Value) Marshal() (dAtA []byte, err error)
- func (m *Summary_Value) MarshalTo(dAtA []byte) (int, error)
- func (*Summary_Value) ProtoMessage()
- func (m *Summary_Value) Reset()
- func (m *Summary_Value) Size() (n int)
- func (m *Summary_Value) String() string
- func (m *Summary_Value) Unmarshal(dAtA []byte) error
- func (*Summary_Value) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, ...)
- type Summary_Value_Audio
- type Summary_Value_Histo
- type Summary_Value_Image
- type Summary_Value_ObsoleteOldStyleHistogram
- type Summary_Value_SimpleValue
- type Summary_Value_Tensor
- type TensorDescription
- func (*TensorDescription) Descriptor() ([]byte, []int)
- func (m *TensorDescription) GetAllocationDescription() *AllocationDescription
- func (m *TensorDescription) GetDtype() DataType
- func (m *TensorDescription) GetShape() *TensorShapeProto
- func (m *TensorDescription) Marshal() (dAtA []byte, err error)
- func (m *TensorDescription) MarshalTo(dAtA []byte) (int, error)
- func (*TensorDescription) ProtoMessage()
- func (m *TensorDescription) Reset()
- func (m *TensorDescription) Size() (n int)
- func (m *TensorDescription) String() string
- func (m *TensorDescription) Unmarshal(dAtA []byte) error
- type TensorInfo
- func (*TensorInfo) Descriptor() ([]byte, []int)
- func (m *TensorInfo) GetDtype() DataType
- func (m *TensorInfo) GetName() string
- func (m *TensorInfo) GetTensorShape() *TensorShapeProto
- func (m *TensorInfo) Marshal() (dAtA []byte, err error)
- func (m *TensorInfo) MarshalTo(dAtA []byte) (int, error)
- func (*TensorInfo) ProtoMessage()
- func (m *TensorInfo) Reset()
- func (m *TensorInfo) Size() (n int)
- func (m *TensorInfo) String() string
- func (m *TensorInfo) Unmarshal(dAtA []byte) error
- type TensorProto
- func (*TensorProto) Descriptor() ([]byte, []int)
- func (m *TensorProto) GetBoolVal() []bool
- func (m *TensorProto) GetDcomplexVal() []float64
- func (m *TensorProto) GetDoubleVal() []float64
- func (m *TensorProto) GetDtype() DataType
- func (m *TensorProto) GetFloatVal() []float32
- func (m *TensorProto) GetHalfVal() []int32
- func (m *TensorProto) GetInt64Val() []int64
- func (m *TensorProto) GetIntVal() []int32
- func (m *TensorProto) GetResourceHandleVal() []*ResourceHandle
- func (m *TensorProto) GetScomplexVal() []float32
- func (m *TensorProto) GetStringVal() [][]byte
- func (m *TensorProto) GetTensorContent() []byte
- func (m *TensorProto) GetTensorShape() *TensorShapeProto
- func (m *TensorProto) GetVersionNumber() int32
- func (m *TensorProto) Marshal() (dAtA []byte, err error)
- func (m *TensorProto) MarshalTo(dAtA []byte) (int, error)
- func (*TensorProto) ProtoMessage()
- func (m *TensorProto) Reset()
- func (m *TensorProto) Size() (n int)
- func (m *TensorProto) String() string
- func (m *TensorProto) Unmarshal(dAtA []byte) error
- type TensorShapeProto
- func (*TensorShapeProto) Descriptor() ([]byte, []int)
- func (m *TensorShapeProto) GetDim() []*TensorShapeProto_Dim
- func (m *TensorShapeProto) GetUnknownRank() bool
- func (m *TensorShapeProto) Marshal() (dAtA []byte, err error)
- func (m *TensorShapeProto) MarshalTo(dAtA []byte) (int, error)
- func (*TensorShapeProto) ProtoMessage()
- func (m *TensorShapeProto) Reset()
- func (m *TensorShapeProto) Size() (n int)
- func (m *TensorShapeProto) String() string
- func (m *TensorShapeProto) Unmarshal(dAtA []byte) error
- type TensorShapeProto_Dim
- func (*TensorShapeProto_Dim) Descriptor() ([]byte, []int)
- func (m *TensorShapeProto_Dim) GetName() string
- func (m *TensorShapeProto_Dim) GetSize_() int64
- func (m *TensorShapeProto_Dim) Marshal() (dAtA []byte, err error)
- func (m *TensorShapeProto_Dim) MarshalTo(dAtA []byte) (int, error)
- func (*TensorShapeProto_Dim) ProtoMessage()
- func (m *TensorShapeProto_Dim) Reset()
- func (m *TensorShapeProto_Dim) Size() (n int)
- func (m *TensorShapeProto_Dim) String() string
- func (m *TensorShapeProto_Dim) Unmarshal(dAtA []byte) error
- type TensorSliceProto
- func (*TensorSliceProto) Descriptor() ([]byte, []int)
- func (m *TensorSliceProto) GetExtent() []*TensorSliceProto_Extent
- func (m *TensorSliceProto) Marshal() (dAtA []byte, err error)
- func (m *TensorSliceProto) MarshalTo(dAtA []byte) (int, error)
- func (*TensorSliceProto) ProtoMessage()
- func (m *TensorSliceProto) Reset()
- func (m *TensorSliceProto) Size() (n int)
- func (m *TensorSliceProto) String() string
- func (m *TensorSliceProto) Unmarshal(dAtA []byte) error
- type TensorSliceProto_Extent
- func (*TensorSliceProto_Extent) Descriptor() ([]byte, []int)
- func (m *TensorSliceProto_Extent) GetHasLength() isTensorSliceProto_Extent_HasLength
- func (m *TensorSliceProto_Extent) GetLength() int64
- func (m *TensorSliceProto_Extent) GetStart() int64
- func (m *TensorSliceProto_Extent) Marshal() (dAtA []byte, err error)
- func (m *TensorSliceProto_Extent) MarshalTo(dAtA []byte) (int, error)
- func (*TensorSliceProto_Extent) ProtoMessage()
- func (m *TensorSliceProto_Extent) Reset()
- func (m *TensorSliceProto_Extent) Size() (n int)
- func (m *TensorSliceProto_Extent) String() string
- func (m *TensorSliceProto_Extent) Unmarshal(dAtA []byte) error
- func (*TensorSliceProto_Extent) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, ...)
- type TensorSliceProto_Extent_Length
- type ThreadPoolOptionProto
- func (*ThreadPoolOptionProto) Descriptor() ([]byte, []int)
- func (m *ThreadPoolOptionProto) GetNumThreads() int32
- func (m *ThreadPoolOptionProto) Marshal() (dAtA []byte, err error)
- func (m *ThreadPoolOptionProto) MarshalTo(dAtA []byte) (int, error)
- func (*ThreadPoolOptionProto) ProtoMessage()
- func (m *ThreadPoolOptionProto) Reset()
- func (m *ThreadPoolOptionProto) Size() (n int)
- func (m *ThreadPoolOptionProto) String() string
- func (m *ThreadPoolOptionProto) Unmarshal(dAtA []byte) error
- type TraceOpts
- func (*TraceOpts) Descriptor() ([]byte, []int)
- func (m *TraceOpts) GetDuration() float64
- func (m *TraceOpts) GetUseExtendedProfiler() bool
- func (m *TraceOpts) GetUseGpuProfiler() bool
- func (m *TraceOpts) GetUseKernelProfiler() bool
- func (m *TraceOpts) GetUseSampleProfiler() bool
- func (m *TraceOpts) GetUseStepProfiler() bool
- func (m *TraceOpts) Marshal() (dAtA []byte, err error)
- func (m *TraceOpts) MarshalTo(dAtA []byte) (int, error)
- func (*TraceOpts) ProtoMessage()
- func (m *TraceOpts) Reset()
- func (m *TraceOpts) Size() (n int)
- func (m *TraceOpts) String() string
- func (m *TraceOpts) Unmarshal(dAtA []byte) error
- type TracingRequest
- func (*TracingRequest) Descriptor() ([]byte, []int)
- func (m *TracingRequest) GetOptions() *TraceOpts
- func (m *TracingRequest) Marshal() (dAtA []byte, err error)
- func (m *TracingRequest) MarshalTo(dAtA []byte) (int, error)
- func (*TracingRequest) ProtoMessage()
- func (m *TracingRequest) Reset()
- func (m *TracingRequest) Size() (n int)
- func (m *TracingRequest) String() string
- func (m *TracingRequest) Unmarshal(dAtA []byte) error
- type TracingResponse
- func (*TracingResponse) Descriptor() ([]byte, []int)
- func (m *TracingResponse) Marshal() (dAtA []byte, err error)
- func (m *TracingResponse) MarshalTo(dAtA []byte) (int, error)
- func (*TracingResponse) ProtoMessage()
- func (m *TracingResponse) Reset()
- func (m *TracingResponse) Size() (n int)
- func (m *TracingResponse) String() string
- func (m *TracingResponse) Unmarshal(dAtA []byte) error
- type ValuesDef
- func (*ValuesDef) Descriptor() ([]byte, []int)
- func (m *ValuesDef) GetExternalValues() map[string]string
- func (m *ValuesDef) GetValues() []string
- func (m *ValuesDef) Marshal() (dAtA []byte, err error)
- func (m *ValuesDef) MarshalTo(dAtA []byte) (int, error)
- func (*ValuesDef) ProtoMessage()
- func (m *ValuesDef) Reset()
- func (m *ValuesDef) Size() (n int)
- func (m *ValuesDef) String() string
- func (m *ValuesDef) Unmarshal(dAtA []byte) error
- type VariableDef
- func (*VariableDef) Descriptor() ([]byte, []int)
- func (m *VariableDef) GetInitializerName() string
- func (m *VariableDef) GetIsResource() bool
- func (m *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef
- func (m *VariableDef) GetSnapshotName() string
- func (m *VariableDef) GetVariableName() string
- func (m *VariableDef) Marshal() (dAtA []byte, err error)
- func (m *VariableDef) MarshalTo(dAtA []byte) (int, error)
- func (*VariableDef) ProtoMessage()
- func (m *VariableDef) Reset()
- func (m *VariableDef) Size() (n int)
- func (m *VariableDef) String() string
- func (m *VariableDef) Unmarshal(dAtA []byte) error
- type VersionDef
- func (*VersionDef) Descriptor() ([]byte, []int)
- func (m *VersionDef) GetBadConsumers() []int32
- func (m *VersionDef) GetMinConsumer() int32
- func (m *VersionDef) GetProducer() int32
- func (m *VersionDef) Marshal() (dAtA []byte, err error)
- func (m *VersionDef) MarshalTo(dAtA []byte) (int, error)
- func (*VersionDef) ProtoMessage()
- func (m *VersionDef) Reset()
- func (m *VersionDef) Size() (n int)
- func (m *VersionDef) String() string
- func (m *VersionDef) Unmarshal(dAtA []byte) error
- type WhileContextDef
- func (*WhileContextDef) Descriptor() ([]byte, []int)
- func (m *WhileContextDef) GetBackProp() bool
- func (m *WhileContextDef) GetContextName() string
- func (m *WhileContextDef) GetLoopEnterNames() []string
- func (m *WhileContextDef) GetLoopExitNames() []string
- func (m *WhileContextDef) GetParallelIterations() int32
- func (m *WhileContextDef) GetPivotForBodyName() string
- func (m *WhileContextDef) GetPivotForPredName() string
- func (m *WhileContextDef) GetPivotName() string
- func (m *WhileContextDef) GetSwapMemory() bool
- func (m *WhileContextDef) GetValuesDef() *ValuesDef
- func (m *WhileContextDef) Marshal() (dAtA []byte, err error)
- func (m *WhileContextDef) MarshalTo(dAtA []byte) (int, error)
- func (*WhileContextDef) ProtoMessage()
- func (m *WhileContextDef) Reset()
- func (m *WhileContextDef) Size() (n int)
- func (m *WhileContextDef) String() string
- func (m *WhileContextDef) Unmarshal(dAtA []byte) error
- type WorkerServiceClient
- type WorkerServiceServer
Constants ¶
This section is empty.
Variables ¶
var ( ErrInvalidLengthAllocationDescription = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowAllocationDescription = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthAttrValue = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowAttrValue = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthCluster = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowCluster = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthConfig = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowConfig = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthControlFlow = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowControlFlow = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthCostGraph = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowCostGraph = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthDebug = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowDebug = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthDeviceAttributes = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowDeviceAttributes = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthDeviceProperties = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowDeviceProperties = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthErrorCodes = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowErrorCodes = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthFunction = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowFunction = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthGraph = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowGraph = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthGraphTransferInfo = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowGraphTransferInfo = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthKernelDef = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowKernelDef = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthLogMemory = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowLogMemory = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthMaster = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowMaster = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthMetaGraph = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowMetaGraph = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthNamedTensor = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowNamedTensor = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthNodeDef = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowNodeDef = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthOpDef = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowOpDef = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthOpGenOverrides = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowOpGenOverrides = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthQueueRunner = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowQueueRunner = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthReaderBase = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowReaderBase = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthRemoteFusedGraphExecuteInfo = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowRemoteFusedGraphExecuteInfo = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthResourceHandle = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowResourceHandle = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthRewriterConfig = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowRewriterConfig = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthSavedModel = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowSavedModel = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthSaver = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowSaver = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthStepStats = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowStepStats = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthSummary = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowSummary = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthTensor = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowTensor = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthTensorBundle = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowTensorBundle = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthTensorDescription = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowTensorDescription = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthTensorShape = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowTensorShape = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthTensorSlice = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowTensorSlice = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthTensorflowServer = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowTensorflowServer = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthVariable = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowVariable = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthVersions = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowVersions = fmt.Errorf("proto: integer overflow") )
var ( ErrInvalidLengthWorker = fmt.Errorf("proto: negative length found during unmarshaling") ErrIntOverflowWorker = fmt.Errorf("proto: integer overflow") )
var BundleHeaderProto_Endianness_name = map[int32]string{
0: "LITTLE",
1: "BIG",
}
var BundleHeaderProto_Endianness_value = map[string]int32{
"LITTLE": 0,
"BIG": 1,
}
var DataType_name = map[int32]string{
0: "DT_INVALID",
1: "DT_FLOAT",
2: "DT_DOUBLE",
3: "DT_INT32",
4: "DT_UINT8",
5: "DT_INT16",
6: "DT_INT8",
7: "DT_STRING",
8: "DT_COMPLEX64",
9: "DT_INT64",
10: "DT_BOOL",
11: "DT_QINT8",
12: "DT_QUINT8",
13: "DT_QINT32",
14: "DT_BFLOAT16",
15: "DT_QINT16",
16: "DT_QUINT16",
17: "DT_UINT16",
18: "DT_COMPLEX128",
19: "DT_HALF",
20: "DT_RESOURCE",
101: "DT_FLOAT_REF",
102: "DT_DOUBLE_REF",
103: "DT_INT32_REF",
104: "DT_UINT8_REF",
105: "DT_INT16_REF",
106: "DT_INT8_REF",
107: "DT_STRING_REF",
108: "DT_COMPLEX64_REF",
109: "DT_INT64_REF",
110: "DT_BOOL_REF",
111: "DT_QINT8_REF",
112: "DT_QUINT8_REF",
113: "DT_QINT32_REF",
114: "DT_BFLOAT16_REF",
115: "DT_QINT16_REF",
116: "DT_QUINT16_REF",
117: "DT_UINT16_REF",
118: "DT_COMPLEX128_REF",
119: "DT_HALF_REF",
120: "DT_RESOURCE_REF",
}
var DataType_value = map[string]int32{
"DT_INVALID": 0,
"DT_FLOAT": 1,
"DT_DOUBLE": 2,
"DT_INT32": 3,
"DT_UINT8": 4,
"DT_INT16": 5,
"DT_INT8": 6,
"DT_STRING": 7,
"DT_COMPLEX64": 8,
"DT_INT64": 9,
"DT_BOOL": 10,
"DT_QINT8": 11,
"DT_QUINT8": 12,
"DT_QINT32": 13,
"DT_BFLOAT16": 14,
"DT_QINT16": 15,
"DT_QUINT16": 16,
"DT_UINT16": 17,
"DT_COMPLEX128": 18,
"DT_HALF": 19,
"DT_RESOURCE": 20,
"DT_FLOAT_REF": 101,
"DT_DOUBLE_REF": 102,
"DT_INT32_REF": 103,
"DT_UINT8_REF": 104,
"DT_INT16_REF": 105,
"DT_INT8_REF": 106,
"DT_STRING_REF": 107,
"DT_COMPLEX64_REF": 108,
"DT_INT64_REF": 109,
"DT_BOOL_REF": 110,
"DT_QINT8_REF": 111,
"DT_QUINT8_REF": 112,
"DT_QINT32_REF": 113,
"DT_BFLOAT16_REF": 114,
"DT_QINT16_REF": 115,
"DT_QUINT16_REF": 116,
"DT_UINT16_REF": 117,
"DT_COMPLEX128_REF": 118,
"DT_HALF_REF": 119,
"DT_RESOURCE_REF": 120,
}
var Error_Code_name = map[int32]string{
0: "OK",
1: "CANCELLED",
2: "UNKNOWN",
3: "INVALID_ARGUMENT",
4: "DEADLINE_EXCEEDED",
5: "NOT_FOUND",
6: "ALREADY_EXISTS",
7: "PERMISSION_DENIED",
16: "UNAUTHENTICATED",
8: "RESOURCE_EXHAUSTED",
9: "FAILED_PRECONDITION",
10: "ABORTED",
11: "OUT_OF_RANGE",
12: "UNIMPLEMENTED",
13: "INTERNAL",
14: "UNAVAILABLE",
15: "DATA_LOSS",
20: "DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_",
}
var Error_Code_value = map[string]int32{
"OK": 0,
"CANCELLED": 1,
"UNKNOWN": 2,
"INVALID_ARGUMENT": 3,
"DEADLINE_EXCEEDED": 4,
"NOT_FOUND": 5,
"ALREADY_EXISTS": 6,
"PERMISSION_DENIED": 7,
"UNAUTHENTICATED": 16,
"RESOURCE_EXHAUSTED": 8,
"FAILED_PRECONDITION": 9,
"ABORTED": 10,
"OUT_OF_RANGE": 11,
"UNIMPLEMENTED": 12,
"INTERNAL": 13,
"UNAVAILABLE": 14,
"DATA_LOSS": 15,
"DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_": 20,
}
var FrameworkManifest = dlframework.FrameworkManifest{ Name: "Tensorflow", Version: "1.2", Container: map[string]*dlframework.ContainerHardware{ "amd64": { Cpu: "raiproject/carml-tensorflow:amd64-cpu", Gpu: "raiproject/carml-tensorflow:amd64-gpu", }, "ppc64le": { Cpu: "raiproject/carml-tensorflow:ppc64le-gpu", Gpu: "raiproject/carml-tensorflow:ppc64le-gpu", }, }, }
var GraphTransferInfo_Destination_name = map[int32]string{
0: "NOP",
1: "HEXAGON",
}
var GraphTransferInfo_Destination_value = map[string]int32{
"NOP": 0,
"HEXAGON": 1,
}
var OptimizerOptions_GlobalJitLevel_name = map[int32]string{
0: "DEFAULT",
-1: "OFF",
1: "ON_1",
2: "ON_2",
}
var OptimizerOptions_GlobalJitLevel_value = map[string]int32{
"DEFAULT": 0,
"OFF": -1,
"ON_1": 1,
"ON_2": 2,
}
var OptimizerOptions_Level_name = map[int32]string{
0: "L1",
-1: "L0",
}
var OptimizerOptions_Level_value = map[string]int32{
"L1": 0,
"L0": -1,
}
var RewriterConfig_MemOptType_name = map[int32]string{
0: "NO_MEM_OPT",
1: "MANUAL",
}
var RewriterConfig_MemOptType_value = map[string]int32{
"NO_MEM_OPT": 0,
"MANUAL": 1,
}
var RunOptions_TraceLevel_name = map[int32]string{
0: "NO_TRACE",
1: "SOFTWARE_TRACE",
2: "HARDWARE_TRACE",
3: "FULL_TRACE",
}
var RunOptions_TraceLevel_value = map[string]int32{
"NO_TRACE": 0,
"SOFTWARE_TRACE": 1,
"HARDWARE_TRACE": 2,
"FULL_TRACE": 3,
}
var SaverDef_CheckpointFormatVersion_name = map[int32]string{
0: "LEGACY",
1: "V1",
2: "V2",
}
var SaverDef_CheckpointFormatVersion_value = map[string]int32{
"LEGACY": 0,
"V1": 1,
"V2": 2,
}
Functions ¶
func Asset ¶
Asset loads and returns the asset for the given name. It returns an error if the asset could not be found or could not be loaded.
func AssetDir ¶
AssetDir returns the file names below a certain directory embedded in the file by go-bindata. For example if you run go-bindata on data/... and data contains the following hierarchy:
data/
foo.txt
img/
a.png
b.png
then AssetDir("data") would return []string{"foo.txt", "img"} AssetDir("data/img") would return []string{"a.png", "b.png"} AssetDir("foo.txt") and AssetDir("notexist") would return an error AssetDir("") will return []string{"data"}.
func AssetInfo ¶
AssetInfo loads and returns the asset info for the given name. It returns an error if the asset could not be found or could not be loaded.
func MustAsset ¶
MustAsset is like Asset but panics when Asset would return an error. It simplifies safe initialization of global variables.
func RegisterMasterServiceServer ¶
func RegisterMasterServiceServer(s *grpc.Server, srv MasterServiceServer)
func RegisterWorkerServiceServer ¶
func RegisterWorkerServiceServer(s *grpc.Server, srv WorkerServiceServer)
func RestoreAsset ¶
RestoreAsset restores an asset under the given directory
func RestoreAssets ¶
RestoreAssets restores an asset under the given directory recursively
Types ¶
type AllocationDescription ¶
type AllocationDescription struct {
// Total number of bytes requested
RequestedBytes int64 `protobuf:"varint,1,opt,name=requested_bytes,json=requestedBytes,proto3" json:"requested_bytes,omitempty"`
// Total number of bytes allocated if known
AllocatedBytes int64 `protobuf:"varint,2,opt,name=allocated_bytes,json=allocatedBytes,proto3" json:"allocated_bytes,omitempty"`
// Name of the allocator used
AllocatorName string `protobuf:"bytes,3,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
// Identifier of the allocated buffer if known
AllocationId int64 `protobuf:"varint,4,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
// Set if this tensor only has one remaining reference
HasSingleReference bool `protobuf:"varint,5,opt,name=has_single_reference,json=hasSingleReference,proto3" json:"has_single_reference,omitempty"`
// Address of the allocation.
Ptr uint64 `protobuf:"varint,6,opt,name=ptr,proto3" json:"ptr,omitempty"`
}
func (*AllocationDescription) Descriptor ¶
func (*AllocationDescription) Descriptor() ([]byte, []int)
func (*AllocationDescription) GetAllocatedBytes ¶
func (m *AllocationDescription) GetAllocatedBytes() int64
func (*AllocationDescription) GetAllocationId ¶
func (m *AllocationDescription) GetAllocationId() int64
func (*AllocationDescription) GetAllocatorName ¶
func (m *AllocationDescription) GetAllocatorName() string
func (*AllocationDescription) GetHasSingleReference ¶
func (m *AllocationDescription) GetHasSingleReference() bool
func (*AllocationDescription) GetPtr ¶
func (m *AllocationDescription) GetPtr() uint64
func (*AllocationDescription) GetRequestedBytes ¶
func (m *AllocationDescription) GetRequestedBytes() int64
func (*AllocationDescription) Marshal ¶
func (m *AllocationDescription) Marshal() (dAtA []byte, err error)
func (*AllocationDescription) MarshalTo ¶
func (m *AllocationDescription) MarshalTo(dAtA []byte) (int, error)
func (*AllocationDescription) ProtoMessage ¶
func (*AllocationDescription) ProtoMessage()
func (*AllocationDescription) Reset ¶
func (m *AllocationDescription) Reset()
func (*AllocationDescription) Size ¶
func (m *AllocationDescription) Size() (n int)
func (*AllocationDescription) String ¶
func (m *AllocationDescription) String() string
func (*AllocationDescription) Unmarshal ¶
func (m *AllocationDescription) Unmarshal(dAtA []byte) error
type AllocatorMemoryUsed ¶
type AllocatorMemoryUsed struct {
AllocatorName string `protobuf:"bytes,1,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
TotalBytes int64 `protobuf:"varint,2,opt,name=total_bytes,json=totalBytes,proto3" json:"total_bytes,omitempty"`
PeakBytes int64 `protobuf:"varint,3,opt,name=peak_bytes,json=peakBytes,proto3" json:"peak_bytes,omitempty"`
// The bytes that are not deallocated.
LiveBytes int64 `protobuf:"varint,4,opt,name=live_bytes,json=liveBytes,proto3" json:"live_bytes,omitempty"`
}
func (*AllocatorMemoryUsed) Descriptor ¶
func (*AllocatorMemoryUsed) Descriptor() ([]byte, []int)
func (*AllocatorMemoryUsed) GetAllocatorName ¶
func (m *AllocatorMemoryUsed) GetAllocatorName() string
func (*AllocatorMemoryUsed) GetLiveBytes ¶
func (m *AllocatorMemoryUsed) GetLiveBytes() int64
func (*AllocatorMemoryUsed) GetPeakBytes ¶
func (m *AllocatorMemoryUsed) GetPeakBytes() int64
func (*AllocatorMemoryUsed) GetTotalBytes ¶
func (m *AllocatorMemoryUsed) GetTotalBytes() int64
func (*AllocatorMemoryUsed) Marshal ¶
func (m *AllocatorMemoryUsed) Marshal() (dAtA []byte, err error)
func (*AllocatorMemoryUsed) MarshalTo ¶
func (m *AllocatorMemoryUsed) MarshalTo(dAtA []byte) (int, error)
func (*AllocatorMemoryUsed) ProtoMessage ¶
func (*AllocatorMemoryUsed) ProtoMessage()
func (*AllocatorMemoryUsed) Reset ¶
func (m *AllocatorMemoryUsed) Reset()
func (*AllocatorMemoryUsed) Size ¶
func (m *AllocatorMemoryUsed) Size() (n int)
func (*AllocatorMemoryUsed) String ¶
func (m *AllocatorMemoryUsed) String() string
func (*AllocatorMemoryUsed) Unmarshal ¶
func (m *AllocatorMemoryUsed) Unmarshal(dAtA []byte) error
type AssetFileDef ¶
type AssetFileDef struct {
// The tensor to bind the asset filename to.
TensorInfo *TensorInfo `protobuf:"bytes,1,opt,name=tensor_info,json=tensorInfo" 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"`
}
An asset file def for a single file or a set of sharded files with the same name.
func (*AssetFileDef) Descriptor ¶
func (*AssetFileDef) Descriptor() ([]byte, []int)
func (*AssetFileDef) GetFilename ¶
func (m *AssetFileDef) GetFilename() string
func (*AssetFileDef) GetTensorInfo ¶
func (m *AssetFileDef) GetTensorInfo() *TensorInfo
func (*AssetFileDef) Marshal ¶
func (m *AssetFileDef) Marshal() (dAtA []byte, err error)
func (*AssetFileDef) ProtoMessage ¶
func (*AssetFileDef) ProtoMessage()
func (*AssetFileDef) Reset ¶
func (m *AssetFileDef) Reset()
func (*AssetFileDef) Size ¶
func (m *AssetFileDef) Size() (n int)
func (*AssetFileDef) String ¶
func (m *AssetFileDef) String() string
func (*AssetFileDef) Unmarshal ¶
func (m *AssetFileDef) Unmarshal(dAtA []byte) error
type AttrValue ¶
type AttrValue struct {
// Types that are valid to be assigned to Value:
// *AttrValue_S
// *AttrValue_I
// *AttrValue_F
// *AttrValue_B
// *AttrValue_Type
// *AttrValue_Shape
// *AttrValue_Tensor
// *AttrValue_List
// *AttrValue_Func
// *AttrValue_Placeholder
Value isAttrValue_Value `protobuf_oneof:"value"`
}
Protocol buffer representing the value for an attr used to configure an Op. Comment indicates the corresponding attr type. Only the field matching the attr type may be filled.
func (*AttrValue) Descriptor ¶
func (*AttrValue) GetFunc ¶
func (m *AttrValue) GetFunc() *NameAttrList
func (*AttrValue) GetList ¶
func (m *AttrValue) GetList() *AttrValue_ListValue
func (*AttrValue) GetPlaceholder ¶
func (*AttrValue) GetShape ¶
func (m *AttrValue) GetShape() *TensorShapeProto
func (*AttrValue) GetTensor ¶
func (m *AttrValue) GetTensor() *TensorProto
func (*AttrValue) ProtoMessage ¶
func (*AttrValue) ProtoMessage()
type AttrValue_B ¶
type AttrValue_B struct {
B bool `protobuf:"varint,5,opt,name=b,proto3,oneof"`
}
func (*AttrValue_B) Size ¶
func (m *AttrValue_B) Size() (n int)
type AttrValue_F ¶
type AttrValue_F struct {
F float32 `protobuf:"fixed32,4,opt,name=f,proto3,oneof"`
}
func (*AttrValue_F) Size ¶
func (m *AttrValue_F) Size() (n int)
type AttrValue_Func ¶
type AttrValue_Func struct {
Func *NameAttrList `protobuf:"bytes,10,opt,name=func,oneof"`
}
func (*AttrValue_Func) Size ¶
func (m *AttrValue_Func) Size() (n int)
type AttrValue_I ¶
type AttrValue_I struct {
I int64 `protobuf:"varint,3,opt,name=i,proto3,oneof"`
}
func (*AttrValue_I) Size ¶
func (m *AttrValue_I) Size() (n int)
type AttrValue_List ¶
type AttrValue_List struct {
List *AttrValue_ListValue `protobuf:"bytes,1,opt,name=list,oneof"`
}
func (*AttrValue_List) Size ¶
func (m *AttrValue_List) Size() (n int)
type AttrValue_ListValue ¶
type AttrValue_ListValue struct {
S [][]byte `protobuf:"bytes,2,rep,name=s" json:"s,omitempty"`
I []int64 `protobuf:"varint,3,rep,packed,name=i" json:"i,omitempty"`
F []float32 `protobuf:"fixed32,4,rep,packed,name=f" json:"f,omitempty"`
B []bool `protobuf:"varint,5,rep,packed,name=b" json:"b,omitempty"`
Type []DataType `protobuf:"varint,6,rep,packed,name=type,enum=tensorflow.DataType" json:"type,omitempty"`
Shape []*TensorShapeProto `protobuf:"bytes,7,rep,name=shape" json:"shape,omitempty"`
Tensor []*TensorProto `protobuf:"bytes,8,rep,name=tensor" json:"tensor,omitempty"`
Func []*NameAttrList `protobuf:"bytes,9,rep,name=func" json:"func,omitempty"`
}
LINT.IfChange
func (*AttrValue_ListValue) Descriptor ¶
func (*AttrValue_ListValue) Descriptor() ([]byte, []int)
func (*AttrValue_ListValue) GetB ¶
func (m *AttrValue_ListValue) GetB() []bool
func (*AttrValue_ListValue) GetF ¶
func (m *AttrValue_ListValue) GetF() []float32
func (*AttrValue_ListValue) GetFunc ¶
func (m *AttrValue_ListValue) GetFunc() []*NameAttrList
func (*AttrValue_ListValue) GetI ¶
func (m *AttrValue_ListValue) GetI() []int64
func (*AttrValue_ListValue) GetS ¶
func (m *AttrValue_ListValue) GetS() [][]byte
func (*AttrValue_ListValue) GetShape ¶
func (m *AttrValue_ListValue) GetShape() []*TensorShapeProto
func (*AttrValue_ListValue) GetTensor ¶
func (m *AttrValue_ListValue) GetTensor() []*TensorProto
func (*AttrValue_ListValue) GetType ¶
func (m *AttrValue_ListValue) GetType() []DataType
func (*AttrValue_ListValue) Marshal ¶
func (m *AttrValue_ListValue) Marshal() (dAtA []byte, err error)
func (*AttrValue_ListValue) MarshalTo ¶
func (m *AttrValue_ListValue) MarshalTo(dAtA []byte) (int, error)
func (*AttrValue_ListValue) ProtoMessage ¶
func (*AttrValue_ListValue) ProtoMessage()
func (*AttrValue_ListValue) Reset ¶
func (m *AttrValue_ListValue) Reset()
func (*AttrValue_ListValue) Size ¶
func (m *AttrValue_ListValue) Size() (n int)
func (*AttrValue_ListValue) String ¶
func (m *AttrValue_ListValue) String() string
func (*AttrValue_ListValue) Unmarshal ¶
func (m *AttrValue_ListValue) Unmarshal(dAtA []byte) error
type AttrValue_Placeholder ¶
type AttrValue_Placeholder struct {
Placeholder string `protobuf:"bytes,9,opt,name=placeholder,proto3,oneof"`
}
func (*AttrValue_Placeholder) MarshalTo ¶
func (m *AttrValue_Placeholder) MarshalTo(dAtA []byte) (int, error)
func (*AttrValue_Placeholder) Size ¶
func (m *AttrValue_Placeholder) Size() (n int)
type AttrValue_S ¶
type AttrValue_S struct {
S []byte `protobuf:"bytes,2,opt,name=s,proto3,oneof"`
}
func (*AttrValue_S) Size ¶
func (m *AttrValue_S) Size() (n int)
type AttrValue_Shape ¶
type AttrValue_Shape struct {
Shape *TensorShapeProto `protobuf:"bytes,7,opt,name=shape,oneof"`
}
func (*AttrValue_Shape) Size ¶
func (m *AttrValue_Shape) Size() (n int)
type AttrValue_Tensor ¶
type AttrValue_Tensor struct {
Tensor *TensorProto `protobuf:"bytes,8,opt,name=tensor,oneof"`
}
func (*AttrValue_Tensor) Size ¶
func (m *AttrValue_Tensor) Size() (n int)
type AttrValue_Type ¶
type AttrValue_Type struct {
Type DataType `protobuf:"varint,6,opt,name=type,proto3,enum=tensorflow.DataType,oneof"`
}
func (*AttrValue_Type) Size ¶
func (m *AttrValue_Type) Size() (n int)
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"`
}
func (*AutoParallelOptions) Descriptor ¶
func (*AutoParallelOptions) Descriptor() ([]byte, []int)
func (*AutoParallelOptions) GetEnable ¶
func (m *AutoParallelOptions) GetEnable() bool
func (*AutoParallelOptions) GetNumReplicas ¶
func (m *AutoParallelOptions) GetNumReplicas() int32
func (*AutoParallelOptions) Marshal ¶
func (m *AutoParallelOptions) Marshal() (dAtA []byte, err error)
func (*AutoParallelOptions) MarshalTo ¶
func (m *AutoParallelOptions) MarshalTo(dAtA []byte) (int, error)
func (*AutoParallelOptions) ProtoMessage ¶
func (*AutoParallelOptions) ProtoMessage()
func (*AutoParallelOptions) Reset ¶
func (m *AutoParallelOptions) Reset()
func (*AutoParallelOptions) Size ¶
func (m *AutoParallelOptions) Size() (n int)
func (*AutoParallelOptions) String ¶
func (m *AutoParallelOptions) String() string
func (*AutoParallelOptions) Unmarshal ¶
func (m *AutoParallelOptions) Unmarshal(dAtA []byte) error
type BundleEntryProto ¶
type BundleEntryProto struct {
// The tensor dtype and shape.
Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape" 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 []*TensorSliceProto `protobuf:"bytes,7,rep,name=slices" json:"slices,omitempty"`
}
Describes the metadata related to a checkpointed tensor.
func (*BundleEntryProto) Descriptor ¶
func (*BundleEntryProto) Descriptor() ([]byte, []int)
func (*BundleEntryProto) GetCrc32C ¶
func (m *BundleEntryProto) GetCrc32C() uint32
func (*BundleEntryProto) GetDtype ¶
func (m *BundleEntryProto) GetDtype() DataType
func (*BundleEntryProto) GetOffset ¶
func (m *BundleEntryProto) GetOffset() int64
func (*BundleEntryProto) GetShape ¶
func (m *BundleEntryProto) GetShape() *TensorShapeProto
func (*BundleEntryProto) GetShardId ¶
func (m *BundleEntryProto) GetShardId() int32
func (*BundleEntryProto) GetSize_ ¶
func (m *BundleEntryProto) GetSize_() int64
func (*BundleEntryProto) GetSlices ¶
func (m *BundleEntryProto) GetSlices() []*TensorSliceProto
func (*BundleEntryProto) Marshal ¶
func (m *BundleEntryProto) Marshal() (dAtA []byte, err error)
func (*BundleEntryProto) ProtoMessage ¶
func (*BundleEntryProto) ProtoMessage()
func (*BundleEntryProto) Reset ¶
func (m *BundleEntryProto) Reset()
func (*BundleEntryProto) Size ¶
func (m *BundleEntryProto) Size() (n int)
func (*BundleEntryProto) String ¶
func (m *BundleEntryProto) String() string
func (*BundleEntryProto) Unmarshal ¶
func (m *BundleEntryProto) Unmarshal(dAtA []byte) error
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 *VersionDef `protobuf:"bytes,3,opt,name=version" json:"version,omitempty"`
}
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 ¶
func (*BundleHeaderProto) Descriptor() ([]byte, []int)
func (*BundleHeaderProto) GetEndianness ¶
func (m *BundleHeaderProto) GetEndianness() BundleHeaderProto_Endianness
func (*BundleHeaderProto) GetNumShards ¶
func (m *BundleHeaderProto) GetNumShards() int32
func (*BundleHeaderProto) GetVersion ¶
func (m *BundleHeaderProto) GetVersion() *VersionDef
func (*BundleHeaderProto) Marshal ¶
func (m *BundleHeaderProto) Marshal() (dAtA []byte, err error)
func (*BundleHeaderProto) MarshalTo ¶
func (m *BundleHeaderProto) MarshalTo(dAtA []byte) (int, error)
func (*BundleHeaderProto) ProtoMessage ¶
func (*BundleHeaderProto) ProtoMessage()
func (*BundleHeaderProto) Reset ¶
func (m *BundleHeaderProto) Reset()
func (*BundleHeaderProto) Size ¶
func (m *BundleHeaderProto) Size() (n int)
func (*BundleHeaderProto) String ¶
func (m *BundleHeaderProto) String() string
func (*BundleHeaderProto) Unmarshal ¶
func (m *BundleHeaderProto) Unmarshal(dAtA []byte) error
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) EnumDescriptor ¶
func (BundleHeaderProto_Endianness) EnumDescriptor() ([]byte, []int)
func (BundleHeaderProto_Endianness) String ¶
func (x BundleHeaderProto_Endianness) String() string
type CleanupAllRequest ¶
type CleanupAllRequest struct {
// A list of container names.
//
// If 'container' is not empty, releases resoures 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" json:"container,omitempty"`
}
func (*CleanupAllRequest) Descriptor ¶
func (*CleanupAllRequest) Descriptor() ([]byte, []int)
func (*CleanupAllRequest) GetContainer ¶
func (m *CleanupAllRequest) GetContainer() []string
func (*CleanupAllRequest) Marshal ¶
func (m *CleanupAllRequest) Marshal() (dAtA []byte, err error)
func (*CleanupAllRequest) MarshalTo ¶
func (m *CleanupAllRequest) MarshalTo(dAtA []byte) (int, error)
func (*CleanupAllRequest) ProtoMessage ¶
func (*CleanupAllRequest) ProtoMessage()
func (*CleanupAllRequest) Reset ¶
func (m *CleanupAllRequest) Reset()
func (*CleanupAllRequest) Size ¶
func (m *CleanupAllRequest) Size() (n int)
func (*CleanupAllRequest) String ¶
func (m *CleanupAllRequest) String() string
func (*CleanupAllRequest) Unmarshal ¶
func (m *CleanupAllRequest) Unmarshal(dAtA []byte) error
type CleanupAllResponse ¶
type CleanupAllResponse struct {
}
func (*CleanupAllResponse) Descriptor ¶
func (*CleanupAllResponse) Descriptor() ([]byte, []int)
func (*CleanupAllResponse) Marshal ¶
func (m *CleanupAllResponse) Marshal() (dAtA []byte, err error)
func (*CleanupAllResponse) MarshalTo ¶
func (m *CleanupAllResponse) MarshalTo(dAtA []byte) (int, error)
func (*CleanupAllResponse) ProtoMessage ¶
func (*CleanupAllResponse) ProtoMessage()
func (*CleanupAllResponse) Reset ¶
func (m *CleanupAllResponse) Reset()
func (*CleanupAllResponse) Size ¶
func (m *CleanupAllResponse) Size() (n int)
func (*CleanupAllResponse) String ¶
func (m *CleanupAllResponse) String() string
func (*CleanupAllResponse) Unmarshal ¶
func (m *CleanupAllResponse) Unmarshal(dAtA []byte) error
type CleanupGraphRequest ¶
type CleanupGraphRequest struct {
StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
}
func (*CleanupGraphRequest) Descriptor ¶
func (*CleanupGraphRequest) Descriptor() ([]byte, []int)
func (*CleanupGraphRequest) GetStepId ¶
func (m *CleanupGraphRequest) GetStepId() int64
func (*CleanupGraphRequest) Marshal ¶
func (m *CleanupGraphRequest) Marshal() (dAtA []byte, err error)
func (*CleanupGraphRequest) MarshalTo ¶
func (m *CleanupGraphRequest) MarshalTo(dAtA []byte) (int, error)
func (*CleanupGraphRequest) ProtoMessage ¶
func (*CleanupGraphRequest) ProtoMessage()
func (*CleanupGraphRequest) Reset ¶
func (m *CleanupGraphRequest) Reset()
func (*CleanupGraphRequest) Size ¶
func (m *CleanupGraphRequest) Size() (n int)
func (*CleanupGraphRequest) String ¶
func (m *CleanupGraphRequest) String() string
func (*CleanupGraphRequest) Unmarshal ¶
func (m *CleanupGraphRequest) Unmarshal(dAtA []byte) error
type CleanupGraphResponse ¶
type CleanupGraphResponse struct {
}
func (*CleanupGraphResponse) Descriptor ¶
func (*CleanupGraphResponse) Descriptor() ([]byte, []int)
func (*CleanupGraphResponse) Marshal ¶
func (m *CleanupGraphResponse) Marshal() (dAtA []byte, err error)
func (*CleanupGraphResponse) MarshalTo ¶
func (m *CleanupGraphResponse) MarshalTo(dAtA []byte) (int, error)
func (*CleanupGraphResponse) ProtoMessage ¶
func (*CleanupGraphResponse) ProtoMessage()
func (*CleanupGraphResponse) Reset ¶
func (m *CleanupGraphResponse) Reset()
func (*CleanupGraphResponse) Size ¶
func (m *CleanupGraphResponse) Size() (n int)
func (*CleanupGraphResponse) String ¶
func (m *CleanupGraphResponse) String() string
func (*CleanupGraphResponse) Unmarshal ¶
func (m *CleanupGraphResponse) Unmarshal(dAtA []byte) error
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"`
}
func (*CloseSessionRequest) Descriptor ¶
func (*CloseSessionRequest) Descriptor() ([]byte, []int)
func (*CloseSessionRequest) GetSessionHandle ¶
func (m *CloseSessionRequest) GetSessionHandle() string
func (*CloseSessionRequest) Marshal ¶
func (m *CloseSessionRequest) Marshal() (dAtA []byte, err error)
func (*CloseSessionRequest) MarshalTo ¶
func (m *CloseSessionRequest) MarshalTo(dAtA []byte) (int, error)
func (*CloseSessionRequest) ProtoMessage ¶
func (*CloseSessionRequest) ProtoMessage()
func (*CloseSessionRequest) Reset ¶
func (m *CloseSessionRequest) Reset()
func (*CloseSessionRequest) Size ¶
func (m *CloseSessionRequest) Size() (n int)
func (*CloseSessionRequest) String ¶
func (m *CloseSessionRequest) String() string
func (*CloseSessionRequest) Unmarshal ¶
func (m *CloseSessionRequest) Unmarshal(dAtA []byte) error
type CloseSessionResponse ¶
type CloseSessionResponse struct {
}
func (*CloseSessionResponse) Descriptor ¶
func (*CloseSessionResponse) Descriptor() ([]byte, []int)
func (*CloseSessionResponse) Marshal ¶
func (m *CloseSessionResponse) Marshal() (dAtA []byte, err error)
func (*CloseSessionResponse) MarshalTo ¶
func (m *CloseSessionResponse) MarshalTo(dAtA []byte) (int, error)
func (*CloseSessionResponse) ProtoMessage ¶
func (*CloseSessionResponse) ProtoMessage()
func (*CloseSessionResponse) Reset ¶
func (m *CloseSessionResponse) Reset()
func (*CloseSessionResponse) Size ¶
func (m *CloseSessionResponse) Size() (n int)
func (*CloseSessionResponse) String ¶
func (m *CloseSessionResponse) String() string
func (*CloseSessionResponse) Unmarshal ¶
func (m *CloseSessionResponse) Unmarshal(dAtA []byte) error
type ClusterDef ¶
type ClusterDef struct {
// The jobs that comprise the cluster.
Job []*JobDef `protobuf:"bytes,1,rep,name=job" json:"job,omitempty"`
}
Defines a TensorFlow cluster as a set of jobs.
func (*ClusterDef) Descriptor ¶
func (*ClusterDef) Descriptor() ([]byte, []int)
func (*ClusterDef) GetJob ¶
func (m *ClusterDef) GetJob() []*JobDef
func (*ClusterDef) Marshal ¶
func (m *ClusterDef) Marshal() (dAtA []byte, err error)
func (*ClusterDef) ProtoMessage ¶
func (*ClusterDef) ProtoMessage()
func (*ClusterDef) Reset ¶
func (m *ClusterDef) Reset()
func (*ClusterDef) Size ¶
func (m *ClusterDef) Size() (n int)
func (*ClusterDef) String ¶
func (m *ClusterDef) String() string
func (*ClusterDef) Unmarshal ¶
func (m *ClusterDef) Unmarshal(dAtA []byte) error
type CollectionDef ¶
type CollectionDef struct {
// Types that are valid to be assigned to Kind:
// *CollectionDef_NodeList_
// *CollectionDef_BytesList_
// *CollectionDef_Int64List_
// *CollectionDef_FloatList_
// *CollectionDef_AnyList_
Kind isCollectionDef_Kind `protobuf_oneof:"kind"`
}
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 ¶
func (*CollectionDef) Descriptor() ([]byte, []int)
func (*CollectionDef) GetAnyList ¶
func (m *CollectionDef) GetAnyList() *CollectionDef_AnyList
func (*CollectionDef) GetBytesList ¶
func (m *CollectionDef) GetBytesList() *CollectionDef_BytesList
func (*CollectionDef) GetFloatList ¶
func (m *CollectionDef) GetFloatList() *CollectionDef_FloatList
func (*CollectionDef) GetInt64List ¶
func (m *CollectionDef) GetInt64List() *CollectionDef_Int64List
func (*CollectionDef) GetKind ¶
func (m *CollectionDef) GetKind() isCollectionDef_Kind
func (*CollectionDef) GetNodeList ¶
func (m *CollectionDef) GetNodeList() *CollectionDef_NodeList
func (*CollectionDef) Marshal ¶
func (m *CollectionDef) Marshal() (dAtA []byte, err error)
func (*CollectionDef) ProtoMessage ¶
func (*CollectionDef) ProtoMessage()
func (*CollectionDef) Reset ¶
func (m *CollectionDef) Reset()
func (*CollectionDef) Size ¶
func (m *CollectionDef) Size() (n int)
func (*CollectionDef) String ¶
func (m *CollectionDef) String() string
func (*CollectionDef) Unmarshal ¶
func (m *CollectionDef) Unmarshal(dAtA []byte) error
func (*CollectionDef) XXX_OneofFuncs ¶
func (*CollectionDef) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, func(msg proto.Message, tag, wire int, b *proto.Buffer) (bool, error), func(msg proto.Message) (n int), []interface{})
XXX_OneofFuncs is for the internal use of the proto package.
type CollectionDef_AnyList ¶
type CollectionDef_AnyList struct {
Value []*google_protobuf.Any `protobuf:"bytes,1,rep,name=value" json:"value,omitempty"`
}
AnyList is used for collecting Any protos.
func (*CollectionDef_AnyList) Descriptor ¶
func (*CollectionDef_AnyList) Descriptor() ([]byte, []int)
func (*CollectionDef_AnyList) GetValue ¶
func (m *CollectionDef_AnyList) GetValue() []*google_protobuf.Any
func (*CollectionDef_AnyList) Marshal ¶
func (m *CollectionDef_AnyList) Marshal() (dAtA []byte, err error)
func (*CollectionDef_AnyList) MarshalTo ¶
func (m *CollectionDef_AnyList) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_AnyList) ProtoMessage ¶
func (*CollectionDef_AnyList) ProtoMessage()
func (*CollectionDef_AnyList) Reset ¶
func (m *CollectionDef_AnyList) Reset()
func (*CollectionDef_AnyList) Size ¶
func (m *CollectionDef_AnyList) Size() (n int)
func (*CollectionDef_AnyList) String ¶
func (m *CollectionDef_AnyList) String() string
func (*CollectionDef_AnyList) Unmarshal ¶
func (m *CollectionDef_AnyList) Unmarshal(dAtA []byte) error
type CollectionDef_AnyList_ ¶
type CollectionDef_AnyList_ struct {
AnyList *CollectionDef_AnyList `protobuf:"bytes,5,opt,name=any_list,json=anyList,oneof"`
}
func (*CollectionDef_AnyList_) MarshalTo ¶
func (m *CollectionDef_AnyList_) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_AnyList_) Size ¶
func (m *CollectionDef_AnyList_) Size() (n int)
type CollectionDef_BytesList ¶
type CollectionDef_BytesList struct {
Value [][]byte `protobuf:"bytes,1,rep,name=value" json:"value,omitempty"`
}
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 ¶
func (*CollectionDef_BytesList) Descriptor() ([]byte, []int)
func (*CollectionDef_BytesList) GetValue ¶
func (m *CollectionDef_BytesList) GetValue() [][]byte
func (*CollectionDef_BytesList) Marshal ¶
func (m *CollectionDef_BytesList) Marshal() (dAtA []byte, err error)
func (*CollectionDef_BytesList) MarshalTo ¶
func (m *CollectionDef_BytesList) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_BytesList) ProtoMessage ¶
func (*CollectionDef_BytesList) ProtoMessage()
func (*CollectionDef_BytesList) Reset ¶
func (m *CollectionDef_BytesList) Reset()
func (*CollectionDef_BytesList) Size ¶
func (m *CollectionDef_BytesList) Size() (n int)
func (*CollectionDef_BytesList) String ¶
func (m *CollectionDef_BytesList) String() string
func (*CollectionDef_BytesList) Unmarshal ¶
func (m *CollectionDef_BytesList) Unmarshal(dAtA []byte) error
type CollectionDef_BytesList_ ¶
type CollectionDef_BytesList_ struct {
BytesList *CollectionDef_BytesList `protobuf:"bytes,2,opt,name=bytes_list,json=bytesList,oneof"`
}
func (*CollectionDef_BytesList_) MarshalTo ¶
func (m *CollectionDef_BytesList_) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_BytesList_) Size ¶
func (m *CollectionDef_BytesList_) Size() (n int)
type CollectionDef_FloatList ¶
type CollectionDef_FloatList struct {
Value []float32 `protobuf:"fixed32,1,rep,packed,name=value" json:"value,omitempty"`
}
FloatList is used for collecting float values.
func (*CollectionDef_FloatList) Descriptor ¶
func (*CollectionDef_FloatList) Descriptor() ([]byte, []int)
func (*CollectionDef_FloatList) GetValue ¶
func (m *CollectionDef_FloatList) GetValue() []float32
func (*CollectionDef_FloatList) Marshal ¶
func (m *CollectionDef_FloatList) Marshal() (dAtA []byte, err error)
func (*CollectionDef_FloatList) MarshalTo ¶
func (m *CollectionDef_FloatList) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_FloatList) ProtoMessage ¶
func (*CollectionDef_FloatList) ProtoMessage()
func (*CollectionDef_FloatList) Reset ¶
func (m *CollectionDef_FloatList) Reset()
func (*CollectionDef_FloatList) Size ¶
func (m *CollectionDef_FloatList) Size() (n int)
func (*CollectionDef_FloatList) String ¶
func (m *CollectionDef_FloatList) String() string
func (*CollectionDef_FloatList) Unmarshal ¶
func (m *CollectionDef_FloatList) Unmarshal(dAtA []byte) error
type CollectionDef_FloatList_ ¶
type CollectionDef_FloatList_ struct {
FloatList *CollectionDef_FloatList `protobuf:"bytes,4,opt,name=float_list,json=floatList,oneof"`
}
func (*CollectionDef_FloatList_) MarshalTo ¶
func (m *CollectionDef_FloatList_) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_FloatList_) Size ¶
func (m *CollectionDef_FloatList_) Size() (n int)
type CollectionDef_Int64List ¶
type CollectionDef_Int64List struct {
Value []int64 `protobuf:"varint,1,rep,packed,name=value" json:"value,omitempty"`
}
Int64List is used for collecting int, int64 and long values.
func (*CollectionDef_Int64List) Descriptor ¶
func (*CollectionDef_Int64List) Descriptor() ([]byte, []int)
func (*CollectionDef_Int64List) GetValue ¶
func (m *CollectionDef_Int64List) GetValue() []int64
func (*CollectionDef_Int64List) Marshal ¶
func (m *CollectionDef_Int64List) Marshal() (dAtA []byte, err error)
func (*CollectionDef_Int64List) MarshalTo ¶
func (m *CollectionDef_Int64List) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_Int64List) ProtoMessage ¶
func (*CollectionDef_Int64List) ProtoMessage()
func (*CollectionDef_Int64List) Reset ¶
func (m *CollectionDef_Int64List) Reset()
func (*CollectionDef_Int64List) Size ¶
func (m *CollectionDef_Int64List) Size() (n int)
func (*CollectionDef_Int64List) String ¶
func (m *CollectionDef_Int64List) String() string
func (*CollectionDef_Int64List) Unmarshal ¶
func (m *CollectionDef_Int64List) Unmarshal(dAtA []byte) error
type CollectionDef_Int64List_ ¶
type CollectionDef_Int64List_ struct {
Int64List *CollectionDef_Int64List `protobuf:"bytes,3,opt,name=int64_list,json=int64List,oneof"`
}
func (*CollectionDef_Int64List_) MarshalTo ¶
func (m *CollectionDef_Int64List_) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_Int64List_) Size ¶
func (m *CollectionDef_Int64List_) Size() (n int)
type CollectionDef_NodeList ¶
type CollectionDef_NodeList struct {
Value []string `protobuf:"bytes,1,rep,name=value" json:"value,omitempty"`
}
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 ¶
func (*CollectionDef_NodeList) Descriptor() ([]byte, []int)
func (*CollectionDef_NodeList) GetValue ¶
func (m *CollectionDef_NodeList) GetValue() []string
func (*CollectionDef_NodeList) Marshal ¶
func (m *CollectionDef_NodeList) Marshal() (dAtA []byte, err error)
func (*CollectionDef_NodeList) MarshalTo ¶
func (m *CollectionDef_NodeList) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_NodeList) ProtoMessage ¶
func (*CollectionDef_NodeList) ProtoMessage()
func (*CollectionDef_NodeList) Reset ¶
func (m *CollectionDef_NodeList) Reset()
func (*CollectionDef_NodeList) Size ¶
func (m *CollectionDef_NodeList) Size() (n int)
func (*CollectionDef_NodeList) String ¶
func (m *CollectionDef_NodeList) String() string
func (*CollectionDef_NodeList) Unmarshal ¶
func (m *CollectionDef_NodeList) Unmarshal(dAtA []byte) error
type CollectionDef_NodeList_ ¶
type CollectionDef_NodeList_ struct {
NodeList *CollectionDef_NodeList `protobuf:"bytes,1,opt,name=node_list,json=nodeList,oneof"`
}
func (*CollectionDef_NodeList_) MarshalTo ¶
func (m *CollectionDef_NodeList_) MarshalTo(dAtA []byte) (int, error)
func (*CollectionDef_NodeList_) Size ¶
func (m *CollectionDef_NodeList_) Size() (n int)
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" json:"values_def,omitempty"`
}
Protocol buffer representing a CondContext object.
func (*CondContextDef) Descriptor ¶
func (*CondContextDef) Descriptor() ([]byte, []int)
func (*CondContextDef) GetBranch ¶
func (m *CondContextDef) GetBranch() int32
func (*CondContextDef) GetContextName ¶
func (m *CondContextDef) GetContextName() string
func (*CondContextDef) GetPivotName ¶
func (m *CondContextDef) GetPivotName() string
func (*CondContextDef) GetPredName ¶
func (m *CondContextDef) GetPredName() string
func (*CondContextDef) GetValuesDef ¶
func (m *CondContextDef) GetValuesDef() *ValuesDef
func (*CondContextDef) Marshal ¶
func (m *CondContextDef) Marshal() (dAtA []byte, err error)
func (*CondContextDef) ProtoMessage ¶
func (*CondContextDef) ProtoMessage()
func (*CondContextDef) Reset ¶
func (m *CondContextDef) Reset()
func (*CondContextDef) Size ¶
func (m *CondContextDef) Size() (n int)
func (*CondContextDef) String ¶
func (m *CondContextDef) String() string
func (*CondContextDef) Unmarshal ¶
func (m *CondContextDef) Unmarshal(dAtA []byte) error
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 `` /* 176-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. The intended use is for when some session invocations need
// to run in a background pool limited to a small number of threads.
//
// Configures session thread pools. If this is configured, then RunOptions for
// a Run call can select the thread pool to use.
//
// If a pool's num_threads is 0, then inter_op_parallelism_threads is used.
SessionInterOpThreadPool []*ThreadPoolOptionProto `` /* 133-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" json:"device_filters,omitempty"`
// Options that apply to all GPUs.
GpuOptions *GPUOptions `protobuf:"bytes,6,opt,name=gpu_options,json=gpuOptions" 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" 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" 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" json:"cluster_def,omitempty"`
}
Session configuration parameters. The system picks appropriate values for fields that are not set.
func (*ConfigProto) Descriptor ¶
func (*ConfigProto) Descriptor() ([]byte, []int)
func (*ConfigProto) GetAllowSoftPlacement ¶
func (m *ConfigProto) GetAllowSoftPlacement() bool
func (*ConfigProto) GetClusterDef ¶
func (m *ConfigProto) GetClusterDef() *ClusterDef
func (*ConfigProto) GetDeviceCount ¶
func (m *ConfigProto) GetDeviceCount() map[string]int32
func (*ConfigProto) GetDeviceFilters ¶
func (m *ConfigProto) GetDeviceFilters() []string
func (*ConfigProto) GetGpuOptions ¶
func (m *ConfigProto) GetGpuOptions() *GPUOptions
func (*ConfigProto) GetGraphOptions ¶
func (m *ConfigProto) GetGraphOptions() *GraphOptions
func (*ConfigProto) GetInterOpParallelismThreads ¶
func (m *ConfigProto) GetInterOpParallelismThreads() int32
func (*ConfigProto) GetIntraOpParallelismThreads ¶
func (m *ConfigProto) GetIntraOpParallelismThreads() int32
func (*ConfigProto) GetLogDevicePlacement ¶
func (m *ConfigProto) GetLogDevicePlacement() bool
func (*ConfigProto) GetOperationTimeoutInMs ¶
func (m *ConfigProto) GetOperationTimeoutInMs() int64
func (*ConfigProto) GetPlacementPeriod ¶
func (m *ConfigProto) GetPlacementPeriod() int32
func (*ConfigProto) GetRpcOptions ¶
func (m *ConfigProto) GetRpcOptions() *RPCOptions
func (*ConfigProto) GetSessionInterOpThreadPool ¶
func (m *ConfigProto) GetSessionInterOpThreadPool() []*ThreadPoolOptionProto
func (*ConfigProto) GetUsePerSessionThreads ¶
func (m *ConfigProto) GetUsePerSessionThreads() bool
func (*ConfigProto) Marshal ¶
func (m *ConfigProto) Marshal() (dAtA []byte, err error)
func (*ConfigProto) ProtoMessage ¶
func (*ConfigProto) ProtoMessage()
func (*ConfigProto) Reset ¶
func (m *ConfigProto) Reset()
func (*ConfigProto) Size ¶
func (m *ConfigProto) Size() (n int)
func (*ConfigProto) String ¶
func (m *ConfigProto) String() string
func (*ConfigProto) Unmarshal ¶
func (m *ConfigProto) Unmarshal(dAtA []byte) error
type CostGraphDef ¶
type CostGraphDef struct {
Node []*CostGraphDef_Node `protobuf:"bytes,1,rep,name=node" json:"node,omitempty"`
}
func (*CostGraphDef) Descriptor ¶
func (*CostGraphDef) Descriptor() ([]byte, []int)
func (*CostGraphDef) GetNode ¶
func (m *CostGraphDef) GetNode() []*CostGraphDef_Node
func (*CostGraphDef) Marshal ¶
func (m *CostGraphDef) Marshal() (dAtA []byte, err error)
func (*CostGraphDef) ProtoMessage ¶
func (*CostGraphDef) ProtoMessage()
func (*CostGraphDef) Reset ¶
func (m *CostGraphDef) Reset()
func (*CostGraphDef) Size ¶
func (m *CostGraphDef) Size() (n int)
func (*CostGraphDef) String ¶
func (m *CostGraphDef) String() string
func (*CostGraphDef) Unmarshal ¶
func (m *CostGraphDef) Unmarshal(dAtA []byte) error
type CostGraphDef_Node ¶
type CostGraphDef_Node struct {
// The name of the node. Names are globally unique.
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// The device of the node. Can be empty if the node is mapped to the
// default partition or partitioning hasn't been run yet.
Device string `protobuf:"bytes,2,opt,name=device,proto3" json:"device,omitempty"`
// The id of the node. Node ids are only unique inside a partition.
Id int32 `protobuf:"varint,3,opt,name=id,proto3" json:"id,omitempty"`
InputInfo []*CostGraphDef_Node_InputInfo `protobuf:"bytes,4,rep,name=input_info,json=inputInfo" json:"input_info,omitempty"`
OutputInfo []*CostGraphDef_Node_OutputInfo `protobuf:"bytes,5,rep,name=output_info,json=outputInfo" json:"output_info,omitempty"`
// Temporary memory used by this node.
TemporaryMemorySize int64 `protobuf:"varint,6,opt,name=temporary_memory_size,json=temporaryMemorySize,proto3" json:"temporary_memory_size,omitempty"`
HostTempMemorySize int64 `protobuf:"varint,10,opt,name=host_temp_memory_size,json=hostTempMemorySize,proto3" json:"host_temp_memory_size,omitempty"`
DeviceTempMemorySize int64 `` /* 127-byte string literal not displayed */
HostPersistentMemorySize int64 `` /* 139-byte string literal not displayed */
DevicePersistentMemorySize int64 `` /* 145-byte string literal not displayed */
// Estimate of the computational cost of this node, in microseconds.
ComputeCost int64 `protobuf:"varint,9,opt,name=compute_cost,json=computeCost,proto3" json:"compute_cost,omitempty"`
// Analytical estimate of the computational cost of this node, in
// microseconds.
ComputeTime int64 `protobuf:"varint,14,opt,name=compute_time,json=computeTime,proto3" json:"compute_time,omitempty"`
// Analytical estimate of the memory access cost of this node, in
// microseconds.
MemoryTime int64 `protobuf:"varint,15,opt,name=memory_time,json=memoryTime,proto3" json:"memory_time,omitempty"`
// If true, the output is permanent: it can't be discarded, because this
// node is part of the "final output". Nodes may depend on final nodes.
IsFinal bool `protobuf:"varint,7,opt,name=is_final,json=isFinal,proto3" json:"is_final,omitempty"`
// Ids of the control inputs for this node.
ControlInput []int32 `protobuf:"varint,8,rep,packed,name=control_input,json=controlInput" json:"control_input,omitempty"`
}
func (*CostGraphDef_Node) Descriptor ¶
func (*CostGraphDef_Node) Descriptor() ([]byte, []int)
func (*CostGraphDef_Node) GetComputeCost ¶
func (m *CostGraphDef_Node) GetComputeCost() int64
func (*CostGraphDef_Node) GetComputeTime ¶
func (m *CostGraphDef_Node) GetComputeTime() int64
func (*CostGraphDef_Node) GetControlInput ¶
func (m *CostGraphDef_Node) GetControlInput() []int32
func (*CostGraphDef_Node) GetDevice ¶
func (m *CostGraphDef_Node) GetDevice() string
func (*CostGraphDef_Node) GetDevicePersistentMemorySize ¶
func (m *CostGraphDef_Node) GetDevicePersistentMemorySize() int64
func (*CostGraphDef_Node) GetDeviceTempMemorySize ¶
func (m *CostGraphDef_Node) GetDeviceTempMemorySize() int64
func (*CostGraphDef_Node) GetHostPersistentMemorySize ¶
func (m *CostGraphDef_Node) GetHostPersistentMemorySize() int64
func (*CostGraphDef_Node) GetHostTempMemorySize ¶
func (m *CostGraphDef_Node) GetHostTempMemorySize() int64
func (*CostGraphDef_Node) GetId ¶
func (m *CostGraphDef_Node) GetId() int32
func (*CostGraphDef_Node) GetInputInfo ¶
func (m *CostGraphDef_Node) GetInputInfo() []*CostGraphDef_Node_InputInfo
func (*CostGraphDef_Node) GetIsFinal ¶
func (m *CostGraphDef_Node) GetIsFinal() bool
func (*CostGraphDef_Node) GetMemoryTime ¶
func (m *CostGraphDef_Node) GetMemoryTime() int64
func (*CostGraphDef_Node) GetName ¶
func (m *CostGraphDef_Node) GetName() string
func (*CostGraphDef_Node) GetOutputInfo ¶
func (m *CostGraphDef_Node) GetOutputInfo() []*CostGraphDef_Node_OutputInfo
func (*CostGraphDef_Node) GetTemporaryMemorySize ¶
func (m *CostGraphDef_Node) GetTemporaryMemorySize() int64
func (*CostGraphDef_Node) Marshal ¶
func (m *CostGraphDef_Node) Marshal() (dAtA []byte, err error)
func (*CostGraphDef_Node) MarshalTo ¶
func (m *CostGraphDef_Node) MarshalTo(dAtA []byte) (int, error)
func (*CostGraphDef_Node) ProtoMessage ¶
func (*CostGraphDef_Node) ProtoMessage()
func (*CostGraphDef_Node) Reset ¶
func (m *CostGraphDef_Node) Reset()
func (*CostGraphDef_Node) Size ¶
func (m *CostGraphDef_Node) Size() (n int)
func (*CostGraphDef_Node) String ¶
func (m *CostGraphDef_Node) String() string
func (*CostGraphDef_Node) Unmarshal ¶
func (m *CostGraphDef_Node) Unmarshal(dAtA []byte) error
type CostGraphDef_Node_InputInfo ¶
type CostGraphDef_Node_InputInfo struct {
PrecedingNode int32 `protobuf:"varint,1,opt,name=preceding_node,json=precedingNode,proto3" json:"preceding_node,omitempty"`
PrecedingPort int32 `protobuf:"varint,2,opt,name=preceding_port,json=precedingPort,proto3" json:"preceding_port,omitempty"`
}
Inputs of this node. They must be executed before this node can be executed. An input is a particular output of another node, specified by the node id and the output index.
func (*CostGraphDef_Node_InputInfo) Descriptor ¶
func (*CostGraphDef_Node_InputInfo) Descriptor() ([]byte, []int)
func (*CostGraphDef_Node_InputInfo) GetPrecedingNode ¶
func (m *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32
func (*CostGraphDef_Node_InputInfo) GetPrecedingPort ¶
func (m *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32
func (*CostGraphDef_Node_InputInfo) Marshal ¶
func (m *CostGraphDef_Node_InputInfo) Marshal() (dAtA []byte, err error)
func (*CostGraphDef_Node_InputInfo) MarshalTo ¶
func (m *CostGraphDef_Node_InputInfo) MarshalTo(dAtA []byte) (int, error)
func (*CostGraphDef_Node_InputInfo) ProtoMessage ¶
func (*CostGraphDef_Node_InputInfo) ProtoMessage()
func (*CostGraphDef_Node_InputInfo) Reset ¶
func (m *CostGraphDef_Node_InputInfo) Reset()
func (*CostGraphDef_Node_InputInfo) Size ¶
func (m *CostGraphDef_Node_InputInfo) Size() (n int)
func (*CostGraphDef_Node_InputInfo) String ¶
func (m *CostGraphDef_Node_InputInfo) String() string
func (*CostGraphDef_Node_InputInfo) Unmarshal ¶
func (m *CostGraphDef_Node_InputInfo) Unmarshal(dAtA []byte) error
type CostGraphDef_Node_OutputInfo ¶
type CostGraphDef_Node_OutputInfo struct {
Size_ int64 `protobuf:"varint,1,opt,name=size,proto3" json:"size,omitempty"`
// If >= 0, the output is an alias of an input. Note that an alias input
// may itself be an alias. The algorithm will therefore need to follow
// those pointers.
AliasInputPort int64 `protobuf:"varint,2,opt,name=alias_input_port,json=aliasInputPort,proto3" json:"alias_input_port,omitempty"`
Shape *TensorShapeProto `protobuf:"bytes,3,opt,name=shape" json:"shape,omitempty"`
Dtype DataType `protobuf:"varint,4,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
}
Outputs of this node.
func (*CostGraphDef_Node_OutputInfo) Descriptor ¶
func (*CostGraphDef_Node_OutputInfo) Descriptor() ([]byte, []int)
func (*CostGraphDef_Node_OutputInfo) GetAliasInputPort ¶
func (m *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64
func (*CostGraphDef_Node_OutputInfo) GetDtype ¶
func (m *CostGraphDef_Node_OutputInfo) GetDtype() DataType
func (*CostGraphDef_Node_OutputInfo) GetShape ¶
func (m *CostGraphDef_Node_OutputInfo) GetShape() *TensorShapeProto
func (*CostGraphDef_Node_OutputInfo) GetSize_ ¶
func (m *CostGraphDef_Node_OutputInfo) GetSize_() int64
func (*CostGraphDef_Node_OutputInfo) Marshal ¶
func (m *CostGraphDef_Node_OutputInfo) Marshal() (dAtA []byte, err error)
func (*CostGraphDef_Node_OutputInfo) MarshalTo ¶
func (m *CostGraphDef_Node_OutputInfo) MarshalTo(dAtA []byte) (int, error)
func (*CostGraphDef_Node_OutputInfo) ProtoMessage ¶
func (*CostGraphDef_Node_OutputInfo) ProtoMessage()
func (*CostGraphDef_Node_OutputInfo) Reset ¶
func (m *CostGraphDef_Node_OutputInfo) Reset()
func (*CostGraphDef_Node_OutputInfo) Size ¶
func (m *CostGraphDef_Node_OutputInfo) Size() (n int)
func (*CostGraphDef_Node_OutputInfo) String ¶
func (m *CostGraphDef_Node_OutputInfo) String() string
func (*CostGraphDef_Node_OutputInfo) Unmarshal ¶
func (m *CostGraphDef_Node_OutputInfo) Unmarshal(dAtA []byte) error
type CreateSessionRequest ¶
type CreateSessionRequest struct {
// The initial graph definition.
GraphDef *GraphDef `protobuf:"bytes,1,opt,name=graph_def,json=graphDef" json:"graph_def,omitempty"`
// Configuration options.
Config *ConfigProto `protobuf:"bytes,2,opt,name=config" json:"config,omitempty"`
// The target string used from the client's perspective.
Target string `protobuf:"bytes,3,opt,name=target,proto3" json:"target,omitempty"`
}
func (*CreateSessionRequest) Descriptor ¶
func (*CreateSessionRequest) Descriptor() ([]byte, []int)
func (*CreateSessionRequest) GetConfig ¶
func (m *CreateSessionRequest) GetConfig() *ConfigProto
func (*CreateSessionRequest) GetGraphDef ¶
func (m *CreateSessionRequest) GetGraphDef() *GraphDef
func (*CreateSessionRequest) GetTarget ¶
func (m *CreateSessionRequest) GetTarget() string
func (*CreateSessionRequest) Marshal ¶
func (m *CreateSessionRequest) Marshal() (dAtA []byte, err error)
func (*CreateSessionRequest) MarshalTo ¶
func (m *CreateSessionRequest) MarshalTo(dAtA []byte) (int, error)
func (*CreateSessionRequest) ProtoMessage ¶
func (*CreateSessionRequest) ProtoMessage()
func (*CreateSessionRequest) Reset ¶
func (m *CreateSessionRequest) Reset()
func (*CreateSessionRequest) Size ¶
func (m *CreateSessionRequest) Size() (n int)
func (*CreateSessionRequest) String ¶
func (m *CreateSessionRequest) String() string
func (*CreateSessionRequest) Unmarshal ¶
func (m *CreateSessionRequest) Unmarshal(dAtA []byte) error
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"`
}
func (*CreateSessionResponse) Descriptor ¶
func (*CreateSessionResponse) Descriptor() ([]byte, []int)
func (*CreateSessionResponse) GetGraphVersion ¶
func (m *CreateSessionResponse) GetGraphVersion() int64
func (*CreateSessionResponse) GetSessionHandle ¶
func (m *CreateSessionResponse) GetSessionHandle() string
func (*CreateSessionResponse) Marshal ¶
func (m *CreateSessionResponse) Marshal() (dAtA []byte, err error)
func (*CreateSessionResponse) MarshalTo ¶
func (m *CreateSessionResponse) MarshalTo(dAtA []byte) (int, error)
func (*CreateSessionResponse) ProtoMessage ¶
func (*CreateSessionResponse) ProtoMessage()
func (*CreateSessionResponse) Reset ¶
func (m *CreateSessionResponse) Reset()
func (*CreateSessionResponse) Size ¶
func (m *CreateSessionResponse) Size() (n int)
func (*CreateSessionResponse) String ¶
func (m *CreateSessionResponse) String() string
func (*CreateSessionResponse) Unmarshal ¶
func (m *CreateSessionResponse) Unmarshal(dAtA []byte) error
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" json:"server_def,omitempty"`
}
func (*CreateWorkerSessionRequest) Descriptor ¶
func (*CreateWorkerSessionRequest) Descriptor() ([]byte, []int)
func (*CreateWorkerSessionRequest) GetServerDef ¶
func (m *CreateWorkerSessionRequest) GetServerDef() *ServerDef
func (*CreateWorkerSessionRequest) GetSessionHandle ¶
func (m *CreateWorkerSessionRequest) GetSessionHandle() string
func (*CreateWorkerSessionRequest) Marshal ¶
func (m *CreateWorkerSessionRequest) Marshal() (dAtA []byte, err error)
func (*CreateWorkerSessionRequest) MarshalTo ¶
func (m *CreateWorkerSessionRequest) MarshalTo(dAtA []byte) (int, error)
func (*CreateWorkerSessionRequest) ProtoMessage ¶
func (*CreateWorkerSessionRequest) ProtoMessage()
func (*CreateWorkerSessionRequest) Reset ¶
func (m *CreateWorkerSessionRequest) Reset()
func (*CreateWorkerSessionRequest) Size ¶
func (m *CreateWorkerSessionRequest) Size() (n int)
func (*CreateWorkerSessionRequest) String ¶
func (m *CreateWorkerSessionRequest) String() string
func (*CreateWorkerSessionRequest) Unmarshal ¶
func (m *CreateWorkerSessionRequest) Unmarshal(dAtA []byte) error
type CreateWorkerSessionResponse ¶
type CreateWorkerSessionResponse struct {
}
func (*CreateWorkerSessionResponse) Descriptor ¶
func (*CreateWorkerSessionResponse) Descriptor() ([]byte, []int)
func (*CreateWorkerSessionResponse) Marshal ¶
func (m *CreateWorkerSessionResponse) Marshal() (dAtA []byte, err error)
func (*CreateWorkerSessionResponse) MarshalTo ¶
func (m *CreateWorkerSessionResponse) MarshalTo(dAtA []byte) (int, error)
func (*CreateWorkerSessionResponse) ProtoMessage ¶
func (*CreateWorkerSessionResponse) ProtoMessage()
func (*CreateWorkerSessionResponse) Reset ¶
func (m *CreateWorkerSessionResponse) Reset()
func (*CreateWorkerSessionResponse) Size ¶
func (m *CreateWorkerSessionResponse) Size() (n int)
func (*CreateWorkerSessionResponse) String ¶
func (m *CreateWorkerSessionResponse) String() string
func (*CreateWorkerSessionResponse) Unmarshal ¶
func (m *CreateWorkerSessionResponse) Unmarshal(dAtA []byte) error
type DataType ¶
type DataType int32
LINT.IfChange
const ( // Not a legal value for DataType. Used to indicate a DataType field // has not been set. DataType_DT_INVALID DataType = 0 // Data types that all computation devices are expected to be // capable to support. DataType_DT_FLOAT DataType = 1 DataType_DT_DOUBLE DataType = 2 DataType_DT_INT32 DataType = 3 DataType_DT_UINT8 DataType = 4 DataType_DT_INT16 DataType = 5 DataType_DT_INT8 DataType = 6 DataType_DT_STRING DataType = 7 DataType_DT_COMPLEX64 DataType = 8 DataType_DT_INT64 DataType = 9 DataType_DT_BOOL DataType = 10 DataType_DT_QINT8 DataType = 11 DataType_DT_QUINT8 DataType = 12 DataType_DT_QINT32 DataType = 13 DataType_DT_BFLOAT16 DataType = 14 DataType_DT_QINT16 DataType = 15 DataType_DT_QUINT16 DataType = 16 DataType_DT_UINT16 DataType = 17 DataType_DT_COMPLEX128 DataType = 18 DataType_DT_HALF DataType = 19 DataType_DT_RESOURCE DataType = 20 // Do not use! These are only for parameters. Every enum above // should have a corresponding value below (verified by types_test). DataType_DT_FLOAT_REF DataType = 101 DataType_DT_DOUBLE_REF DataType = 102 DataType_DT_INT32_REF DataType = 103 DataType_DT_UINT8_REF DataType = 104 DataType_DT_INT16_REF DataType = 105 DataType_DT_INT8_REF DataType = 106 DataType_DT_STRING_REF DataType = 107 DataType_DT_COMPLEX64_REF DataType = 108 DataType_DT_INT64_REF DataType = 109 DataType_DT_BOOL_REF DataType = 110 DataType_DT_QINT8_REF DataType = 111 DataType_DT_QUINT8_REF DataType = 112 DataType_DT_QINT32_REF DataType = 113 DataType_DT_BFLOAT16_REF DataType = 114 DataType_DT_QINT16_REF DataType = 115 DataType_DT_QUINT16_REF DataType = 116 DataType_DT_UINT16_REF DataType = 117 DataType_DT_COMPLEX128_REF DataType = 118 DataType_DT_HALF_REF DataType = 119 DataType_DT_RESOURCE_REF DataType = 120 )
func (DataType) EnumDescriptor ¶
type DebugOptions ¶
type DebugOptions struct {
// Debugging options
DebugTensorWatchOpts []*DebugTensorWatch `protobuf:"bytes,4,rep,name=debug_tensor_watch_opts,json=debugTensorWatchOpts" 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"`
}
EXPERIMENTAL. Options for initializing DebuggerState.
func (*DebugOptions) Descriptor ¶
func (*DebugOptions) Descriptor() ([]byte, []int)
func (*DebugOptions) GetDebugTensorWatchOpts ¶
func (m *DebugOptions) GetDebugTensorWatchOpts() []*DebugTensorWatch
func (*DebugOptions) GetGlobalStep ¶
func (m *DebugOptions) GetGlobalStep() int64
func (*DebugOptions) Marshal ¶
func (m *DebugOptions) Marshal() (dAtA []byte, err error)
func (*DebugOptions) ProtoMessage ¶
func (*DebugOptions) ProtoMessage()
func (*DebugOptions) Reset ¶
func (m *DebugOptions) Reset()
func (*DebugOptions) Size ¶
func (m *DebugOptions) Size() (n int)
func (*DebugOptions) String ¶
func (m *DebugOptions) String() string
func (*DebugOptions) Unmarshal ¶
func (m *DebugOptions) Unmarshal(dAtA []byte) error
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" json:"debug_ops,omitempty"`
// URL(s) for debug targets(s).
// E.g., "file:///foo/tfdbg_dump", "grpc://localhost:11011"
// 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" 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 */
}
EXPERIMENTAL. Option for watching a node.
func (*DebugTensorWatch) Descriptor ¶
func (*DebugTensorWatch) Descriptor() ([]byte, []int)
func (*DebugTensorWatch) GetDebugOps ¶
func (m *DebugTensorWatch) GetDebugOps() []string
func (*DebugTensorWatch) GetDebugUrls ¶
func (m *DebugTensorWatch) GetDebugUrls() []string
func (*DebugTensorWatch) GetNodeName ¶
func (m *DebugTensorWatch) GetNodeName() string
func (*DebugTensorWatch) GetOutputSlot ¶
func (m *DebugTensorWatch) GetOutputSlot() int32
func (*DebugTensorWatch) GetTolerateDebugOpCreationFailures ¶
func (m *DebugTensorWatch) GetTolerateDebugOpCreationFailures() bool
func (*DebugTensorWatch) Marshal ¶
func (m *DebugTensorWatch) Marshal() (dAtA []byte, err error)
func (*DebugTensorWatch) ProtoMessage ¶
func (*DebugTensorWatch) ProtoMessage()
func (*DebugTensorWatch) Reset ¶
func (m *DebugTensorWatch) Reset()
func (*DebugTensorWatch) Size ¶
func (m *DebugTensorWatch) Size() (n int)
func (*DebugTensorWatch) String ¶
func (m *DebugTensorWatch) String() string
func (*DebugTensorWatch) Unmarshal ¶
func (m *DebugTensorWatch) Unmarshal(dAtA []byte) error
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"`
// 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"`
}
func (*DeregisterGraphRequest) Descriptor ¶
func (*DeregisterGraphRequest) Descriptor() ([]byte, []int)
func (*DeregisterGraphRequest) GetGraphHandle ¶
func (m *DeregisterGraphRequest) GetGraphHandle() string
func (*DeregisterGraphRequest) GetSessionHandle ¶
func (m *DeregisterGraphRequest) GetSessionHandle() string
func (*DeregisterGraphRequest) Marshal ¶
func (m *DeregisterGraphRequest) Marshal() (dAtA []byte, err error)
func (*DeregisterGraphRequest) MarshalTo ¶
func (m *DeregisterGraphRequest) MarshalTo(dAtA []byte) (int, error)
func (*DeregisterGraphRequest) ProtoMessage ¶
func (*DeregisterGraphRequest) ProtoMessage()
func (*DeregisterGraphRequest) Reset ¶
func (m *DeregisterGraphRequest) Reset()
func (*DeregisterGraphRequest) Size ¶
func (m *DeregisterGraphRequest) Size() (n int)
func (*DeregisterGraphRequest) String ¶
func (m *DeregisterGraphRequest) String() string
func (*DeregisterGraphRequest) Unmarshal ¶
func (m *DeregisterGraphRequest) Unmarshal(dAtA []byte) error
type DeregisterGraphResponse ¶
type DeregisterGraphResponse struct {
}
func (*DeregisterGraphResponse) Descriptor ¶
func (*DeregisterGraphResponse) Descriptor() ([]byte, []int)
func (*DeregisterGraphResponse) Marshal ¶
func (m *DeregisterGraphResponse) Marshal() (dAtA []byte, err error)
func (*DeregisterGraphResponse) MarshalTo ¶
func (m *DeregisterGraphResponse) MarshalTo(dAtA []byte) (int, error)
func (*DeregisterGraphResponse) ProtoMessage ¶
func (*DeregisterGraphResponse) ProtoMessage()
func (*DeregisterGraphResponse) Reset ¶
func (m *DeregisterGraphResponse) Reset()
func (*DeregisterGraphResponse) Size ¶
func (m *DeregisterGraphResponse) Size() (n int)
func (*DeregisterGraphResponse) String ¶
func (m *DeregisterGraphResponse) String() string
func (*DeregisterGraphResponse) Unmarshal ¶
func (m *DeregisterGraphResponse) Unmarshal(dAtA []byte) error
type DeviceAttributes ¶
type DeviceAttributes struct {
// Fully specified name of the device within a cluster.
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// String representation of device_type.
DeviceType string `protobuf:"bytes,2,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"`
// Memory capacity of device in bytes.
MemoryLimit int64 `protobuf:"varint,4,opt,name=memory_limit,json=memoryLimit,proto3" json:"memory_limit,omitempty"`
// Platform-specific data about device that may be useful
// for supporting efficient data transfers.
Locality *DeviceLocality `protobuf:"bytes,5,opt,name=locality" json:"locality,omitempty"`
// A device is assigned a global unique number each time it is
// initialized. "incarnation" should never be 0.
Incarnation uint64 `protobuf:"fixed64,6,opt,name=incarnation,proto3" json:"incarnation,omitempty"`
// String representation of the physical device that this device maps to.
PhysicalDeviceDesc string `protobuf:"bytes,7,opt,name=physical_device_desc,json=physicalDeviceDesc,proto3" json:"physical_device_desc,omitempty"`
}
func (*DeviceAttributes) Descriptor ¶
func (*DeviceAttributes) Descriptor() ([]byte, []int)
func (*DeviceAttributes) GetDeviceType ¶
func (m *DeviceAttributes) GetDeviceType() string
func (*DeviceAttributes) GetIncarnation ¶
func (m *DeviceAttributes) GetIncarnation() uint64
func (*DeviceAttributes) GetLocality ¶
func (m *DeviceAttributes) GetLocality() *DeviceLocality
func (*DeviceAttributes) GetMemoryLimit ¶
func (m *DeviceAttributes) GetMemoryLimit() int64
func (*DeviceAttributes) GetName ¶
func (m *DeviceAttributes) GetName() string
func (*DeviceAttributes) GetPhysicalDeviceDesc ¶
func (m *DeviceAttributes) GetPhysicalDeviceDesc() string
func (*DeviceAttributes) Marshal ¶
func (m *DeviceAttributes) Marshal() (dAtA []byte, err error)
func (*DeviceAttributes) ProtoMessage ¶
func (*DeviceAttributes) ProtoMessage()
func (*DeviceAttributes) Reset ¶
func (m *DeviceAttributes) Reset()
func (*DeviceAttributes) Size ¶
func (m *DeviceAttributes) Size() (n int)
func (*DeviceAttributes) String ¶
func (m *DeviceAttributes) String() string
func (*DeviceAttributes) Unmarshal ¶
func (m *DeviceAttributes) Unmarshal(dAtA []byte) error
type DeviceLocality ¶
type DeviceLocality struct {
// Optional bus locality of device. Default value of 0 means
// no specific locality. Specific localities are indexed from 1.
BusId int32 `protobuf:"varint,1,opt,name=bus_id,json=busId,proto3" json:"bus_id,omitempty"`
}
func (*DeviceLocality) Descriptor ¶
func (*DeviceLocality) Descriptor() ([]byte, []int)
func (*DeviceLocality) GetBusId ¶
func (m *DeviceLocality) GetBusId() int32
func (*DeviceLocality) Marshal ¶
func (m *DeviceLocality) Marshal() (dAtA []byte, err error)
func (*DeviceLocality) ProtoMessage ¶
func (*DeviceLocality) ProtoMessage()
func (*DeviceLocality) Reset ¶
func (m *DeviceLocality) Reset()
func (*DeviceLocality) Size ¶
func (m *DeviceLocality) Size() (n int)
func (*DeviceLocality) String ¶
func (m *DeviceLocality) String() string
func (*DeviceLocality) Unmarshal ¶
func (m *DeviceLocality) Unmarshal(dAtA []byte) error
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 `` /* 156-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"`
}
func (*DeviceProperties) Descriptor ¶
func (*DeviceProperties) Descriptor() ([]byte, []int)
func (*DeviceProperties) GetBandwidth ¶
func (m *DeviceProperties) GetBandwidth() int64
func (*DeviceProperties) GetEnvironment ¶
func (m *DeviceProperties) GetEnvironment() map[string]string
func (*DeviceProperties) GetFrequency ¶
func (m *DeviceProperties) GetFrequency() int64
func (*DeviceProperties) GetL1CacheSize ¶
func (m *DeviceProperties) GetL1CacheSize() int64
func (*DeviceProperties) GetL2CacheSize ¶
func (m *DeviceProperties) GetL2CacheSize() int64
func (*DeviceProperties) GetL3CacheSize ¶
func (m *DeviceProperties) GetL3CacheSize() int64
func (*DeviceProperties) GetMemorySize ¶
func (m *DeviceProperties) GetMemorySize() int64
func (*DeviceProperties) GetModel ¶
func (m *DeviceProperties) GetModel() string
func (*DeviceProperties) GetNumCores ¶
func (m *DeviceProperties) GetNumCores() int64
func (*DeviceProperties) GetNumRegisters ¶
func (m *DeviceProperties) GetNumRegisters() int64
func (*DeviceProperties) GetSharedMemorySizePerMultiprocessor ¶
func (m *DeviceProperties) GetSharedMemorySizePerMultiprocessor() int64
func (*DeviceProperties) GetType ¶
func (m *DeviceProperties) GetType() string
func (*DeviceProperties) GetVendor ¶
func (m *DeviceProperties) GetVendor() string
func (*DeviceProperties) Marshal ¶
func (m *DeviceProperties) Marshal() (dAtA []byte, err error)
func (*DeviceProperties) ProtoMessage ¶
func (*DeviceProperties) ProtoMessage()
func (*DeviceProperties) Reset ¶
func (m *DeviceProperties) Reset()
func (*DeviceProperties) Size ¶
func (m *DeviceProperties) Size() (n int)
func (*DeviceProperties) String ¶
func (m *DeviceProperties) String() string
func (*DeviceProperties) Unmarshal ¶
func (m *DeviceProperties) Unmarshal(dAtA []byte) error
type DeviceStepStats ¶
type DeviceStepStats struct {
Device string `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"`
NodeStats []*NodeExecStats `protobuf:"bytes,2,rep,name=node_stats,json=nodeStats" json:"node_stats,omitempty"`
}
func (*DeviceStepStats) Descriptor ¶
func (*DeviceStepStats) Descriptor() ([]byte, []int)
func (*DeviceStepStats) GetDevice ¶
func (m *DeviceStepStats) GetDevice() string
func (*DeviceStepStats) GetNodeStats ¶
func (m *DeviceStepStats) GetNodeStats() []*NodeExecStats
func (*DeviceStepStats) Marshal ¶
func (m *DeviceStepStats) Marshal() (dAtA []byte, err error)
func (*DeviceStepStats) ProtoMessage ¶
func (*DeviceStepStats) ProtoMessage()
func (*DeviceStepStats) Reset ¶
func (m *DeviceStepStats) Reset()
func (*DeviceStepStats) Size ¶
func (m *DeviceStepStats) Size() (n int)
func (*DeviceStepStats) String ¶
func (m *DeviceStepStats) String() string
func (*DeviceStepStats) Unmarshal ¶
func (m *DeviceStepStats) Unmarshal(dAtA []byte) error
type Error ¶
type Error struct {
}
func (*Error) Descriptor ¶
func (*Error) ProtoMessage ¶
func (*Error) ProtoMessage()
type Error_Code ¶
type Error_Code int32
The canonical error codes for TensorFlow APIs.
Warnings:
- Do not change any numeric assignments.
- Changes to this list should only be made if there is a compelling need that can't be satisfied in another way. Such changes must be approved by at least two OWNERS.
Sometimes multiple error codes may apply. Services should return the most specific error code that applies. For example, prefer OUT_OF_RANGE over FAILED_PRECONDITION if both codes apply. Similarly prefer NOT_FOUND or ALREADY_EXISTS over FAILED_PRECONDITION.
const ( // Not an error; returned on success Error_OK Error_Code = 0 // The operation was cancelled (typically by the caller). Error_CANCELLED Error_Code = 1 // Unknown error. An example of where this error may be returned is // if a Status value received from another address space belongs to // an error-space that is not known in this address space. Also // errors raised by APIs that do not return enough error information // may be converted to this error. Error_UNKNOWN Error_Code = 2 // Client specified an invalid argument. Note that this differs // from FAILED_PRECONDITION. INVALID_ARGUMENT indicates arguments // that are problematic regardless of the state of the system // (e.g., a malformed file name). Error_INVALID_ARGUMENT Error_Code = 3 // Deadline expired before operation could complete. For operations // that change the state of the system, this error may be returned // even if the operation has completed successfully. For example, a // successful response from a server could have been delayed long // enough for the deadline to expire. Error_DEADLINE_EXCEEDED Error_Code = 4 // Some requested entity (e.g., file or directory) was not found. // For privacy reasons, this code *may* be returned when the client // does not have the access right to the entity. Error_NOT_FOUND Error_Code = 5 // Some entity that we attempted to create (e.g., file or directory) // already exists. Error_ALREADY_EXISTS Error_Code = 6 // The caller does not have permission to execute the specified // operation. PERMISSION_DENIED must not be used for rejections // caused by exhausting some resource (use RESOURCE_EXHAUSTED // instead for those errors). PERMISSION_DENIED must not be // used if the caller can not be identified (use UNAUTHENTICATED // instead for those errors). Error_PERMISSION_DENIED Error_Code = 7 // The request does not have valid authentication credentials for the // operation. Error_UNAUTHENTICATED Error_Code = 16 // Some resource has been exhausted, perhaps a per-user quota, or // perhaps the entire file system is out of space. Error_RESOURCE_EXHAUSTED Error_Code = 8 // Operation was rejected because the system is not in a state // required for the operation's execution. For example, directory // to be deleted may be non-empty, an rmdir operation is applied to // a non-directory, etc. // // A litmus test that may help a service implementor in deciding // between FAILED_PRECONDITION, ABORTED, and UNAVAILABLE: // (a) Use UNAVAILABLE if the client can retry just the failing call. // (b) Use ABORTED if the client should retry at a higher-level // (e.g., restarting a read-modify-write sequence). // (c) Use FAILED_PRECONDITION if the client should not retry until // the system state has been explicitly fixed. E.g., if an "rmdir" // fails because the directory is non-empty, FAILED_PRECONDITION // should be returned since the client should not retry unless // they have first fixed up the directory by deleting files from it. // (d) Use FAILED_PRECONDITION if the client performs conditional // REST Get/Update/Delete on a resource and the resource on the // server does not match the condition. E.g., conflicting // read-modify-write on the same resource. Error_FAILED_PRECONDITION Error_Code = 9 // The operation was aborted, typically due to a concurrency issue // like sequencer check failures, transaction aborts, etc. // // See litmus test above for deciding between FAILED_PRECONDITION, // ABORTED, and UNAVAILABLE. Error_ABORTED Error_Code = 10 // Operation tried to iterate past the valid input range. E.g., seeking or // reading past end of file. // // Unlike INVALID_ARGUMENT, this error indicates a problem that may // be fixed if the system state changes. For example, a 32-bit file // system will generate INVALID_ARGUMENT if asked to read at an // offset that is not in the range [0,2^32-1], but it will generate // OUT_OF_RANGE if asked to read from an offset past the current // file size. // // There is a fair bit of overlap between FAILED_PRECONDITION and // OUT_OF_RANGE. We recommend using OUT_OF_RANGE (the more specific // error) when it applies so that callers who are iterating through // a space can easily look for an OUT_OF_RANGE error to detect when // they are done. Error_OUT_OF_RANGE Error_Code = 11 // Operation is not implemented or not supported/enabled in this service. Error_UNIMPLEMENTED Error_Code = 12 // Internal errors. Means some invariants expected by underlying // system has been broken. If you see one of these errors, // something is very broken. Error_INTERNAL Error_Code = 13 // The service is currently unavailable. This is a most likely a // transient condition and may be corrected by retrying with // a backoff. // // See litmus test above for deciding between FAILED_PRECONDITION, // ABORTED, and UNAVAILABLE. Error_UNAVAILABLE Error_Code = 14 // Unrecoverable data loss or corruption. Error_DATA_LOSS Error_Code = 15 // An extra enum entry to prevent people from writing code that // fails to compile when a new code is added. // // Nobody should ever reference this enumeration entry. In particular, // if you write C++ code that switches on this enumeration, add a default: // case instead of a case that mentions this enumeration entry. // // Nobody should rely on the value (currently 20) listed here. It // may change in the future. Error_DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_ Error_Code = 20 )
func (Error_Code) EnumDescriptor ¶
func (Error_Code) EnumDescriptor() ([]byte, []int)
func (Error_Code) String ¶
func (x Error_Code) 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"`
}
Options specific to the execution of a single step.
func (*ExecutorOpts) Descriptor ¶
func (*ExecutorOpts) Descriptor() ([]byte, []int)
func (*ExecutorOpts) GetRecordCosts ¶
func (m *ExecutorOpts) GetRecordCosts() bool
func (*ExecutorOpts) GetRecordTimeline ¶
func (m *ExecutorOpts) GetRecordTimeline() bool
func (*ExecutorOpts) Marshal ¶
func (m *ExecutorOpts) Marshal() (dAtA []byte, err error)
func (*ExecutorOpts) ProtoMessage ¶
func (*ExecutorOpts) ProtoMessage()
func (*ExecutorOpts) Reset ¶
func (m *ExecutorOpts) Reset()
func (*ExecutorOpts) Size ¶
func (m *ExecutorOpts) Size() (n int)
func (*ExecutorOpts) String ¶
func (m *ExecutorOpts) String() string
func (*ExecutorOpts) Unmarshal ¶
func (m *ExecutorOpts) Unmarshal(dAtA []byte) error
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 *GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef" 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"`
}
func (*ExtendSessionRequest) Descriptor ¶
func (*ExtendSessionRequest) Descriptor() ([]byte, []int)
func (*ExtendSessionRequest) GetCurrentGraphVersion ¶
func (m *ExtendSessionRequest) GetCurrentGraphVersion() int64
func (*ExtendSessionRequest) GetGraphDef ¶
func (m *ExtendSessionRequest) GetGraphDef() *GraphDef
func (*ExtendSessionRequest) GetSessionHandle ¶
func (m *ExtendSessionRequest) GetSessionHandle() string
func (*ExtendSessionRequest) Marshal ¶
func (m *ExtendSessionRequest) Marshal() (dAtA []byte, err error)
func (*ExtendSessionRequest) MarshalTo ¶
func (m *ExtendSessionRequest) MarshalTo(dAtA []byte) (int, error)
func (*ExtendSessionRequest) ProtoMessage ¶
func (*ExtendSessionRequest) ProtoMessage()
func (*ExtendSessionRequest) Reset ¶
func (m *ExtendSessionRequest) Reset()
func (*ExtendSessionRequest) Size ¶
func (m *ExtendSessionRequest) Size() (n int)
func (*ExtendSessionRequest) String ¶
func (m *ExtendSessionRequest) String() string
func (*ExtendSessionRequest) Unmarshal ¶
func (m *ExtendSessionRequest) Unmarshal(dAtA []byte) error
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"`
}
func (*ExtendSessionResponse) Descriptor ¶
func (*ExtendSessionResponse) Descriptor() ([]byte, []int)
func (*ExtendSessionResponse) GetNewGraphVersion ¶
func (m *ExtendSessionResponse) GetNewGraphVersion() int64
func (*ExtendSessionResponse) Marshal ¶
func (m *ExtendSessionResponse) Marshal() (dAtA []byte, err error)
func (*ExtendSessionResponse) MarshalTo ¶
func (m *ExtendSessionResponse) MarshalTo(dAtA []byte) (int, error)
func (*ExtendSessionResponse) ProtoMessage ¶
func (*ExtendSessionResponse) ProtoMessage()
func (*ExtendSessionResponse) Reset ¶
func (m *ExtendSessionResponse) Reset()
func (*ExtendSessionResponse) Size ¶
func (m *ExtendSessionResponse) Size() (n int)
func (*ExtendSessionResponse) String ¶
func (m *ExtendSessionResponse) String() string
func (*ExtendSessionResponse) Unmarshal ¶
func (m *ExtendSessionResponse) Unmarshal(dAtA []byte) error
type FunctionDef ¶
type FunctionDef struct {
// The definition of the function's name, arguments, return values,
// attrs etc.
Signature *OpDef `protobuf:"bytes,1,opt,name=signature" json:"signature,omitempty"`
// Attributes specific to this function definition.
Attr map[string]*AttrValue `` /* 135-byte string literal not displayed */
// By convention, "op" in node_def is resolved by consulting with a
// user-defined library first. If not resolved, "func" is assumed to
// be a builtin op.
NodeDef []*NodeDef `protobuf:"bytes,3,rep,name=node_def,json=nodeDef" json:"node_def,omitempty"`
// A mapping from the output arg names from `signature` to the
// outputs from `node_def` that should be returned by the function.
Ret map[string]string `` /* 140-byte string literal not displayed */
}
A function can be instantiated when the runtime can bind every attr with a value. When a GraphDef has a call to a function, it must have binding for every attr defined in the signature.
TODO(zhifengc):
- device spec, etc.
func (*FunctionDef) Descriptor ¶
func (*FunctionDef) Descriptor() ([]byte, []int)
func (*FunctionDef) GetAttr ¶
func (m *FunctionDef) GetAttr() map[string]*AttrValue
func (*FunctionDef) GetNodeDef ¶
func (m *FunctionDef) GetNodeDef() []*NodeDef
func (*FunctionDef) GetRet ¶
func (m *FunctionDef) GetRet() map[string]string
func (*FunctionDef) GetSignature ¶
func (m *FunctionDef) GetSignature() *OpDef
func (*FunctionDef) Marshal ¶
func (m *FunctionDef) Marshal() (dAtA []byte, err error)
func (*FunctionDef) ProtoMessage ¶
func (*FunctionDef) ProtoMessage()
func (*FunctionDef) Reset ¶
func (m *FunctionDef) Reset()
func (*FunctionDef) Size ¶
func (m *FunctionDef) Size() (n int)
func (*FunctionDef) String ¶
func (m *FunctionDef) String() string
func (*FunctionDef) Unmarshal ¶
func (m *FunctionDef) Unmarshal(dAtA []byte) error
type FunctionDefLibrary ¶
type FunctionDefLibrary struct {
Function []*FunctionDef `protobuf:"bytes,1,rep,name=function" json:"function,omitempty"`
Gradient []*GradientDef `protobuf:"bytes,2,rep,name=gradient" json:"gradient,omitempty"`
}
A library is a set of named functions.
func (*FunctionDefLibrary) Descriptor ¶
func (*FunctionDefLibrary) Descriptor() ([]byte, []int)
func (*FunctionDefLibrary) GetFunction ¶
func (m *FunctionDefLibrary) GetFunction() []*FunctionDef
func (*FunctionDefLibrary) GetGradient ¶
func (m *FunctionDefLibrary) GetGradient() []*GradientDef
func (*FunctionDefLibrary) Marshal ¶
func (m *FunctionDefLibrary) Marshal() (dAtA []byte, err error)
func (*FunctionDefLibrary) MarshalTo ¶
func (m *FunctionDefLibrary) MarshalTo(dAtA []byte) (int, error)
func (*FunctionDefLibrary) ProtoMessage ¶
func (*FunctionDefLibrary) ProtoMessage()
func (*FunctionDefLibrary) Reset ¶
func (m *FunctionDefLibrary) Reset()
func (*FunctionDefLibrary) Size ¶
func (m *FunctionDefLibrary) Size() (n int)
func (*FunctionDefLibrary) String ¶
func (m *FunctionDefLibrary) String() string
func (*FunctionDefLibrary) Unmarshal ¶
func (m *FunctionDefLibrary) Unmarshal(dAtA []byte) error
type GPUOptions ¶
type GPUOptions struct {
// A value between 0 and 1 that indicates what fraction of the
// available GPU memory to pre-allocate for each process. 1 means
// to pre-allocate all of the GPU memory, 0.5 means the process
// allocates ~50% of the available GPU memory.
PerProcessGpuMemoryFraction float64 `` /* 150-byte string literal not displayed */
// 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 */
// 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"`
// 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 "/gpu:0", and "/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: 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.
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 */
// In the event polling loop sleep this many millisconds between
// PollEvents calls, when the queue is empty. If value is not
// set or set to 0, gets set to a non-zero default.
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"`
}
func (*GPUOptions) Descriptor ¶
func (*GPUOptions) Descriptor() ([]byte, []int)
func (*GPUOptions) GetAllocatorType ¶
func (m *GPUOptions) GetAllocatorType() string
func (*GPUOptions) GetAllowGrowth ¶
func (m *GPUOptions) GetAllowGrowth() bool
func (*GPUOptions) GetDeferredDeletionBytes ¶
func (m *GPUOptions) GetDeferredDeletionBytes() int64
func (*GPUOptions) GetForceGpuCompatible ¶
func (m *GPUOptions) GetForceGpuCompatible() bool
func (*GPUOptions) GetPerProcessGpuMemoryFraction ¶
func (m *GPUOptions) GetPerProcessGpuMemoryFraction() float64
func (*GPUOptions) GetPollingActiveDelayUsecs ¶
func (m *GPUOptions) GetPollingActiveDelayUsecs() int32
func (*GPUOptions) GetPollingInactiveDelayMsecs ¶
func (m *GPUOptions) GetPollingInactiveDelayMsecs() int32
func (*GPUOptions) GetVisibleDeviceList ¶
func (m *GPUOptions) GetVisibleDeviceList() string
func (*GPUOptions) Marshal ¶
func (m *GPUOptions) Marshal() (dAtA []byte, err error)
func (*GPUOptions) ProtoMessage ¶
func (*GPUOptions) ProtoMessage()
func (*GPUOptions) Reset ¶
func (m *GPUOptions) Reset()
func (*GPUOptions) Size ¶
func (m *GPUOptions) Size() (n int)
func (*GPUOptions) String ¶
func (m *GPUOptions) String() string
func (*GPUOptions) Unmarshal ¶
func (m *GPUOptions) Unmarshal(dAtA []byte) error
type GetStatusRequest ¶
type GetStatusRequest struct {
}
func (*GetStatusRequest) Descriptor ¶
func (*GetStatusRequest) Descriptor() ([]byte, []int)
func (*GetStatusRequest) Marshal ¶
func (m *GetStatusRequest) Marshal() (dAtA []byte, err error)
func (*GetStatusRequest) ProtoMessage ¶
func (*GetStatusRequest) ProtoMessage()
func (*GetStatusRequest) Reset ¶
func (m *GetStatusRequest) Reset()
func (*GetStatusRequest) Size ¶
func (m *GetStatusRequest) Size() (n int)
func (*GetStatusRequest) String ¶
func (m *GetStatusRequest) String() string
func (*GetStatusRequest) Unmarshal ¶
func (m *GetStatusRequest) Unmarshal(dAtA []byte) error
type GetStatusResponse ¶
type GetStatusResponse struct {
DeviceAttributes []*DeviceAttributes `protobuf:"bytes,1,rep,name=device_attributes,json=deviceAttributes" json:"device_attributes,omitempty"`
}
func (*GetStatusResponse) Descriptor ¶
func (*GetStatusResponse) Descriptor() ([]byte, []int)
func (*GetStatusResponse) GetDeviceAttributes ¶
func (m *GetStatusResponse) GetDeviceAttributes() []*DeviceAttributes
func (*GetStatusResponse) Marshal ¶
func (m *GetStatusResponse) Marshal() (dAtA []byte, err error)
func (*GetStatusResponse) MarshalTo ¶
func (m *GetStatusResponse) MarshalTo(dAtA []byte) (int, error)
func (*GetStatusResponse) ProtoMessage ¶
func (*GetStatusResponse) ProtoMessage()
func (*GetStatusResponse) Reset ¶
func (m *GetStatusResponse) Reset()
func (*GetStatusResponse) Size ¶
func (m *GetStatusResponse) Size() (n int)
func (*GetStatusResponse) String ¶
func (m *GetStatusResponse) String() string
func (*GetStatusResponse) Unmarshal ¶
func (m *GetStatusResponse) Unmarshal(dAtA []byte) error
type GradientDef ¶
type GradientDef struct {
FunctionName string `protobuf:"bytes,1,opt,name=function_name,json=functionName,proto3" json:"function_name,omitempty"`
GradientFunc string `protobuf:"bytes,2,opt,name=gradient_func,json=gradientFunc,proto3" json:"gradient_func,omitempty"`
}
GradientDef defines the gradient function of a function defined in a function library.
A gradient function g (specified by gradient_func) for a function f (specified by function_name) must follow the following:
The function 'f' must be a numerical function which takes N inputs and produces M outputs. Its gradient function 'g', which is a function taking N + M inputs and produces N outputs.
I.e. if we have
(y1, y2, ..., y_M) = f(x1, x2, ..., x_N),
then, g is
(dL/dx1, dL/dx2, ..., dL/dx_N) = g(x1, x2, ..., x_N,
dL/dy1, dL/dy2, ..., dL/dy_M),
where L is a scalar-value function of (x1, x2, ..., xN) (e.g., the loss function). dL/dx_i is the partial derivative of L with respect to x_i.
func (*GradientDef) Descriptor ¶
func (*GradientDef) Descriptor() ([]byte, []int)
func (*GradientDef) GetFunctionName ¶
func (m *GradientDef) GetFunctionName() string
func (*GradientDef) GetGradientFunc ¶
func (m *GradientDef) GetGradientFunc() string
func (*GradientDef) Marshal ¶
func (m *GradientDef) Marshal() (dAtA []byte, err error)
func (*GradientDef) ProtoMessage ¶
func (*GradientDef) ProtoMessage()
func (*GradientDef) Reset ¶
func (m *GradientDef) Reset()
func (*GradientDef) Size ¶
func (m *GradientDef) Size() (n int)
func (*GradientDef) String ¶
func (m *GradientDef) String() string
func (*GradientDef) Unmarshal ¶
func (m *GradientDef) Unmarshal(dAtA []byte) error
type GraphDef ¶
type GraphDef struct {
Node []*NodeDef `protobuf:"bytes,1,rep,name=node" json:"node,omitempty"`
// Compatibility versions of the graph. See core/public/version.h for version
// history. The GraphDef version is distinct from the TensorFlow version, and
// each release of TensorFlow will support a range of GraphDef versions.
Versions *VersionDef `protobuf:"bytes,4,opt,name=versions" json:"versions,omitempty"`
// Deprecated single version field; use versions above instead. Since all
// GraphDef changes before "versions" was introduced were forward
// compatible, this field is entirely ignored.
Version int32 `protobuf:"varint,3,opt,name=version,proto3" json:"version,omitempty"`
// EXPERIMENTAL. DO NOT USE OR DEPEND ON THIS YET.
//
// "library" provides user-defined functions.
//
// Naming:
// * library.function.name are in a flat namespace.
// NOTE: We may need to change it to be hierarchical to support
// different orgs. E.g.,
// { "/google/nn", { ... }},
// { "/google/vision", { ... }}
// { "/org_foo/module_bar", { ... }}
// map<string, FunctionDefLib> named_lib;
// * If node[i].op is the name of one function in "library",
// node[i] is deemed as a function call. Otherwise, node[i].op
// must be a primitive operation supported by the runtime.
//
//
// Function call semantics:
//
// * The callee may start execution as soon as some of its inputs
// are ready. The caller may want to use Tuple() mechanism to
// ensure all inputs are ready in the same time.
//
// * The consumer of return values may start executing as soon as
// the return values the consumer depends on are ready. The
// consumer may want to use Tuple() mechanism to ensure the
// consumer does not start until all return values of the callee
// function are ready.
Library *FunctionDefLibrary `protobuf:"bytes,2,opt,name=library" json:"library,omitempty"`
}
Represents the graph of operations
func (*GraphDef) Descriptor ¶
func (*GraphDef) GetLibrary ¶
func (m *GraphDef) GetLibrary() *FunctionDefLibrary
func (*GraphDef) GetVersion ¶
func (*GraphDef) GetVersions ¶
func (m *GraphDef) GetVersions() *VersionDef
func (*GraphDef) ProtoMessage ¶
func (*GraphDef) ProtoMessage()
type 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" 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.
RewriteOptions *RewriterConfig `protobuf:"bytes,10,opt,name=rewrite_options,json=rewriteOptions" json:"rewrite_options,omitempty"`
}
func (*GraphOptions) Descriptor ¶
func (*GraphOptions) Descriptor() ([]byte, []int)
func (*GraphOptions) GetBuildCostModel ¶
func (m *GraphOptions) GetBuildCostModel() int64
func (*GraphOptions) GetBuildCostModelAfter ¶
func (m *GraphOptions) GetBuildCostModelAfter() int64
func (*GraphOptions) GetEnableBfloat16Sendrecv ¶
func (m *GraphOptions) GetEnableBfloat16Sendrecv() bool
func (*GraphOptions) GetEnableRecvScheduling ¶
func (m *GraphOptions) GetEnableRecvScheduling() bool
func (*GraphOptions) GetInferShapes ¶
func (m *GraphOptions) GetInferShapes() bool
func (*GraphOptions) GetOptimizerOptions ¶
func (m *GraphOptions) GetOptimizerOptions() *OptimizerOptions
func (*GraphOptions) GetPlacePrunedGraph ¶
func (m *GraphOptions) GetPlacePrunedGraph() bool
func (*GraphOptions) GetRewriteOptions ¶
func (m *GraphOptions) GetRewriteOptions() *RewriterConfig
func (*GraphOptions) GetTimelineStep ¶
func (m *GraphOptions) GetTimelineStep() int32
func (*GraphOptions) Marshal ¶
func (m *GraphOptions) Marshal() (dAtA []byte, err error)
func (*GraphOptions) ProtoMessage ¶
func (*GraphOptions) ProtoMessage()
func (*GraphOptions) Reset ¶
func (m *GraphOptions) Reset()
func (*GraphOptions) Size ¶
func (m *GraphOptions) Size() (n int)
func (*GraphOptions) String ¶
func (m *GraphOptions) String() string
func (*GraphOptions) Unmarshal ¶
func (m *GraphOptions) Unmarshal(dAtA []byte) error
type GraphTransferInfo ¶
type GraphTransferInfo struct {
NodeInfo []*GraphTransferInfo_NodeInfo `protobuf:"bytes,1,rep,name=node_info,json=nodeInfo" json:"node_info,omitempty"`
ConstNodeInfo []*GraphTransferInfo_ConstNodeInfo `protobuf:"bytes,2,rep,name=const_node_info,json=constNodeInfo" json:"const_node_info,omitempty"`
NodeInputInfo []*GraphTransferInfo_NodeInputInfo `protobuf:"bytes,3,rep,name=node_input_info,json=nodeInputInfo" json:"node_input_info,omitempty"`
NodeOutputInfo []*GraphTransferInfo_NodeOutputInfo `protobuf:"bytes,4,rep,name=node_output_info,json=nodeOutputInfo" json:"node_output_info,omitempty"`
// Input Node parameters of transferred graph
GraphInputNodeInfo []*GraphTransferInfo_GraphInputNodeInfo `protobuf:"bytes,5,rep,name=graph_input_node_info,json=graphInputNodeInfo" json:"graph_input_node_info,omitempty"`
GraphOutputNodeInfo []*GraphTransferInfo_GraphOutputNodeInfo `protobuf:"bytes,6,rep,name=graph_output_node_info,json=graphOutputNodeInfo" json:"graph_output_node_info,omitempty"`
// Destination of graph transfer
Destination GraphTransferInfo_Destination `protobuf:"varint,7,opt,name=destination,proto3,enum=tensorflow.GraphTransferInfo_Destination" json:"destination,omitempty"`
}
Protocol buffer representing a handle to a tensorflow resource. Handles are not valid across executions, but can be serialized back and forth from within a single run.
func (*GraphTransferInfo) Descriptor ¶
func (*GraphTransferInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo) GetConstNodeInfo ¶
func (m *GraphTransferInfo) GetConstNodeInfo() []*GraphTransferInfo_ConstNodeInfo
func (*GraphTransferInfo) GetDestination ¶
func (m *GraphTransferInfo) GetDestination() GraphTransferInfo_Destination
func (*GraphTransferInfo) GetGraphInputNodeInfo ¶
func (m *GraphTransferInfo) GetGraphInputNodeInfo() []*GraphTransferInfo_GraphInputNodeInfo
func (*GraphTransferInfo) GetGraphOutputNodeInfo ¶
func (m *GraphTransferInfo) GetGraphOutputNodeInfo() []*GraphTransferInfo_GraphOutputNodeInfo
func (*GraphTransferInfo) GetNodeInfo ¶
func (m *GraphTransferInfo) GetNodeInfo() []*GraphTransferInfo_NodeInfo
func (*GraphTransferInfo) GetNodeInputInfo ¶
func (m *GraphTransferInfo) GetNodeInputInfo() []*GraphTransferInfo_NodeInputInfo
func (*GraphTransferInfo) GetNodeOutputInfo ¶
func (m *GraphTransferInfo) GetNodeOutputInfo() []*GraphTransferInfo_NodeOutputInfo
func (*GraphTransferInfo) Marshal ¶
func (m *GraphTransferInfo) Marshal() (dAtA []byte, err error)
func (*GraphTransferInfo) MarshalTo ¶
func (m *GraphTransferInfo) MarshalTo(dAtA []byte) (int, error)
func (*GraphTransferInfo) ProtoMessage ¶
func (*GraphTransferInfo) ProtoMessage()
func (*GraphTransferInfo) Reset ¶
func (m *GraphTransferInfo) Reset()
func (*GraphTransferInfo) Size ¶
func (m *GraphTransferInfo) Size() (n int)
func (*GraphTransferInfo) String ¶
func (m *GraphTransferInfo) String() string
func (*GraphTransferInfo) Unmarshal ¶
func (m *GraphTransferInfo) Unmarshal(dAtA []byte) error
type GraphTransferInfo_ConstNodeInfo ¶
type GraphTransferInfo_ConstNodeInfo struct {
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
NodeId int32 `protobuf:"varint,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
Shape []int64 `protobuf:"varint,3,rep,packed,name=shape" json:"shape,omitempty"`
Data []byte `protobuf:"bytes,4,opt,name=data,proto3" json:"data,omitempty"`
Dtype DataType `protobuf:"varint,5,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
}
func (*GraphTransferInfo_ConstNodeInfo) Descriptor ¶
func (*GraphTransferInfo_ConstNodeInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_ConstNodeInfo) GetData ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetData() []byte
func (*GraphTransferInfo_ConstNodeInfo) GetDtype ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetDtype() DataType
func (*GraphTransferInfo_ConstNodeInfo) GetName ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetName() string
func (*GraphTransferInfo_ConstNodeInfo) GetNodeId ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetNodeId() int32
func (*GraphTransferInfo_ConstNodeInfo) GetShape ¶
func (m *GraphTransferInfo_ConstNodeInfo) GetShape() []int64
func (*GraphTransferInfo_ConstNodeInfo) Marshal ¶
func (m *GraphTransferInfo_ConstNodeInfo) Marshal() (dAtA []byte, err error)
func (*GraphTransferInfo_ConstNodeInfo) MarshalTo ¶
func (m *GraphTransferInfo_ConstNodeInfo) MarshalTo(dAtA []byte) (int, error)
func (*GraphTransferInfo_ConstNodeInfo) ProtoMessage ¶
func (*GraphTransferInfo_ConstNodeInfo) ProtoMessage()
func (*GraphTransferInfo_ConstNodeInfo) Reset ¶
func (m *GraphTransferInfo_ConstNodeInfo) Reset()
func (*GraphTransferInfo_ConstNodeInfo) Size ¶
func (m *GraphTransferInfo_ConstNodeInfo) Size() (n int)
func (*GraphTransferInfo_ConstNodeInfo) String ¶
func (m *GraphTransferInfo_ConstNodeInfo) String() string
func (*GraphTransferInfo_ConstNodeInfo) Unmarshal ¶
func (m *GraphTransferInfo_ConstNodeInfo) Unmarshal(dAtA []byte) error
type GraphTransferInfo_Destination ¶
type GraphTransferInfo_Destination int32
const ( GraphTransferInfo_NOP GraphTransferInfo_Destination = 0 GraphTransferInfo_HEXAGON GraphTransferInfo_Destination = 1 )
func (GraphTransferInfo_Destination) EnumDescriptor ¶
func (GraphTransferInfo_Destination) EnumDescriptor() ([]byte, []int)
func (GraphTransferInfo_Destination) String ¶
func (x GraphTransferInfo_Destination) String() string
type GraphTransferInfo_GraphInputNodeInfo ¶
type GraphTransferInfo_GraphInputNodeInfo struct {
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
Shape []int64 `protobuf:"varint,2,rep,packed,name=shape" json:"shape,omitempty"`
Dtype DataType `protobuf:"varint,3,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
}
func (*GraphTransferInfo_GraphInputNodeInfo) Descriptor ¶
func (*GraphTransferInfo_GraphInputNodeInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_GraphInputNodeInfo) GetDtype ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) GetDtype() DataType
func (*GraphTransferInfo_GraphInputNodeInfo) GetName ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) GetName() string
func (*GraphTransferInfo_GraphInputNodeInfo) GetShape ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) GetShape() []int64
func (*GraphTransferInfo_GraphInputNodeInfo) Marshal ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) Marshal() (dAtA []byte, err error)
func (*GraphTransferInfo_GraphInputNodeInfo) MarshalTo ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) MarshalTo(dAtA []byte) (int, error)
func (*GraphTransferInfo_GraphInputNodeInfo) ProtoMessage ¶
func (*GraphTransferInfo_GraphInputNodeInfo) ProtoMessage()
func (*GraphTransferInfo_GraphInputNodeInfo) Reset ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) Reset()
func (*GraphTransferInfo_GraphInputNodeInfo) Size ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) Size() (n int)
func (*GraphTransferInfo_GraphInputNodeInfo) String ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) String() string
func (*GraphTransferInfo_GraphInputNodeInfo) Unmarshal ¶
func (m *GraphTransferInfo_GraphInputNodeInfo) Unmarshal(dAtA []byte) error
type GraphTransferInfo_GraphOutputNodeInfo ¶
type GraphTransferInfo_GraphOutputNodeInfo struct {
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
Shape []int64 `protobuf:"varint,2,rep,packed,name=shape" json:"shape,omitempty"`
Dtype DataType `protobuf:"varint,3,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
}
func (*GraphTransferInfo_GraphOutputNodeInfo) Descriptor ¶
func (*GraphTransferInfo_GraphOutputNodeInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_GraphOutputNodeInfo) GetDtype ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) GetDtype() DataType
func (*GraphTransferInfo_GraphOutputNodeInfo) GetName ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) GetName() string
func (*GraphTransferInfo_GraphOutputNodeInfo) GetShape ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) GetShape() []int64
func (*GraphTransferInfo_GraphOutputNodeInfo) Marshal ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) Marshal() (dAtA []byte, err error)
func (*GraphTransferInfo_GraphOutputNodeInfo) MarshalTo ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) MarshalTo(dAtA []byte) (int, error)
func (*GraphTransferInfo_GraphOutputNodeInfo) ProtoMessage ¶
func (*GraphTransferInfo_GraphOutputNodeInfo) ProtoMessage()
func (*GraphTransferInfo_GraphOutputNodeInfo) Reset ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) Reset()
func (*GraphTransferInfo_GraphOutputNodeInfo) Size ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) Size() (n int)
func (*GraphTransferInfo_GraphOutputNodeInfo) String ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) String() string
func (*GraphTransferInfo_GraphOutputNodeInfo) Unmarshal ¶
func (m *GraphTransferInfo_GraphOutputNodeInfo) Unmarshal(dAtA []byte) error
type GraphTransferInfo_NodeInfo ¶
type GraphTransferInfo_NodeInfo struct {
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
NodeId int32 `protobuf:"varint,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
TypeName string `protobuf:"bytes,3,opt,name=type_name,json=typeName,proto3" json:"type_name,omitempty"`
SocOpId int32 `protobuf:"varint,4,opt,name=soc_op_id,json=socOpId,proto3" json:"soc_op_id,omitempty"`
PaddingId int32 `protobuf:"varint,5,opt,name=padding_id,json=paddingId,proto3" json:"padding_id,omitempty"`
InputCount int32 `protobuf:"varint,6,opt,name=input_count,json=inputCount,proto3" json:"input_count,omitempty"`
OutputCount int32 `protobuf:"varint,7,opt,name=output_count,json=outputCount,proto3" json:"output_count,omitempty"`
}
func (*GraphTransferInfo_NodeInfo) Descriptor ¶
func (*GraphTransferInfo_NodeInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_NodeInfo) GetInputCount ¶
func (m *GraphTransferInfo_NodeInfo) GetInputCount() int32
func (*GraphTransferInfo_NodeInfo) GetName ¶
func (m *GraphTransferInfo_NodeInfo) GetName() string
func (*GraphTransferInfo_NodeInfo) GetNodeId ¶
func (m *GraphTransferInfo_NodeInfo) GetNodeId() int32
func (*GraphTransferInfo_NodeInfo) GetOutputCount ¶
func (m *GraphTransferInfo_NodeInfo) GetOutputCount() int32
func (*GraphTransferInfo_NodeInfo) GetPaddingId ¶
func (m *GraphTransferInfo_NodeInfo) GetPaddingId() int32
func (*GraphTransferInfo_NodeInfo) GetSocOpId ¶
func (m *GraphTransferInfo_NodeInfo) GetSocOpId() int32
func (*GraphTransferInfo_NodeInfo) GetTypeName ¶
func (m *GraphTransferInfo_NodeInfo) GetTypeName() string
func (*GraphTransferInfo_NodeInfo) Marshal ¶
func (m *GraphTransferInfo_NodeInfo) Marshal() (dAtA []byte, err error)
func (*GraphTransferInfo_NodeInfo) MarshalTo ¶
func (m *GraphTransferInfo_NodeInfo) MarshalTo(dAtA []byte) (int, error)
func (*GraphTransferInfo_NodeInfo) ProtoMessage ¶
func (*GraphTransferInfo_NodeInfo) ProtoMessage()
func (*GraphTransferInfo_NodeInfo) Reset ¶
func (m *GraphTransferInfo_NodeInfo) Reset()
func (*GraphTransferInfo_NodeInfo) Size ¶
func (m *GraphTransferInfo_NodeInfo) Size() (n int)
func (*GraphTransferInfo_NodeInfo) String ¶
func (m *GraphTransferInfo_NodeInfo) String() string
func (*GraphTransferInfo_NodeInfo) Unmarshal ¶
func (m *GraphTransferInfo_NodeInfo) Unmarshal(dAtA []byte) error
type GraphTransferInfo_NodeInput ¶
type GraphTransferInfo_NodeInput struct {
NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
OutputPort int32 `protobuf:"varint,2,opt,name=output_port,json=outputPort,proto3" json:"output_port,omitempty"`
}
func (*GraphTransferInfo_NodeInput) Descriptor ¶
func (*GraphTransferInfo_NodeInput) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_NodeInput) GetNodeId ¶
func (m *GraphTransferInfo_NodeInput) GetNodeId() int32
func (*GraphTransferInfo_NodeInput) GetOutputPort ¶
func (m *GraphTransferInfo_NodeInput) GetOutputPort() int32
func (*GraphTransferInfo_NodeInput) Marshal ¶
func (m *GraphTransferInfo_NodeInput) Marshal() (dAtA []byte, err error)
func (*GraphTransferInfo_NodeInput) MarshalTo ¶
func (m *GraphTransferInfo_NodeInput) MarshalTo(dAtA []byte) (int, error)
func (*GraphTransferInfo_NodeInput) ProtoMessage ¶
func (*GraphTransferInfo_NodeInput) ProtoMessage()
func (*GraphTransferInfo_NodeInput) Reset ¶
func (m *GraphTransferInfo_NodeInput) Reset()
func (*GraphTransferInfo_NodeInput) Size ¶
func (m *GraphTransferInfo_NodeInput) Size() (n int)
func (*GraphTransferInfo_NodeInput) String ¶
func (m *GraphTransferInfo_NodeInput) String() string
func (*GraphTransferInfo_NodeInput) Unmarshal ¶
func (m *GraphTransferInfo_NodeInput) Unmarshal(dAtA []byte) error
type GraphTransferInfo_NodeInputInfo ¶
type GraphTransferInfo_NodeInputInfo struct {
NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
NodeInput []*GraphTransferInfo_NodeInput `protobuf:"bytes,2,rep,name=node_input,json=nodeInput" json:"node_input,omitempty"`
}
func (*GraphTransferInfo_NodeInputInfo) Descriptor ¶
func (*GraphTransferInfo_NodeInputInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_NodeInputInfo) GetNodeId ¶
func (m *GraphTransferInfo_NodeInputInfo) GetNodeId() int32
func (*GraphTransferInfo_NodeInputInfo) GetNodeInput ¶
func (m *GraphTransferInfo_NodeInputInfo) GetNodeInput() []*GraphTransferInfo_NodeInput
func (*GraphTransferInfo_NodeInputInfo) Marshal ¶
func (m *GraphTransferInfo_NodeInputInfo) Marshal() (dAtA []byte, err error)
func (*GraphTransferInfo_NodeInputInfo) MarshalTo ¶
func (m *GraphTransferInfo_NodeInputInfo) MarshalTo(dAtA []byte) (int, error)
func (*GraphTransferInfo_NodeInputInfo) ProtoMessage ¶
func (*GraphTransferInfo_NodeInputInfo) ProtoMessage()
func (*GraphTransferInfo_NodeInputInfo) Reset ¶
func (m *GraphTransferInfo_NodeInputInfo) Reset()
func (*GraphTransferInfo_NodeInputInfo) Size ¶
func (m *GraphTransferInfo_NodeInputInfo) Size() (n int)
func (*GraphTransferInfo_NodeInputInfo) String ¶
func (m *GraphTransferInfo_NodeInputInfo) String() string
func (*GraphTransferInfo_NodeInputInfo) Unmarshal ¶
func (m *GraphTransferInfo_NodeInputInfo) Unmarshal(dAtA []byte) error
type GraphTransferInfo_NodeOutputInfo ¶
type GraphTransferInfo_NodeOutputInfo struct {
NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
MaxByteSize []int32 `protobuf:"varint,2,rep,packed,name=max_byte_size,json=maxByteSize" json:"max_byte_size,omitempty"`
}
func (*GraphTransferInfo_NodeOutputInfo) Descriptor ¶
func (*GraphTransferInfo_NodeOutputInfo) Descriptor() ([]byte, []int)
func (*GraphTransferInfo_NodeOutputInfo) GetMaxByteSize ¶
func (m *GraphTransferInfo_NodeOutputInfo) GetMaxByteSize() []int32
func (*GraphTransferInfo_NodeOutputInfo) GetNodeId ¶
func (m *GraphTransferInfo_NodeOutputInfo) GetNodeId() int32
func (*GraphTransferInfo_NodeOutputInfo) Marshal ¶
func (m *GraphTransferInfo_NodeOutputInfo) Marshal() (dAtA []byte, err error)
func (*GraphTransferInfo_NodeOutputInfo) MarshalTo ¶
func (m *GraphTransferInfo_NodeOutputInfo) MarshalTo(dAtA []byte) (int, error)
func (*GraphTransferInfo_NodeOutputInfo) ProtoMessage ¶
func (*GraphTransferInfo_NodeOutputInfo) ProtoMessage()
func (*GraphTransferInfo_NodeOutputInfo) Reset ¶
func (m *GraphTransferInfo_NodeOutputInfo) Reset()
func (*GraphTransferInfo_NodeOutputInfo) Size ¶
func (m *GraphTransferInfo_NodeOutputInfo) Size() (n int)
func (*GraphTransferInfo_NodeOutputInfo) String ¶
func (m *GraphTransferInfo_NodeOutputInfo) String() string
func (*GraphTransferInfo_NodeOutputInfo) Unmarshal ¶
func (m *GraphTransferInfo_NodeOutputInfo) Unmarshal(dAtA []byte) error
type HistogramProto ¶
type HistogramProto struct {
Min float64 `protobuf:"fixed64,1,opt,name=min,proto3" json:"min,omitempty"`
Max float64 `protobuf:"fixed64,2,opt,name=max,proto3" json:"max,omitempty"`
Num float64 `protobuf:"fixed64,3,opt,name=num,proto3" json:"num,omitempty"`
Sum float64 `protobuf:"fixed64,4,opt,name=sum,proto3" json:"sum,omitempty"`
SumSquares float64 `protobuf:"fixed64,5,opt,name=sum_squares,json=sumSquares,proto3" json:"sum_squares,omitempty"`
// Parallel arrays encoding the bucket boundaries and the bucket values.
// bucket(i) is the count for the bucket i. The range for
// a bucket is:
// i == 0: -DBL_MAX .. bucket_limit(0)
// i != 0: bucket_limit(i-1) .. bucket_limit(i)
BucketLimit []float64 `protobuf:"fixed64,6,rep,packed,name=bucket_limit,json=bucketLimit" json:"bucket_limit,omitempty"`
Bucket []float64 `protobuf:"fixed64,7,rep,packed,name=bucket" json:"bucket,omitempty"`
}
Serialization format for histogram module in core/lib/histogram/histogram.h
func (*HistogramProto) Descriptor ¶
func (*HistogramProto) Descriptor() ([]byte, []int)
func (*HistogramProto) GetBucket ¶
func (m *HistogramProto) GetBucket() []float64
func (*HistogramProto) GetBucketLimit ¶
func (m *HistogramProto) GetBucketLimit() []float64
func (*HistogramProto) GetMax ¶
func (m *HistogramProto) GetMax() float64
func (*HistogramProto) GetMin ¶
func (m *HistogramProto) GetMin() float64
func (*HistogramProto) GetNum ¶
func (m *HistogramProto) GetNum() float64
func (*HistogramProto) GetSum ¶
func (m *HistogramProto) GetSum() float64
func (*HistogramProto) GetSumSquares ¶
func (m *HistogramProto) GetSumSquares() float64
func (*HistogramProto) Marshal ¶
func (m *HistogramProto) Marshal() (dAtA []byte, err error)
func (*HistogramProto) ProtoMessage ¶
func (*HistogramProto) ProtoMessage()
func (*HistogramProto) Reset ¶
func (m *HistogramProto) Reset()
func (*HistogramProto) Size ¶
func (m *HistogramProto) Size() (n int)
func (*HistogramProto) String ¶
func (m *HistogramProto) String() string
func (*HistogramProto) Unmarshal ¶
func (m *HistogramProto) Unmarshal(dAtA []byte) error
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 `` /* 145-byte string literal not displayed */
}
Defines a single job in a TensorFlow cluster.
func (*JobDef) Descriptor ¶
func (*JobDef) ProtoMessage ¶
func (*JobDef) ProtoMessage()
type KernelDef ¶
type KernelDef struct {
// Must match the name of an Op.
Op string `protobuf:"bytes,1,opt,name=op,proto3" json:"op,omitempty"`
// Type of device this kernel runs on.
DeviceType string `protobuf:"bytes,2,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"`
Constraint []*KernelDef_AttrConstraint `protobuf:"bytes,3,rep,name=constraint" json:"constraint,omitempty"`
// Names of the Op's input_/output_args that reside in host memory
// instead of device memory.
HostMemoryArg []string `protobuf:"bytes,4,rep,name=host_memory_arg,json=hostMemoryArg" json:"host_memory_arg,omitempty"`
// This allows experimental kernels to be registered for an op that
// won't be used unless the user specifies a "_kernel" attr with
// value matching this.
Label string `protobuf:"bytes,5,opt,name=label,proto3" json:"label,omitempty"`
}
func (*KernelDef) Descriptor ¶
func (*KernelDef) GetConstraint ¶
func (m *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
func (*KernelDef) GetDeviceType ¶
func (*KernelDef) GetHostMemoryArg ¶
func (*KernelDef) ProtoMessage ¶
func (*KernelDef) ProtoMessage()
type KernelDef_AttrConstraint ¶
type KernelDef_AttrConstraint struct {
// Name of an attr from the Op.
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// A list of values that this kernel supports for this attr.
// Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
AllowedValues *AttrValue `protobuf:"bytes,2,opt,name=allowed_values,json=allowedValues" json:"allowed_values,omitempty"`
}
func (*KernelDef_AttrConstraint) Descriptor ¶
func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)
func (*KernelDef_AttrConstraint) GetAllowedValues ¶
func (m *KernelDef_AttrConstraint) GetAllowedValues() *AttrValue
func (*KernelDef_AttrConstraint) GetName ¶
func (m *KernelDef_AttrConstraint) GetName() string
func (*KernelDef_AttrConstraint) Marshal ¶
func (m *KernelDef_AttrConstraint) Marshal() (dAtA []byte, err error)
func (*KernelDef_AttrConstraint) MarshalTo ¶
func (m *KernelDef_AttrConstraint) MarshalTo(dAtA []byte) (int, error)
func (*KernelDef_AttrConstraint) ProtoMessage ¶
func (*KernelDef_AttrConstraint) ProtoMessage()
func (*KernelDef_AttrConstraint) Reset ¶
func (m *KernelDef_AttrConstraint) Reset()
func (*KernelDef_AttrConstraint) Size ¶
func (m *KernelDef_AttrConstraint) Size() (n int)
func (*KernelDef_AttrConstraint) String ¶
func (m *KernelDef_AttrConstraint) String() string
func (*KernelDef_AttrConstraint) Unmarshal ¶
func (m *KernelDef_AttrConstraint) Unmarshal(dAtA []byte) error
type LabeledStepStats ¶
type LabeledStepStats struct {
StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
StepStats *StepStats `protobuf:"bytes,2,opt,name=step_stats,json=stepStats" json:"step_stats,omitempty"`
}
func (*LabeledStepStats) Descriptor ¶
func (*LabeledStepStats) Descriptor() ([]byte, []int)
func (*LabeledStepStats) GetStepId ¶
func (m *LabeledStepStats) GetStepId() int64
func (*LabeledStepStats) GetStepStats ¶
func (m *LabeledStepStats) GetStepStats() *StepStats
func (*LabeledStepStats) Marshal ¶
func (m *LabeledStepStats) Marshal() (dAtA []byte, err error)
func (*LabeledStepStats) ProtoMessage ¶
func (*LabeledStepStats) ProtoMessage()
func (*LabeledStepStats) Reset ¶
func (m *LabeledStepStats) Reset()
func (*LabeledStepStats) Size ¶
func (m *LabeledStepStats) Size() (n int)
func (*LabeledStepStats) String ¶
func (m *LabeledStepStats) String() string
func (*LabeledStepStats) Unmarshal ¶
func (m *LabeledStepStats) Unmarshal(dAtA []byte) error
type ListDevicesRequest ¶
type ListDevicesRequest struct {
}
func (*ListDevicesRequest) Descriptor ¶
func (*ListDevicesRequest) Descriptor() ([]byte, []int)
func (*ListDevicesRequest) Marshal ¶
func (m *ListDevicesRequest) Marshal() (dAtA []byte, err error)
func (*ListDevicesRequest) MarshalTo ¶
func (m *ListDevicesRequest) MarshalTo(dAtA []byte) (int, error)
func (*ListDevicesRequest) ProtoMessage ¶
func (*ListDevicesRequest) ProtoMessage()
func (*ListDevicesRequest) Reset ¶
func (m *ListDevicesRequest) Reset()
func (*ListDevicesRequest) Size ¶
func (m *ListDevicesRequest) Size() (n int)
func (*ListDevicesRequest) String ¶
func (m *ListDevicesRequest) String() string
func (*ListDevicesRequest) Unmarshal ¶
func (m *ListDevicesRequest) Unmarshal(dAtA []byte) error
type ListDevicesResponse ¶
type ListDevicesResponse struct {
LocalDevice []*DeviceAttributes `protobuf:"bytes,1,rep,name=local_device,json=localDevice" json:"local_device,omitempty"`
RemoteDevice []*DeviceAttributes `protobuf:"bytes,2,rep,name=remote_device,json=remoteDevice" json:"remote_device,omitempty"`
}
func (*ListDevicesResponse) Descriptor ¶
func (*ListDevicesResponse) Descriptor() ([]byte, []int)
func (*ListDevicesResponse) GetLocalDevice ¶
func (m *ListDevicesResponse) GetLocalDevice() []*DeviceAttributes
func (*ListDevicesResponse) GetRemoteDevice ¶
func (m *ListDevicesResponse) GetRemoteDevice() []*DeviceAttributes
func (*ListDevicesResponse) Marshal ¶
func (m *ListDevicesResponse) Marshal() (dAtA []byte, err error)
func (*ListDevicesResponse) MarshalTo ¶
func (m *ListDevicesResponse) MarshalTo(dAtA []byte) (int, error)
func (*ListDevicesResponse) ProtoMessage ¶
func (*ListDevicesResponse) ProtoMessage()
func (*ListDevicesResponse) Reset ¶
func (m *ListDevicesResponse) Reset()
func (*ListDevicesResponse) Size ¶
func (m *ListDevicesResponse) Size() (n int)
func (*ListDevicesResponse) String ¶
func (m *ListDevicesResponse) String() string
func (*ListDevicesResponse) Unmarshal ¶
func (m *ListDevicesResponse) Unmarshal(dAtA []byte) error
type LoggingRequest ¶
type LoggingRequest struct {
// If true, RPC logging will be activated.
RpcLogging bool `protobuf:"varint,1,opt,name=rpc_logging,json=rpcLogging,proto3" json:"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" json:"fetch_step_id,omitempty"`
}
Out-of-band request to begin or end logging, or to retrieve logs for particular steps.
func (*LoggingRequest) Descriptor ¶
func (*LoggingRequest) Descriptor() ([]byte, []int)
func (*LoggingRequest) GetClear ¶
func (m *LoggingRequest) GetClear() bool
func (*LoggingRequest) GetFetchStepId ¶
func (m *LoggingRequest) GetFetchStepId() []int64
func (*LoggingRequest) GetRpcLogging ¶
func (m *LoggingRequest) GetRpcLogging() bool
func (*LoggingRequest) Marshal ¶
func (m *LoggingRequest) Marshal() (dAtA []byte, err error)
func (*LoggingRequest) ProtoMessage ¶
func (*LoggingRequest) ProtoMessage()
func (*LoggingRequest) Reset ¶
func (m *LoggingRequest) Reset()
func (*LoggingRequest) Size ¶
func (m *LoggingRequest) Size() (n int)
func (*LoggingRequest) String ¶
func (m *LoggingRequest) String() string
func (*LoggingRequest) Unmarshal ¶
func (m *LoggingRequest) Unmarshal(dAtA []byte) error
type LoggingResponse ¶
type LoggingResponse struct {
Step []*LabeledStepStats `protobuf:"bytes,1,rep,name=step" json:"step,omitempty"`
}
func (*LoggingResponse) Descriptor ¶
func (*LoggingResponse) Descriptor() ([]byte, []int)
func (*LoggingResponse) GetStep ¶
func (m *LoggingResponse) GetStep() []*LabeledStepStats
func (*LoggingResponse) Marshal ¶
func (m *LoggingResponse) Marshal() (dAtA []byte, err error)
func (*LoggingResponse) ProtoMessage ¶
func (*LoggingResponse) ProtoMessage()
func (*LoggingResponse) Reset ¶
func (m *LoggingResponse) Reset()
func (*LoggingResponse) Size ¶
func (m *LoggingResponse) Size() (n int)
func (*LoggingResponse) String ¶
func (m *LoggingResponse) String() string
func (*LoggingResponse) Unmarshal ¶
func (m *LoggingResponse) Unmarshal(dAtA []byte) error
type MasterServiceClient ¶
type MasterServiceClient interface {
// Creates a session.
CreateSession(ctx context.Context, in *CreateSessionRequest, opts ...grpc.CallOption) (*CreateSessionResponse, error)
// Extends a session.
ExtendSession(ctx context.Context, in *ExtendSessionRequest, opts ...grpc.CallOption) (*ExtendSessionResponse, error)
// Prepares future partial run calls.
PartialRunSetup(ctx context.Context, in *PartialRunSetupRequest, opts ...grpc.CallOption) (*PartialRunSetupResponse, error)
// Drives the graph computation.
RunStep(ctx context.Context, in *RunStepRequest, opts ...grpc.CallOption) (*RunStepResponse, error)
// Closes a session.
CloseSession(ctx context.Context, in *CloseSessionRequest, opts ...grpc.CallOption) (*CloseSessionResponse, error)
// List the devices usable by the master.
ListDevices(ctx context.Context, in *ListDevicesRequest, opts ...grpc.CallOption) (*ListDevicesResponse, error)
// Close and abandon all existing sessions. Ongoing computations
// will no longer affect fresh ones via the resources in containers listed in
// the ResetRequest. See ResetRequest for more details.
Reset(ctx context.Context, in *ResetRequest, opts ...grpc.CallOption) (*ResetResponse, error)
}
func NewMasterServiceClient ¶
func NewMasterServiceClient(cc *grpc.ClientConn) MasterServiceClient
type MasterServiceServer ¶
type MasterServiceServer interface {
// Creates a session.
CreateSession(context.Context, *CreateSessionRequest) (*CreateSessionResponse, error)
// Extends a session.
ExtendSession(context.Context, *ExtendSessionRequest) (*ExtendSessionResponse, error)
// Prepares future partial run calls.
PartialRunSetup(context.Context, *PartialRunSetupRequest) (*PartialRunSetupResponse, error)
// Drives the graph computation.
RunStep(context.Context, *RunStepRequest) (*RunStepResponse, error)
// Closes a session.
CloseSession(context.Context, *CloseSessionRequest) (*CloseSessionResponse, error)
// List the devices usable by the master.
ListDevices(context.Context, *ListDevicesRequest) (*ListDevicesResponse, error)
// Close and abandon all existing sessions. Ongoing computations
// will no longer affect fresh ones via the resources in containers listed in
// the ResetRequest. See ResetRequest for more details.
Reset(context.Context, *ResetRequest) (*ResetResponse, error)
}
type MemoryLogRawAllocation ¶
type MemoryLogRawAllocation struct {
// Process-unique step id.
StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
// Name of the operation making the allocation.
Operation string `protobuf:"bytes,2,opt,name=operation,proto3" json:"operation,omitempty"`
// Number of bytes in the allocation.
NumBytes int64 `protobuf:"varint,3,opt,name=num_bytes,json=numBytes,proto3" json:"num_bytes,omitempty"`
// Address of the allocation.
Ptr uint64 `protobuf:"varint,4,opt,name=ptr,proto3" json:"ptr,omitempty"`
// Id of the tensor buffer being allocated, used to match to a
// corresponding deallocation.
AllocationId int64 `protobuf:"varint,5,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
// Name of the allocator used.
AllocatorName string `protobuf:"bytes,6,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
}
func (*MemoryLogRawAllocation) Descriptor ¶
func (*MemoryLogRawAllocation) Descriptor() ([]byte, []int)
func (*MemoryLogRawAllocation) GetAllocationId ¶
func (m *MemoryLogRawAllocation) GetAllocationId() int64
func (*MemoryLogRawAllocation) GetAllocatorName ¶
func (m *MemoryLogRawAllocation) GetAllocatorName() string
func (*MemoryLogRawAllocation) GetNumBytes ¶
func (m *MemoryLogRawAllocation) GetNumBytes() int64
func (*MemoryLogRawAllocation) GetOperation ¶
func (m *MemoryLogRawAllocation) GetOperation() string
func (*MemoryLogRawAllocation) GetPtr ¶
func (m *MemoryLogRawAllocation) GetPtr() uint64
func (*MemoryLogRawAllocation) GetStepId ¶
func (m *MemoryLogRawAllocation) GetStepId() int64
func (*MemoryLogRawAllocation) Marshal ¶
func (m *MemoryLogRawAllocation) Marshal() (dAtA []byte, err error)
func (*MemoryLogRawAllocation) MarshalTo ¶
func (m *MemoryLogRawAllocation) MarshalTo(dAtA []byte) (int, error)
func (*MemoryLogRawAllocation) ProtoMessage ¶
func (*MemoryLogRawAllocation) ProtoMessage()
func (*MemoryLogRawAllocation) Reset ¶
func (m *MemoryLogRawAllocation) Reset()
func (*MemoryLogRawAllocation) Size ¶
func (m *MemoryLogRawAllocation) Size() (n int)
func (*MemoryLogRawAllocation) String ¶
func (m *MemoryLogRawAllocation) String() string
func (*MemoryLogRawAllocation) Unmarshal ¶
func (m *MemoryLogRawAllocation) Unmarshal(dAtA []byte) error
type MemoryLogRawDeallocation ¶
type MemoryLogRawDeallocation struct {
// Process-unique step id.
StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
// Name of the operation making the deallocation.
Operation string `protobuf:"bytes,2,opt,name=operation,proto3" json:"operation,omitempty"`
// Id of the tensor buffer being deallocated, used to match to a
// corresponding allocation.
AllocationId int64 `protobuf:"varint,3,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
// Name of the allocator used.
AllocatorName string `protobuf:"bytes,4,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
// True if the deallocation is queued and will be performed later,
// e.g. for GPU lazy freeing of buffers.
Deferred bool `protobuf:"varint,5,opt,name=deferred,proto3" json:"deferred,omitempty"`
}
func (*MemoryLogRawDeallocation) Descriptor ¶
func (*MemoryLogRawDeallocation) Descriptor() ([]byte, []int)
func (*MemoryLogRawDeallocation) GetAllocationId ¶
func (m *MemoryLogRawDeallocation) GetAllocationId() int64
func (*MemoryLogRawDeallocation) GetAllocatorName ¶
func (m *MemoryLogRawDeallocation) GetAllocatorName() string
func (*MemoryLogRawDeallocation) GetDeferred ¶
func (m *MemoryLogRawDeallocation) GetDeferred() bool
func (*MemoryLogRawDeallocation) GetOperation ¶
func (m *MemoryLogRawDeallocation) GetOperation() string
func (*MemoryLogRawDeallocation) GetStepId ¶
func (m *MemoryLogRawDeallocation) GetStepId() int64
func (*MemoryLogRawDeallocation) Marshal ¶
func (m *MemoryLogRawDeallocation) Marshal() (dAtA []byte, err error)
func (*MemoryLogRawDeallocation) MarshalTo ¶
func (m *MemoryLogRawDeallocation) MarshalTo(dAtA []byte) (int, error)
func (*MemoryLogRawDeallocation) ProtoMessage ¶
func (*MemoryLogRawDeallocation) ProtoMessage()
func (*MemoryLogRawDeallocation) Reset ¶
func (m *MemoryLogRawDeallocation) Reset()
func (*MemoryLogRawDeallocation) Size ¶
func (m *MemoryLogRawDeallocation) Size() (n int)
func (*MemoryLogRawDeallocation) String ¶
func (m *MemoryLogRawDeallocation) String() string
func (*MemoryLogRawDeallocation) Unmarshal ¶
func (m *MemoryLogRawDeallocation) Unmarshal(dAtA []byte) error
type MemoryLogStep ¶
type MemoryLogStep struct {
// Process-unique step id.
StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
// Handle describing the feeds and fetches of the step.
Handle string `protobuf:"bytes,2,opt,name=handle,proto3" json:"handle,omitempty"`
}
func (*MemoryLogStep) Descriptor ¶
func (*MemoryLogStep) Descriptor() ([]byte, []int)
func (*MemoryLogStep) GetHandle ¶
func (m *MemoryLogStep) GetHandle() string
func (*MemoryLogStep) GetStepId ¶
func (m *MemoryLogStep) GetStepId() int64
func (*MemoryLogStep) Marshal ¶
func (m *MemoryLogStep) Marshal() (dAtA []byte, err error)
func (*MemoryLogStep) ProtoMessage ¶
func (*MemoryLogStep) ProtoMessage()
func (*MemoryLogStep) Reset ¶
func (m *MemoryLogStep) Reset()
func (*MemoryLogStep) Size ¶
func (m *MemoryLogStep) Size() (n int)
func (*MemoryLogStep) String ¶
func (m *MemoryLogStep) String() string
func (*MemoryLogStep) Unmarshal ¶
func (m *MemoryLogStep) Unmarshal(dAtA []byte) error
type MemoryLogTensorAllocation ¶
type MemoryLogTensorAllocation struct {
// Process-unique step id.
StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
// Name of the kernel making the allocation as set in GraphDef,
// e.g., "affine2/weights/Assign".
KernelName string `protobuf:"bytes,2,opt,name=kernel_name,json=kernelName,proto3" json:"kernel_name,omitempty"`
// Allocated tensor details.
Tensor *TensorDescription `protobuf:"bytes,3,opt,name=tensor" json:"tensor,omitempty"`
}
func (*MemoryLogTensorAllocation) Descriptor ¶
func (*MemoryLogTensorAllocation) Descriptor() ([]byte, []int)
func (*MemoryLogTensorAllocation) GetKernelName ¶
func (m *MemoryLogTensorAllocation) GetKernelName() string
func (*MemoryLogTensorAllocation) GetStepId ¶
func (m *MemoryLogTensorAllocation) GetStepId() int64
func (*MemoryLogTensorAllocation) GetTensor ¶
func (m *MemoryLogTensorAllocation) GetTensor() *TensorDescription
func (*MemoryLogTensorAllocation) Marshal ¶
func (m *MemoryLogTensorAllocation) Marshal() (dAtA []byte, err error)
func (*MemoryLogTensorAllocation) MarshalTo ¶
func (m *MemoryLogTensorAllocation) MarshalTo(dAtA []byte) (int, error)
func (*MemoryLogTensorAllocation) ProtoMessage ¶
func (*MemoryLogTensorAllocation) ProtoMessage()
func (*MemoryLogTensorAllocation) Reset ¶
func (m *MemoryLogTensorAllocation) Reset()
func (*MemoryLogTensorAllocation) Size ¶
func (m *MemoryLogTensorAllocation) Size() (n int)
func (*MemoryLogTensorAllocation) String ¶
func (m *MemoryLogTensorAllocation) String() string
func (*MemoryLogTensorAllocation) Unmarshal ¶
func (m *MemoryLogTensorAllocation) Unmarshal(dAtA []byte) error
type MemoryLogTensorDeallocation ¶
type MemoryLogTensorDeallocation struct {
// Id of the tensor buffer being deallocated, used to match to a
// corresponding allocation.
AllocationId int64 `protobuf:"varint,1,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
// Name of the allocator used.
AllocatorName string `protobuf:"bytes,2,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
}
func (*MemoryLogTensorDeallocation) Descriptor ¶
func (*MemoryLogTensorDeallocation) Descriptor() ([]byte, []int)
func (*MemoryLogTensorDeallocation) GetAllocationId ¶
func (m *MemoryLogTensorDeallocation) GetAllocationId() int64
func (*MemoryLogTensorDeallocation) GetAllocatorName ¶
func (m *MemoryLogTensorDeallocation) GetAllocatorName() string
func (*MemoryLogTensorDeallocation) Marshal ¶
func (m *MemoryLogTensorDeallocation) Marshal() (dAtA []byte, err error)
func (*MemoryLogTensorDeallocation) MarshalTo ¶
func (m *MemoryLogTensorDeallocation) MarshalTo(dAtA []byte) (int, error)
func (*MemoryLogTensorDeallocation) ProtoMessage ¶
func (*MemoryLogTensorDeallocation) ProtoMessage()
func (*MemoryLogTensorDeallocation) Reset ¶
func (m *MemoryLogTensorDeallocation) Reset()
func (*MemoryLogTensorDeallocation) Size ¶
func (m *MemoryLogTensorDeallocation) Size() (n int)
func (*MemoryLogTensorDeallocation) String ¶
func (m *MemoryLogTensorDeallocation) String() string
func (*MemoryLogTensorDeallocation) Unmarshal ¶
func (m *MemoryLogTensorDeallocation) Unmarshal(dAtA []byte) error
type MemoryLogTensorOutput ¶
type MemoryLogTensorOutput struct {
// Process-unique step id.
StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
// Name of the kernel producing an output as set in GraphDef, e.g.,
// "affine2/weights/Assign".
KernelName string `protobuf:"bytes,2,opt,name=kernel_name,json=kernelName,proto3" json:"kernel_name,omitempty"`
// Index of the output being set.
Index int32 `protobuf:"varint,3,opt,name=index,proto3" json:"index,omitempty"`
// Output tensor details.
Tensor *TensorDescription `protobuf:"bytes,4,opt,name=tensor" json:"tensor,omitempty"`
}
func (*MemoryLogTensorOutput) Descriptor ¶
func (*MemoryLogTensorOutput) Descriptor() ([]byte, []int)
func (*MemoryLogTensorOutput) GetIndex ¶
func (m *MemoryLogTensorOutput) GetIndex() int32
func (*MemoryLogTensorOutput) GetKernelName ¶
func (m *MemoryLogTensorOutput) GetKernelName() string
func (*MemoryLogTensorOutput) GetStepId ¶
func (m *MemoryLogTensorOutput) GetStepId() int64
func (*MemoryLogTensorOutput) GetTensor ¶
func (m *MemoryLogTensorOutput) GetTensor() *TensorDescription
func (*MemoryLogTensorOutput) Marshal ¶
func (m *MemoryLogTensorOutput) Marshal() (dAtA []byte, err error)
func (*MemoryLogTensorOutput) MarshalTo ¶
func (m *MemoryLogTensorOutput) MarshalTo(dAtA []byte) (int, error)
func (*MemoryLogTensorOutput) ProtoMessage ¶
func (*MemoryLogTensorOutput) ProtoMessage()
func (*MemoryLogTensorOutput) Reset ¶
func (m *MemoryLogTensorOutput) Reset()
func (*MemoryLogTensorOutput) Size ¶
func (m *MemoryLogTensorOutput) Size() (n int)
func (*MemoryLogTensorOutput) String ¶
func (m *MemoryLogTensorOutput) String() string
func (*MemoryLogTensorOutput) Unmarshal ¶
func (m *MemoryLogTensorOutput) Unmarshal(dAtA []byte) error
type MemoryStats ¶
type MemoryStats struct {
HostTempMemorySize int64 `protobuf:"varint,1,opt,name=host_temp_memory_size,json=hostTempMemorySize,proto3" json:"host_temp_memory_size,omitempty"`
DeviceTempMemorySize int64 `` /* 126-byte string literal not displayed */
HostPersistentMemorySize int64 `` /* 138-byte string literal not displayed */
DevicePersistentMemorySize int64 `` /* 144-byte string literal not displayed */
HostPersistentTensorAllocIds []int64 `` /* 152-byte string literal not displayed */
DevicePersistentTensorAllocIds []int64 `` /* 158-byte string literal not displayed */
}
For memory tracking.
func (*MemoryStats) Descriptor ¶
func (*MemoryStats) Descriptor() ([]byte, []int)
func (*MemoryStats) GetDevicePersistentMemorySize ¶
func (m *MemoryStats) GetDevicePersistentMemorySize() int64
func (*MemoryStats) GetDevicePersistentTensorAllocIds ¶
func (m *MemoryStats) GetDevicePersistentTensorAllocIds() []int64
func (*MemoryStats) GetDeviceTempMemorySize ¶
func (m *MemoryStats) GetDeviceTempMemorySize() int64
func (*MemoryStats) GetHostPersistentMemorySize ¶
func (m *MemoryStats) GetHostPersistentMemorySize() int64
func (*MemoryStats) GetHostPersistentTensorAllocIds ¶
func (m *MemoryStats) GetHostPersistentTensorAllocIds() []int64
func (*MemoryStats) GetHostTempMemorySize ¶
func (m *MemoryStats) GetHostTempMemorySize() int64
func (*MemoryStats) Marshal ¶
func (m *MemoryStats) Marshal() (dAtA []byte, err error)
func (*MemoryStats) ProtoMessage ¶
func (*MemoryStats) ProtoMessage()
func (*MemoryStats) Reset ¶
func (m *MemoryStats) Reset()
func (*MemoryStats) Size ¶
func (m *MemoryStats) Size() (n int)
func (*MemoryStats) String ¶
func (m *MemoryStats) String() string
func (*MemoryStats) Unmarshal ¶
func (m *MemoryStats) Unmarshal(dAtA []byte) error
type MetaGraphDef ¶
type MetaGraphDef struct {
MetaInfoDef *MetaGraphDef_MetaInfoDef `protobuf:"bytes,1,opt,name=meta_info_def,json=metaInfoDef" json:"meta_info_def,omitempty"`
// GraphDef.
GraphDef *GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef" json:"graph_def,omitempty"`
// SaverDef.
SaverDef *SaverDef `protobuf:"bytes,3,opt,name=saver_def,json=saverDef" json:"saver_def,omitempty"`
// collection_def: Map from collection name to collections.
// See CollectionDef section for details.
CollectionDef map[string]*CollectionDef `` /* 174-byte string literal not displayed */
// signature_def: Map from user supplied key for a signature to a single
// SignatureDef.
SignatureDef map[string]*SignatureDef `` /* 171-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" json:"asset_file_def,omitempty"`
}
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 FromCheckpoint ¶
func FromCheckpoint(r io.Reader) (*MetaGraphDef, error)
func (*MetaGraphDef) Descriptor ¶
func (*MetaGraphDef) Descriptor() ([]byte, []int)
func (*MetaGraphDef) GetAssetFileDef ¶
func (m *MetaGraphDef) GetAssetFileDef() []*AssetFileDef
func (*MetaGraphDef) GetCollectionDef ¶
func (m *MetaGraphDef) GetCollectionDef() map[string]*CollectionDef
func (*MetaGraphDef) GetGraphDef ¶
func (m *MetaGraphDef) GetGraphDef() *GraphDef
func (*MetaGraphDef) GetMetaInfoDef ¶
func (m *MetaGraphDef) GetMetaInfoDef() *MetaGraphDef_MetaInfoDef
func (*MetaGraphDef) GetSaverDef ¶
func (m *MetaGraphDef) GetSaverDef() *SaverDef
func (*MetaGraphDef) GetSignatureDef ¶
func (m *MetaGraphDef) GetSignatureDef() map[string]*SignatureDef
func (*MetaGraphDef) Marshal ¶
func (m *MetaGraphDef) Marshal() (dAtA []byte, err error)
func (*MetaGraphDef) ProtoMessage ¶
func (*MetaGraphDef) ProtoMessage()
func (*MetaGraphDef) Reset ¶
func (m *MetaGraphDef) Reset()
func (*MetaGraphDef) Size ¶
func (m *MetaGraphDef) Size() (n int)
func (*MetaGraphDef) String ¶
func (m *MetaGraphDef) String() string
func (*MetaGraphDef) Unmarshal ¶
func (m *MetaGraphDef) Unmarshal(dAtA []byte) error
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 *OpList `protobuf:"bytes,2,opt,name=stripped_op_list,json=strippedOpList" 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 *google_protobuf.Any `protobuf:"bytes,3,opt,name=any_info,json=anyInfo" 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" 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"`
}
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 ¶
func (*MetaGraphDef_MetaInfoDef) Descriptor() ([]byte, []int)
func (*MetaGraphDef_MetaInfoDef) GetAnyInfo ¶
func (m *MetaGraphDef_MetaInfoDef) GetAnyInfo() *google_protobuf.Any
func (*MetaGraphDef_MetaInfoDef) GetMetaGraphVersion ¶
func (m *MetaGraphDef_MetaInfoDef) GetMetaGraphVersion() string
func (*MetaGraphDef_MetaInfoDef) GetStrippedOpList ¶
func (m *MetaGraphDef_MetaInfoDef) GetStrippedOpList() *OpList
func (*MetaGraphDef_MetaInfoDef) GetTags ¶
func (m *MetaGraphDef_MetaInfoDef) GetTags() []string
func (*MetaGraphDef_MetaInfoDef) GetTensorflowGitVersion ¶
func (m *MetaGraphDef_MetaInfoDef) GetTensorflowGitVersion() string
func (*MetaGraphDef_MetaInfoDef) GetTensorflowVersion ¶
func (m *MetaGraphDef_MetaInfoDef) GetTensorflowVersion() string
func (*MetaGraphDef_MetaInfoDef) Marshal ¶
func (m *MetaGraphDef_MetaInfoDef) Marshal() (dAtA []byte, err error)
func (*MetaGraphDef_MetaInfoDef) MarshalTo ¶
func (m *MetaGraphDef_MetaInfoDef) MarshalTo(dAtA []byte) (int, error)
func (*MetaGraphDef_MetaInfoDef) ProtoMessage ¶
func (*MetaGraphDef_MetaInfoDef) ProtoMessage()
func (*MetaGraphDef_MetaInfoDef) Reset ¶
func (m *MetaGraphDef_MetaInfoDef) Reset()
func (*MetaGraphDef_MetaInfoDef) Size ¶
func (m *MetaGraphDef_MetaInfoDef) Size() (n int)
func (*MetaGraphDef_MetaInfoDef) String ¶
func (m *MetaGraphDef_MetaInfoDef) String() string
func (*MetaGraphDef_MetaInfoDef) Unmarshal ¶
func (m *MetaGraphDef_MetaInfoDef) Unmarshal(dAtA []byte) error
type NameAttrList ¶
type NameAttrList struct {
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
Attr map[string]*AttrValue `` /* 135-byte string literal not displayed */
}
A list of attr names and their values. The whole list is attached with a string name. E.g., MatMul[T=float].
func (*NameAttrList) Descriptor ¶
func (*NameAttrList) Descriptor() ([]byte, []int)
func (*NameAttrList) GetAttr ¶
func (m *NameAttrList) GetAttr() map[string]*AttrValue
func (*NameAttrList) GetName ¶
func (m *NameAttrList) GetName() string
func (*NameAttrList) Marshal ¶
func (m *NameAttrList) Marshal() (dAtA []byte, err error)
func (*NameAttrList) ProtoMessage ¶
func (*NameAttrList) ProtoMessage()
func (*NameAttrList) Reset ¶
func (m *NameAttrList) Reset()
func (*NameAttrList) Size ¶
func (m *NameAttrList) Size() (n int)
func (*NameAttrList) String ¶
func (m *NameAttrList) String() string
func (*NameAttrList) Unmarshal ¶
func (m *NameAttrList) Unmarshal(dAtA []byte) error
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 *TensorProto `protobuf:"bytes,2,opt,name=tensor" json:"tensor,omitempty"`
}
A pair of tensor name and tensor values.
func (*NamedTensorProto) Descriptor ¶
func (*NamedTensorProto) Descriptor() ([]byte, []int)
func (*NamedTensorProto) GetName ¶
func (m *NamedTensorProto) GetName() string
func (*NamedTensorProto) GetTensor ¶
func (m *NamedTensorProto) GetTensor() *TensorProto
func (*NamedTensorProto) Marshal ¶
func (m *NamedTensorProto) Marshal() (dAtA []byte, err error)
func (*NamedTensorProto) ProtoMessage ¶
func (*NamedTensorProto) ProtoMessage()
func (*NamedTensorProto) Reset ¶
func (m *NamedTensorProto) Reset()
func (*NamedTensorProto) Size ¶
func (m *NamedTensorProto) Size() (n int)
func (*NamedTensorProto) String ¶
func (m *NamedTensorProto) String() string
func (*NamedTensorProto) Unmarshal ¶
func (m *NamedTensorProto) Unmarshal(dAtA []byte) error
type NodeDef ¶
type NodeDef struct {
// The name given to this operator. Used for naming inputs,
// logging, visualization, etc. Unique within a single GraphDef.
// Must match the regexp "[A-Za-z0-9.][A-Za-z0-9_./]*".
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// The operation name. There may be custom parameters in attrs.
// Op names starting with an underscore are reserved for internal use.
Op string `protobuf:"bytes,2,opt,name=op,proto3" json:"op,omitempty"`
// Each input is "node:src_output" with "node" being a string name and
// "src_output" indicating which output tensor to use from "node". If
// "src_output" is 0 the ":0" suffix can be omitted. Regular inputs
// may optionally be followed by control inputs that have the format
// "^node".
Input []string `protobuf:"bytes,3,rep,name=input" json:"input,omitempty"`
// A (possibly partial) specification for the device on which this
// node should be placed.
// The expected syntax for this string is as follows:
//
// DEVICE_SPEC ::= PARTIAL_SPEC
//
// PARTIAL_SPEC ::= ("/" CONSTRAINT) *
// CONSTRAINT ::= ("job:" JOB_NAME)
// | ("replica:" [1-9][0-9]*)
// | ("task:" [1-9][0-9]*)
// | ( ("gpu" | "cpu") ":" ([1-9][0-9]* | "*") )
//
// Valid values for this string include:
// * "/job:worker/replica:0/task:1/gpu:3" (full specification)
// * "/job:worker/gpu:3" (partial specification)
// * "" (no specification)
//
// If the constraints do not resolve to a single device (or if this
// field is empty or not present), the runtime will attempt to
// choose a device automatically.
Device string `protobuf:"bytes,4,opt,name=device,proto3" json:"device,omitempty"`
// Operation-specific graph-construction-time configuration.
// Note that this should include all attrs defined in the
// corresponding OpDef, including those with a value matching
// the default -- this allows the default to change and makes
// NodeDefs easier to interpret on their own. However, if
// an attr with a default is not specified in this list, the
// default will be used.
// The "names" (keys) must match the regexp "[a-z][a-z0-9_]+" (and
// one of the names from the corresponding OpDef's attr field).
// The values must have a type matching the corresponding OpDef
// attr's type field.
// TODO(josh11b): Add some examples here showing best practices.
Attr map[string]*AttrValue `` /* 135-byte string literal not displayed */
}
func (*NodeDef) Descriptor ¶
func (*NodeDef) ProtoMessage ¶
func (*NodeDef) ProtoMessage()
type NodeExecStats ¶
type NodeExecStats struct {
// TODO(tucker): Use some more compact form of node identity than
// the full string name. Either all processes should agree on a
// global id (cost_id?) for each node, or we should use a hash of
// the name.
NodeName string `protobuf:"bytes,1,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"`
AllStartMicros int64 `protobuf:"varint,2,opt,name=all_start_micros,json=allStartMicros,proto3" json:"all_start_micros,omitempty"`
OpStartRelMicros int64 `protobuf:"varint,3,opt,name=op_start_rel_micros,json=opStartRelMicros,proto3" json:"op_start_rel_micros,omitempty"`
OpEndRelMicros int64 `protobuf:"varint,4,opt,name=op_end_rel_micros,json=opEndRelMicros,proto3" json:"op_end_rel_micros,omitempty"`
AllEndRelMicros int64 `protobuf:"varint,5,opt,name=all_end_rel_micros,json=allEndRelMicros,proto3" json:"all_end_rel_micros,omitempty"`
Memory []*AllocatorMemoryUsed `protobuf:"bytes,6,rep,name=memory" json:"memory,omitempty"`
Output []*NodeOutput `protobuf:"bytes,7,rep,name=output" json:"output,omitempty"`
TimelineLabel string `protobuf:"bytes,8,opt,name=timeline_label,json=timelineLabel,proto3" json:"timeline_label,omitempty"`
ScheduledMicros int64 `protobuf:"varint,9,opt,name=scheduled_micros,json=scheduledMicros,proto3" json:"scheduled_micros,omitempty"`
ThreadId uint32 `protobuf:"varint,10,opt,name=thread_id,json=threadId,proto3" json:"thread_id,omitempty"`
ReferencedTensor []*AllocationDescription `protobuf:"bytes,11,rep,name=referenced_tensor,json=referencedTensor" json:"referenced_tensor,omitempty"`
MemoryStats *MemoryStats `protobuf:"bytes,12,opt,name=memory_stats,json=memoryStats" json:"memory_stats,omitempty"`
}
Time/size stats recorded for a single execution of a graph node.
func (*NodeExecStats) Descriptor ¶
func (*NodeExecStats) Descriptor() ([]byte, []int)
func (*NodeExecStats) GetAllEndRelMicros ¶
func (m *NodeExecStats) GetAllEndRelMicros() int64
func (*NodeExecStats) GetAllStartMicros ¶
func (m *NodeExecStats) GetAllStartMicros() int64
func (*NodeExecStats) GetMemory ¶
func (m *NodeExecStats) GetMemory() []*AllocatorMemoryUsed
func (*NodeExecStats) GetMemoryStats ¶
func (m *NodeExecStats) GetMemoryStats() *MemoryStats
func (*NodeExecStats) GetNodeName ¶
func (m *NodeExecStats) GetNodeName() string
func (*NodeExecStats) GetOpEndRelMicros ¶
func (m *NodeExecStats) GetOpEndRelMicros() int64
func (*NodeExecStats) GetOpStartRelMicros ¶
func (m *NodeExecStats) GetOpStartRelMicros() int64
func (*NodeExecStats) GetOutput ¶
func (m *NodeExecStats) GetOutput() []*NodeOutput
func (*NodeExecStats) GetReferencedTensor ¶
func (m *NodeExecStats) GetReferencedTensor() []*AllocationDescription
func (*NodeExecStats) GetScheduledMicros ¶
func (m *NodeExecStats) GetScheduledMicros() int64
func (*NodeExecStats) GetThreadId ¶
func (m *NodeExecStats) GetThreadId() uint32
func (*NodeExecStats) GetTimelineLabel ¶
func (m *NodeExecStats) GetTimelineLabel() string
func (*NodeExecStats) Marshal ¶
func (m *NodeExecStats) Marshal() (dAtA []byte, err error)
func (*NodeExecStats) ProtoMessage ¶
func (*NodeExecStats) ProtoMessage()
func (*NodeExecStats) Reset ¶
func (m *NodeExecStats) Reset()
func (*NodeExecStats) Size ¶
func (m *NodeExecStats) Size() (n int)
func (*NodeExecStats) String ¶
func (m *NodeExecStats) String() string
func (*NodeExecStats) Unmarshal ¶
func (m *NodeExecStats) Unmarshal(dAtA []byte) error
type NodeOutput ¶
type NodeOutput struct {
Slot int32 `protobuf:"varint,1,opt,name=slot,proto3" json:"slot,omitempty"`
TensorDescription *TensorDescription `protobuf:"bytes,3,opt,name=tensor_description,json=tensorDescription" json:"tensor_description,omitempty"`
}
Output sizes recorded for a single execution of a graph node.
func (*NodeOutput) Descriptor ¶
func (*NodeOutput) Descriptor() ([]byte, []int)
func (*NodeOutput) GetSlot ¶
func (m *NodeOutput) GetSlot() int32
func (*NodeOutput) GetTensorDescription ¶
func (m *NodeOutput) GetTensorDescription() *TensorDescription
func (*NodeOutput) Marshal ¶
func (m *NodeOutput) Marshal() (dAtA []byte, err error)
func (*NodeOutput) ProtoMessage ¶
func (*NodeOutput) ProtoMessage()
func (*NodeOutput) Reset ¶
func (m *NodeOutput) Reset()
func (*NodeOutput) Size ¶
func (m *NodeOutput) Size() (n int)
func (*NodeOutput) String ¶
func (m *NodeOutput) String() string
func (*NodeOutput) Unmarshal ¶
func (m *NodeOutput) Unmarshal(dAtA []byte) error
type OpDef ¶
type OpDef struct {
// Op names starting with an underscore are reserved for internal use.
// Names should be CamelCase and match the regexp "[A-Z][a-zA-Z0-9_]*".
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// Description of the input(s).
InputArg []*OpDef_ArgDef `protobuf:"bytes,2,rep,name=input_arg,json=inputArg" json:"input_arg,omitempty"`
// Description of the output(s).
OutputArg []*OpDef_ArgDef `protobuf:"bytes,3,rep,name=output_arg,json=outputArg" json:"output_arg,omitempty"`
Attr []*OpDef_AttrDef `protobuf:"bytes,4,rep,name=attr" json:"attr,omitempty"`
// Optional deprecation based on GraphDef versions.
Deprecation *OpDeprecation `protobuf:"bytes,8,opt,name=deprecation" json:"deprecation,omitempty"`
// One-line human-readable description of what the Op does.
Summary string `protobuf:"bytes,5,opt,name=summary,proto3" json:"summary,omitempty"`
// Additional, longer human-readable description of what the Op does.
Description string `protobuf:"bytes,6,opt,name=description,proto3" json:"description,omitempty"`
// True if the operation is commutative ("op(a,b) == op(b,a)" for all inputs)
IsCommutative bool `protobuf:"varint,18,opt,name=is_commutative,json=isCommutative,proto3" json:"is_commutative,omitempty"`
// If is_aggregate is true, then this operation accepts N >= 2
// inputs and produces 1 output all of the same type. Should be
// associative and commutative, and produce output with the same
// shape as the input. The optimizer may replace an aggregate op
// taking input from multiple devices with a tree of aggregate ops
// that aggregate locally within each device (and possibly within
// groups of nearby devices) before communicating.
// TODO(josh11b): Implement that optimization.
IsAggregate bool `protobuf:"varint,16,opt,name=is_aggregate,json=isAggregate,proto3" json:"is_aggregate,omitempty"`
// By default Ops may be moved between devices. Stateful ops should
// either not be moved, or should only be moved if that state can also
// be moved (e.g. via some sort of save / restore).
// Stateful ops are guaranteed to never be optimized away by Common
// Subexpression Elimination (CSE).
IsStateful bool `protobuf:"varint,17,opt,name=is_stateful,json=isStateful,proto3" json:"is_stateful,omitempty"`
// By default, all inputs to an Op must be initialized Tensors. Ops
// that may initialize tensors for the first time should set this
// field to true, to allow the Op to take an uninitialized Tensor as
// input.
AllowsUninitializedInput bool `` /* 137-byte string literal not displayed */
}
Defines an operation. A NodeDef in a GraphDef specifies an Op by using the "op" field which should match the name of a OpDef.
func (*OpDef) Descriptor ¶
func (*OpDef) GetAllowsUninitializedInput ¶
func (*OpDef) GetAttr ¶
func (m *OpDef) GetAttr() []*OpDef_AttrDef
func (*OpDef) GetDeprecation ¶
func (m *OpDef) GetDeprecation() *OpDeprecation
func (*OpDef) GetDescription ¶
func (*OpDef) GetInputArg ¶
func (m *OpDef) GetInputArg() []*OpDef_ArgDef
func (*OpDef) GetIsAggregate ¶
func (*OpDef) GetIsCommutative ¶
func (*OpDef) GetIsStateful ¶
func (*OpDef) GetOutputArg ¶
func (m *OpDef) GetOutputArg() []*OpDef_ArgDef
func (*OpDef) GetSummary ¶
func (*OpDef) ProtoMessage ¶
func (*OpDef) ProtoMessage()
type OpDef_ArgDef ¶
type OpDef_ArgDef struct {
// Name for the input/output. Should match the regexp "[a-z][a-z0-9_]*".
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// Human readable description.
Description string `protobuf:"bytes,2,opt,name=description,proto3" json:"description,omitempty"`
// Describes the type of one or more tensors that are accepted/produced
// by this input/output arg. The only legal combinations are:
// * For a single tensor: either the "type" field is set or the
// "type_attr" field is set to the name of an attr with type "type".
// * For a sequence of tensors with the same type: the "number_attr"
// field will be set to the name of an attr with type "int", and
// either the "type" or "type_attr" field will be set as for
// single tensors.
// * For a sequence of tensors, the "type_list_attr" field will be set
// to the name of an attr with type "list(type)".
Type DataType `protobuf:"varint,3,opt,name=type,proto3,enum=tensorflow.DataType" json:"type,omitempty"`
TypeAttr string `protobuf:"bytes,4,opt,name=type_attr,json=typeAttr,proto3" json:"type_attr,omitempty"`
NumberAttr string `protobuf:"bytes,5,opt,name=number_attr,json=numberAttr,proto3" json:"number_attr,omitempty"`
// If specified, attr must have type "list(type)", and none of
// type, type_attr, and number_attr may be specified.
TypeListAttr string `protobuf:"bytes,6,opt,name=type_list_attr,json=typeListAttr,proto3" json:"type_list_attr,omitempty"`
// For inputs: if true, the inputs are required to be refs.
// By default, inputs can be either refs or non-refs.
// For outputs: if true, outputs are refs, otherwise they are not.
IsRef bool `protobuf:"varint,16,opt,name=is_ref,json=isRef,proto3" json:"is_ref,omitempty"`
}
For describing inputs and outputs.
func (*OpDef_ArgDef) Descriptor ¶
func (*OpDef_ArgDef) Descriptor() ([]byte, []int)
func (*OpDef_ArgDef) GetDescription ¶
func (m *OpDef_ArgDef) GetDescription() string
func (*OpDef_ArgDef) GetIsRef ¶
func (m *OpDef_ArgDef) GetIsRef() bool
func (*OpDef_ArgDef) GetName ¶
func (m *OpDef_ArgDef) GetName() string
func (*OpDef_ArgDef) GetNumberAttr ¶
func (m *OpDef_ArgDef) GetNumberAttr() string
func (*OpDef_ArgDef) GetType ¶
func (m *OpDef_ArgDef) GetType() DataType
func (*OpDef_ArgDef) GetTypeAttr ¶
func (m *OpDef_ArgDef) GetTypeAttr() string
func (*OpDef_ArgDef) GetTypeListAttr ¶
func (m *OpDef_ArgDef) GetTypeListAttr() string
func (*OpDef_ArgDef) Marshal ¶
func (m *OpDef_ArgDef) Marshal() (dAtA []byte, err error)
func (*OpDef_ArgDef) ProtoMessage ¶
func (*OpDef_ArgDef) ProtoMessage()
func (*OpDef_ArgDef) Reset ¶
func (m *OpDef_ArgDef) Reset()
func (*OpDef_ArgDef) Size ¶
func (m *OpDef_ArgDef) Size() (n int)
func (*OpDef_ArgDef) String ¶
func (m *OpDef_ArgDef) String() string
func (*OpDef_ArgDef) Unmarshal ¶
func (m *OpDef_ArgDef) Unmarshal(dAtA []byte) error
type OpDef_AttrDef ¶
type OpDef_AttrDef struct {
// A descriptive name for the argument. May be used, e.g. by the
// Python client, as a keyword argument name, and so should match
// the regexp "[a-z][a-z0-9_]+".
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// One of the type names from attr_value.proto ("string", "list(string)",
// "int", etc.).
Type string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"`
// A reasonable default for this attribute if the user does not supply
// a value. If not specified, the user must supply a value.
DefaultValue *AttrValue `protobuf:"bytes,3,opt,name=default_value,json=defaultValue" json:"default_value,omitempty"`
// Human-readable description.
Description string `protobuf:"bytes,4,opt,name=description,proto3" json:"description,omitempty"`
// For type == "int", this is a minimum value. For "list(___)"
// types, this is the minimum length.
HasMinimum bool `protobuf:"varint,5,opt,name=has_minimum,json=hasMinimum,proto3" json:"has_minimum,omitempty"`
Minimum int64 `protobuf:"varint,6,opt,name=minimum,proto3" json:"minimum,omitempty"`
// The set of allowed values. Has type that is the "list" version
// of the "type" field above (uses the "list" field of AttrValue).
// If type == "type" or "list(type)" above, then the "type" field
// of "allowed_values.list" has the set of allowed DataTypes.
// If type == "string" or "list(string)", then the "s" field of
// "allowed_values.list" has the set of allowed strings.
AllowedValues *AttrValue `protobuf:"bytes,7,opt,name=allowed_values,json=allowedValues" json:"allowed_values,omitempty"`
}
Description of the graph-construction-time configuration of this Op. That is to say, this describes the attr fields that will be specified in the NodeDef.
func (*OpDef_AttrDef) Descriptor ¶
func (*OpDef_AttrDef) Descriptor() ([]byte, []int)
func (*OpDef_AttrDef) GetAllowedValues ¶
func (m *OpDef_AttrDef) GetAllowedValues() *AttrValue
func (*OpDef_AttrDef) GetDefaultValue ¶
func (m *OpDef_AttrDef) GetDefaultValue() *AttrValue
func (*OpDef_AttrDef) GetDescription ¶
func (m *OpDef_AttrDef) GetDescription() string
func (*OpDef_AttrDef) GetHasMinimum ¶
func (m *OpDef_AttrDef) GetHasMinimum() bool
func (*OpDef_AttrDef) GetMinimum ¶
func (m *OpDef_AttrDef) GetMinimum() int64
func (*OpDef_AttrDef) GetName ¶
func (m *OpDef_AttrDef) GetName() string
func (*OpDef_AttrDef) GetType ¶
func (m *OpDef_AttrDef) GetType() string
func (*OpDef_AttrDef) Marshal ¶
func (m *OpDef_AttrDef) Marshal() (dAtA []byte, err error)
func (*OpDef_AttrDef) ProtoMessage ¶
func (*OpDef_AttrDef) ProtoMessage()
func (*OpDef_AttrDef) Reset ¶
func (m *OpDef_AttrDef) Reset()
func (*OpDef_AttrDef) Size ¶
func (m *OpDef_AttrDef) Size() (n int)
func (*OpDef_AttrDef) String ¶
func (m *OpDef_AttrDef) String() string
func (*OpDef_AttrDef) Unmarshal ¶
func (m *OpDef_AttrDef) Unmarshal(dAtA []byte) error
type OpDeprecation ¶
type OpDeprecation struct {
// First GraphDef version at which the op is disallowed.
Version int32 `protobuf:"varint,1,opt,name=version,proto3" json:"version,omitempty"`
// Explanation of why it was deprecated and what to use instead.
Explanation string `protobuf:"bytes,2,opt,name=explanation,proto3" json:"explanation,omitempty"`
}
Information about version-dependent deprecation of an op
func (*OpDeprecation) Descriptor ¶
func (*OpDeprecation) Descriptor() ([]byte, []int)
func (*OpDeprecation) GetExplanation ¶
func (m *OpDeprecation) GetExplanation() string
func (*OpDeprecation) GetVersion ¶
func (m *OpDeprecation) GetVersion() int32
func (*OpDeprecation) Marshal ¶
func (m *OpDeprecation) Marshal() (dAtA []byte, err error)
func (*OpDeprecation) ProtoMessage ¶
func (*OpDeprecation) ProtoMessage()
func (*OpDeprecation) Reset ¶
func (m *OpDeprecation) Reset()
func (*OpDeprecation) Size ¶
func (m *OpDeprecation) Size() (n int)
func (*OpDeprecation) String ¶
func (m *OpDeprecation) String() string
func (*OpDeprecation) Unmarshal ¶
func (m *OpDeprecation) Unmarshal(dAtA []byte) error
type OpGenOverride ¶
type OpGenOverride struct {
// Name of the op to apply overrides to.
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
// Do not include this op in the generated API.
// If `skip` is true, all other overrides are ignored for this op.
Skip bool `protobuf:"varint,2,opt,name=skip,proto3" json:"skip,omitempty"`
// Hide this op by putting it into an internal namespace (or whatever
// is appropriate in the target language).
Hide bool `protobuf:"varint,3,opt,name=hide,proto3" json:"hide,omitempty"`
// Use a different name in the API than the op's name. Note that
// the op's name in `backticks` will also be replaced in the docs.
RenameTo string `protobuf:"bytes,4,opt,name=rename_to,json=renameTo,proto3" json:"rename_to,omitempty"`
// Create *additional* API endpoints with different names (contrast
// with rename_to, which affects the original name).
Alias []string `protobuf:"bytes,5,rep,name=alias" json:"alias,omitempty"`
AttrDefault []*OpGenOverride_AttrDefault `protobuf:"bytes,6,rep,name=attr_default,json=attrDefault" json:"attr_default,omitempty"`
AttrRename []*OpGenOverride_Rename `protobuf:"bytes,7,rep,name=attr_rename,json=attrRename" json:"attr_rename,omitempty"`
InputRename []*OpGenOverride_Rename `protobuf:"bytes,8,rep,name=input_rename,json=inputRename" json:"input_rename,omitempty"`
OutputRename []*OpGenOverride_Rename `protobuf:"bytes,9,rep,name=output_rename,json=outputRename" json:"output_rename,omitempty"`
}
Used to override the default API & behavior in the generated code for client languages, from what you would get from the OpDef alone. This is so we can evolve the API while remaining backwards compatible when interpretting old graphs. Overrides go in an "op_gen_overrides.pbtxt" file with a text-format OpGenOverrides message. Right now these only apply to the C++ API. TODO(josh11b): In the future there will be a common set of overrides and per-client-language overrides.
WARNING: Be *very* careful using these features -- these overrides can change the semantics of existing code. These changes may need to wait until a major release of TensorFlow to avoid breaking our compatibility promises.
func (*OpGenOverride) Descriptor ¶
func (*OpGenOverride) Descriptor() ([]byte, []int)
func (*OpGenOverride) GetAlias ¶
func (m *OpGenOverride) GetAlias() []string
func (*OpGenOverride) GetAttrDefault ¶
func (m *OpGenOverride) GetAttrDefault() []*OpGenOverride_AttrDefault
func (*OpGenOverride) GetAttrRename ¶
func (m *OpGenOverride) GetAttrRename() []*OpGenOverride_Rename
func (*OpGenOverride) GetHide ¶
func (m *OpGenOverride) GetHide() bool
func (*OpGenOverride) GetInputRename ¶
func (m *OpGenOverride) GetInputRename() []*OpGenOverride_Rename
func (*OpGenOverride) GetName ¶
func (m *OpGenOverride) GetName() string
func (*OpGenOverride) GetOutputRename ¶
func (m *OpGenOverride) GetOutputRename() []*OpGenOverride_Rename
func (*OpGenOverride) GetRenameTo ¶
func (m *OpGenOverride) GetRenameTo() string
func (*OpGenOverride) GetSkip ¶
func (m *OpGenOverride) GetSkip() bool
func (*OpGenOverride) Marshal ¶
func (m *OpGenOverride) Marshal() (dAtA []byte, err error)
func (*OpGenOverride) ProtoMessage ¶
func (*OpGenOverride) ProtoMessage()
func (*OpGenOverride) Reset ¶
func (m *OpGenOverride) Reset()
func (*OpGenOverride) Size ¶
func (m *OpGenOverride) Size() (n int)
func (*OpGenOverride) String ¶
func (m *OpGenOverride) String() string
func (*OpGenOverride) Unmarshal ¶
func (m *OpGenOverride) Unmarshal(dAtA []byte) error
type OpGenOverride_AttrDefault ¶
type OpGenOverride_AttrDefault struct {
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
Value *AttrValue `protobuf:"bytes,2,opt,name=value" json:"value,omitempty"`
}
Map the name of an attr to a new default value to use. This default will be used when creating new graphs, as opposed to the default in the OpDef, which will be used when interpreting old GraphDefs. If this attr is also renamed (using attr_rename below), use the original name of the attr.
func (*OpGenOverride_AttrDefault) Descriptor ¶
func (*OpGenOverride_AttrDefault) Descriptor() ([]byte, []int)
func (*OpGenOverride_AttrDefault) GetName ¶
func (m *OpGenOverride_AttrDefault) GetName() string
func (*OpGenOverride_AttrDefault) GetValue ¶
func (m *OpGenOverride_AttrDefault) GetValue() *AttrValue
func (*OpGenOverride_AttrDefault) Marshal ¶
func (m *OpGenOverride_AttrDefault) Marshal() (dAtA []byte, err error)
func (*OpGenOverride_AttrDefault) MarshalTo ¶
func (m *OpGenOverride_AttrDefault) MarshalTo(dAtA []byte) (int, error)
func (*OpGenOverride_AttrDefault) ProtoMessage ¶
func (*OpGenOverride_AttrDefault) ProtoMessage()
func (*OpGenOverride_AttrDefault) Reset ¶
func (m *OpGenOverride_AttrDefault) Reset()
func (*OpGenOverride_AttrDefault) Size ¶
func (m *OpGenOverride_AttrDefault) Size() (n int)
func (*OpGenOverride_AttrDefault) String ¶
func (m *OpGenOverride_AttrDefault) String() string
func (*OpGenOverride_AttrDefault) Unmarshal ¶
func (m *OpGenOverride_AttrDefault) Unmarshal(dAtA []byte) error
type OpGenOverride_Rename ¶
type OpGenOverride_Rename struct {
From string `protobuf:"bytes,1,opt,name=from,proto3" json:"from,omitempty"`
To string `protobuf:"bytes,2,opt,name=to,proto3" json:"to,omitempty"`
}
Change the name used to access attrs/inputs/outputs in the API from what is used in the GraphDef. Note that these names in `backticks` will also be replaced in the docs.
func (*OpGenOverride_Rename) Descriptor ¶
func (*OpGenOverride_Rename) Descriptor() ([]byte, []int)
func (*OpGenOverride_Rename) GetFrom ¶
func (m *OpGenOverride_Rename) GetFrom() string
func (*OpGenOverride_Rename) GetTo ¶
func (m *OpGenOverride_Rename) GetTo() string
func (*OpGenOverride_Rename) Marshal ¶
func (m *OpGenOverride_Rename) Marshal() (dAtA []byte, err error)
func (*OpGenOverride_Rename) MarshalTo ¶
func (m *OpGenOverride_Rename) MarshalTo(dAtA []byte) (int, error)
func (*OpGenOverride_Rename) ProtoMessage ¶
func (*OpGenOverride_Rename) ProtoMessage()
func (*OpGenOverride_Rename) Reset ¶
func (m *OpGenOverride_Rename) Reset()
func (*OpGenOverride_Rename) Size ¶
func (m *OpGenOverride_Rename) Size() (n int)
func (*OpGenOverride_Rename) String ¶
func (m *OpGenOverride_Rename) String() string
func (*OpGenOverride_Rename) Unmarshal ¶
func (m *OpGenOverride_Rename) Unmarshal(dAtA []byte) error
type OpGenOverrides ¶
type OpGenOverrides struct {
Op []*OpGenOverride `protobuf:"bytes,1,rep,name=op" json:"op,omitempty"`
}
func (*OpGenOverrides) Descriptor ¶
func (*OpGenOverrides) Descriptor() ([]byte, []int)
func (*OpGenOverrides) GetOp ¶
func (m *OpGenOverrides) GetOp() []*OpGenOverride
func (*OpGenOverrides) Marshal ¶
func (m *OpGenOverrides) Marshal() (dAtA []byte, err error)
func (*OpGenOverrides) ProtoMessage ¶
func (*OpGenOverrides) ProtoMessage()
func (*OpGenOverrides) Reset ¶
func (m *OpGenOverrides) Reset()
func (*OpGenOverrides) Size ¶
func (m *OpGenOverrides) Size() (n int)
func (*OpGenOverrides) String ¶
func (m *OpGenOverrides) String() string
func (*OpGenOverrides) Unmarshal ¶
func (m *OpGenOverrides) Unmarshal(dAtA []byte) error
type OpList ¶
type OpList struct {
Op []*OpDef `protobuf:"bytes,1,rep,name=op" json:"op,omitempty"`
}
A collection of OpDefs
func (*OpList) Descriptor ¶
func (*OpList) ProtoMessage ¶
func (*OpList) ProtoMessage()
type 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"`
// 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"`
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 */
}
Options passed to the graph optimizer
func (*OptimizerOptions) Descriptor ¶
func (*OptimizerOptions) Descriptor() ([]byte, []int)
func (*OptimizerOptions) GetDoCommonSubexpressionElimination ¶
func (m *OptimizerOptions) GetDoCommonSubexpressionElimination() bool
func (*OptimizerOptions) GetDoConstantFolding ¶
func (m *OptimizerOptions) GetDoConstantFolding() bool
func (*OptimizerOptions) GetDoFunctionInlining ¶
func (m *OptimizerOptions) GetDoFunctionInlining() bool
func (*OptimizerOptions) GetGlobalJitLevel ¶
func (m *OptimizerOptions) GetGlobalJitLevel() OptimizerOptions_GlobalJitLevel
func (*OptimizerOptions) GetOptLevel ¶
func (m *OptimizerOptions) GetOptLevel() OptimizerOptions_Level
func (*OptimizerOptions) Marshal ¶
func (m *OptimizerOptions) Marshal() (dAtA []byte, err error)
func (*OptimizerOptions) ProtoMessage ¶
func (*OptimizerOptions) ProtoMessage()
func (*OptimizerOptions) Reset ¶
func (m *OptimizerOptions) Reset()
func (*OptimizerOptions) Size ¶
func (m *OptimizerOptions) Size() (n int)
func (*OptimizerOptions) String ¶
func (m *OptimizerOptions) String() string
func (*OptimizerOptions) Unmarshal ¶
func (m *OptimizerOptions) Unmarshal(dAtA []byte) error
type OptimizerOptions_GlobalJitLevel ¶
type OptimizerOptions_GlobalJitLevel int32
Control the use of the compiler/jit. Experimental.
const ( OptimizerOptions_DEFAULT OptimizerOptions_GlobalJitLevel = 0 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) EnumDescriptor ¶
func (OptimizerOptions_GlobalJitLevel) EnumDescriptor() ([]byte, []int)
func (OptimizerOptions_GlobalJitLevel) String ¶
func (x OptimizerOptions_GlobalJitLevel) String() string
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) EnumDescriptor ¶
func (OptimizerOptions_Level) EnumDescriptor() ([]byte, []int)
func (OptimizerOptions_Level) String ¶
func (x OptimizerOptions_Level) String() string
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" 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" 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" json:"target,omitempty"`
}
func (*PartialRunSetupRequest) Descriptor ¶
func (*PartialRunSetupRequest) Descriptor() ([]byte, []int)
func (*PartialRunSetupRequest) GetFeed ¶
func (m *PartialRunSetupRequest) GetFeed() []string
func (*PartialRunSetupRequest) GetFetch ¶
func (m *PartialRunSetupRequest) GetFetch() []string
func (*PartialRunSetupRequest) GetSessionHandle ¶
func (m *PartialRunSetupRequest) GetSessionHandle() string
func (*PartialRunSetupRequest) GetTarget ¶
func (m *PartialRunSetupRequest) GetTarget() []string
func (*PartialRunSetupRequest) Marshal ¶
func (m *PartialRunSetupRequest) Marshal() (dAtA []byte, err error)
func (*PartialRunSetupRequest) MarshalTo ¶
func (m *PartialRunSetupRequest) MarshalTo(dAtA []byte) (int, error)
func (*PartialRunSetupRequest) ProtoMessage ¶
func (*PartialRunSetupRequest) ProtoMessage()
func (*PartialRunSetupRequest) Reset ¶
func (m *PartialRunSetupRequest) Reset()
func (*PartialRunSetupRequest) Size ¶
func (m *PartialRunSetupRequest) Size() (n int)
func (*PartialRunSetupRequest) String ¶
func (m *PartialRunSetupRequest) String() string
func (*PartialRunSetupRequest) Unmarshal ¶
func (m *PartialRunSetupRequest) Unmarshal(dAtA []byte) error
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"`
}
func (*PartialRunSetupResponse) Descriptor ¶
func (*PartialRunSetupResponse) Descriptor() ([]byte, []int)
func (*PartialRunSetupResponse) GetPartialRunHandle ¶
func (m *PartialRunSetupResponse) GetPartialRunHandle() string
func (*PartialRunSetupResponse) Marshal ¶
func (m *PartialRunSetupResponse) Marshal() (dAtA []byte, err error)
func (*PartialRunSetupResponse) MarshalTo ¶
func (m *PartialRunSetupResponse) MarshalTo(dAtA []byte) (int, error)
func (*PartialRunSetupResponse) ProtoMessage ¶
func (*PartialRunSetupResponse) ProtoMessage()
func (*PartialRunSetupResponse) Reset ¶
func (m *PartialRunSetupResponse) Reset()
func (*PartialRunSetupResponse) Size ¶
func (m *PartialRunSetupResponse) Size() (n int)
func (*PartialRunSetupResponse) String ¶
func (m *PartialRunSetupResponse) String() string
func (*PartialRunSetupResponse) Unmarshal ¶
func (m *PartialRunSetupResponse) Unmarshal(dAtA []byte) error
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" 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 []Error_Code `` /* 168-byte string literal not displayed */
}
Protocol buffer representing a QueueRunner.
func (*QueueRunnerDef) Descriptor ¶
func (*QueueRunnerDef) Descriptor() ([]byte, []int)
func (*QueueRunnerDef) GetCancelOpName ¶
func (m *QueueRunnerDef) GetCancelOpName() string
func (*QueueRunnerDef) GetCloseOpName ¶
func (m *QueueRunnerDef) GetCloseOpName() string
func (*QueueRunnerDef) GetEnqueueOpName ¶
func (m *QueueRunnerDef) GetEnqueueOpName() []string
func (*QueueRunnerDef) GetQueueClosedExceptionTypes ¶
func (m *QueueRunnerDef) GetQueueClosedExceptionTypes() []Error_Code
func (*QueueRunnerDef) GetQueueName ¶
func (m *QueueRunnerDef) GetQueueName() string
func (*QueueRunnerDef) Marshal ¶
func (m *QueueRunnerDef) Marshal() (dAtA []byte, err error)
func (*QueueRunnerDef) ProtoMessage ¶
func (*QueueRunnerDef) ProtoMessage()
func (*QueueRunnerDef) Reset ¶
func (m *QueueRunnerDef) Reset()
func (*QueueRunnerDef) Size ¶
func (m *QueueRunnerDef) Size() (n int)
func (*QueueRunnerDef) String ¶
func (m *QueueRunnerDef) String() string
func (*QueueRunnerDef) Unmarshal ¶
func (m *QueueRunnerDef) Unmarshal(dAtA []byte) error
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 */
}
func (*RPCOptions) Descriptor ¶
func (*RPCOptions) Descriptor() ([]byte, []int)
func (*RPCOptions) GetUseRpcForInprocessMaster ¶
func (m *RPCOptions) GetUseRpcForInprocessMaster() bool
func (*RPCOptions) Marshal ¶
func (m *RPCOptions) Marshal() (dAtA []byte, err error)
func (*RPCOptions) ProtoMessage ¶
func (*RPCOptions) ProtoMessage()
func (*RPCOptions) Reset ¶
func (m *RPCOptions) Reset()
func (*RPCOptions) Size ¶
func (m *RPCOptions) Size() (n int)
func (*RPCOptions) String ¶
func (m *RPCOptions) String() string
func (*RPCOptions) Unmarshal ¶
func (m *RPCOptions) Unmarshal(dAtA []byte) error
type ReaderBaseState ¶
type ReaderBaseState struct {
WorkStarted int64 `protobuf:"varint,1,opt,name=work_started,json=workStarted,proto3" json:"work_started,omitempty"`
WorkFinished int64 `protobuf:"varint,2,opt,name=work_finished,json=workFinished,proto3" json:"work_finished,omitempty"`
NumRecordsProduced int64 `protobuf:"varint,3,opt,name=num_records_produced,json=numRecordsProduced,proto3" json:"num_records_produced,omitempty"`
CurrentWork []byte `protobuf:"bytes,4,opt,name=current_work,json=currentWork,proto3" json:"current_work,omitempty"`
}
For serializing and restoring the state of ReaderBase, see reader_base.h for details.
func (*ReaderBaseState) Descriptor ¶
func (*ReaderBaseState) Descriptor() ([]byte, []int)
func (*ReaderBaseState) GetCurrentWork ¶
func (m *ReaderBaseState) GetCurrentWork() []byte
func (*ReaderBaseState) GetNumRecordsProduced ¶
func (m *ReaderBaseState) GetNumRecordsProduced() int64
func (*ReaderBaseState) GetWorkFinished ¶
func (m *ReaderBaseState) GetWorkFinished() int64
func (*ReaderBaseState) GetWorkStarted ¶
func (m *ReaderBaseState) GetWorkStarted() int64
func (*ReaderBaseState) Marshal ¶
func (m *ReaderBaseState) Marshal() (dAtA []byte, err error)
func (*ReaderBaseState) ProtoMessage ¶
func (*ReaderBaseState) ProtoMessage()
func (*ReaderBaseState) Reset ¶
func (m *ReaderBaseState) Reset()
func (*ReaderBaseState) Size ¶
func (m *ReaderBaseState) Size() (n int)
func (*ReaderBaseState) String ¶
func (m *ReaderBaseState) String() string
func (*ReaderBaseState) Unmarshal ¶
func (m *ReaderBaseState) Unmarshal(dAtA []byte) error
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 that identifies the tensor to be received.
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 *DeviceLocality `protobuf:"bytes,4,opt,name=client_locality,json=clientLocality" json:"client_locality,omitempty"`
// Optional information on server-side device locality.
ServerLocality *DeviceLocality `protobuf:"bytes,5,opt,name=server_locality,json=serverLocality" json:"server_locality,omitempty"`
// Optional information needed by the RPC subsystem.
TransportOptions *google_protobuf.Any `protobuf:"bytes,6,opt,name=transport_options,json=transportOptions" json:"transport_options,omitempty"`
}
func (*RecvTensorRequest) Descriptor ¶
func (*RecvTensorRequest) Descriptor() ([]byte, []int)
func (*RecvTensorRequest) GetClientLocality ¶
func (m *RecvTensorRequest) GetClientLocality() *DeviceLocality
func (*RecvTensorRequest) GetDmaOk ¶
func (m *RecvTensorRequest) GetDmaOk() bool
func (*RecvTensorRequest) GetRendezvousKey ¶
func (m *RecvTensorRequest) GetRendezvousKey() string
func (*RecvTensorRequest) GetServerLocality ¶
func (m *RecvTensorRequest) GetServerLocality() *DeviceLocality
func (*RecvTensorRequest) GetStepId ¶
func (m *RecvTensorRequest) GetStepId() int64
func (*RecvTensorRequest) GetTransportOptions ¶
func (m *RecvTensorRequest) GetTransportOptions() *google_protobuf.Any
func (*RecvTensorRequest) Marshal ¶
func (m *RecvTensorRequest) Marshal() (dAtA []byte, err error)
func (*RecvTensorRequest) MarshalTo ¶
func (m *RecvTensorRequest) MarshalTo(dAtA []byte) (int, error)
func (*RecvTensorRequest) ProtoMessage ¶
func (*RecvTensorRequest) ProtoMessage()
func (*RecvTensorRequest) Reset ¶
func (m *RecvTensorRequest) Reset()
func (*RecvTensorRequest) Size ¶
func (m *RecvTensorRequest) Size() (n int)
func (*RecvTensorRequest) String ¶
func (m *RecvTensorRequest) String() string
func (*RecvTensorRequest) Unmarshal ¶
func (m *RecvTensorRequest) Unmarshal(dAtA []byte) error
type RecvTensorResponse ¶
type RecvTensorResponse struct {
// The tensor as a proto.
Tensor *TensorProto `protobuf:"bytes,1,opt,name=tensor" 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 *google_protobuf.Any `protobuf:"bytes,4,opt,name=transport_options,json=transportOptions" json:"transport_options,omitempty"`
}
func (*RecvTensorResponse) Descriptor ¶
func (*RecvTensorResponse) Descriptor() ([]byte, []int)
func (*RecvTensorResponse) GetIsDead ¶
func (m *RecvTensorResponse) GetIsDead() bool
func (*RecvTensorResponse) GetSendStartMicros ¶
func (m *RecvTensorResponse) GetSendStartMicros() int64
func (*RecvTensorResponse) GetTensor ¶
func (m *RecvTensorResponse) GetTensor() *TensorProto
func (*RecvTensorResponse) GetTransportOptions ¶
func (m *RecvTensorResponse) GetTransportOptions() *google_protobuf.Any
func (*RecvTensorResponse) Marshal ¶
func (m *RecvTensorResponse) Marshal() (dAtA []byte, err error)
func (*RecvTensorResponse) MarshalTo ¶
func (m *RecvTensorResponse) MarshalTo(dAtA []byte) (int, error)
func (*RecvTensorResponse) ProtoMessage ¶
func (*RecvTensorResponse) ProtoMessage()
func (*RecvTensorResponse) Reset ¶
func (m *RecvTensorResponse) Reset()
func (*RecvTensorResponse) Size ¶
func (m *RecvTensorResponse) Size() (n int)
func (*RecvTensorResponse) String ¶
func (m *RecvTensorResponse) String() string
func (*RecvTensorResponse) Unmarshal ¶
func (m *RecvTensorResponse) Unmarshal(dAtA []byte) error
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"`
// "graph_def" has the subgraph of nodes for this worker, with each node
// having its device_name filled in.
GraphDef *GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef" 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.
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" json:"graph_options,omitempty"`
// Field(s) used by TensorFlow Debugger (tfdbg).
DebugOptions *DebugOptions `protobuf:"bytes,5,opt,name=debug_options,json=debugOptions" json:"debug_options,omitempty"`
}
func (*RegisterGraphRequest) Descriptor ¶
func (*RegisterGraphRequest) Descriptor() ([]byte, []int)
func (*RegisterGraphRequest) GetDebugOptions ¶
func (m *RegisterGraphRequest) GetDebugOptions() *DebugOptions
func (*RegisterGraphRequest) GetGraphDef ¶
func (m *RegisterGraphRequest) GetGraphDef() *GraphDef
func (*RegisterGraphRequest) GetGraphOptions ¶
func (m *RegisterGraphRequest) GetGraphOptions() *GraphOptions
func (*RegisterGraphRequest) GetHasControlFlow ¶
func (m *RegisterGraphRequest) GetHasControlFlow() bool
func (*RegisterGraphRequest) GetSessionHandle ¶
func (m *RegisterGraphRequest) GetSessionHandle() string
func (*RegisterGraphRequest) Marshal ¶
func (m *RegisterGraphRequest) Marshal() (dAtA []byte, err error)
func (*RegisterGraphRequest) MarshalTo ¶
func (m *RegisterGraphRequest) MarshalTo(dAtA []byte) (int, error)
func (*RegisterGraphRequest) ProtoMessage ¶
func (*RegisterGraphRequest) ProtoMessage()
func (*RegisterGraphRequest) Reset ¶
func (m *RegisterGraphRequest) Reset()
func (*RegisterGraphRequest) Size ¶
func (m *RegisterGraphRequest) Size() (n int)
func (*RegisterGraphRequest) String ¶
func (m *RegisterGraphRequest) String() string
func (*RegisterGraphRequest) Unmarshal ¶
func (m *RegisterGraphRequest) Unmarshal(dAtA []byte) error
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"`
}
func (*RegisterGraphResponse) Descriptor ¶
func (*RegisterGraphResponse) Descriptor() ([]byte, []int)
func (*RegisterGraphResponse) GetGraphHandle ¶
func (m *RegisterGraphResponse) GetGraphHandle() string
func (*RegisterGraphResponse) Marshal ¶
func (m *RegisterGraphResponse) Marshal() (dAtA []byte, err error)
func (*RegisterGraphResponse) MarshalTo ¶
func (m *RegisterGraphResponse) MarshalTo(dAtA []byte) (int, error)
func (*RegisterGraphResponse) ProtoMessage ¶
func (*RegisterGraphResponse) ProtoMessage()
func (*RegisterGraphResponse) Reset ¶
func (m *RegisterGraphResponse) Reset()
func (*RegisterGraphResponse) Size ¶
func (m *RegisterGraphResponse) Size() (n int)
func (*RegisterGraphResponse) String ¶
func (m *RegisterGraphResponse) String() string
func (*RegisterGraphResponse) Unmarshal ¶
func (m *RegisterGraphResponse) Unmarshal(dAtA []byte) error
type RemoteFusedGraphExecuteInfo ¶
type RemoteFusedGraphExecuteInfo struct {
// Definition of remote graph
RemoteGraph *GraphDef `protobuf:"bytes,1,opt,name=remote_graph,json=remoteGraph" json:"remote_graph,omitempty"`
// Remote fused graph input node name
GraphInputNodeName []string `protobuf:"bytes,2,rep,name=graph_input_node_name,json=graphInputNodeName" json:"graph_input_node_name,omitempty"`
// Remote fused graph output node name
GraphOutputNodeName []string `protobuf:"bytes,3,rep,name=graph_output_node_name,json=graphOutputNodeName" json:"graph_output_node_name,omitempty"`
// Executor's name
ExecutorName string `protobuf:"bytes,4,opt,name=executor_name,json=executorName,proto3" json:"executor_name,omitempty"`
// Optional: Parameters given to the executor
SerializedExecutorParameters []byte `` /* 147-byte string literal not displayed */
// Optional: Default graph input tensor shape used to allocate memory
// before executing op
DefaultGraphInputTensorShape []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto `` /* 144-byte string literal not displayed */
// Optional: Default graph input tensor shape used to allocate memory
// before executing op
// TODO(satok): Remote output tensor shape once shape information is stored
// in NodeDef
DefaultGraphOutputTensorShape []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto `` /* 147-byte string literal not displayed */
}
Protocol buffer representing a handle to a tensorflow resource. Handles are not valid across executions, but can be serialized back and forth from within a single run.
func (*RemoteFusedGraphExecuteInfo) Descriptor ¶
func (*RemoteFusedGraphExecuteInfo) Descriptor() ([]byte, []int)
func (*RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape ¶
func (m *RemoteFusedGraphExecuteInfo) GetDefaultGraphInputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
func (*RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape ¶
func (m *RemoteFusedGraphExecuteInfo) GetDefaultGraphOutputTensorShape() []*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto
func (*RemoteFusedGraphExecuteInfo) GetExecutorName ¶
func (m *RemoteFusedGraphExecuteInfo) GetExecutorName() string
func (*RemoteFusedGraphExecuteInfo) GetGraphInputNodeName ¶
func (m *RemoteFusedGraphExecuteInfo) GetGraphInputNodeName() []string
func (*RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName ¶
func (m *RemoteFusedGraphExecuteInfo) GetGraphOutputNodeName() []string
func (*RemoteFusedGraphExecuteInfo) GetRemoteGraph ¶
func (m *RemoteFusedGraphExecuteInfo) GetRemoteGraph() *GraphDef
func (*RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters ¶
func (m *RemoteFusedGraphExecuteInfo) GetSerializedExecutorParameters() []byte
func (*RemoteFusedGraphExecuteInfo) Marshal ¶
func (m *RemoteFusedGraphExecuteInfo) Marshal() (dAtA []byte, err error)
func (*RemoteFusedGraphExecuteInfo) MarshalTo ¶
func (m *RemoteFusedGraphExecuteInfo) MarshalTo(dAtA []byte) (int, error)
func (*RemoteFusedGraphExecuteInfo) ProtoMessage ¶
func (*RemoteFusedGraphExecuteInfo) ProtoMessage()
func (*RemoteFusedGraphExecuteInfo) Reset ¶
func (m *RemoteFusedGraphExecuteInfo) Reset()
func (*RemoteFusedGraphExecuteInfo) Size ¶
func (m *RemoteFusedGraphExecuteInfo) Size() (n int)
func (*RemoteFusedGraphExecuteInfo) String ¶
func (m *RemoteFusedGraphExecuteInfo) String() string
func (*RemoteFusedGraphExecuteInfo) Unmarshal ¶
func (m *RemoteFusedGraphExecuteInfo) Unmarshal(dAtA []byte) error
type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto ¶
type RemoteFusedGraphExecuteInfo_TensorShapeTypeProto struct {
Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape" json:"shape,omitempty"`
}
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor ¶
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Descriptor() ([]byte, []int)
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetDtype() DataType
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) GetShape() *TensorShapeProto
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Marshal ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Marshal() (dAtA []byte, err error)
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) MarshalTo ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) MarshalTo(dAtA []byte) (int, error)
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage ¶
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) ProtoMessage()
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Reset()
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Size ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Size() (n int)
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) String ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) String() string
func (*RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Unmarshal ¶
func (m *RemoteFusedGraphExecuteInfo_TensorShapeTypeProto) Unmarshal(dAtA []byte) error
type ResetRequest ¶
type ResetRequest struct {
// A list of container names, which may be empty.
//
// If 'container' is not empty, releases resoures 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" 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" json:"device_filters,omitempty"`
}
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 behaviour 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 ¶
func (*ResetRequest) Descriptor() ([]byte, []int)
func (*ResetRequest) GetContainer ¶
func (m *ResetRequest) GetContainer() []string
func (*ResetRequest) GetDeviceFilters ¶
func (m *ResetRequest) GetDeviceFilters() []string
func (*ResetRequest) Marshal ¶
func (m *ResetRequest) Marshal() (dAtA []byte, err error)
func (*ResetRequest) ProtoMessage ¶
func (*ResetRequest) ProtoMessage()
func (*ResetRequest) Reset ¶
func (m *ResetRequest) Reset()
func (*ResetRequest) Size ¶
func (m *ResetRequest) Size() (n int)
func (*ResetRequest) String ¶
func (m *ResetRequest) String() string
func (*ResetRequest) Unmarshal ¶
func (m *ResetRequest) Unmarshal(dAtA []byte) error
type ResetResponse ¶
type ResetResponse struct {
}
func (*ResetResponse) Descriptor ¶
func (*ResetResponse) Descriptor() ([]byte, []int)
func (*ResetResponse) Marshal ¶
func (m *ResetResponse) Marshal() (dAtA []byte, err error)
func (*ResetResponse) ProtoMessage ¶
func (*ResetResponse) ProtoMessage()
func (*ResetResponse) Reset ¶
func (m *ResetResponse) Reset()
func (*ResetResponse) Size ¶
func (m *ResetResponse) Size() (n int)
func (*ResetResponse) String ¶
func (m *ResetResponse) String() string
func (*ResetResponse) Unmarshal ¶
func (m *ResetResponse) Unmarshal(dAtA []byte) error
type ResourceHandle ¶
type ResourceHandle struct {
// Unique name for the device containing the resource.
Device string `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"`
// Container in which this resource is placed.
Container string `protobuf:"bytes,2,opt,name=container,proto3" json:"container,omitempty"`
// Unique name of this resource.
Name string `protobuf:"bytes,3,opt,name=name,proto3" json:"name,omitempty"`
// Hash code for the type of the resource. Is only valid in the same device
// and in the same execution.
HashCode uint64 `protobuf:"varint,4,opt,name=hash_code,json=hashCode,proto3" json:"hash_code,omitempty"`
// For debug-only, the name of the type pointed to by this handle, if
// available.
MaybeTypeName string `protobuf:"bytes,5,opt,name=maybe_type_name,json=maybeTypeName,proto3" json:"maybe_type_name,omitempty"`
}
Protocol buffer representing a handle to a tensorflow resource. Handles are not valid across executions, but can be serialized back and forth from within a single run.
func (*ResourceHandle) Descriptor ¶
func (*ResourceHandle) Descriptor() ([]byte, []int)
func (*ResourceHandle) GetContainer ¶
func (m *ResourceHandle) GetContainer() string
func (*ResourceHandle) GetDevice ¶
func (m *ResourceHandle) GetDevice() string
func (*ResourceHandle) GetHashCode ¶
func (m *ResourceHandle) GetHashCode() uint64
func (*ResourceHandle) GetMaybeTypeName ¶
func (m *ResourceHandle) GetMaybeTypeName() string
func (*ResourceHandle) GetName ¶
func (m *ResourceHandle) GetName() string
func (*ResourceHandle) Marshal ¶
func (m *ResourceHandle) Marshal() (dAtA []byte, err error)
func (*ResourceHandle) ProtoMessage ¶
func (*ResourceHandle) ProtoMessage()
func (*ResourceHandle) Reset ¶
func (m *ResourceHandle) Reset()
func (*ResourceHandle) Size ¶
func (m *ResourceHandle) Size() (n int)
func (*ResourceHandle) String ¶
func (m *ResourceHandle) String() string
func (*ResourceHandle) Unmarshal ¶
func (m *ResourceHandle) Unmarshal(dAtA []byte) error
type RewriterConfig ¶
type RewriterConfig struct {
OptimizeTensorLayout bool `protobuf:"varint,1,opt,name=optimize_tensor_layout,json=optimizeTensorLayout,proto3" json:"optimize_tensor_layout,omitempty"`
DisableModelPruning bool `protobuf:"varint,2,opt,name=disable_model_pruning,json=disableModelPruning,proto3" json:"disable_model_pruning,omitempty"`
ConstantFolding bool `protobuf:"varint,3,opt,name=constant_folding,json=constantFolding,proto3" json:"constant_folding,omitempty"`
MemoryOptimization RewriterConfig_MemOptType `` /* 158-byte string literal not displayed */
AutoParallel *AutoParallelOptions `protobuf:"bytes,5,opt,name=auto_parallel,json=autoParallel" json:"auto_parallel,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.
Optimizers []string `protobuf:"bytes,100,rep,name=optimizers" json:"optimizers,omitempty"`
}
func (*RewriterConfig) Descriptor ¶
func (*RewriterConfig) Descriptor() ([]byte, []int)
func (*RewriterConfig) GetAutoParallel ¶
func (m *RewriterConfig) GetAutoParallel() *AutoParallelOptions
func (*RewriterConfig) GetConstantFolding ¶
func (m *RewriterConfig) GetConstantFolding() bool
func (*RewriterConfig) GetDisableModelPruning ¶
func (m *RewriterConfig) GetDisableModelPruning() bool
func (*RewriterConfig) GetMemoryOptimization ¶
func (m *RewriterConfig) GetMemoryOptimization() RewriterConfig_MemOptType
func (*RewriterConfig) GetOptimizeTensorLayout ¶
func (m *RewriterConfig) GetOptimizeTensorLayout() bool
func (*RewriterConfig) GetOptimizers ¶
func (m *RewriterConfig) GetOptimizers() []string
func (*RewriterConfig) Marshal ¶
func (m *RewriterConfig) Marshal() (dAtA []byte, err error)
func (*RewriterConfig) ProtoMessage ¶
func (*RewriterConfig) ProtoMessage()
func (*RewriterConfig) Reset ¶
func (m *RewriterConfig) Reset()
func (*RewriterConfig) Size ¶
func (m *RewriterConfig) Size() (n int)
func (*RewriterConfig) String ¶
func (m *RewriterConfig) String() string
func (*RewriterConfig) Unmarshal ¶
func (m *RewriterConfig) Unmarshal(dAtA []byte) error
type RewriterConfig_MemOptType ¶
type RewriterConfig_MemOptType int32
const ( // Fully disabled RewriterConfig_NO_MEM_OPT RewriterConfig_MemOptType = 0 // Driven by manual annotations RewriterConfig_MANUAL RewriterConfig_MemOptType = 1 )
func (RewriterConfig_MemOptType) EnumDescriptor ¶
func (RewriterConfig_MemOptType) EnumDescriptor() ([]byte, []int)
func (RewriterConfig_MemOptType) String ¶
func (x RewriterConfig_MemOptType) String() string
type RunGraphRequest ¶
type RunGraphRequest struct {
// session_handle is the 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"`
// 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" 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" json:"send,omitempty"`
RecvKey []string `protobuf:"bytes,4,rep,name=recv_key,json=recvKey" 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"`
}
func (*RunGraphRequest) Descriptor ¶
func (*RunGraphRequest) Descriptor() ([]byte, []int)
func (*RunGraphRequest) GetExecOpts ¶
func (m *RunGraphRequest) GetExecOpts() *ExecutorOpts
func (*RunGraphRequest) GetGraphHandle ¶
func (m *RunGraphRequest) GetGraphHandle() string
func (*RunGraphRequest) GetIsLastPartialRun ¶
func (m *RunGraphRequest) GetIsLastPartialRun() bool
func (*RunGraphRequest) GetIsPartial ¶
func (m *RunGraphRequest) GetIsPartial() bool
func (*RunGraphRequest) GetRecvKey ¶
func (m *RunGraphRequest) GetRecvKey() []string
func (*RunGraphRequest) GetSend ¶
func (m *RunGraphRequest) GetSend() []*NamedTensorProto
func (*RunGraphRequest) GetSessionHandle ¶
func (m *RunGraphRequest) GetSessionHandle() string
func (*RunGraphRequest) GetStepId ¶
func (m *RunGraphRequest) GetStepId() int64
func (*RunGraphRequest) Marshal ¶
func (m *RunGraphRequest) Marshal() (dAtA []byte, err error)
func (*RunGraphRequest) ProtoMessage ¶
func (*RunGraphRequest) ProtoMessage()
func (*RunGraphRequest) Reset ¶
func (m *RunGraphRequest) Reset()
func (*RunGraphRequest) Size ¶
func (m *RunGraphRequest) Size() (n int)
func (*RunGraphRequest) String ¶
func (m *RunGraphRequest) String() string
func (*RunGraphRequest) Unmarshal ¶
func (m *RunGraphRequest) Unmarshal(dAtA []byte) error
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" json:"recv,omitempty"`
// If the request asked for execution stats or cost graph, these are returned
// here.
StepStats *StepStats `protobuf:"bytes,2,opt,name=step_stats,json=stepStats" json:"step_stats,omitempty"`
CostGraph *CostGraphDef `protobuf:"bytes,3,opt,name=cost_graph,json=costGraph" json:"cost_graph,omitempty"`
}
func (*RunGraphResponse) Descriptor ¶
func (*RunGraphResponse) Descriptor() ([]byte, []int)
func (*RunGraphResponse) GetCostGraph ¶
func (m *RunGraphResponse) GetCostGraph() *CostGraphDef
func (*RunGraphResponse) GetRecv ¶
func (m *RunGraphResponse) GetRecv() []*NamedTensorProto
func (*RunGraphResponse) GetStepStats ¶
func (m *RunGraphResponse) GetStepStats() *StepStats
func (*RunGraphResponse) Marshal ¶
func (m *RunGraphResponse) Marshal() (dAtA []byte, err error)
func (*RunGraphResponse) ProtoMessage ¶
func (*RunGraphResponse) ProtoMessage()
func (*RunGraphResponse) Reset ¶
func (m *RunGraphResponse) Reset()
func (*RunGraphResponse) Size ¶
func (m *RunGraphResponse) Size() (n int)
func (*RunGraphResponse) String ¶
func (m *RunGraphResponse) String() string
func (*RunGraphResponse) Unmarshal ¶
func (m *RunGraphResponse) Unmarshal(dAtA []byte) error
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 *StepStats `protobuf:"bytes,1,opt,name=step_stats,json=stepStats" json:"step_stats,omitempty"`
// The cost graph for the computation defined by the run call.
CostGraph *CostGraphDef `protobuf:"bytes,2,opt,name=cost_graph,json=costGraph" json:"cost_graph,omitempty"`
// Graphs of the partitions executed by executors.
PartitionGraphs []*GraphDef `protobuf:"bytes,3,rep,name=partition_graphs,json=partitionGraphs" json:"partition_graphs,omitempty"`
}
Metadata output (i.e., non-Tensor) for a single Run() call.
func (*RunMetadata) Descriptor ¶
func (*RunMetadata) Descriptor() ([]byte, []int)
func (*RunMetadata) GetCostGraph ¶
func (m *RunMetadata) GetCostGraph() *CostGraphDef
func (*RunMetadata) GetPartitionGraphs ¶
func (m *RunMetadata) GetPartitionGraphs() []*GraphDef
func (*RunMetadata) GetStepStats ¶
func (m *RunMetadata) GetStepStats() *StepStats
func (*RunMetadata) Marshal ¶
func (m *RunMetadata) Marshal() (dAtA []byte, err error)
func (*RunMetadata) ProtoMessage ¶
func (*RunMetadata) ProtoMessage()
func (*RunMetadata) Reset ¶
func (m *RunMetadata) Reset()
func (*RunMetadata) Size ¶
func (m *RunMetadata) Size() (n int)
func (*RunMetadata) String ¶
func (m *RunMetadata) String() string
func (*RunMetadata) Unmarshal ¶
func (m *RunMetadata) Unmarshal(dAtA []byte) error
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.
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" json:"debug_options,omitempty"`
}
Options for a single Run() call.
func (*RunOptions) Descriptor ¶
func (*RunOptions) Descriptor() ([]byte, []int)
func (*RunOptions) GetDebugOptions ¶
func (m *RunOptions) GetDebugOptions() *DebugOptions
func (*RunOptions) GetInterOpThreadPool ¶
func (m *RunOptions) GetInterOpThreadPool() int32
func (*RunOptions) GetOutputPartitionGraphs ¶
func (m *RunOptions) GetOutputPartitionGraphs() bool
func (*RunOptions) GetTimeoutInMs ¶
func (m *RunOptions) GetTimeoutInMs() int64
func (*RunOptions) GetTraceLevel ¶
func (m *RunOptions) GetTraceLevel() RunOptions_TraceLevel
func (*RunOptions) Marshal ¶
func (m *RunOptions) Marshal() (dAtA []byte, err error)
func (*RunOptions) ProtoMessage ¶
func (*RunOptions) ProtoMessage()
func (*RunOptions) Reset ¶
func (m *RunOptions) Reset()
func (*RunOptions) Size ¶
func (m *RunOptions) Size() (n int)
func (*RunOptions) String ¶
func (m *RunOptions) String() string
func (*RunOptions) Unmarshal ¶
func (m *RunOptions) Unmarshal(dAtA []byte) error
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) EnumDescriptor ¶
func (RunOptions_TraceLevel) EnumDescriptor() ([]byte, []int)
func (RunOptions_TraceLevel) String ¶
func (x RunOptions_TraceLevel) String() string
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" 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" 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" json:"target,omitempty"`
// Options for the run call.
Options *RunOptions `protobuf:"bytes,5,opt,name=options" 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"`
}
func (*RunStepRequest) Descriptor ¶
func (*RunStepRequest) Descriptor() ([]byte, []int)
func (*RunStepRequest) GetFeed ¶
func (m *RunStepRequest) GetFeed() []*NamedTensorProto
func (*RunStepRequest) GetFetch ¶
func (m *RunStepRequest) GetFetch() []string
func (*RunStepRequest) GetOptions ¶
func (m *RunStepRequest) GetOptions() *RunOptions
func (*RunStepRequest) GetPartialRunHandle ¶
func (m *RunStepRequest) GetPartialRunHandle() string
func (*RunStepRequest) GetSessionHandle ¶
func (m *RunStepRequest) GetSessionHandle() string
func (*RunStepRequest) GetTarget ¶
func (m *RunStepRequest) GetTarget() []string
func (*RunStepRequest) Marshal ¶
func (m *RunStepRequest) Marshal() (dAtA []byte, err error)
func (*RunStepRequest) ProtoMessage ¶
func (*RunStepRequest) ProtoMessage()
func (*RunStepRequest) Reset ¶
func (m *RunStepRequest) Reset()
func (*RunStepRequest) Size ¶
func (m *RunStepRequest) Size() (n int)
func (*RunStepRequest) String ¶
func (m *RunStepRequest) String() string
func (*RunStepRequest) Unmarshal ¶
func (m *RunStepRequest) Unmarshal(dAtA []byte) error
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" json:"tensor,omitempty"`
// Returned metadata if requested in the options.
Metadata *RunMetadata `protobuf:"bytes,2,opt,name=metadata" json:"metadata,omitempty"`
}
func (*RunStepResponse) Descriptor ¶
func (*RunStepResponse) Descriptor() ([]byte, []int)
func (*RunStepResponse) GetMetadata ¶
func (m *RunStepResponse) GetMetadata() *RunMetadata
func (*RunStepResponse) GetTensor ¶
func (m *RunStepResponse) GetTensor() []*NamedTensorProto
func (*RunStepResponse) Marshal ¶
func (m *RunStepResponse) Marshal() (dAtA []byte, err error)
func (*RunStepResponse) ProtoMessage ¶
func (*RunStepResponse) ProtoMessage()
func (*RunStepResponse) Reset ¶
func (m *RunStepResponse) Reset()
func (*RunStepResponse) Size ¶
func (m *RunStepResponse) Size() (n int)
func (*RunStepResponse) String ¶
func (m *RunStepResponse) String() string
func (*RunStepResponse) Unmarshal ¶
func (m *RunStepResponse) Unmarshal(dAtA []byte) error
type SaveSliceInfoDef ¶
type SaveSliceInfoDef struct {
// Name of the full variable of which this is a slice.
FullName string `protobuf:"bytes,1,opt,name=full_name,json=fullName,proto3" json:"full_name,omitempty"`
// Shape of the full variable.
FullShape []int64 `protobuf:"varint,2,rep,packed,name=full_shape,json=fullShape" json:"full_shape,omitempty"`
// Offset of this variable into the full variable.
VarOffset []int64 `protobuf:"varint,3,rep,packed,name=var_offset,json=varOffset" json:"var_offset,omitempty"`
// Shape of this variable.
VarShape []int64 `protobuf:"varint,4,rep,packed,name=var_shape,json=varShape" json:"var_shape,omitempty"`
}
func (*SaveSliceInfoDef) Descriptor ¶
func (*SaveSliceInfoDef) Descriptor() ([]byte, []int)
func (*SaveSliceInfoDef) GetFullName ¶
func (m *SaveSliceInfoDef) GetFullName() string
func (*SaveSliceInfoDef) GetFullShape ¶
func (m *SaveSliceInfoDef) GetFullShape() []int64
func (*SaveSliceInfoDef) GetVarOffset ¶
func (m *SaveSliceInfoDef) GetVarOffset() []int64
func (*SaveSliceInfoDef) GetVarShape ¶
func (m *SaveSliceInfoDef) GetVarShape() []int64
func (*SaveSliceInfoDef) Marshal ¶
func (m *SaveSliceInfoDef) Marshal() (dAtA []byte, err error)
func (*SaveSliceInfoDef) ProtoMessage ¶
func (*SaveSliceInfoDef) ProtoMessage()
func (*SaveSliceInfoDef) Reset ¶
func (m *SaveSliceInfoDef) Reset()
func (*SaveSliceInfoDef) Size ¶
func (m *SaveSliceInfoDef) Size() (n int)
func (*SaveSliceInfoDef) String ¶
func (m *SaveSliceInfoDef) String() string
func (*SaveSliceInfoDef) Unmarshal ¶
func (m *SaveSliceInfoDef) Unmarshal(dAtA []byte) error
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" json:"meta_graphs,omitempty"`
}
SavedModel is the high level serialization format for TensorFlow Models. See [todo: doc links, similar to session_bundle] for more information.
func (*SavedModel) Descriptor ¶
func (*SavedModel) Descriptor() ([]byte, []int)
func (*SavedModel) GetMetaGraphs ¶
func (m *SavedModel) GetMetaGraphs() []*MetaGraphDef
func (*SavedModel) GetSavedModelSchemaVersion ¶
func (m *SavedModel) GetSavedModelSchemaVersion() int64
func (*SavedModel) Marshal ¶
func (m *SavedModel) Marshal() (dAtA []byte, err error)
func (*SavedModel) ProtoMessage ¶
func (*SavedModel) ProtoMessage()
func (*SavedModel) Reset ¶
func (m *SavedModel) Reset()
func (*SavedModel) Size ¶
func (m *SavedModel) Size() (n int)
func (*SavedModel) String ¶
func (m *SavedModel) String() string
func (*SavedModel) Unmarshal ¶
func (m *SavedModel) Unmarshal(dAtA []byte) error
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"`
}
Protocol buffer representing the configuration of a Saver.
func (*SaverDef) Descriptor ¶
func (*SaverDef) GetFilenameTensorName ¶
func (*SaverDef) GetKeepCheckpointEveryNHours ¶
func (*SaverDef) GetMaxToKeep ¶
func (*SaverDef) GetRestoreOpName ¶
func (*SaverDef) GetSaveTensorName ¶
func (*SaverDef) GetSharded ¶
func (*SaverDef) GetVersion ¶
func (m *SaverDef) GetVersion() SaverDef_CheckpointFormatVersion
func (*SaverDef) ProtoMessage ¶
func (*SaverDef) ProtoMessage()
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 // Current format: tf.Saver() which works with tensorflow::table::Table. SaverDef_V1 SaverDef_CheckpointFormatVersion = 1 // Experimental format under development. SaverDef_V2 SaverDef_CheckpointFormatVersion = 2 )
func (SaverDef_CheckpointFormatVersion) EnumDescriptor ¶
func (SaverDef_CheckpointFormatVersion) EnumDescriptor() ([]byte, []int)
func (SaverDef_CheckpointFormatVersion) String ¶
func (x SaverDef_CheckpointFormatVersion) String() string
type ServerDef ¶
type ServerDef struct {
// The cluster of which this server is a member.
Cluster *ClusterDef `protobuf:"bytes,1,opt,name=cluster" 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" json:"default_session_config,omitempty"`
// The protocol to be used by this server.
//
// Acceptable values include: "grpc".
Protocol string `protobuf:"bytes,5,opt,name=protocol,proto3" json:"protocol,omitempty"`
}
Defines the configuration of a single TensorFlow server.
func (*ServerDef) Descriptor ¶
func (*ServerDef) GetCluster ¶
func (m *ServerDef) GetCluster() *ClusterDef
func (*ServerDef) GetDefaultSessionConfig ¶
func (m *ServerDef) GetDefaultSessionConfig() *ConfigProto
func (*ServerDef) GetJobName ¶
func (*ServerDef) GetProtocol ¶
func (*ServerDef) GetTaskIndex ¶
func (*ServerDef) ProtoMessage ¶
func (*ServerDef) ProtoMessage()
type SignatureDef ¶
type SignatureDef struct {
// Named input parameters.
Inputs map[string]*TensorInfo `` /* 139-byte string literal not displayed */
// Named output parameters.
Outputs map[string]*TensorInfo `` /* 141-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"`
}
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 ¶
func (*SignatureDef) Descriptor() ([]byte, []int)
func (*SignatureDef) GetInputs ¶
func (m *SignatureDef) GetInputs() map[string]*TensorInfo
func (*SignatureDef) GetMethodName ¶
func (m *SignatureDef) GetMethodName() string
func (*SignatureDef) GetOutputs ¶
func (m *SignatureDef) GetOutputs() map[string]*TensorInfo
func (*SignatureDef) Marshal ¶
func (m *SignatureDef) Marshal() (dAtA []byte, err error)
func (*SignatureDef) ProtoMessage ¶
func (*SignatureDef) ProtoMessage()
func (*SignatureDef) Reset ¶
func (m *SignatureDef) Reset()
func (*SignatureDef) Size ¶
func (m *SignatureDef) Size() (n int)
func (*SignatureDef) String ¶
func (m *SignatureDef) String() string
func (*SignatureDef) Unmarshal ¶
func (m *SignatureDef) Unmarshal(dAtA []byte) error
type StepStats ¶
type StepStats struct {
DevStats []*DeviceStepStats `protobuf:"bytes,1,rep,name=dev_stats,json=devStats" json:"dev_stats,omitempty"`
}
func (*StepStats) Descriptor ¶
func (*StepStats) GetDevStats ¶
func (m *StepStats) GetDevStats() []*DeviceStepStats
func (*StepStats) ProtoMessage ¶
func (*StepStats) ProtoMessage()
type Summary ¶
type Summary struct {
// Set of values for the summary.
Value []*Summary_Value `protobuf:"bytes,1,rep,name=value" json:"value,omitempty"`
}
A Summary is a set of named values to be displayed by the visualizer.
Summaries are produced regularly during training, as controlled by the "summary_interval_secs" attribute of the training operation. Summaries are also produced at the end of an evaluation.
func (*Summary) Descriptor ¶
func (*Summary) GetValue ¶
func (m *Summary) GetValue() []*Summary_Value
func (*Summary) ProtoMessage ¶
func (*Summary) ProtoMessage()
type SummaryDescription ¶
type SummaryDescription struct {
// Hint on how plugins should process the data in this series.
// Supported values include "scalar", "histogram", "image", "audio"
TypeHint string `protobuf:"bytes,1,opt,name=type_hint,json=typeHint,proto3" json:"type_hint,omitempty"`
}
Metadata associated with a series of Summary data
func (*SummaryDescription) Descriptor ¶
func (*SummaryDescription) Descriptor() ([]byte, []int)
func (*SummaryDescription) GetTypeHint ¶
func (m *SummaryDescription) GetTypeHint() string
func (*SummaryDescription) Marshal ¶
func (m *SummaryDescription) Marshal() (dAtA []byte, err error)
func (*SummaryDescription) MarshalTo ¶
func (m *SummaryDescription) MarshalTo(dAtA []byte) (int, error)
func (*SummaryDescription) ProtoMessage ¶
func (*SummaryDescription) ProtoMessage()
func (*SummaryDescription) Reset ¶
func (m *SummaryDescription) Reset()
func (*SummaryDescription) Size ¶
func (m *SummaryDescription) Size() (n int)
func (*SummaryDescription) String ¶
func (m *SummaryDescription) String() string
func (*SummaryDescription) Unmarshal ¶
func (m *SummaryDescription) Unmarshal(dAtA []byte) error
type Summary_Audio ¶
type Summary_Audio struct {
// Sample rate of the audio in Hz.
SampleRate float32 `protobuf:"fixed32,1,opt,name=sample_rate,json=sampleRate,proto3" json:"sample_rate,omitempty"`
// Number of channels of audio.
NumChannels int64 `protobuf:"varint,2,opt,name=num_channels,json=numChannels,proto3" json:"num_channels,omitempty"`
// Length of the audio in frames (samples per channel).
LengthFrames int64 `protobuf:"varint,3,opt,name=length_frames,json=lengthFrames,proto3" json:"length_frames,omitempty"`
// Encoded audio data and its associated RFC 2045 content type (e.g.
// "audio/wav").
EncodedAudioString []byte `protobuf:"bytes,4,opt,name=encoded_audio_string,json=encodedAudioString,proto3" json:"encoded_audio_string,omitempty"`
ContentType string `protobuf:"bytes,5,opt,name=content_type,json=contentType,proto3" json:"content_type,omitempty"`
}
func (*Summary_Audio) Descriptor ¶
func (*Summary_Audio) Descriptor() ([]byte, []int)
func (*Summary_Audio) GetContentType ¶
func (m *Summary_Audio) GetContentType() string
func (*Summary_Audio) GetEncodedAudioString ¶
func (m *Summary_Audio) GetEncodedAudioString() []byte
func (*Summary_Audio) GetLengthFrames ¶
func (m *Summary_Audio) GetLengthFrames() int64
func (*Summary_Audio) GetNumChannels ¶
func (m *Summary_Audio) GetNumChannels() int64
func (*Summary_Audio) GetSampleRate ¶
func (m *Summary_Audio) GetSampleRate() float32
func (*Summary_Audio) Marshal ¶
func (m *Summary_Audio) Marshal() (dAtA []byte, err error)
func (*Summary_Audio) ProtoMessage ¶
func (*Summary_Audio) ProtoMessage()
func (*Summary_Audio) Reset ¶
func (m *Summary_Audio) Reset()
func (*Summary_Audio) Size ¶
func (m *Summary_Audio) Size() (n int)
func (*Summary_Audio) String ¶
func (m *Summary_Audio) String() string
func (*Summary_Audio) Unmarshal ¶
func (m *Summary_Audio) Unmarshal(dAtA []byte) error
type Summary_Image ¶
type Summary_Image struct {
// Dimensions of the image.
Height int32 `protobuf:"varint,1,opt,name=height,proto3" json:"height,omitempty"`
Width int32 `protobuf:"varint,2,opt,name=width,proto3" json:"width,omitempty"`
// Valid colorspace values are
// 1 - grayscale
// 2 - grayscale + alpha
// 3 - RGB
// 4 - RGBA
// 5 - DIGITAL_YUV
// 6 - BGRA
Colorspace int32 `protobuf:"varint,3,opt,name=colorspace,proto3" json:"colorspace,omitempty"`
// Image data in encoded format. All image formats supported by
// image_codec::CoderUtil can be stored here.
EncodedImageString []byte `protobuf:"bytes,4,opt,name=encoded_image_string,json=encodedImageString,proto3" json:"encoded_image_string,omitempty"`
}
func (*Summary_Image) Descriptor ¶
func (*Summary_Image) Descriptor() ([]byte, []int)
func (*Summary_Image) GetColorspace ¶
func (m *Summary_Image) GetColorspace() int32
func (*Summary_Image) GetEncodedImageString ¶
func (m *Summary_Image) GetEncodedImageString() []byte
func (*Summary_Image) GetHeight ¶
func (m *Summary_Image) GetHeight() int32
func (*Summary_Image) GetWidth ¶
func (m *Summary_Image) GetWidth() int32
func (*Summary_Image) Marshal ¶
func (m *Summary_Image) Marshal() (dAtA []byte, err error)
func (*Summary_Image) ProtoMessage ¶
func (*Summary_Image) ProtoMessage()
func (*Summary_Image) Reset ¶
func (m *Summary_Image) Reset()
func (*Summary_Image) Size ¶
func (m *Summary_Image) Size() (n int)
func (*Summary_Image) String ¶
func (m *Summary_Image) String() string
func (*Summary_Image) Unmarshal ¶
func (m *Summary_Image) Unmarshal(dAtA []byte) error
type Summary_Value ¶
type Summary_Value struct {
// Name of the node that output this summary; in general, the name of a
// TensorSummary node. If the node in question has multiple outputs, then
// a ":\d+" suffix will be appended, like "some_op:13".
// Might not be set for legacy summaries (i.e. those not using the tensor
// value field)
NodeName string `protobuf:"bytes,7,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"`
// Tag name for the data. Will only be used by legacy summaries
// (ie. those not using the tensor value field)
// For legacy summaries, will be used as the title of the graph
// in the visualizer.
//
// Tag is usually "op_name:value_name", where "op_name" itself can have
// structure to indicate grouping.
Tag string `protobuf:"bytes,1,opt,name=tag,proto3" json:"tag,omitempty"`
// Value associated with the tag.
//
// Types that are valid to be assigned to Value:
// *Summary_Value_SimpleValue
// *Summary_Value_ObsoleteOldStyleHistogram
// *Summary_Value_Image
// *Summary_Value_Histo
// *Summary_Value_Audio
// *Summary_Value_Tensor
Value isSummary_Value_Value `protobuf_oneof:"value"`
}
func (*Summary_Value) Descriptor ¶
func (*Summary_Value) Descriptor() ([]byte, []int)
func (*Summary_Value) GetAudio ¶
func (m *Summary_Value) GetAudio() *Summary_Audio
func (*Summary_Value) GetHisto ¶
func (m *Summary_Value) GetHisto() *HistogramProto
func (*Summary_Value) GetImage ¶
func (m *Summary_Value) GetImage() *Summary_Image
func (*Summary_Value) GetNodeName ¶
func (m *Summary_Value) GetNodeName() string
func (*Summary_Value) GetObsoleteOldStyleHistogram ¶
func (m *Summary_Value) GetObsoleteOldStyleHistogram() []byte
func (*Summary_Value) GetSimpleValue ¶
func (m *Summary_Value) GetSimpleValue() float32
func (*Summary_Value) GetTag ¶
func (m *Summary_Value) GetTag() string
func (*Summary_Value) GetTensor ¶
func (m *Summary_Value) GetTensor() *TensorProto
func (*Summary_Value) GetValue ¶
func (m *Summary_Value) GetValue() isSummary_Value_Value
func (*Summary_Value) Marshal ¶
func (m *Summary_Value) Marshal() (dAtA []byte, err error)
func (*Summary_Value) ProtoMessage ¶
func (*Summary_Value) ProtoMessage()
func (*Summary_Value) Reset ¶
func (m *Summary_Value) Reset()
func (*Summary_Value) Size ¶
func (m *Summary_Value) Size() (n int)
func (*Summary_Value) String ¶
func (m *Summary_Value) String() string
func (*Summary_Value) Unmarshal ¶
func (m *Summary_Value) Unmarshal(dAtA []byte) error
func (*Summary_Value) XXX_OneofFuncs ¶
func (*Summary_Value) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, func(msg proto.Message, tag, wire int, b *proto.Buffer) (bool, error), func(msg proto.Message) (n int), []interface{})
XXX_OneofFuncs is for the internal use of the proto package.
type Summary_Value_Audio ¶
type Summary_Value_Audio struct {
Audio *Summary_Audio `protobuf:"bytes,6,opt,name=audio,oneof"`
}
func (*Summary_Value_Audio) MarshalTo ¶
func (m *Summary_Value_Audio) MarshalTo(dAtA []byte) (int, error)
func (*Summary_Value_Audio) Size ¶
func (m *Summary_Value_Audio) Size() (n int)
type Summary_Value_Histo ¶
type Summary_Value_Histo struct {
Histo *HistogramProto `protobuf:"bytes,5,opt,name=histo,oneof"`
}
func (*Summary_Value_Histo) MarshalTo ¶
func (m *Summary_Value_Histo) MarshalTo(dAtA []byte) (int, error)
func (*Summary_Value_Histo) Size ¶
func (m *Summary_Value_Histo) Size() (n int)
type Summary_Value_Image ¶
type Summary_Value_Image struct {
Image *Summary_Image `protobuf:"bytes,4,opt,name=image,oneof"`
}
func (*Summary_Value_Image) MarshalTo ¶
func (m *Summary_Value_Image) MarshalTo(dAtA []byte) (int, error)
func (*Summary_Value_Image) Size ¶
func (m *Summary_Value_Image) Size() (n int)
type Summary_Value_ObsoleteOldStyleHistogram ¶
type Summary_Value_ObsoleteOldStyleHistogram struct {
ObsoleteOldStyleHistogram []byte `protobuf:"bytes,3,opt,name=obsolete_old_style_histogram,json=obsoleteOldStyleHistogram,proto3,oneof"`
}
func (*Summary_Value_ObsoleteOldStyleHistogram) MarshalTo ¶
func (m *Summary_Value_ObsoleteOldStyleHistogram) MarshalTo(dAtA []byte) (int, error)
func (*Summary_Value_ObsoleteOldStyleHistogram) Size ¶
func (m *Summary_Value_ObsoleteOldStyleHistogram) Size() (n int)
type Summary_Value_SimpleValue ¶
type Summary_Value_SimpleValue struct {
SimpleValue float32 `protobuf:"fixed32,2,opt,name=simple_value,json=simpleValue,proto3,oneof"`
}
func (*Summary_Value_SimpleValue) MarshalTo ¶
func (m *Summary_Value_SimpleValue) MarshalTo(dAtA []byte) (int, error)
func (*Summary_Value_SimpleValue) Size ¶
func (m *Summary_Value_SimpleValue) Size() (n int)
type Summary_Value_Tensor ¶
type Summary_Value_Tensor struct {
Tensor *TensorProto `protobuf:"bytes,8,opt,name=tensor,oneof"`
}
func (*Summary_Value_Tensor) MarshalTo ¶
func (m *Summary_Value_Tensor) MarshalTo(dAtA []byte) (int, error)
func (*Summary_Value_Tensor) Size ¶
func (m *Summary_Value_Tensor) Size() (n int)
type TensorDescription ¶
type TensorDescription struct {
// Data type of tensor elements
Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
// Shape of the tensor.
Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape" json:"shape,omitempty"`
// Information about the size and allocator used for the data
AllocationDescription *AllocationDescription `protobuf:"bytes,4,opt,name=allocation_description,json=allocationDescription" json:"allocation_description,omitempty"`
}
func (*TensorDescription) Descriptor ¶
func (*TensorDescription) Descriptor() ([]byte, []int)
func (*TensorDescription) GetAllocationDescription ¶
func (m *TensorDescription) GetAllocationDescription() *AllocationDescription
func (*TensorDescription) GetDtype ¶
func (m *TensorDescription) GetDtype() DataType
func (*TensorDescription) GetShape ¶
func (m *TensorDescription) GetShape() *TensorShapeProto
func (*TensorDescription) Marshal ¶
func (m *TensorDescription) Marshal() (dAtA []byte, err error)
func (*TensorDescription) MarshalTo ¶
func (m *TensorDescription) MarshalTo(dAtA []byte) (int, error)
func (*TensorDescription) ProtoMessage ¶
func (*TensorDescription) ProtoMessage()
func (*TensorDescription) Reset ¶
func (m *TensorDescription) Reset()
func (*TensorDescription) Size ¶
func (m *TensorDescription) Size() (n int)
func (*TensorDescription) String ¶
func (m *TensorDescription) String() string
func (*TensorDescription) Unmarshal ¶
func (m *TensorDescription) Unmarshal(dAtA []byte) error
type TensorInfo ¶
type TensorInfo struct {
Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
Dtype DataType `protobuf:"varint,2,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
TensorShape *TensorShapeProto `protobuf:"bytes,3,opt,name=tensor_shape,json=tensorShape" json:"tensor_shape,omitempty"`
}
Information about a Tensor necessary for feeding or retrieval.
func (*TensorInfo) Descriptor ¶
func (*TensorInfo) Descriptor() ([]byte, []int)
func (*TensorInfo) GetDtype ¶
func (m *TensorInfo) GetDtype() DataType
func (*TensorInfo) GetName ¶
func (m *TensorInfo) GetName() string
func (*TensorInfo) GetTensorShape ¶
func (m *TensorInfo) GetTensorShape() *TensorShapeProto
func (*TensorInfo) Marshal ¶
func (m *TensorInfo) Marshal() (dAtA []byte, err error)
func (*TensorInfo) ProtoMessage ¶
func (*TensorInfo) ProtoMessage()
func (*TensorInfo) Reset ¶
func (m *TensorInfo) Reset()
func (*TensorInfo) Size ¶
func (m *TensorInfo) Size() (n int)
func (*TensorInfo) String ¶
func (m *TensorInfo) String() string
func (*TensorInfo) Unmarshal ¶
func (m *TensorInfo) Unmarshal(dAtA []byte) error
type TensorProto ¶
type TensorProto struct {
Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
// Shape of the tensor. TODO(touts): sort out the 0-rank issues.
TensorShape *TensorShapeProto `protobuf:"bytes,2,opt,name=tensor_shape,json=tensorShape" json:"tensor_shape,omitempty"`
// Version number.
//
// In version 0, if the "repeated xxx" representations contain only one
// element, that element is repeated to fill the shape. This makes it easy
// to represent a constant Tensor with a single value.
VersionNumber int32 `protobuf:"varint,3,opt,name=version_number,json=versionNumber,proto3" json:"version_number,omitempty"`
// Serialized raw tensor content from either Tensor::AsProtoTensorContent or
// memcpy in tensorflow::grpc::EncodeTensorToByteBuffer. This representation
// can be used for all tensor types. The purpose of this representation is to
// reduce serialization overhead during RPC call by avoiding serialization of
// many repeated small items.
TensorContent []byte `protobuf:"bytes,4,opt,name=tensor_content,json=tensorContent,proto3" json:"tensor_content,omitempty"`
// DT_HALF. Note that since protobuf has no int16 type, we'll have some
// pointless zero padding for each value here.
HalfVal []int32 `protobuf:"varint,13,rep,packed,name=half_val,json=halfVal" json:"half_val,omitempty"`
// DT_FLOAT.
FloatVal []float32 `protobuf:"fixed32,5,rep,packed,name=float_val,json=floatVal" json:"float_val,omitempty"`
// DT_DOUBLE.
DoubleVal []float64 `protobuf:"fixed64,6,rep,packed,name=double_val,json=doubleVal" json:"double_val,omitempty"`
// DT_INT32, DT_INT16, DT_INT8, DT_UINT8.
IntVal []int32 `protobuf:"varint,7,rep,packed,name=int_val,json=intVal" json:"int_val,omitempty"`
// DT_STRING
StringVal [][]byte `protobuf:"bytes,8,rep,name=string_val,json=stringVal" json:"string_val,omitempty"`
// DT_COMPLEX64. scomplex_val(2*i) and scomplex_val(2*i+1) are real
// and imaginary parts of i-th single precision complex.
ScomplexVal []float32 `protobuf:"fixed32,9,rep,packed,name=scomplex_val,json=scomplexVal" json:"scomplex_val,omitempty"`
// DT_INT64
Int64Val []int64 `protobuf:"varint,10,rep,packed,name=int64_val,json=int64Val" json:"int64_val,omitempty"`
// DT_BOOL
BoolVal []bool `protobuf:"varint,11,rep,packed,name=bool_val,json=boolVal" json:"bool_val,omitempty"`
// DT_COMPLEX128. dcomplex_val(2*i) and dcomplex_val(2*i+1) are real
// and imaginary parts of i-th double precision complex.
DcomplexVal []float64 `protobuf:"fixed64,12,rep,packed,name=dcomplex_val,json=dcomplexVal" json:"dcomplex_val,omitempty"`
// DT_RESOURCE
ResourceHandleVal []*ResourceHandle `protobuf:"bytes,14,rep,name=resource_handle_val,json=resourceHandleVal" json:"resource_handle_val,omitempty"`
}
Protocol buffer representing a tensor.
func (*TensorProto) Descriptor ¶
func (*TensorProto) Descriptor() ([]byte, []int)
func (*TensorProto) GetBoolVal ¶
func (m *TensorProto) GetBoolVal() []bool
func (*TensorProto) GetDcomplexVal ¶
func (m *TensorProto) GetDcomplexVal() []float64
func (*TensorProto) GetDoubleVal ¶
func (m *TensorProto) GetDoubleVal() []float64
func (*TensorProto) GetDtype ¶
func (m *TensorProto) GetDtype() DataType
func (*TensorProto) GetFloatVal ¶
func (m *TensorProto) GetFloatVal() []float32
func (*TensorProto) GetHalfVal ¶
func (m *TensorProto) GetHalfVal() []int32
func (*TensorProto) GetInt64Val ¶
func (m *TensorProto) GetInt64Val() []int64
func (*TensorProto) GetIntVal ¶
func (m *TensorProto) GetIntVal() []int32
func (*TensorProto) GetResourceHandleVal ¶
func (m *TensorProto) GetResourceHandleVal() []*ResourceHandle
func (*TensorProto) GetScomplexVal ¶
func (m *TensorProto) GetScomplexVal() []float32
func (*TensorProto) GetStringVal ¶
func (m *TensorProto) GetStringVal() [][]byte
func (*TensorProto) GetTensorContent ¶
func (m *TensorProto) GetTensorContent() []byte
func (*TensorProto) GetTensorShape ¶
func (m *TensorProto) GetTensorShape() *TensorShapeProto
func (*TensorProto) GetVersionNumber ¶
func (m *TensorProto) GetVersionNumber() int32
func (*TensorProto) Marshal ¶
func (m *TensorProto) Marshal() (dAtA []byte, err error)
func (*TensorProto) ProtoMessage ¶
func (*TensorProto) ProtoMessage()
func (*TensorProto) Reset ¶
func (m *TensorProto) Reset()
func (*TensorProto) Size ¶
func (m *TensorProto) Size() (n int)
func (*TensorProto) String ¶
func (m *TensorProto) String() string
func (*TensorProto) Unmarshal ¶
func (m *TensorProto) Unmarshal(dAtA []byte) error
type TensorShapeProto ¶
type TensorShapeProto struct {
// Dimensions of the tensor, such as {"input", 30}, {"output", 40}
// for a 30 x 40 2D tensor. If an entry has size -1, this
// corresponds to a dimension of unknown size. The names are
// optional.
//
// The order of entries in "dim" matters: It indicates the layout of the
// values in the tensor in-memory representation.
//
// The first entry in "dim" is the outermost dimension used to layout the
// values, the last entry is the innermost dimension. This matches the
// in-memory layout of RowMajor Eigen tensors.
//
// If "dim.size()" > 0, "unknown_rank" must be false.
Dim []*TensorShapeProto_Dim `protobuf:"bytes,2,rep,name=dim" json:"dim,omitempty"`
// If true, the number of dimensions in the shape is unknown.
//
// If true, "dim.size()" must be 0.
UnknownRank bool `protobuf:"varint,3,opt,name=unknown_rank,json=unknownRank,proto3" json:"unknown_rank,omitempty"`
}
Dimensions of a tensor.
func (*TensorShapeProto) Descriptor ¶
func (*TensorShapeProto) Descriptor() ([]byte, []int)
func (*TensorShapeProto) GetDim ¶
func (m *TensorShapeProto) GetDim() []*TensorShapeProto_Dim
func (*TensorShapeProto) GetUnknownRank ¶
func (m *TensorShapeProto) GetUnknownRank() bool
func (*TensorShapeProto) Marshal ¶
func (m *TensorShapeProto) Marshal() (dAtA []byte, err error)
func (*TensorShapeProto) ProtoMessage ¶
func (*TensorShapeProto) ProtoMessage()
func (*TensorShapeProto) Reset ¶
func (m *TensorShapeProto) Reset()
func (*TensorShapeProto) Size ¶
func (m *TensorShapeProto) Size() (n int)
func (*TensorShapeProto) String ¶
func (m *TensorShapeProto) String() string
func (*TensorShapeProto) Unmarshal ¶
func (m *TensorShapeProto) Unmarshal(dAtA []byte) error
type TensorShapeProto_Dim ¶
type TensorShapeProto_Dim struct {
// Size of the tensor in that dimension.
// This value must be >= -1, but values of -1 are reserved for "unknown"
// shapes (values of -1 mean "unknown" dimension). Certain wrappers
// that work with TensorShapeProto may fail at runtime when deserializing
// a TensorShapeProto containing a dim value of -1.
Size_ int64 `protobuf:"varint,1,opt,name=size,proto3" json:"size,omitempty"`
// Optional name of the tensor dimension.
Name string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
}
One dimension of the tensor.
func (*TensorShapeProto_Dim) Descriptor ¶
func (*TensorShapeProto_Dim) Descriptor() ([]byte, []int)
func (*TensorShapeProto_Dim) GetName ¶
func (m *TensorShapeProto_Dim) GetName() string
func (*TensorShapeProto_Dim) GetSize_ ¶
func (m *TensorShapeProto_Dim) GetSize_() int64
func (*TensorShapeProto_Dim) Marshal ¶
func (m *TensorShapeProto_Dim) Marshal() (dAtA []byte, err error)
func (*TensorShapeProto_Dim) MarshalTo ¶
func (m *TensorShapeProto_Dim) MarshalTo(dAtA []byte) (int, error)
func (*TensorShapeProto_Dim) ProtoMessage ¶
func (*TensorShapeProto_Dim) ProtoMessage()
func (*TensorShapeProto_Dim) Reset ¶
func (m *TensorShapeProto_Dim) Reset()
func (*TensorShapeProto_Dim) Size ¶
func (m *TensorShapeProto_Dim) Size() (n int)
func (*TensorShapeProto_Dim) String ¶
func (m *TensorShapeProto_Dim) String() string
func (*TensorShapeProto_Dim) Unmarshal ¶
func (m *TensorShapeProto_Dim) Unmarshal(dAtA []byte) error
type TensorSliceProto ¶
type TensorSliceProto struct {
// Extent of the slice in all tensor dimensions.
//
// Must have one entry for each of the dimension of the tensor that this
// slice belongs to. The order of sizes is the same as the order of
// dimensions in the TensorShape.
Extent []*TensorSliceProto_Extent `protobuf:"bytes,1,rep,name=extent" json:"extent,omitempty"`
}
Can only be interpreted if you know the corresponding TensorShape.
func (*TensorSliceProto) Descriptor ¶
func (*TensorSliceProto) Descriptor() ([]byte, []int)
func (*TensorSliceProto) GetExtent ¶
func (m *TensorSliceProto) GetExtent() []*TensorSliceProto_Extent
func (*TensorSliceProto) Marshal ¶
func (m *TensorSliceProto) Marshal() (dAtA []byte, err error)
func (*TensorSliceProto) ProtoMessage ¶
func (*TensorSliceProto) ProtoMessage()
func (*TensorSliceProto) Reset ¶
func (m *TensorSliceProto) Reset()
func (*TensorSliceProto) Size ¶
func (m *TensorSliceProto) Size() (n int)
func (*TensorSliceProto) String ¶
func (m *TensorSliceProto) String() string
func (*TensorSliceProto) Unmarshal ¶
func (m *TensorSliceProto) Unmarshal(dAtA []byte) error
type TensorSliceProto_Extent ¶
type TensorSliceProto_Extent struct {
// Start index of the slice, starting at 0.
Start int64 `protobuf:"varint,1,opt,name=start,proto3" json:"start,omitempty"`
// Length of the slice: if the length is missing or -1 we will
// interpret this as "everything in this dimension". We use
// "oneof" to preserve information about whether the length is
// present without changing the serialization format from the
// prior proto2 version of this proto.
//
// Types that are valid to be assigned to HasLength:
// *TensorSliceProto_Extent_Length
HasLength isTensorSliceProto_Extent_HasLength `protobuf_oneof:"has_length"`
}
Extent of the slice in one dimension.
func (*TensorSliceProto_Extent) Descriptor ¶
func (*TensorSliceProto_Extent) Descriptor() ([]byte, []int)
func (*TensorSliceProto_Extent) GetHasLength ¶
func (m *TensorSliceProto_Extent) GetHasLength() isTensorSliceProto_Extent_HasLength
func (*TensorSliceProto_Extent) GetLength ¶
func (m *TensorSliceProto_Extent) GetLength() int64
func (*TensorSliceProto_Extent) GetStart ¶
func (m *TensorSliceProto_Extent) GetStart() int64
func (*TensorSliceProto_Extent) Marshal ¶
func (m *TensorSliceProto_Extent) Marshal() (dAtA []byte, err error)
func (*TensorSliceProto_Extent) MarshalTo ¶
func (m *TensorSliceProto_Extent) MarshalTo(dAtA []byte) (int, error)
func (*TensorSliceProto_Extent) ProtoMessage ¶
func (*TensorSliceProto_Extent) ProtoMessage()
func (*TensorSliceProto_Extent) Reset ¶
func (m *TensorSliceProto_Extent) Reset()
func (*TensorSliceProto_Extent) Size ¶
func (m *TensorSliceProto_Extent) Size() (n int)
func (*TensorSliceProto_Extent) String ¶
func (m *TensorSliceProto_Extent) String() string
func (*TensorSliceProto_Extent) Unmarshal ¶
func (m *TensorSliceProto_Extent) Unmarshal(dAtA []byte) error
func (*TensorSliceProto_Extent) XXX_OneofFuncs ¶
func (*TensorSliceProto_Extent) XXX_OneofFuncs() (func(msg proto.Message, b *proto.Buffer) error, func(msg proto.Message, tag, wire int, b *proto.Buffer) (bool, error), func(msg proto.Message) (n int), []interface{})
XXX_OneofFuncs is for the internal use of the proto package.
type TensorSliceProto_Extent_Length ¶
type TensorSliceProto_Extent_Length struct {
Length int64 `protobuf:"varint,2,opt,name=length,proto3,oneof"`
}
func (*TensorSliceProto_Extent_Length) MarshalTo ¶
func (m *TensorSliceProto_Extent_Length) MarshalTo(dAtA []byte) (int, error)
func (*TensorSliceProto_Extent_Length) Size ¶
func (m *TensorSliceProto_Extent_Length) Size() (n int)
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"`
}
func (*ThreadPoolOptionProto) Descriptor ¶
func (*ThreadPoolOptionProto) Descriptor() ([]byte, []int)
func (*ThreadPoolOptionProto) GetNumThreads ¶
func (m *ThreadPoolOptionProto) GetNumThreads() int32
func (*ThreadPoolOptionProto) Marshal ¶
func (m *ThreadPoolOptionProto) Marshal() (dAtA []byte, err error)
func (*ThreadPoolOptionProto) MarshalTo ¶
func (m *ThreadPoolOptionProto) MarshalTo(dAtA []byte) (int, error)
func (*ThreadPoolOptionProto) ProtoMessage ¶
func (*ThreadPoolOptionProto) ProtoMessage()
func (*ThreadPoolOptionProto) Reset ¶
func (m *ThreadPoolOptionProto) Reset()
func (*ThreadPoolOptionProto) Size ¶
func (m *ThreadPoolOptionProto) Size() (n int)
func (*ThreadPoolOptionProto) String ¶
func (m *ThreadPoolOptionProto) String() string
func (*ThreadPoolOptionProto) Unmarshal ¶
func (m *ThreadPoolOptionProto) Unmarshal(dAtA []byte) error
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"`
}
func (*TraceOpts) Descriptor ¶
func (*TraceOpts) GetDuration ¶
func (*TraceOpts) GetUseExtendedProfiler ¶
func (*TraceOpts) GetUseGpuProfiler ¶
func (*TraceOpts) GetUseKernelProfiler ¶
func (*TraceOpts) GetUseSampleProfiler ¶
func (*TraceOpts) GetUseStepProfiler ¶
func (*TraceOpts) ProtoMessage ¶
func (*TraceOpts) ProtoMessage()
type TracingRequest ¶
type TracingRequest struct {
Options *TraceOpts `protobuf:"bytes,1,opt,name=options" json:"options,omitempty"`
}
Out-of-band request to configure distributed tracing.
func (*TracingRequest) Descriptor ¶
func (*TracingRequest) Descriptor() ([]byte, []int)
func (*TracingRequest) GetOptions ¶
func (m *TracingRequest) GetOptions() *TraceOpts
func (*TracingRequest) Marshal ¶
func (m *TracingRequest) Marshal() (dAtA []byte, err error)
func (*TracingRequest) ProtoMessage ¶
func (*TracingRequest) ProtoMessage()
func (*TracingRequest) Reset ¶
func (m *TracingRequest) Reset()
func (*TracingRequest) Size ¶
func (m *TracingRequest) Size() (n int)
func (*TracingRequest) String ¶
func (m *TracingRequest) String() string
func (*TracingRequest) Unmarshal ¶
func (m *TracingRequest) Unmarshal(dAtA []byte) error
type TracingResponse ¶
type TracingResponse struct {
}
func (*TracingResponse) Descriptor ¶
func (*TracingResponse) Descriptor() ([]byte, []int)
func (*TracingResponse) Marshal ¶
func (m *TracingResponse) Marshal() (dAtA []byte, err error)
func (*TracingResponse) ProtoMessage ¶
func (*TracingResponse) ProtoMessage()
func (*TracingResponse) Reset ¶
func (m *TracingResponse) Reset()
func (*TracingResponse) Size ¶
func (m *TracingResponse) Size() (n int)
func (*TracingResponse) String ¶
func (m *TracingResponse) String() string
func (*TracingResponse) Unmarshal ¶
func (m *TracingResponse) Unmarshal(dAtA []byte) error
type ValuesDef ¶
type ValuesDef struct {
// Value names that have been seen in this context.
Values []string `protobuf:"bytes,1,rep,name=values" json:"values,omitempty"`
// Value names referenced by but external to this context.
ExternalValues map[string]string `` /* 184-byte string literal not displayed */
}
Protocol buffer representing the values in ControlFlowContext.
func (*ValuesDef) Descriptor ¶
func (*ValuesDef) GetExternalValues ¶
func (*ValuesDef) ProtoMessage ¶
func (*ValuesDef) ProtoMessage()
type VariableDef ¶
type VariableDef struct {
// Name of the variable tensor.
VariableName string `protobuf:"bytes,1,opt,name=variable_name,json=variableName,proto3" json:"variable_name,omitempty"`
// Name of the initializer op.
InitializerName string `protobuf:"bytes,2,opt,name=initializer_name,json=initializerName,proto3" json:"initializer_name,omitempty"`
// Name of the snapshot tensor.
SnapshotName string `protobuf:"bytes,3,opt,name=snapshot_name,json=snapshotName,proto3" json:"snapshot_name,omitempty"`
// Support for saving variables as slices of a larger variable.
SaveSliceInfoDef *SaveSliceInfoDef `protobuf:"bytes,4,opt,name=save_slice_info_def,json=saveSliceInfoDef" json:"save_slice_info_def,omitempty"`
// Whether to represent this as a ResourceVariable.
IsResource bool `protobuf:"varint,5,opt,name=is_resource,json=isResource,proto3" json:"is_resource,omitempty"`
}
Protocol buffer representing a Variable.
func (*VariableDef) Descriptor ¶
func (*VariableDef) Descriptor() ([]byte, []int)
func (*VariableDef) GetInitializerName ¶
func (m *VariableDef) GetInitializerName() string
func (*VariableDef) GetIsResource ¶
func (m *VariableDef) GetIsResource() bool
func (*VariableDef) GetSaveSliceInfoDef ¶
func (m *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef
func (*VariableDef) GetSnapshotName ¶
func (m *VariableDef) GetSnapshotName() string
func (*VariableDef) GetVariableName ¶
func (m *VariableDef) GetVariableName() string
func (*VariableDef) Marshal ¶
func (m *VariableDef) Marshal() (dAtA []byte, err error)
func (*VariableDef) ProtoMessage ¶
func (*VariableDef) ProtoMessage()
func (*VariableDef) Reset ¶
func (m *VariableDef) Reset()
func (*VariableDef) Size ¶
func (m *VariableDef) Size() (n int)
func (*VariableDef) String ¶
func (m *VariableDef) String() string
func (*VariableDef) Unmarshal ¶
func (m *VariableDef) Unmarshal(dAtA []byte) error
type VersionDef ¶
type VersionDef struct {
// The version of the code that produced this data.
Producer int32 `protobuf:"varint,1,opt,name=producer,proto3" json:"producer,omitempty"`
// Any consumer below this version is not allowed to consume this data.
MinConsumer int32 `protobuf:"varint,2,opt,name=min_consumer,json=minConsumer,proto3" json:"min_consumer,omitempty"`
// Specific consumer versions which are disallowed (e.g. due to bugs).
BadConsumers []int32 `protobuf:"varint,3,rep,packed,name=bad_consumers,json=badConsumers" json:"bad_consumers,omitempty"`
}
Version information for a piece of serialized data
There are different types of versions for each type of data (GraphDef, etc.), but they all have the same common shape described here.
Each consumer has "consumer" and "min_producer" versions (specified elsewhere). A consumer is allowed to consume this data if
producer >= min_producer consumer >= min_consumer consumer not in bad_consumers
func (*VersionDef) Descriptor ¶
func (*VersionDef) Descriptor() ([]byte, []int)
func (*VersionDef) GetBadConsumers ¶
func (m *VersionDef) GetBadConsumers() []int32
func (*VersionDef) GetMinConsumer ¶
func (m *VersionDef) GetMinConsumer() int32
func (*VersionDef) GetProducer ¶
func (m *VersionDef) GetProducer() int32
func (*VersionDef) Marshal ¶
func (m *VersionDef) Marshal() (dAtA []byte, err error)
func (*VersionDef) ProtoMessage ¶
func (*VersionDef) ProtoMessage()
func (*VersionDef) Reset ¶
func (m *VersionDef) Reset()
func (*VersionDef) Size ¶
func (m *VersionDef) Size() (n int)
func (*VersionDef) String ¶
func (m *VersionDef) String() string
func (*VersionDef) Unmarshal ¶
func (m *VersionDef) Unmarshal(dAtA []byte) error
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" json:"loop_exit_names,omitempty"`
// List of names for enter tensors.
LoopEnterNames []string `protobuf:"bytes,10,rep,name=loop_enter_names,json=loopEnterNames" json:"loop_enter_names,omitempty"`
// Values and external values in control flow context.
ValuesDef *ValuesDef `protobuf:"bytes,9,opt,name=values_def,json=valuesDef" json:"values_def,omitempty"`
}
Protocol buffer representing a WhileContext object.
func (*WhileContextDef) Descriptor ¶
func (*WhileContextDef) Descriptor() ([]byte, []int)
func (*WhileContextDef) GetBackProp ¶
func (m *WhileContextDef) GetBackProp() bool
func (*WhileContextDef) GetContextName ¶
func (m *WhileContextDef) GetContextName() string
func (*WhileContextDef) GetLoopEnterNames ¶
func (m *WhileContextDef) GetLoopEnterNames() []string
func (*WhileContextDef) GetLoopExitNames ¶
func (m *WhileContextDef) GetLoopExitNames() []string
func (*WhileContextDef) GetParallelIterations ¶
func (m *WhileContextDef) GetParallelIterations() int32
func (*WhileContextDef) GetPivotForBodyName ¶
func (m *WhileContextDef) GetPivotForBodyName() string
func (*WhileContextDef) GetPivotForPredName ¶
func (m *WhileContextDef) GetPivotForPredName() string
func (*WhileContextDef) GetPivotName ¶
func (m *WhileContextDef) GetPivotName() string
func (*WhileContextDef) GetSwapMemory ¶
func (m *WhileContextDef) GetSwapMemory() bool
func (*WhileContextDef) GetValuesDef ¶
func (m *WhileContextDef) GetValuesDef() *ValuesDef
func (*WhileContextDef) Marshal ¶
func (m *WhileContextDef) Marshal() (dAtA []byte, err error)
func (*WhileContextDef) ProtoMessage ¶
func (*WhileContextDef) ProtoMessage()
func (*WhileContextDef) Reset ¶
func (m *WhileContextDef) Reset()
func (*WhileContextDef) Size ¶
func (m *WhileContextDef) Size() (n int)
func (*WhileContextDef) String ¶
func (m *WhileContextDef) String() string
func (*WhileContextDef) Unmarshal ¶
func (m *WhileContextDef) Unmarshal(dAtA []byte) error
type WorkerServiceClient ¶
type WorkerServiceClient interface {
// See worker.proto for details.
GetStatus(ctx context.Context, in *GetStatusRequest, opts ...grpc.CallOption) (*GetStatusResponse, error)
// See worker.proto for details.
CreateWorkerSession(ctx context.Context, in *CreateWorkerSessionRequest, opts ...grpc.CallOption) (*CreateWorkerSessionResponse, error)
// See worker.proto for details.
RegisterGraph(ctx context.Context, in *RegisterGraphRequest, opts ...grpc.CallOption) (*RegisterGraphResponse, error)
// See worker.proto for details.
DeregisterGraph(ctx context.Context, in *DeregisterGraphRequest, opts ...grpc.CallOption) (*DeregisterGraphResponse, error)
// See worker.proto for details.
RunGraph(ctx context.Context, in *RunGraphRequest, opts ...grpc.CallOption) (*RunGraphResponse, error)
// See worker.proto for details.
CleanupGraph(ctx context.Context, in *CleanupGraphRequest, opts ...grpc.CallOption) (*CleanupGraphResponse, error)
// See worker.proto for details.
CleanupAll(ctx context.Context, in *CleanupAllRequest, opts ...grpc.CallOption) (*CleanupAllResponse, error)
// See worker.proto for details.
RecvTensor(ctx context.Context, in *RecvTensorRequest, opts ...grpc.CallOption) (*RecvTensorResponse, error)
// See worker.proto for details.
Logging(ctx context.Context, in *LoggingRequest, opts ...grpc.CallOption) (*LoggingResponse, error)
// See worker.proto for details.
Tracing(ctx context.Context, in *TracingRequest, opts ...grpc.CallOption) (*TracingResponse, error)
}
func NewWorkerServiceClient ¶
func NewWorkerServiceClient(cc *grpc.ClientConn) WorkerServiceClient
type WorkerServiceServer ¶
type WorkerServiceServer interface {
// See worker.proto for details.
GetStatus(context.Context, *GetStatusRequest) (*GetStatusResponse, error)
// See worker.proto for details.
CreateWorkerSession(context.Context, *CreateWorkerSessionRequest) (*CreateWorkerSessionResponse, error)
// See worker.proto for details.
RegisterGraph(context.Context, *RegisterGraphRequest) (*RegisterGraphResponse, error)
// See worker.proto for details.
DeregisterGraph(context.Context, *DeregisterGraphRequest) (*DeregisterGraphResponse, error)
// See worker.proto for details.
RunGraph(context.Context, *RunGraphRequest) (*RunGraphResponse, error)
// See worker.proto for details.
CleanupGraph(context.Context, *CleanupGraphRequest) (*CleanupGraphResponse, error)
// See worker.proto for details.
CleanupAll(context.Context, *CleanupAllRequest) (*CleanupAllResponse, error)
// See worker.proto for details.
RecvTensor(context.Context, *RecvTensorRequest) (*RecvTensorResponse, error)
// See worker.proto for details.
Logging(context.Context, *LoggingRequest) (*LoggingResponse, error)
// See worker.proto for details.
Tracing(context.Context, *TracingRequest) (*TracingResponse, error)
}
Source Files
¶
- allocation_description.pb.go
- attr_value.pb.go
- builtin_models_static.go
- checkpoint.go
- cluster.pb.go
- config.pb.go
- control_flow.pb.go
- cost_graph.pb.go
- debug.pb.go
- device_attributes.pb.go
- device_properties.pb.go
- error_codes.pb.go
- function.pb.go
- graph.pb.go
- graph_transfer_info.pb.go
- init.go
- kernel_def.pb.go
- log_memory.pb.go
- master.pb.go
- master_service.pb.go
- meta_graph.pb.go
- named_tensor.pb.go
- node_def.pb.go
- op_def.pb.go
- op_gen_overrides.pb.go
- queue_runner.pb.go
- reader_base.pb.go
- register.go
- remote_fused_graph_execute_info.pb.go
- resource_handle.pb.go
- rewriter_config.pb.go
- saved_model.pb.go
- saver.pb.go
- step_stats.pb.go
- summary.pb.go
- tensor.pb.go
- tensor_bundle.pb.go
- tensor_description.pb.go
- tensor_shape.pb.go
- tensor_slice.pb.go
- tensorflow_server.pb.go
- types.pb.go
- variable.pb.go
- versions.pb.go
- worker.pb.go
- worker_service.pb.go