tensorflow

package
v0.0.0-...-9e721c0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Mar 1, 2020 License: Apache-2.0 Imports: 3 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var DataType_name = map[int32]string{
	0:   "DT_INVALID",
	1:   "DT_FLOAT",
	2:   "DT_DOUBLE",
	3:   "DT_INT32",
	4:   "DT_UINT8",
	5:   "DT_INT16",
	6:   "DT_INT8",
	7:   "DT_STRING",
	8:   "DT_COMPLEX64",
	9:   "DT_INT64",
	10:  "DT_BOOL",
	11:  "DT_QINT8",
	12:  "DT_QUINT8",
	13:  "DT_QINT32",
	14:  "DT_BFLOAT16",
	15:  "DT_QINT16",
	16:  "DT_QUINT16",
	17:  "DT_UINT16",
	18:  "DT_COMPLEX128",
	19:  "DT_HALF",
	20:  "DT_RESOURCE",
	21:  "DT_VARIANT",
	22:  "DT_UINT32",
	23:  "DT_UINT64",
	101: "DT_FLOAT_REF",
	102: "DT_DOUBLE_REF",
	103: "DT_INT32_REF",
	104: "DT_UINT8_REF",
	105: "DT_INT16_REF",
	106: "DT_INT8_REF",
	107: "DT_STRING_REF",
	108: "DT_COMPLEX64_REF",
	109: "DT_INT64_REF",
	110: "DT_BOOL_REF",
	111: "DT_QINT8_REF",
	112: "DT_QUINT8_REF",
	113: "DT_QINT32_REF",
	114: "DT_BFLOAT16_REF",
	115: "DT_QINT16_REF",
	116: "DT_QUINT16_REF",
	117: "DT_UINT16_REF",
	118: "DT_COMPLEX128_REF",
	119: "DT_HALF_REF",
	120: "DT_RESOURCE_REF",
	121: "DT_VARIANT_REF",
	122: "DT_UINT32_REF",
	123: "DT_UINT64_REF",
}
View Source 
var DataType_value = map[string]int32{
	"DT_INVALID":        0,
	"DT_FLOAT":          1,
	"DT_DOUBLE":         2,
	"DT_INT32":          3,
	"DT_UINT8":          4,
	"DT_INT16":          5,
	"DT_INT8":           6,
	"DT_STRING":         7,
	"DT_COMPLEX64":      8,
	"DT_INT64":          9,
	"DT_BOOL":           10,
	"DT_QINT8":          11,
	"DT_QUINT8":         12,
	"DT_QINT32":         13,
	"DT_BFLOAT16":       14,
	"DT_QINT16":         15,
	"DT_QUINT16":        16,
	"DT_UINT16":         17,
	"DT_COMPLEX128":     18,
	"DT_HALF":           19,
	"DT_RESOURCE":       20,
	"DT_VARIANT":        21,
	"DT_UINT32":         22,
	"DT_UINT64":         23,
	"DT_FLOAT_REF":      101,
	"DT_DOUBLE_REF":     102,
	"DT_INT32_REF":      103,
	"DT_UINT8_REF":      104,
	"DT_INT16_REF":      105,
	"DT_INT8_REF":       106,
	"DT_STRING_REF":     107,
	"DT_COMPLEX64_REF":  108,
	"DT_INT64_REF":      109,
	"DT_BOOL_REF":       110,
	"DT_QINT8_REF":      111,
	"DT_QUINT8_REF":     112,
	"DT_QINT32_REF":     113,
	"DT_BFLOAT16_REF":   114,
	"DT_QINT16_REF":     115,
	"DT_QUINT16_REF":    116,
	"DT_UINT16_REF":     117,
	"DT_COMPLEX128_REF": 118,
	"DT_HALF_REF":       119,
	"DT_RESOURCE_REF":   120,
	"DT_VARIANT_REF":    121,
	"DT_UINT32_REF":     122,
	"DT_UINT64_REF":     123,
}
View Source 
var VariableAggregation_name = map[int32]string{
	0: "VARIABLE_AGGREGATION_NONE",
	1: "VARIABLE_AGGREGATION_SUM",
	2: "VARIABLE_AGGREGATION_MEAN",
	3: "VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA",
}
View Source 
var VariableAggregation_value = map[string]int32{
	"VARIABLE_AGGREGATION_NONE":               0,
	"VARIABLE_AGGREGATION_SUM":                1,
	"VARIABLE_AGGREGATION_MEAN":               2,
	"VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA": 3,
}
View Source 
var VariableSynchronization_name = map[int32]string{
	0: "VARIABLE_SYNCHRONIZATION_AUTO",
	1: "VARIABLE_SYNCHRONIZATION_NONE",
	2: "VARIABLE_SYNCHRONIZATION_ON_WRITE",
	3: "VARIABLE_SYNCHRONIZATION_ON_READ",
}
View Source 
var VariableSynchronization_value = map[string]int32{
	"VARIABLE_SYNCHRONIZATION_AUTO":     0,
	"VARIABLE_SYNCHRONIZATION_NONE":     1,
	"VARIABLE_SYNCHRONIZATION_ON_WRITE": 2,
	"VARIABLE_SYNCHRONIZATION_ON_READ":  3,
}

Functions

This section is empty.

Types

type AllocationDescription 

type AllocationDescription struct {
	// Total number of bytes requested
	RequestedBytes int64 `protobuf:"varint,1,opt,name=requested_bytes,json=requestedBytes,proto3" json:"requested_bytes,omitempty"`
	// Total number of bytes allocated if known
	AllocatedBytes int64 `protobuf:"varint,2,opt,name=allocated_bytes,json=allocatedBytes,proto3" json:"allocated_bytes,omitempty"`
	// Name of the allocator used
	AllocatorName string `protobuf:"bytes,3,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
	// Identifier of the allocated buffer if known
	AllocationId int64 `protobuf:"varint,4,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
	// Set if this tensor only has one remaining reference
	HasSingleReference bool `protobuf:"varint,5,opt,name=has_single_reference,json=hasSingleReference,proto3" json:"has_single_reference,omitempty"`
	// Address of the allocation.
	Ptr                  uint64   `protobuf:"varint,6,opt,name=ptr,proto3" json:"ptr,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

func (*AllocationDescription) Descriptor 

func (*AllocationDescription) Descriptor() ([]byte, []int)

func (*AllocationDescription) GetAllocatedBytes 

func (m *AllocationDescription) GetAllocatedBytes() int64

func (*AllocationDescription) GetAllocationId 

func (m *AllocationDescription) GetAllocationId() int64

func (*AllocationDescription) GetAllocatorName 

func (m *AllocationDescription) GetAllocatorName() string

func (*AllocationDescription) GetHasSingleReference 

func (m *AllocationDescription) GetHasSingleReference() bool

func (*AllocationDescription) GetPtr 

func (m *AllocationDescription) GetPtr() uint64

func (*AllocationDescription) GetRequestedBytes 

func (m *AllocationDescription) GetRequestedBytes() int64

func (*AllocationDescription) ProtoMessage 

func (*AllocationDescription) ProtoMessage()

func (*AllocationDescription) Reset 

func (m *AllocationDescription) Reset()

func (*AllocationDescription) String 

func (m *AllocationDescription) String() string

func (*AllocationDescription) XXX_DiscardUnknown 

func (m *AllocationDescription) XXX_DiscardUnknown()

func (*AllocationDescription) XXX_Marshal 

func (m *AllocationDescription) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*AllocationDescription) XXX_Merge 

func (m *AllocationDescription) XXX_Merge(src proto.Message)

func (*AllocationDescription) XXX_Size 

func (m *AllocationDescription) XXX_Size() int

func (*AllocationDescription) XXX_Unmarshal 

func (m *AllocationDescription) XXX_Unmarshal(b []byte) error

type AllocationRecord 

type AllocationRecord struct {
	// The timestamp of the operation.
	AllocMicros int64 `protobuf:"varint,1,opt,name=alloc_micros,json=allocMicros,proto3" json:"alloc_micros,omitempty"`
	// Number of bytes allocated, or de-allocated if negative.
	AllocBytes           int64    `protobuf:"varint,2,opt,name=alloc_bytes,json=allocBytes,proto3" json:"alloc_bytes,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

An allocation/de-allocation operation performed by the allocator.

func (*AllocationRecord) Descriptor 

func (*AllocationRecord) Descriptor() ([]byte, []int)

func (*AllocationRecord) GetAllocBytes 

func (m *AllocationRecord) GetAllocBytes() int64

func (*AllocationRecord) GetAllocMicros 

func (m *AllocationRecord) GetAllocMicros() int64

func (*AllocationRecord) ProtoMessage 

func (*AllocationRecord) ProtoMessage()

func (*AllocationRecord) Reset 

func (m *AllocationRecord) Reset()

func (*AllocationRecord) String 

func (m *AllocationRecord) String() string

func (*AllocationRecord) XXX_DiscardUnknown 

func (m *AllocationRecord) XXX_DiscardUnknown()

func (*AllocationRecord) XXX_Marshal 

func (m *AllocationRecord) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*AllocationRecord) XXX_Merge 

func (m *AllocationRecord) XXX_Merge(src proto.Message)

func (*AllocationRecord) XXX_Size 

func (m *AllocationRecord) XXX_Size() int

func (*AllocationRecord) XXX_Unmarshal 

func (m *AllocationRecord) XXX_Unmarshal(b []byte) error

type AllocatorMemoryUsed 

type AllocatorMemoryUsed struct {
	AllocatorName string `protobuf:"bytes,1,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
	// These are per-node allocator memory stats.
	TotalBytes int64 `protobuf:"varint,2,opt,name=total_bytes,json=totalBytes,proto3" json:"total_bytes,omitempty"`
	PeakBytes  int64 `protobuf:"varint,3,opt,name=peak_bytes,json=peakBytes,proto3" json:"peak_bytes,omitempty"`
	// The bytes that are not deallocated.
	LiveBytes int64 `protobuf:"varint,4,opt,name=live_bytes,json=liveBytes,proto3" json:"live_bytes,omitempty"`
	// The allocation and deallocation timeline.
	AllocationRecords []*AllocationRecord `protobuf:"bytes,6,rep,name=allocation_records,json=allocationRecords,proto3" json:"allocation_records,omitempty"`
	// These are snapshots of the overall allocator memory stats.
	// The number of live bytes currently allocated by the allocator.
	AllocatorBytesInUse  int64    `protobuf:"varint,5,opt,name=allocator_bytes_in_use,json=allocatorBytesInUse,proto3" json:"allocator_bytes_in_use,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

func (*AllocatorMemoryUsed) Descriptor 

func (*AllocatorMemoryUsed) Descriptor() ([]byte, []int)

func (*AllocatorMemoryUsed) GetAllocationRecords 

func (m *AllocatorMemoryUsed) GetAllocationRecords() []*AllocationRecord

func (*AllocatorMemoryUsed) GetAllocatorBytesInUse 

func (m *AllocatorMemoryUsed) GetAllocatorBytesInUse() int64

func (*AllocatorMemoryUsed) GetAllocatorName 

func (m *AllocatorMemoryUsed) GetAllocatorName() string

func (*AllocatorMemoryUsed) GetLiveBytes 

func (m *AllocatorMemoryUsed) GetLiveBytes() int64

func (*AllocatorMemoryUsed) GetPeakBytes 

func (m *AllocatorMemoryUsed) GetPeakBytes() int64

func (*AllocatorMemoryUsed) GetTotalBytes 

func (m *AllocatorMemoryUsed) GetTotalBytes() int64

func (*AllocatorMemoryUsed) ProtoMessage 

func (*AllocatorMemoryUsed) ProtoMessage()

func (*AllocatorMemoryUsed) Reset 

func (m *AllocatorMemoryUsed) Reset()

func (*AllocatorMemoryUsed) String 

func (m *AllocatorMemoryUsed) String() string

func (*AllocatorMemoryUsed) XXX_DiscardUnknown 

func (m *AllocatorMemoryUsed) XXX_DiscardUnknown()

func (*AllocatorMemoryUsed) XXX_Marshal 

func (m *AllocatorMemoryUsed) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*AllocatorMemoryUsed) XXX_Merge 

func (m *AllocatorMemoryUsed) XXX_Merge(src proto.Message)

func (*AllocatorMemoryUsed) XXX_Size 

func (m *AllocatorMemoryUsed) XXX_Size() int

func (*AllocatorMemoryUsed) XXX_Unmarshal 

func (m *AllocatorMemoryUsed) XXX_Unmarshal(b []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"`
	XXX_NoUnkeyedLiteral struct{}          `json:"-"`
	XXX_unrecognized     []byte            `json:"-"`
	XXX_sizecache        int32             `json:"-"`
}

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) Descriptor() ([]byte, []int)

func (*AttrValue) GetB 

func (m *AttrValue) GetB() bool

func (*AttrValue) GetF 

func (m *AttrValue) GetF() float32

func (*AttrValue) GetFunc 

func (m *AttrValue) GetFunc() *NameAttrList

func (*AttrValue) GetI 

func (m *AttrValue) GetI() int64

func (*AttrValue) GetList 

func (m *AttrValue) GetList() *AttrValue_ListValue

func (*AttrValue) GetPlaceholder 

func (m *AttrValue) GetPlaceholder() string

func (*AttrValue) GetS 

func (m *AttrValue) GetS() []byte

func (*AttrValue) GetShape 

func (m *AttrValue) GetShape() *TensorShapeProto

func (*AttrValue) GetTensor 

func (m *AttrValue) GetTensor() *TensorProto

func (*AttrValue) GetType 

func (m *AttrValue) GetType() DataType

func (*AttrValue) GetValue 

func (m *AttrValue) GetValue() isAttrValue_Value

func (*AttrValue) ProtoMessage 

func (*AttrValue) ProtoMessage()

func (*AttrValue) Reset 

func (m *AttrValue) Reset()

func (*AttrValue) String 

func (m *AttrValue) String() string

func (*AttrValue) XXX_DiscardUnknown 

func (m *AttrValue) XXX_DiscardUnknown()

func (*AttrValue) XXX_Marshal 

func (m *AttrValue) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*AttrValue) XXX_Merge 

func (m *AttrValue) XXX_Merge(src proto.Message)

func (*AttrValue) XXX_OneofWrappers 

func (*AttrValue) XXX_OneofWrappers() []interface{}

XXX_OneofWrappers is for the internal use of the proto package.

func (*AttrValue) XXX_Size 

func (m *AttrValue) XXX_Size() int

func (*AttrValue) XXX_Unmarshal 

func (m *AttrValue) XXX_Unmarshal(b []byte) error

type AttrValue_B 

type AttrValue_B struct {
	B bool `protobuf:"varint,5,opt,name=b,proto3,oneof"`
}

type AttrValue_F 

type AttrValue_F struct {
	F float32 `protobuf:"fixed32,4,opt,name=f,proto3,oneof"`
}

type AttrValue_Func 

type AttrValue_Func struct {
	Func *NameAttrList `protobuf:"bytes,10,opt,name=func,proto3,oneof"`
}

type AttrValue_I 

type AttrValue_I struct {
	I int64 `protobuf:"varint,3,opt,name=i,proto3,oneof"`
}

type AttrValue_List 

type AttrValue_List struct {
	List *AttrValue_ListValue `protobuf:"bytes,1,opt,name=list,proto3,oneof"`
}

type AttrValue_ListValue 

type AttrValue_ListValue struct {
	S                    [][]byte            `protobuf:"bytes,2,rep,name=s,proto3" json:"s,omitempty"`
	I                    []int64             `protobuf:"varint,3,rep,packed,name=i,proto3" json:"i,omitempty"`
	F                    []float32           `protobuf:"fixed32,4,rep,packed,name=f,proto3" json:"f,omitempty"`
	B                    []bool              `protobuf:"varint,5,rep,packed,name=b,proto3" json:"b,omitempty"`
	Type                 []DataType          `protobuf:"varint,6,rep,packed,name=type,proto3,enum=tensorflow.DataType" json:"type,omitempty"`
	Shape                []*TensorShapeProto `protobuf:"bytes,7,rep,name=shape,proto3" json:"shape,omitempty"`
	Tensor               []*TensorProto      `protobuf:"bytes,8,rep,name=tensor,proto3" json:"tensor,omitempty"`
	Func                 []*NameAttrList     `protobuf:"bytes,9,rep,name=func,proto3" json:"func,omitempty"`
	XXX_NoUnkeyedLiteral struct{}            `json:"-"`
	XXX_unrecognized     []byte              `json:"-"`
	XXX_sizecache        int32               `json:"-"`
}

LINT.IfChange

func (*AttrValue_ListValue) Descriptor 

func (*AttrValue_ListValue) Descriptor() ([]byte, []int)

func (*AttrValue_ListValue) GetB 

func (m *AttrValue_ListValue) GetB() []bool

func (*AttrValue_ListValue) GetF 

func (m *AttrValue_ListValue) GetF() []float32

func (*AttrValue_ListValue) GetFunc 

func (m *AttrValue_ListValue) GetFunc() []*NameAttrList

func (*AttrValue_ListValue) GetI 

func (m *AttrValue_ListValue) GetI() []int64

func (*AttrValue_ListValue) GetS 

func (m *AttrValue_ListValue) GetS() [][]byte

func (*AttrValue_ListValue) GetShape 

func (m *AttrValue_ListValue) GetShape() []*TensorShapeProto

func (*AttrValue_ListValue) GetTensor 

func (m *AttrValue_ListValue) GetTensor() []*TensorProto

func (*AttrValue_ListValue) GetType 

func (m *AttrValue_ListValue) GetType() []DataType

func (*AttrValue_ListValue) ProtoMessage 

func (*AttrValue_ListValue) ProtoMessage()

func (*AttrValue_ListValue) Reset 

func (m *AttrValue_ListValue) Reset()

func (*AttrValue_ListValue) String 

func (m *AttrValue_ListValue) String() string

func (*AttrValue_ListValue) XXX_DiscardUnknown 

func (m *AttrValue_ListValue) XXX_DiscardUnknown()

func (*AttrValue_ListValue) XXX_Marshal 

func (m *AttrValue_ListValue) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*AttrValue_ListValue) XXX_Merge 

func (m *AttrValue_ListValue) XXX_Merge(src proto.Message)

func (*AttrValue_ListValue) XXX_Size 

func (m *AttrValue_ListValue) XXX_Size() int

func (*AttrValue_ListValue) XXX_Unmarshal 

func (m *AttrValue_ListValue) XXX_Unmarshal(b []byte) error

type AttrValue_Placeholder 

type AttrValue_Placeholder struct {
	Placeholder string `protobuf:"bytes,9,opt,name=placeholder,proto3,oneof"`
}

type AttrValue_S 

type AttrValue_S struct {
	S []byte `protobuf:"bytes,2,opt,name=s,proto3,oneof"`
}

type AttrValue_Shape 

type AttrValue_Shape struct {
	Shape *TensorShapeProto `protobuf:"bytes,7,opt,name=shape,proto3,oneof"`
}

type AttrValue_Tensor 

type AttrValue_Tensor struct {
	Tensor *TensorProto `protobuf:"bytes,8,opt,name=tensor,proto3,oneof"`
}

type AttrValue_Type 

type AttrValue_Type struct {
	Type DataType `protobuf:"varint,6,opt,name=type,proto3,enum=tensorflow.DataType,oneof"`
}

type CostGraphDef 

type CostGraphDef struct {
	Node                 []*CostGraphDef_Node `protobuf:"bytes,1,rep,name=node,proto3" json:"node,omitempty"`
	XXX_NoUnkeyedLiteral struct{}             `json:"-"`
	XXX_unrecognized     []byte               `json:"-"`
	XXX_sizecache        int32                `json:"-"`
}

func (*CostGraphDef) Descriptor 

func (*CostGraphDef) Descriptor() ([]byte, []int)

func (*CostGraphDef) GetNode 

func (m *CostGraphDef) GetNode() []*CostGraphDef_Node

func (*CostGraphDef) ProtoMessage 

func (*CostGraphDef) ProtoMessage()

func (*CostGraphDef) Reset 

func (m *CostGraphDef) Reset()

func (*CostGraphDef) String 

func (m *CostGraphDef) String() string

func (*CostGraphDef) XXX_DiscardUnknown 

func (m *CostGraphDef) XXX_DiscardUnknown()

func (*CostGraphDef) XXX_Marshal 

func (m *CostGraphDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*CostGraphDef) XXX_Merge 

func (m *CostGraphDef) XXX_Merge(src proto.Message)

func (*CostGraphDef) XXX_Size 

func (m *CostGraphDef) XXX_Size() int

func (*CostGraphDef) XXX_Unmarshal 

func (m *CostGraphDef) XXX_Unmarshal(b []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,proto3" json:"input_info,omitempty"`
	OutputInfo []*CostGraphDef_Node_OutputInfo `protobuf:"bytes,5,rep,name=output_info,json=outputInfo,proto3" json:"output_info,omitempty"`
	// Temporary memory used by this node.
	TemporaryMemorySize int64 `protobuf:"varint,6,opt,name=temporary_memory_size,json=temporaryMemorySize,proto3" json:"temporary_memory_size,omitempty"`
	// Persistent memory used by this node.
	PersistentMemorySize int64 `protobuf:"varint,12,opt,name=persistent_memory_size,json=persistentMemorySize,proto3" json:"persistent_memory_size,omitempty"`
	HostTempMemorySize   int64 `protobuf:"varint,10,opt,name=host_temp_memory_size,json=hostTempMemorySize,proto3" json:"host_temp_memory_size,omitempty"` // Deprecated: Do not use.
	DeviceTempMemorySize int64 ``                                                                                                                          // Deprecated: Do not use.
	/* 127-byte string literal not displayed */
	DevicePersistentMemorySize int64 `` // Deprecated: Do not use.
	/* 145-byte string literal not displayed */
	// Estimate of the computational cost of this node, in microseconds.
	ComputeCost int64 `protobuf:"varint,9,opt,name=compute_cost,json=computeCost,proto3" json:"compute_cost,omitempty"`
	// Analytical estimate of the computational cost of this node, in
	// microseconds.
	ComputeTime int64 `protobuf:"varint,14,opt,name=compute_time,json=computeTime,proto3" json:"compute_time,omitempty"`
	// Analytical estimate of the memory access cost of this node, in
	// microseconds.
	MemoryTime int64 `protobuf:"varint,15,opt,name=memory_time,json=memoryTime,proto3" json:"memory_time,omitempty"`
	// If true, the output is permanent: it can't be discarded, because this
	// node is part of the "final output". Nodes may depend on final nodes.
	IsFinal bool `protobuf:"varint,7,opt,name=is_final,json=isFinal,proto3" json:"is_final,omitempty"`
	// Ids of the control inputs for this node.
	ControlInput []int32 `protobuf:"varint,8,rep,packed,name=control_input,json=controlInput,proto3" json:"control_input,omitempty"`
	// Are the costs inaccurate?
	Inaccurate           bool     `protobuf:"varint,17,opt,name=inaccurate,proto3" json:"inaccurate,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

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 deprecated 

func (m *CostGraphDef_Node) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetDeviceTempMemorySize deprecated 

func (m *CostGraphDef_Node) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetHostTempMemorySize deprecated 

func (m *CostGraphDef_Node) GetHostTempMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetId 

func (m *CostGraphDef_Node) GetId() int32

func (*CostGraphDef_Node) GetInaccurate 

func (m *CostGraphDef_Node) GetInaccurate() bool

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) GetPersistentMemorySize 

func (m *CostGraphDef_Node) GetPersistentMemorySize() int64

func (*CostGraphDef_Node) GetTemporaryMemorySize 

func (m *CostGraphDef_Node) GetTemporaryMemorySize() int64

func (*CostGraphDef_Node) ProtoMessage 

func (*CostGraphDef_Node) ProtoMessage()

func (*CostGraphDef_Node) Reset 

func (m *CostGraphDef_Node) Reset()

func (*CostGraphDef_Node) String 

func (m *CostGraphDef_Node) String() string

func (*CostGraphDef_Node) XXX_DiscardUnknown 

func (m *CostGraphDef_Node) XXX_DiscardUnknown()

func (*CostGraphDef_Node) XXX_Marshal 

func (m *CostGraphDef_Node) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*CostGraphDef_Node) XXX_Merge 

func (m *CostGraphDef_Node) XXX_Merge(src proto.Message)

func (*CostGraphDef_Node) XXX_Size 

func (m *CostGraphDef_Node) XXX_Size() int

func (*CostGraphDef_Node) XXX_Unmarshal 

func (m *CostGraphDef_Node) XXX_Unmarshal(b []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"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

Inputs of this node. They must be executed before this node can be executed. An input is a particular output of another node, specified by the node id and the output index.

func (*CostGraphDef_Node_InputInfo) Descriptor 

func (*CostGraphDef_Node_InputInfo) Descriptor() ([]byte, []int)

func (*CostGraphDef_Node_InputInfo) GetPrecedingNode 

func (m *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32

func (*CostGraphDef_Node_InputInfo) GetPrecedingPort 

func (m *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32

func (*CostGraphDef_Node_InputInfo) ProtoMessage 

func (*CostGraphDef_Node_InputInfo) ProtoMessage()

func (*CostGraphDef_Node_InputInfo) Reset 

func (m *CostGraphDef_Node_InputInfo) Reset()

func (*CostGraphDef_Node_InputInfo) String 

func (m *CostGraphDef_Node_InputInfo) String() string

func (*CostGraphDef_Node_InputInfo) XXX_DiscardUnknown 

func (m *CostGraphDef_Node_InputInfo) XXX_DiscardUnknown()

func (*CostGraphDef_Node_InputInfo) XXX_Marshal 

func (m *CostGraphDef_Node_InputInfo) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*CostGraphDef_Node_InputInfo) XXX_Merge 

func (m *CostGraphDef_Node_InputInfo) XXX_Merge(src proto.Message)

func (*CostGraphDef_Node_InputInfo) XXX_Size 

func (m *CostGraphDef_Node_InputInfo) XXX_Size() int

func (*CostGraphDef_Node_InputInfo) XXX_Unmarshal 

func (m *CostGraphDef_Node_InputInfo) XXX_Unmarshal(b []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,proto3" json:"shape,omitempty"`
	Dtype                DataType          `protobuf:"varint,4,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	XXX_NoUnkeyedLiteral struct{}          `json:"-"`
	XXX_unrecognized     []byte            `json:"-"`
	XXX_sizecache        int32             `json:"-"`
}

Outputs of this node.

func (*CostGraphDef_Node_OutputInfo) Descriptor 

func (*CostGraphDef_Node_OutputInfo) Descriptor() ([]byte, []int)

func (*CostGraphDef_Node_OutputInfo) GetAliasInputPort 

func (m *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64

func (*CostGraphDef_Node_OutputInfo) GetDtype 

func (m *CostGraphDef_Node_OutputInfo) GetDtype() DataType

func (*CostGraphDef_Node_OutputInfo) GetShape 

func (m *CostGraphDef_Node_OutputInfo) GetShape() *TensorShapeProto

func (*CostGraphDef_Node_OutputInfo) GetSize 

func (m *CostGraphDef_Node_OutputInfo) GetSize() int64

func (*CostGraphDef_Node_OutputInfo) ProtoMessage 

func (*CostGraphDef_Node_OutputInfo) ProtoMessage()

func (*CostGraphDef_Node_OutputInfo) Reset 

func (m *CostGraphDef_Node_OutputInfo) Reset()

func (*CostGraphDef_Node_OutputInfo) String 

func (m *CostGraphDef_Node_OutputInfo) String() string

func (*CostGraphDef_Node_OutputInfo) XXX_DiscardUnknown 

func (m *CostGraphDef_Node_OutputInfo) XXX_DiscardUnknown()

func (*CostGraphDef_Node_OutputInfo) XXX_Marshal 

func (m *CostGraphDef_Node_OutputInfo) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*CostGraphDef_Node_OutputInfo) XXX_Merge 

func (m *CostGraphDef_Node_OutputInfo) XXX_Merge(src proto.Message)

func (*CostGraphDef_Node_OutputInfo) XXX_Size 

func (m *CostGraphDef_Node_OutputInfo) XXX_Size() int

func (*CostGraphDef_Node_OutputInfo) XXX_Unmarshal 

func (m *CostGraphDef_Node_OutputInfo) XXX_Unmarshal(b []byte) error

type DataType 

type DataType int32

(== suppress_warning documentation-presence ==) LINT.IfChange 

const (
	// Not a legal value for DataType.  Used to indicate a DataType field
	// has not been set.
	DataType_DT_INVALID DataType = 0
	// Data types that all computation devices are expected to be
	// capable to support.
	DataType_DT_FLOAT      DataType = 1
	DataType_DT_DOUBLE     DataType = 2
	DataType_DT_INT32      DataType = 3
	DataType_DT_UINT8      DataType = 4
	DataType_DT_INT16      DataType = 5
	DataType_DT_INT8       DataType = 6
	DataType_DT_STRING     DataType = 7
	DataType_DT_COMPLEX64  DataType = 8
	DataType_DT_INT64      DataType = 9
	DataType_DT_BOOL       DataType = 10
	DataType_DT_QINT8      DataType = 11
	DataType_DT_QUINT8     DataType = 12
	DataType_DT_QINT32     DataType = 13
	DataType_DT_BFLOAT16   DataType = 14
	DataType_DT_QINT16     DataType = 15
	DataType_DT_QUINT16    DataType = 16
	DataType_DT_UINT16     DataType = 17
	DataType_DT_COMPLEX128 DataType = 18
	DataType_DT_HALF       DataType = 19
	DataType_DT_RESOURCE   DataType = 20
	DataType_DT_VARIANT    DataType = 21
	DataType_DT_UINT32     DataType = 22
	DataType_DT_UINT64     DataType = 23
	// Do not use!  These are only for parameters.  Every enum above
	// should have a corresponding value below (verified by types_test).
	DataType_DT_FLOAT_REF      DataType = 101
	DataType_DT_DOUBLE_REF     DataType = 102
	DataType_DT_INT32_REF      DataType = 103
	DataType_DT_UINT8_REF      DataType = 104
	DataType_DT_INT16_REF      DataType = 105
	DataType_DT_INT8_REF       DataType = 106
	DataType_DT_STRING_REF     DataType = 107
	DataType_DT_COMPLEX64_REF  DataType = 108
	DataType_DT_INT64_REF      DataType = 109
	DataType_DT_BOOL_REF       DataType = 110
	DataType_DT_QINT8_REF      DataType = 111
	DataType_DT_QUINT8_REF     DataType = 112
	DataType_DT_QINT32_REF     DataType = 113
	DataType_DT_BFLOAT16_REF   DataType = 114
	DataType_DT_QINT16_REF     DataType = 115
	DataType_DT_QUINT16_REF    DataType = 116
	DataType_DT_UINT16_REF     DataType = 117
	DataType_DT_COMPLEX128_REF DataType = 118
	DataType_DT_HALF_REF       DataType = 119
	DataType_DT_RESOURCE_REF   DataType = 120
	DataType_DT_VARIANT_REF    DataType = 121
	DataType_DT_UINT32_REF     DataType = 122
	DataType_DT_UINT64_REF     DataType = 123
)

func (DataType) EnumDescriptor 

func (DataType) EnumDescriptor() ([]byte, []int)

func (DataType) String 

func (x DataType) String() string

type DeviceStepStats 

type DeviceStepStats struct {
	Device    string           `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"`
	NodeStats []*NodeExecStats `protobuf:"bytes,2,rep,name=node_stats,json=nodeStats,proto3" json:"node_stats,omitempty"`
	// Its key is thread id.
	ThreadNames          map[uint32]string `` /* 183-byte string literal not displayed */
	XXX_NoUnkeyedLiteral struct{}          `json:"-"`
	XXX_unrecognized     []byte            `json:"-"`
	XXX_sizecache        int32             `json:"-"`
}

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) GetThreadNames 

func (m *DeviceStepStats) GetThreadNames() map[uint32]string

func (*DeviceStepStats) ProtoMessage 

func (*DeviceStepStats) ProtoMessage()

func (*DeviceStepStats) Reset 

func (m *DeviceStepStats) Reset()

func (*DeviceStepStats) String 

func (m *DeviceStepStats) String() string

func (*DeviceStepStats) XXX_DiscardUnknown 

func (m *DeviceStepStats) XXX_DiscardUnknown()

func (*DeviceStepStats) XXX_Marshal 

func (m *DeviceStepStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*DeviceStepStats) XXX_Merge 

func (m *DeviceStepStats) XXX_Merge(src proto.Message)

func (*DeviceStepStats) XXX_Size 

func (m *DeviceStepStats) XXX_Size() int

func (*DeviceStepStats) XXX_Unmarshal 

func (m *DeviceStepStats) XXX_Unmarshal(b []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,proto3" json:"signature,omitempty"`
	// Attributes specific to this function definition.
	Attr    map[string]*AttrValue            `` /* 149-byte string literal not displayed */
	ArgAttr map[uint32]*FunctionDef_ArgAttrs `` /* 171-byte string literal not displayed */
	// By convention, "op" in node_def is resolved by consulting with a
	// user-defined library first. If not resolved, "func" is assumed to
	// be a builtin op.
	NodeDef []*NodeDef `protobuf:"bytes,3,rep,name=node_def,json=nodeDef,proto3" json:"node_def,omitempty"`
	// A mapping from the output arg names from `signature` to the
	// outputs from `node_def` that should be returned by the function.
	Ret map[string]string `` /* 147-byte string literal not displayed */
	// A mapping from control output names from `signature` to node names in
	// `node_def` which should be control outputs of this function.
	ControlRet           map[string]string `` /* 179-byte string literal not displayed */
	XXX_NoUnkeyedLiteral struct{}          `json:"-"`
	XXX_unrecognized     []byte            `json:"-"`
	XXX_sizecache        int32             `json:"-"`
}

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) GetArgAttr 

func (m *FunctionDef) GetArgAttr() map[uint32]*FunctionDef_ArgAttrs

func (*FunctionDef) GetAttr 

func (m *FunctionDef) GetAttr() map[string]*AttrValue

func (*FunctionDef) GetControlRet 

func (m *FunctionDef) GetControlRet() map[string]string

func (*FunctionDef) GetNodeDef 

func (m *FunctionDef) GetNodeDef() []*NodeDef

func (*FunctionDef) GetRet 

func (m *FunctionDef) GetRet() map[string]string

func (*FunctionDef) GetSignature 

func (m *FunctionDef) GetSignature() *OpDef

func (*FunctionDef) ProtoMessage 

func (*FunctionDef) ProtoMessage()

func (*FunctionDef) Reset 

func (m *FunctionDef) Reset()

func (*FunctionDef) String 

func (m *FunctionDef) String() string

func (*FunctionDef) XXX_DiscardUnknown 

func (m *FunctionDef) XXX_DiscardUnknown()

func (*FunctionDef) XXX_Marshal 

func (m *FunctionDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*FunctionDef) XXX_Merge 

func (m *FunctionDef) XXX_Merge(src proto.Message)

func (*FunctionDef) XXX_Size 

func (m *FunctionDef) XXX_Size() int

func (*FunctionDef) XXX_Unmarshal 

func (m *FunctionDef) XXX_Unmarshal(b []byte) error

type FunctionDefLibrary 

type FunctionDefLibrary struct {
	Function             []*FunctionDef `protobuf:"bytes,1,rep,name=function,proto3" json:"function,omitempty"`
	Gradient             []*GradientDef `protobuf:"bytes,2,rep,name=gradient,proto3" json:"gradient,omitempty"`
	XXX_NoUnkeyedLiteral struct{}       `json:"-"`
	XXX_unrecognized     []byte         `json:"-"`
	XXX_sizecache        int32          `json:"-"`
}

A library is a set of named functions.

func (*FunctionDefLibrary) Descriptor 

func (*FunctionDefLibrary) Descriptor() ([]byte, []int)

func (*FunctionDefLibrary) GetFunction 

func (m *FunctionDefLibrary) GetFunction() []*FunctionDef

func (*FunctionDefLibrary) GetGradient 

func (m *FunctionDefLibrary) GetGradient() []*GradientDef

func (*FunctionDefLibrary) ProtoMessage 

func (*FunctionDefLibrary) ProtoMessage()

func (*FunctionDefLibrary) Reset 

func (m *FunctionDefLibrary) Reset()

func (*FunctionDefLibrary) String 

func (m *FunctionDefLibrary) String() string

func (*FunctionDefLibrary) XXX_DiscardUnknown 

func (m *FunctionDefLibrary) XXX_DiscardUnknown()

func (*FunctionDefLibrary) XXX_Marshal 

func (m *FunctionDefLibrary) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*FunctionDefLibrary) XXX_Merge 

func (m *FunctionDefLibrary) XXX_Merge(src proto.Message)

func (*FunctionDefLibrary) XXX_Size 

func (m *FunctionDefLibrary) XXX_Size() int

func (*FunctionDefLibrary) XXX_Unmarshal 

func (m *FunctionDefLibrary) XXX_Unmarshal(b []byte) error

type FunctionDef_ArgAttrs 

type FunctionDef_ArgAttrs struct {
	Attr                 map[string]*AttrValue `` /* 149-byte string literal not displayed */
	XXX_NoUnkeyedLiteral struct{}              `json:"-"`
	XXX_unrecognized     []byte                `json:"-"`
	XXX_sizecache        int32                 `json:"-"`
}

Attributes for function arguments. These attributes are the same set of valid attributes as to _Arg nodes.

func (*FunctionDef_ArgAttrs) Descriptor 

func (*FunctionDef_ArgAttrs) Descriptor() ([]byte, []int)

func (*FunctionDef_ArgAttrs) GetAttr 

func (m *FunctionDef_ArgAttrs) GetAttr() map[string]*AttrValue

func (*FunctionDef_ArgAttrs) ProtoMessage 

func (*FunctionDef_ArgAttrs) ProtoMessage()

func (*FunctionDef_ArgAttrs) Reset 

func (m *FunctionDef_ArgAttrs) Reset()

func (*FunctionDef_ArgAttrs) String 

func (m *FunctionDef_ArgAttrs) String() string

func (*FunctionDef_ArgAttrs) XXX_DiscardUnknown 

func (m *FunctionDef_ArgAttrs) XXX_DiscardUnknown()

func (*FunctionDef_ArgAttrs) XXX_Marshal 

func (m *FunctionDef_ArgAttrs) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*FunctionDef_ArgAttrs) XXX_Merge 

func (m *FunctionDef_ArgAttrs) XXX_Merge(src proto.Message)

func (*FunctionDef_ArgAttrs) XXX_Size 

func (m *FunctionDef_ArgAttrs) XXX_Size() int

func (*FunctionDef_ArgAttrs) XXX_Unmarshal 

func (m *FunctionDef_ArgAttrs) XXX_Unmarshal(b []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"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

GradientDef defines the gradient function of a function defined in a function library.

A gradient function g (specified by gradient_func) for a function f (specified by function_name) must follow the following:

The function 'f' must be a numerical function which takes N inputs and produces M outputs. Its gradient function 'g', which is a function taking N + M inputs and produces N outputs.

I.e. if we have 

(y1, y2, ..., y_M) = f(x1, x2, ..., x_N),

then, g is 

(dL/dx1, dL/dx2, ..., dL/dx_N) = g(x1, x2, ..., x_N,
                                  dL/dy1, dL/dy2, ..., dL/dy_M),

where L is a scalar-value function of (x1, x2, ..., xN) (e.g., the loss function). dL/dx_i is the partial derivative of L with respect to x_i.

func (*GradientDef) Descriptor 

func (*GradientDef) Descriptor() ([]byte, []int)

func (*GradientDef) GetFunctionName 

func (m *GradientDef) GetFunctionName() string

func (*GradientDef) GetGradientFunc 

func (m *GradientDef) GetGradientFunc() string

func (*GradientDef) ProtoMessage 

func (*GradientDef) ProtoMessage()

func (*GradientDef) Reset 

func (m *GradientDef) Reset()

func (*GradientDef) String 

func (m *GradientDef) String() string

func (*GradientDef) XXX_DiscardUnknown 

func (m *GradientDef) XXX_DiscardUnknown()

func (*GradientDef) XXX_Marshal 

func (m *GradientDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*GradientDef) XXX_Merge 

func (m *GradientDef) XXX_Merge(src proto.Message)

func (*GradientDef) XXX_Size 

func (m *GradientDef) XXX_Size() int

func (*GradientDef) XXX_Unmarshal 

func (m *GradientDef) XXX_Unmarshal(b []byte) error

type GraphDef 

type GraphDef struct {
	Node []*NodeDef `protobuf:"bytes,1,rep,name=node,proto3" json:"node,omitempty"`
	// Compatibility versions of the graph.  See core/public/version.h for version
	// history.  The GraphDef version is distinct from the TensorFlow version, and
	// each release of TensorFlow will support a range of GraphDef versions.
	Versions *VersionDef `protobuf:"bytes,4,opt,name=versions,proto3" json:"versions,omitempty"`
	// Deprecated single version field; use versions above instead.  Since all
	// GraphDef changes before "versions" was introduced were forward
	// compatible, this field is entirely ignored.
	Version int32 `protobuf:"varint,3,opt,name=version,proto3" json:"version,omitempty"` // Deprecated: Do not use.
	// EXPERIMENTAL. DO NOT USE OR DEPEND ON THIS YET.
	//
	// "library" provides user-defined functions.
	//
	// Naming:
	//   * library.function.name are in a flat namespace.
	//     NOTE: We may need to change it to be hierarchical to support
	//     different orgs. E.g.,
	//     { "/google/nn", { ... }},
	//     { "/google/vision", { ... }}
	//     { "/org_foo/module_bar", { ... }}
	//     map<string, FunctionDefLib> named_lib;
	//   * If node[i].op is the name of one function in "library",
	//     node[i] is deemed as a function call. Otherwise, node[i].op
	//     must be a primitive operation supported by the runtime.
	//
	//
	// Function call semantics:
	//
	//   * The callee may start execution as soon as some of its inputs
	//     are ready. The caller may want to use Tuple() mechanism to
	//     ensure all inputs are ready in the same time.
	//
	//   * The consumer of return values may start executing as soon as
	//     the return values the consumer depends on are ready.  The
	//     consumer may want to use Tuple() mechanism to ensure the
	//     consumer does not start until all return values of the callee
	//     function are ready.
	Library              *FunctionDefLibrary `protobuf:"bytes,2,opt,name=library,proto3" json:"library,omitempty"`
	XXX_NoUnkeyedLiteral struct{}            `json:"-"`
	XXX_unrecognized     []byte              `json:"-"`
	XXX_sizecache        int32               `json:"-"`
}

Represents the graph of operations

func (*GraphDef) Descriptor 

func (*GraphDef) Descriptor() ([]byte, []int)

func (*GraphDef) GetLibrary 

func (m *GraphDef) GetLibrary() *FunctionDefLibrary

func (*GraphDef) GetNode 

func (m *GraphDef) GetNode() []*NodeDef

func (*GraphDef) GetVersion deprecated 

func (m *GraphDef) GetVersion() int32

Deprecated: Do not use.

func (*GraphDef) GetVersions 

func (m *GraphDef) GetVersions() *VersionDef

func (*GraphDef) ProtoMessage 

func (*GraphDef) ProtoMessage()

func (*GraphDef) Reset 

func (m *GraphDef) Reset()

func (*GraphDef) String 

func (m *GraphDef) String() string

func (*GraphDef) XXX_DiscardUnknown 

func (m *GraphDef) XXX_DiscardUnknown()

func (*GraphDef) XXX_Marshal 

func (m *GraphDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*GraphDef) XXX_Merge 

func (m *GraphDef) XXX_Merge(src proto.Message)

func (*GraphDef) XXX_Size 

func (m *GraphDef) XXX_Size() int

func (*GraphDef) XXX_Unmarshal 

func (m *GraphDef) XXX_Unmarshal(b []byte) error

type MemoryStats 

type MemoryStats struct {
	TempMemorySize           int64   `protobuf:"varint,1,opt,name=temp_memory_size,json=tempMemorySize,proto3" json:"temp_memory_size,omitempty"`
	PersistentMemorySize     int64   `protobuf:"varint,3,opt,name=persistent_memory_size,json=persistentMemorySize,proto3" json:"persistent_memory_size,omitempty"`
	PersistentTensorAllocIds []int64 `` /* 145-byte string literal not displayed */
	DeviceTempMemorySize     int64   `` // Deprecated: Do not use.
	/* 126-byte string literal not displayed */
	DevicePersistentMemorySize int64 `` // Deprecated: Do not use.
	/* 144-byte string literal not displayed */
	DevicePersistentTensorAllocIds []int64 `` // Deprecated: Do not use.
	/* 165-byte string literal not displayed */
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

For memory tracking.

func (*MemoryStats) Descriptor 

func (*MemoryStats) Descriptor() ([]byte, []int)

func (*MemoryStats) GetDevicePersistentMemorySize deprecated 

func (m *MemoryStats) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetDevicePersistentTensorAllocIds deprecated 

func (m *MemoryStats) GetDevicePersistentTensorAllocIds() []int64

Deprecated: Do not use.

func (*MemoryStats) GetDeviceTempMemorySize deprecated 

func (m *MemoryStats) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetPersistentMemorySize 

func (m *MemoryStats) GetPersistentMemorySize() int64

func (*MemoryStats) GetPersistentTensorAllocIds 

func (m *MemoryStats) GetPersistentTensorAllocIds() []int64

func (*MemoryStats) GetTempMemorySize 

func (m *MemoryStats) GetTempMemorySize() int64

func (*MemoryStats) ProtoMessage 

func (*MemoryStats) ProtoMessage()

func (*MemoryStats) Reset 

func (m *MemoryStats) Reset()

func (*MemoryStats) String 

func (m *MemoryStats) String() string

func (*MemoryStats) XXX_DiscardUnknown 

func (m *MemoryStats) XXX_DiscardUnknown()

func (*MemoryStats) XXX_Marshal 

func (m *MemoryStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*MemoryStats) XXX_Merge 

func (m *MemoryStats) XXX_Merge(src proto.Message)

func (*MemoryStats) XXX_Size 

func (m *MemoryStats) XXX_Size() int

func (*MemoryStats) XXX_Unmarshal 

func (m *MemoryStats) XXX_Unmarshal(b []byte) error

type NameAttrList 

type NameAttrList struct {
	Name                 string                `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Attr                 map[string]*AttrValue `` /* 149-byte string literal not displayed */
	XXX_NoUnkeyedLiteral struct{}              `json:"-"`
	XXX_unrecognized     []byte                `json:"-"`
	XXX_sizecache        int32                 `json:"-"`
}

A list of attr names and their values. The whole list is attached with a string name. E.g., MatMul[T=float].

func (*NameAttrList) Descriptor 

func (*NameAttrList) Descriptor() ([]byte, []int)

func (*NameAttrList) GetAttr 

func (m *NameAttrList) GetAttr() map[string]*AttrValue

func (*NameAttrList) GetName 

func (m *NameAttrList) GetName() string

func (*NameAttrList) ProtoMessage 

func (*NameAttrList) ProtoMessage()

func (*NameAttrList) Reset 

func (m *NameAttrList) Reset()

func (*NameAttrList) String 

func (m *NameAttrList) String() string

func (*NameAttrList) XXX_DiscardUnknown 

func (m *NameAttrList) XXX_DiscardUnknown()

func (*NameAttrList) XXX_Marshal 

func (m *NameAttrList) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*NameAttrList) XXX_Merge 

func (m *NameAttrList) XXX_Merge(src proto.Message)

func (*NameAttrList) XXX_Size 

func (m *NameAttrList) XXX_Size() int

func (*NameAttrList) XXX_Unmarshal 

func (m *NameAttrList) XXX_Unmarshal(b []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,proto3" json:"input,omitempty"`
	// A (possibly partial) specification for the device on which this
	// node should be placed.
	// The expected syntax for this string is as follows:
	//
	// DEVICE_SPEC ::= PARTIAL_SPEC
	//
	// PARTIAL_SPEC ::= ("/" CONSTRAINT) *
	// CONSTRAINT ::= ("job:" JOB_NAME)
	//              | ("replica:" [1-9][0-9]*)
	//              | ("task:" [1-9][0-9]*)
	//              | ("device:" [A-Za-z]* ":" ([1-9][0-9]* | "*") )
	//
	// Valid values for this string include:
	// * "/job:worker/replica:0/task:1/device:GPU:3"  (full specification)
	// * "/job:worker/device:GPU:3"                   (partial specification)
	// * ""                                    (no specification)
	//
	// If the constraints do not resolve to a single device (or if this
	// field is empty or not present), the runtime will attempt to
	// choose a device automatically.
	Device string `protobuf:"bytes,4,opt,name=device,proto3" json:"device,omitempty"`
	// Operation-specific graph-construction-time configuration.
	// Note that this should include all attrs defined in the
	// corresponding OpDef, including those with a value matching
	// the default -- this allows the default to change and makes
	// NodeDefs easier to interpret on their own.  However, if
	// an attr with a default is not specified in this list, the
	// default will be used.
	// The "names" (keys) must match the regexp "[a-z][a-z0-9_]+" (and
	// one of the names from the corresponding OpDef's attr field).
	// The values must have a type matching the corresponding OpDef
	// attr's type field.
	// TODO(josh11b): Add some examples here showing best practices.
	Attr map[string]*AttrValue `` /* 149-byte string literal not displayed */
	// This stores debug information associated with the node.
	ExperimentalDebugInfo *NodeDef_ExperimentalDebugInfo `` /* 126-byte string literal not displayed */
	XXX_NoUnkeyedLiteral  struct{}                       `json:"-"`
	XXX_unrecognized      []byte                         `json:"-"`
	XXX_sizecache         int32                          `json:"-"`
}

func (*NodeDef) Descriptor 

func (*NodeDef) Descriptor() ([]byte, []int)

func (*NodeDef) GetAttr 

func (m *NodeDef) GetAttr() map[string]*AttrValue

func (*NodeDef) GetDevice 

func (m *NodeDef) GetDevice() string

func (*NodeDef) GetExperimentalDebugInfo 

func (m *NodeDef) GetExperimentalDebugInfo() *NodeDef_ExperimentalDebugInfo

func (*NodeDef) GetInput 

func (m *NodeDef) GetInput() []string

func (*NodeDef) GetName 

func (m *NodeDef) GetName() string

func (*NodeDef) GetOp 

func (m *NodeDef) GetOp() string

func (*NodeDef) ProtoMessage 

func (*NodeDef) ProtoMessage()

func (*NodeDef) Reset 

func (m *NodeDef) Reset()

func (*NodeDef) String 

func (m *NodeDef) String() string

func (*NodeDef) XXX_DiscardUnknown 

func (m *NodeDef) XXX_DiscardUnknown()

func (*NodeDef) XXX_Marshal 

func (m *NodeDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*NodeDef) XXX_Merge 

func (m *NodeDef) XXX_Merge(src proto.Message)

func (*NodeDef) XXX_Size 

func (m *NodeDef) XXX_Size() int

func (*NodeDef) XXX_Unmarshal 

func (m *NodeDef) XXX_Unmarshal(b []byte) error

type NodeDef_ExperimentalDebugInfo 

type NodeDef_ExperimentalDebugInfo struct {
	// Opaque string inserted into error messages created by the runtime.
	//
	// This is intended to store the list of names of the nodes from the
	// original graph that this node was derived. For example if this node, say
	// C, was result of a fusion of 2 nodes A and B, then 'original_node' would
	// be {A, B}. This information can be used to map errors originating at the
	// current node to some top level source code.
	OriginalNodeNames []string `protobuf:"bytes,1,rep,name=original_node_names,json=originalNodeNames,proto3" json:"original_node_names,omitempty"`
	// This is intended to store the list of names of the functions from the
	// original graph that this node was derived. For example if this node, say
	// C, was result of a fusion of node A in function FA and node B in function
	// FB, then `original_funcs` would be {FA, FB}. If the node is in the top
	// level graph, the `original_func` is empty. This information, with the
	// `original_node_names` can be used to map errors originating at the
	// current ndoe to some top level source code.
	OriginalFuncNames    []string `protobuf:"bytes,2,rep,name=original_func_names,json=originalFuncNames,proto3" json:"original_func_names,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

func (*NodeDef_ExperimentalDebugInfo) Descriptor 

func (*NodeDef_ExperimentalDebugInfo) Descriptor() ([]byte, []int)

func (*NodeDef_ExperimentalDebugInfo) GetOriginalFuncNames 

func (m *NodeDef_ExperimentalDebugInfo) GetOriginalFuncNames() []string

func (*NodeDef_ExperimentalDebugInfo) GetOriginalNodeNames 

func (m *NodeDef_ExperimentalDebugInfo) GetOriginalNodeNames() []string

func (*NodeDef_ExperimentalDebugInfo) ProtoMessage 

func (*NodeDef_ExperimentalDebugInfo) ProtoMessage()

func (*NodeDef_ExperimentalDebugInfo) Reset 

func (m *NodeDef_ExperimentalDebugInfo) Reset()

func (*NodeDef_ExperimentalDebugInfo) String 

func (m *NodeDef_ExperimentalDebugInfo) String() string

func (*NodeDef_ExperimentalDebugInfo) XXX_DiscardUnknown 

func (m *NodeDef_ExperimentalDebugInfo) XXX_DiscardUnknown()

func (*NodeDef_ExperimentalDebugInfo) XXX_Marshal 

func (m *NodeDef_ExperimentalDebugInfo) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*NodeDef_ExperimentalDebugInfo) XXX_Merge 

func (m *NodeDef_ExperimentalDebugInfo) XXX_Merge(src proto.Message)

func (*NodeDef_ExperimentalDebugInfo) XXX_Size 

func (m *NodeDef_ExperimentalDebugInfo) XXX_Size() int

func (*NodeDef_ExperimentalDebugInfo) XXX_Unmarshal 

func (m *NodeDef_ExperimentalDebugInfo) XXX_Unmarshal(b []byte) error

type NodeExecStats 

type NodeExecStats struct {
	// TODO(tucker): Use some more compact form of node identity than
	// the full string name.  Either all processes should agree on a
	// global id (cost_id?) for each node, or we should use a hash of
	// the name.
	NodeName             string                   `protobuf:"bytes,1,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"`
	AllStartMicros       int64                    `protobuf:"varint,2,opt,name=all_start_micros,json=allStartMicros,proto3" json:"all_start_micros,omitempty"`
	OpStartRelMicros     int64                    `protobuf:"varint,3,opt,name=op_start_rel_micros,json=opStartRelMicros,proto3" json:"op_start_rel_micros,omitempty"`
	OpEndRelMicros       int64                    `protobuf:"varint,4,opt,name=op_end_rel_micros,json=opEndRelMicros,proto3" json:"op_end_rel_micros,omitempty"`
	AllEndRelMicros      int64                    `protobuf:"varint,5,opt,name=all_end_rel_micros,json=allEndRelMicros,proto3" json:"all_end_rel_micros,omitempty"`
	Memory               []*AllocatorMemoryUsed   `protobuf:"bytes,6,rep,name=memory,proto3" json:"memory,omitempty"`
	Output               []*NodeOutput            `protobuf:"bytes,7,rep,name=output,proto3" json:"output,omitempty"`
	TimelineLabel        string                   `protobuf:"bytes,8,opt,name=timeline_label,json=timelineLabel,proto3" json:"timeline_label,omitempty"`
	ScheduledMicros      int64                    `protobuf:"varint,9,opt,name=scheduled_micros,json=scheduledMicros,proto3" json:"scheduled_micros,omitempty"`
	ThreadId             uint32                   `protobuf:"varint,10,opt,name=thread_id,json=threadId,proto3" json:"thread_id,omitempty"`
	ReferencedTensor     []*AllocationDescription `protobuf:"bytes,11,rep,name=referenced_tensor,json=referencedTensor,proto3" json:"referenced_tensor,omitempty"`
	MemoryStats          *MemoryStats             `protobuf:"bytes,12,opt,name=memory_stats,json=memoryStats,proto3" json:"memory_stats,omitempty"`
	AllStartNanos        int64                    `protobuf:"varint,13,opt,name=all_start_nanos,json=allStartNanos,proto3" json:"all_start_nanos,omitempty"`
	OpStartRelNanos      int64                    `protobuf:"varint,14,opt,name=op_start_rel_nanos,json=opStartRelNanos,proto3" json:"op_start_rel_nanos,omitempty"`
	OpEndRelNanos        int64                    `protobuf:"varint,15,opt,name=op_end_rel_nanos,json=opEndRelNanos,proto3" json:"op_end_rel_nanos,omitempty"`
	AllEndRelNanos       int64                    `protobuf:"varint,16,opt,name=all_end_rel_nanos,json=allEndRelNanos,proto3" json:"all_end_rel_nanos,omitempty"`
	ScheduledNanos       int64                    `protobuf:"varint,17,opt,name=scheduled_nanos,json=scheduledNanos,proto3" json:"scheduled_nanos,omitempty"`
	XXX_NoUnkeyedLiteral struct{}                 `json:"-"`
	XXX_unrecognized     []byte                   `json:"-"`
	XXX_sizecache        int32                    `json:"-"`
}

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) GetAllEndRelNanos 

func (m *NodeExecStats) GetAllEndRelNanos() int64

func (*NodeExecStats) GetAllStartMicros 

func (m *NodeExecStats) GetAllStartMicros() int64

func (*NodeExecStats) GetAllStartNanos 

func (m *NodeExecStats) GetAllStartNanos() 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) GetOpEndRelNanos 

func (m *NodeExecStats) GetOpEndRelNanos() int64

func (*NodeExecStats) GetOpStartRelMicros 

func (m *NodeExecStats) GetOpStartRelMicros() int64

func (*NodeExecStats) GetOpStartRelNanos 

func (m *NodeExecStats) GetOpStartRelNanos() 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) GetScheduledNanos 

func (m *NodeExecStats) GetScheduledNanos() int64

func (*NodeExecStats) GetThreadId 

func (m *NodeExecStats) GetThreadId() uint32

func (*NodeExecStats) GetTimelineLabel 

func (m *NodeExecStats) GetTimelineLabel() string

func (*NodeExecStats) ProtoMessage 

func (*NodeExecStats) ProtoMessage()

func (*NodeExecStats) Reset 

func (m *NodeExecStats) Reset()

func (*NodeExecStats) String 

func (m *NodeExecStats) String() string

func (*NodeExecStats) XXX_DiscardUnknown 

func (m *NodeExecStats) XXX_DiscardUnknown()

func (*NodeExecStats) XXX_Marshal 

func (m *NodeExecStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*NodeExecStats) XXX_Merge 

func (m *NodeExecStats) XXX_Merge(src proto.Message)

func (*NodeExecStats) XXX_Size 

func (m *NodeExecStats) XXX_Size() int

func (*NodeExecStats) XXX_Unmarshal 

func (m *NodeExecStats) XXX_Unmarshal(b []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,proto3" json:"tensor_description,omitempty"`
	XXX_NoUnkeyedLiteral struct{}           `json:"-"`
	XXX_unrecognized     []byte             `json:"-"`
	XXX_sizecache        int32              `json:"-"`
}

Output sizes recorded for a single execution of a graph node.

func (*NodeOutput) Descriptor 

func (*NodeOutput) Descriptor() ([]byte, []int)

func (*NodeOutput) GetSlot 

func (m *NodeOutput) GetSlot() int32

func (*NodeOutput) GetTensorDescription 

func (m *NodeOutput) GetTensorDescription() *TensorDescription

func (*NodeOutput) ProtoMessage 

func (*NodeOutput) ProtoMessage()

func (*NodeOutput) Reset 

func (m *NodeOutput) Reset()

func (*NodeOutput) String 

func (m *NodeOutput) String() string

func (*NodeOutput) XXX_DiscardUnknown 

func (m *NodeOutput) XXX_DiscardUnknown()

func (*NodeOutput) XXX_Marshal 

func (m *NodeOutput) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*NodeOutput) XXX_Merge 

func (m *NodeOutput) XXX_Merge(src proto.Message)

func (*NodeOutput) XXX_Size 

func (m *NodeOutput) XXX_Size() int

func (*NodeOutput) XXX_Unmarshal 

func (m *NodeOutput) XXX_Unmarshal(b []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,proto3" json:"input_arg,omitempty"`
	// Description of the output(s).
	OutputArg []*OpDef_ArgDef `protobuf:"bytes,3,rep,name=output_arg,json=outputArg,proto3" json:"output_arg,omitempty"`
	// Named control outputs for this operation. Useful only for composite
	// operations (i.e. functions) which want to name different control outputs.
	ControlOutput []string         `protobuf:"bytes,20,rep,name=control_output,json=controlOutput,proto3" json:"control_output,omitempty"`
	Attr          []*OpDef_AttrDef `protobuf:"bytes,4,rep,name=attr,proto3" json:"attr,omitempty"`
	// Optional deprecation based on GraphDef versions.
	Deprecation *OpDeprecation `protobuf:"bytes,8,opt,name=deprecation,proto3" json:"deprecation,omitempty"`
	// One-line human-readable description of what the Op does.
	Summary string `protobuf:"bytes,5,opt,name=summary,proto3" json:"summary,omitempty"`
	// Additional, longer human-readable description of what the Op does.
	Description string `protobuf:"bytes,6,opt,name=description,proto3" json:"description,omitempty"`
	// True if the operation is commutative ("op(a,b) == op(b,a)" for all inputs)
	IsCommutative bool `protobuf:"varint,18,opt,name=is_commutative,json=isCommutative,proto3" json:"is_commutative,omitempty"`
	// If is_aggregate is true, then this operation accepts N >= 2
	// inputs and produces 1 output all of the same type.  Should be
	// associative and commutative, and produce output with the same
	// shape as the input.  The optimizer may replace an aggregate op
	// taking input from multiple devices with a tree of aggregate ops
	// that aggregate locally within each device (and possibly within
	// groups of nearby devices) before communicating.
	// TODO(josh11b): Implement that optimization.
	IsAggregate bool `protobuf:"varint,16,opt,name=is_aggregate,json=isAggregate,proto3" json:"is_aggregate,omitempty"`
	// Ops are marked as stateful if their behavior depends on some state beyond
	// their input tensors (e.g. variable reading op) or if they have
	// a side-effect (e.g. printing or asserting ops). Equivalently, stateless ops
	// must always produce the same output for the same input and have
	// no side-effects.
	//
	// By default Ops may be moved between devices.  Stateful ops should
	// either not be moved, or should only be moved if that state can also
	// be moved (e.g. via some sort of save / restore).
	// Stateful ops are guaranteed to never be optimized away by Common
	// Subexpression Elimination (CSE).
	IsStateful bool `protobuf:"varint,17,opt,name=is_stateful,json=isStateful,proto3" json:"is_stateful,omitempty"`
	// 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 */
	XXX_NoUnkeyedLiteral     struct{} `json:"-"`
	XXX_unrecognized         []byte   `json:"-"`
	XXX_sizecache            int32    `json:"-"`
}

Defines an operation. A NodeDef in a GraphDef specifies an Op by using the "op" field which should match the name of a OpDef. LINT.IfChange

func (*OpDef) Descriptor 

func (*OpDef) Descriptor() ([]byte, []int)

func (*OpDef) GetAllowsUninitializedInput 

func (m *OpDef) GetAllowsUninitializedInput() bool

func (*OpDef) GetAttr 

func (m *OpDef) GetAttr() []*OpDef_AttrDef

func (*OpDef) GetControlOutput 

func (m *OpDef) GetControlOutput() []string

func (*OpDef) GetDeprecation 

func (m *OpDef) GetDeprecation() *OpDeprecation

func (*OpDef) GetDescription 

func (m *OpDef) GetDescription() string

func (*OpDef) GetInputArg 

func (m *OpDef) GetInputArg() []*OpDef_ArgDef

func (*OpDef) GetIsAggregate 

func (m *OpDef) GetIsAggregate() bool

func (*OpDef) GetIsCommutative 

func (m *OpDef) GetIsCommutative() bool

func (*OpDef) GetIsStateful 

func (m *OpDef) GetIsStateful() bool

func (*OpDef) GetName 

func (m *OpDef) GetName() string

func (*OpDef) GetOutputArg 

func (m *OpDef) GetOutputArg() []*OpDef_ArgDef

func (*OpDef) GetSummary 

func (m *OpDef) GetSummary() string

func (*OpDef) ProtoMessage 

func (*OpDef) ProtoMessage()

func (*OpDef) Reset 

func (m *OpDef) Reset()

func (*OpDef) String 

func (m *OpDef) String() string

func (*OpDef) XXX_DiscardUnknown 

func (m *OpDef) XXX_DiscardUnknown()

func (*OpDef) XXX_Marshal 

func (m *OpDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*OpDef) XXX_Merge 

func (m *OpDef) XXX_Merge(src proto.Message)

func (*OpDef) XXX_Size 

func (m *OpDef) XXX_Size() int

func (*OpDef) XXX_Unmarshal 

func (m *OpDef) XXX_Unmarshal(b []byte) error

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"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

For describing inputs and outputs.

func (*OpDef_ArgDef) Descriptor 

func (*OpDef_ArgDef) Descriptor() ([]byte, []int)

func (*OpDef_ArgDef) GetDescription 

func (m *OpDef_ArgDef) GetDescription() string

func (*OpDef_ArgDef) GetIsRef 

func (m *OpDef_ArgDef) GetIsRef() bool

func (*OpDef_ArgDef) GetName 

func (m *OpDef_ArgDef) GetName() string

func (*OpDef_ArgDef) GetNumberAttr 

func (m *OpDef_ArgDef) GetNumberAttr() string

func (*OpDef_ArgDef) GetType 

func (m *OpDef_ArgDef) GetType() DataType

func (*OpDef_ArgDef) GetTypeAttr 

func (m *OpDef_ArgDef) GetTypeAttr() string

func (*OpDef_ArgDef) GetTypeListAttr 

func (m *OpDef_ArgDef) GetTypeListAttr() string

func (*OpDef_ArgDef) ProtoMessage 

func (*OpDef_ArgDef) ProtoMessage()

func (*OpDef_ArgDef) Reset 

func (m *OpDef_ArgDef) Reset()

func (*OpDef_ArgDef) String 

func (m *OpDef_ArgDef) String() string

func (*OpDef_ArgDef) XXX_DiscardUnknown 

func (m *OpDef_ArgDef) XXX_DiscardUnknown()

func (*OpDef_ArgDef) XXX_Marshal 

func (m *OpDef_ArgDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*OpDef_ArgDef) XXX_Merge 

func (m *OpDef_ArgDef) XXX_Merge(src proto.Message)

func (*OpDef_ArgDef) XXX_Size 

func (m *OpDef_ArgDef) XXX_Size() int

func (*OpDef_ArgDef) XXX_Unmarshal 

func (m *OpDef_ArgDef) XXX_Unmarshal(b []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,proto3" json:"default_value,omitempty"`
	// Human-readable description.
	Description string `protobuf:"bytes,4,opt,name=description,proto3" json:"description,omitempty"`
	// For type == "int", this is a minimum value.  For "list(___)"
	// types, this is the minimum length.
	HasMinimum bool  `protobuf:"varint,5,opt,name=has_minimum,json=hasMinimum,proto3" json:"has_minimum,omitempty"`
	Minimum    int64 `protobuf:"varint,6,opt,name=minimum,proto3" json:"minimum,omitempty"`
	// The set of allowed values.  Has type that is the "list" version
	// of the "type" field above (uses the "list" field of AttrValue).
	// If type == "type" or "list(type)" above, then the "type" field
	// of "allowed_values.list" has the set of allowed DataTypes.
	// If type == "string" or "list(string)", then the "s" field of
	// "allowed_values.list" has the set of allowed strings.
	AllowedValues        *AttrValue `protobuf:"bytes,7,opt,name=allowed_values,json=allowedValues,proto3" json:"allowed_values,omitempty"`
	XXX_NoUnkeyedLiteral struct{}   `json:"-"`
	XXX_unrecognized     []byte     `json:"-"`
	XXX_sizecache        int32      `json:"-"`
}

Description of the graph-construction-time configuration of this Op. That is to say, this describes the attr fields that will be specified in the NodeDef.

func (*OpDef_AttrDef) Descriptor 

func (*OpDef_AttrDef) Descriptor() ([]byte, []int)

func (*OpDef_AttrDef) GetAllowedValues 

func (m *OpDef_AttrDef) GetAllowedValues() *AttrValue

func (*OpDef_AttrDef) GetDefaultValue 

func (m *OpDef_AttrDef) GetDefaultValue() *AttrValue

func (*OpDef_AttrDef) GetDescription 

func (m *OpDef_AttrDef) GetDescription() string

func (*OpDef_AttrDef) GetHasMinimum 

func (m *OpDef_AttrDef) GetHasMinimum() bool

func (*OpDef_AttrDef) GetMinimum 

func (m *OpDef_AttrDef) GetMinimum() int64

func (*OpDef_AttrDef) GetName 

func (m *OpDef_AttrDef) GetName() string

func (*OpDef_AttrDef) GetType 

func (m *OpDef_AttrDef) GetType() string

func (*OpDef_AttrDef) ProtoMessage 

func (*OpDef_AttrDef) ProtoMessage()

func (*OpDef_AttrDef) Reset 

func (m *OpDef_AttrDef) Reset()

func (*OpDef_AttrDef) String 

func (m *OpDef_AttrDef) String() string

func (*OpDef_AttrDef) XXX_DiscardUnknown 

func (m *OpDef_AttrDef) XXX_DiscardUnknown()

func (*OpDef_AttrDef) XXX_Marshal 

func (m *OpDef_AttrDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*OpDef_AttrDef) XXX_Merge 

func (m *OpDef_AttrDef) XXX_Merge(src proto.Message)

func (*OpDef_AttrDef) XXX_Size 

func (m *OpDef_AttrDef) XXX_Size() int

func (*OpDef_AttrDef) XXX_Unmarshal 

func (m *OpDef_AttrDef) XXX_Unmarshal(b []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"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

Information about version-dependent deprecation of an op

func (*OpDeprecation) Descriptor 

func (*OpDeprecation) Descriptor() ([]byte, []int)

func (*OpDeprecation) GetExplanation 

func (m *OpDeprecation) GetExplanation() string

func (*OpDeprecation) GetVersion 

func (m *OpDeprecation) GetVersion() int32

func (*OpDeprecation) ProtoMessage 

func (*OpDeprecation) ProtoMessage()

func (*OpDeprecation) Reset 

func (m *OpDeprecation) Reset()

func (*OpDeprecation) String 

func (m *OpDeprecation) String() string

func (*OpDeprecation) XXX_DiscardUnknown 

func (m *OpDeprecation) XXX_DiscardUnknown()

func (*OpDeprecation) XXX_Marshal 

func (m *OpDeprecation) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*OpDeprecation) XXX_Merge 

func (m *OpDeprecation) XXX_Merge(src proto.Message)

func (*OpDeprecation) XXX_Size 

func (m *OpDeprecation) XXX_Size() int

func (*OpDeprecation) XXX_Unmarshal 

func (m *OpDeprecation) XXX_Unmarshal(b []byte) error

type OpList 

type OpList struct {
	Op                   []*OpDef `protobuf:"bytes,1,rep,name=op,proto3" json:"op,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

A collection of OpDefs

func (*OpList) Descriptor 

func (*OpList) Descriptor() ([]byte, []int)

func (*OpList) GetOp 

func (m *OpList) GetOp() []*OpDef

func (*OpList) ProtoMessage 

func (*OpList) ProtoMessage()

func (*OpList) Reset 

func (m *OpList) Reset()

func (*OpList) String 

func (m *OpList) String() string

func (*OpList) XXX_DiscardUnknown 

func (m *OpList) XXX_DiscardUnknown()

func (*OpList) XXX_Marshal 

func (m *OpList) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*OpList) XXX_Merge 

func (m *OpList) XXX_Merge(src proto.Message)

func (*OpList) XXX_Size 

func (m *OpList) XXX_Size() int

func (*OpList) XXX_Unmarshal 

func (m *OpList) XXX_Unmarshal(b []byte) error

type ResourceHandleProto 

type ResourceHandleProto struct {
	// Unique name for the device containing the resource.
	Device string `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"`
	// Container in which this resource is placed.
	Container string `protobuf:"bytes,2,opt,name=container,proto3" json:"container,omitempty"`
	// Unique name of this resource.
	Name string `protobuf:"bytes,3,opt,name=name,proto3" json:"name,omitempty"`
	// Hash code for the type of the resource. Is only valid in the same device
	// and in the same execution.
	HashCode uint64 `protobuf:"varint,4,opt,name=hash_code,json=hashCode,proto3" json:"hash_code,omitempty"`
	// For debug-only, the name of the type pointed to by this handle, if
	// available.
	MaybeTypeName string `protobuf:"bytes,5,opt,name=maybe_type_name,json=maybeTypeName,proto3" json:"maybe_type_name,omitempty"`
	// Data types and shapes for the underlying resource.
	DtypesAndShapes      []*ResourceHandleProto_DtypeAndShape `protobuf:"bytes,6,rep,name=dtypes_and_shapes,json=dtypesAndShapes,proto3" json:"dtypes_and_shapes,omitempty"`
	XXX_NoUnkeyedLiteral struct{}                             `json:"-"`
	XXX_unrecognized     []byte                               `json:"-"`
	XXX_sizecache        int32                                `json:"-"`
}

Protocol buffer representing a handle to a tensorflow resource. Handles are not valid across executions, but can be serialized back and forth from within a single run.

func (*ResourceHandleProto) Descriptor 

func (*ResourceHandleProto) Descriptor() ([]byte, []int)

func (*ResourceHandleProto) GetContainer 

func (m *ResourceHandleProto) GetContainer() string

func (*ResourceHandleProto) GetDevice 

func (m *ResourceHandleProto) GetDevice() string

func (*ResourceHandleProto) GetDtypesAndShapes 

func (m *ResourceHandleProto) GetDtypesAndShapes() []*ResourceHandleProto_DtypeAndShape

func (*ResourceHandleProto) GetHashCode 

func (m *ResourceHandleProto) GetHashCode() uint64

func (*ResourceHandleProto) GetMaybeTypeName 

func (m *ResourceHandleProto) GetMaybeTypeName() string

func (*ResourceHandleProto) GetName 

func (m *ResourceHandleProto) GetName() string

func (*ResourceHandleProto) ProtoMessage 

func (*ResourceHandleProto) ProtoMessage()

func (*ResourceHandleProto) Reset 

func (m *ResourceHandleProto) Reset()

func (*ResourceHandleProto) String 

func (m *ResourceHandleProto) String() string

func (*ResourceHandleProto) XXX_DiscardUnknown 

func (m *ResourceHandleProto) XXX_DiscardUnknown()

func (*ResourceHandleProto) XXX_Marshal 

func (m *ResourceHandleProto) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*ResourceHandleProto) XXX_Merge 

func (m *ResourceHandleProto) XXX_Merge(src proto.Message)

func (*ResourceHandleProto) XXX_Size 

func (m *ResourceHandleProto) XXX_Size() int

func (*ResourceHandleProto) XXX_Unmarshal 

func (m *ResourceHandleProto) XXX_Unmarshal(b []byte) error

type ResourceHandleProto_DtypeAndShape 

type ResourceHandleProto_DtypeAndShape struct {
	Dtype                DataType          `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	Shape                *TensorShapeProto `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"`
	XXX_NoUnkeyedLiteral struct{}          `json:"-"`
	XXX_unrecognized     []byte            `json:"-"`
	XXX_sizecache        int32             `json:"-"`
}

Protocol buffer representing a pair of (data type, tensor shape).

func (*ResourceHandleProto_DtypeAndShape) Descriptor 

func (*ResourceHandleProto_DtypeAndShape) Descriptor() ([]byte, []int)

func (*ResourceHandleProto_DtypeAndShape) GetDtype 

func (m *ResourceHandleProto_DtypeAndShape) GetDtype() DataType

func (*ResourceHandleProto_DtypeAndShape) GetShape 

func (m *ResourceHandleProto_DtypeAndShape) GetShape() *TensorShapeProto

func (*ResourceHandleProto_DtypeAndShape) ProtoMessage 

func (*ResourceHandleProto_DtypeAndShape) ProtoMessage()

func (*ResourceHandleProto_DtypeAndShape) Reset 

func (m *ResourceHandleProto_DtypeAndShape) Reset()

func (*ResourceHandleProto_DtypeAndShape) String 

func (m *ResourceHandleProto_DtypeAndShape) String() string

func (*ResourceHandleProto_DtypeAndShape) XXX_DiscardUnknown 

func (m *ResourceHandleProto_DtypeAndShape) XXX_DiscardUnknown()

func (*ResourceHandleProto_DtypeAndShape) XXX_Marshal 

func (m *ResourceHandleProto_DtypeAndShape) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*ResourceHandleProto_DtypeAndShape) XXX_Merge 

func (m *ResourceHandleProto_DtypeAndShape) XXX_Merge(src proto.Message)

func (*ResourceHandleProto_DtypeAndShape) XXX_Size 

func (m *ResourceHandleProto_DtypeAndShape) XXX_Size() int

func (*ResourceHandleProto_DtypeAndShape) XXX_Unmarshal 

func (m *ResourceHandleProto_DtypeAndShape) XXX_Unmarshal(b []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,proto3" json:"full_shape,omitempty"`
	// Offset of this variable into the full variable.
	VarOffset []int64 `protobuf:"varint,3,rep,packed,name=var_offset,json=varOffset,proto3" json:"var_offset,omitempty"`
	// Shape of this variable.
	VarShape             []int64  `protobuf:"varint,4,rep,packed,name=var_shape,json=varShape,proto3" json:"var_shape,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

func (*SaveSliceInfoDef) Descriptor 

func (*SaveSliceInfoDef) Descriptor() ([]byte, []int)

func (*SaveSliceInfoDef) GetFullName 

func (m *SaveSliceInfoDef) GetFullName() string

func (*SaveSliceInfoDef) GetFullShape 

func (m *SaveSliceInfoDef) GetFullShape() []int64

func (*SaveSliceInfoDef) GetVarOffset 

func (m *SaveSliceInfoDef) GetVarOffset() []int64

func (*SaveSliceInfoDef) GetVarShape 

func (m *SaveSliceInfoDef) GetVarShape() []int64

func (*SaveSliceInfoDef) ProtoMessage 

func (*SaveSliceInfoDef) ProtoMessage()

func (*SaveSliceInfoDef) Reset 

func (m *SaveSliceInfoDef) Reset()

func (*SaveSliceInfoDef) String 

func (m *SaveSliceInfoDef) String() string

func (*SaveSliceInfoDef) XXX_DiscardUnknown 

func (m *SaveSliceInfoDef) XXX_DiscardUnknown()

func (*SaveSliceInfoDef) XXX_Marshal 

func (m *SaveSliceInfoDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*SaveSliceInfoDef) XXX_Merge 

func (m *SaveSliceInfoDef) XXX_Merge(src proto.Message)

func (*SaveSliceInfoDef) XXX_Size 

func (m *SaveSliceInfoDef) XXX_Size() int

func (*SaveSliceInfoDef) XXX_Unmarshal 

func (m *SaveSliceInfoDef) XXX_Unmarshal(b []byte) error

type StepStats 

type StepStats struct {
	DevStats             []*DeviceStepStats `protobuf:"bytes,1,rep,name=dev_stats,json=devStats,proto3" json:"dev_stats,omitempty"`
	XXX_NoUnkeyedLiteral struct{}           `json:"-"`
	XXX_unrecognized     []byte             `json:"-"`
	XXX_sizecache        int32              `json:"-"`
}

func (*StepStats) Descriptor 

func (*StepStats) Descriptor() ([]byte, []int)

func (*StepStats) GetDevStats 

func (m *StepStats) GetDevStats() []*DeviceStepStats

func (*StepStats) ProtoMessage 

func (*StepStats) ProtoMessage()

func (*StepStats) Reset 

func (m *StepStats) Reset()

func (*StepStats) String 

func (m *StepStats) String() string

func (*StepStats) XXX_DiscardUnknown 

func (m *StepStats) XXX_DiscardUnknown()

func (*StepStats) XXX_Marshal 

func (m *StepStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*StepStats) XXX_Merge 

func (m *StepStats) XXX_Merge(src proto.Message)

func (*StepStats) XXX_Size 

func (m *StepStats) XXX_Size() int

func (*StepStats) XXX_Unmarshal 

func (m *StepStats) XXX_Unmarshal(b []byte) error

type TensorDescription 

type TensorDescription struct {
	// Data type of tensor elements
	Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// Shape of the tensor.
	Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"`
	// Information about the size and allocator used for the data
	AllocationDescription *AllocationDescription `protobuf:"bytes,4,opt,name=allocation_description,json=allocationDescription,proto3" json:"allocation_description,omitempty"`
	XXX_NoUnkeyedLiteral  struct{}               `json:"-"`
	XXX_unrecognized      []byte                 `json:"-"`
	XXX_sizecache         int32                  `json:"-"`
}

func (*TensorDescription) Descriptor 

func (*TensorDescription) Descriptor() ([]byte, []int)

func (*TensorDescription) GetAllocationDescription 

func (m *TensorDescription) GetAllocationDescription() *AllocationDescription

func (*TensorDescription) GetDtype 

func (m *TensorDescription) GetDtype() DataType

func (*TensorDescription) GetShape 

func (m *TensorDescription) GetShape() *TensorShapeProto

func (*TensorDescription) ProtoMessage 

func (*TensorDescription) ProtoMessage()

func (*TensorDescription) Reset 

func (m *TensorDescription) Reset()

func (*TensorDescription) String 

func (m *TensorDescription) String() string

func (*TensorDescription) XXX_DiscardUnknown 

func (m *TensorDescription) XXX_DiscardUnknown()

func (*TensorDescription) XXX_Marshal 

func (m *TensorDescription) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*TensorDescription) XXX_Merge 

func (m *TensorDescription) XXX_Merge(src proto.Message)

func (*TensorDescription) XXX_Size 

func (m *TensorDescription) XXX_Size() int

func (*TensorDescription) XXX_Unmarshal 

func (m *TensorDescription) XXX_Unmarshal(b []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,proto3" json:"tensor_shape,omitempty"`
	// Version number.
	//
	// In version 0, if the "repeated xxx" representations contain only one
	// element, that element is repeated to fill the shape.  This makes it easy
	// to represent a constant Tensor with a single value.
	VersionNumber int32 `protobuf:"varint,3,opt,name=version_number,json=versionNumber,proto3" json:"version_number,omitempty"`
	// Serialized raw tensor content from either Tensor::AsProtoTensorContent or
	// memcpy in tensorflow::grpc::EncodeTensorToByteBuffer. This representation
	// can be used for all tensor types. The purpose of this representation is to
	// reduce serialization overhead during RPC call by avoiding serialization of
	// many repeated small items.
	TensorContent []byte `protobuf:"bytes,4,opt,name=tensor_content,json=tensorContent,proto3" json:"tensor_content,omitempty"`
	// DT_HALF, DT_BFLOAT16. Note that since protobuf has no int16 type, we'll
	// have some pointless zero padding for each value here.
	HalfVal []int32 `protobuf:"varint,13,rep,packed,name=half_val,json=halfVal,proto3" json:"half_val,omitempty"`
	// DT_FLOAT.
	FloatVal []float32 `protobuf:"fixed32,5,rep,packed,name=float_val,json=floatVal,proto3" json:"float_val,omitempty"`
	// DT_DOUBLE.
	DoubleVal []float64 `protobuf:"fixed64,6,rep,packed,name=double_val,json=doubleVal,proto3" json:"double_val,omitempty"`
	// DT_INT32, DT_INT16, DT_INT8, DT_UINT8.
	IntVal []int32 `protobuf:"varint,7,rep,packed,name=int_val,json=intVal,proto3" json:"int_val,omitempty"`
	// DT_STRING
	StringVal [][]byte `protobuf:"bytes,8,rep,name=string_val,json=stringVal,proto3" json:"string_val,omitempty"`
	// DT_COMPLEX64. scomplex_val(2*i) and scomplex_val(2*i+1) are real
	// and imaginary parts of i-th single precision complex.
	ScomplexVal []float32 `protobuf:"fixed32,9,rep,packed,name=scomplex_val,json=scomplexVal,proto3" json:"scomplex_val,omitempty"`
	// DT_INT64
	Int64Val []int64 `protobuf:"varint,10,rep,packed,name=int64_val,json=int64Val,proto3" json:"int64_val,omitempty"`
	// DT_BOOL
	BoolVal []bool `protobuf:"varint,11,rep,packed,name=bool_val,json=boolVal,proto3" json:"bool_val,omitempty"`
	// DT_COMPLEX128. dcomplex_val(2*i) and dcomplex_val(2*i+1) are real
	// and imaginary parts of i-th double precision complex.
	DcomplexVal []float64 `protobuf:"fixed64,12,rep,packed,name=dcomplex_val,json=dcomplexVal,proto3" json:"dcomplex_val,omitempty"`
	// DT_RESOURCE
	ResourceHandleVal []*ResourceHandleProto `protobuf:"bytes,14,rep,name=resource_handle_val,json=resourceHandleVal,proto3" json:"resource_handle_val,omitempty"`
	// DT_VARIANT
	VariantVal []*VariantTensorDataProto `protobuf:"bytes,15,rep,name=variant_val,json=variantVal,proto3" json:"variant_val,omitempty"`
	// DT_UINT32
	Uint32Val []uint32 `protobuf:"varint,16,rep,packed,name=uint32_val,json=uint32Val,proto3" json:"uint32_val,omitempty"`
	// DT_UINT64
	Uint64Val            []uint64 `protobuf:"varint,17,rep,packed,name=uint64_val,json=uint64Val,proto3" json:"uint64_val,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

Protocol buffer representing a tensor.

func (*TensorProto) Descriptor 

func (*TensorProto) Descriptor() ([]byte, []int)

func (*TensorProto) GetBoolVal 

func (m *TensorProto) GetBoolVal() []bool

func (*TensorProto) GetDcomplexVal 

func (m *TensorProto) GetDcomplexVal() []float64

func (*TensorProto) GetDoubleVal 

func (m *TensorProto) GetDoubleVal() []float64

func (*TensorProto) GetDtype 

func (m *TensorProto) GetDtype() DataType

func (*TensorProto) GetFloatVal 

func (m *TensorProto) GetFloatVal() []float32

func (*TensorProto) GetHalfVal 

func (m *TensorProto) GetHalfVal() []int32

func (*TensorProto) GetInt64Val 

func (m *TensorProto) GetInt64Val() []int64

func (*TensorProto) GetIntVal 

func (m *TensorProto) GetIntVal() []int32

func (*TensorProto) GetResourceHandleVal 

func (m *TensorProto) GetResourceHandleVal() []*ResourceHandleProto

func (*TensorProto) GetScomplexVal 

func (m *TensorProto) GetScomplexVal() []float32

func (*TensorProto) GetStringVal 

func (m *TensorProto) GetStringVal() [][]byte

func (*TensorProto) GetTensorContent 

func (m *TensorProto) GetTensorContent() []byte

func (*TensorProto) GetTensorShape 

func (m *TensorProto) GetTensorShape() *TensorShapeProto

func (*TensorProto) GetUint32Val 

func (m *TensorProto) GetUint32Val() []uint32

func (*TensorProto) GetUint64Val 

func (m *TensorProto) GetUint64Val() []uint64

func (*TensorProto) GetVariantVal 

func (m *TensorProto) GetVariantVal() []*VariantTensorDataProto

func (*TensorProto) GetVersionNumber 

func (m *TensorProto) GetVersionNumber() int32

func (*TensorProto) ProtoMessage 

func (*TensorProto) ProtoMessage()

func (*TensorProto) Reset 

func (m *TensorProto) Reset()

func (*TensorProto) String 

func (m *TensorProto) String() string

func (*TensorProto) XXX_DiscardUnknown 

func (m *TensorProto) XXX_DiscardUnknown()

func (*TensorProto) XXX_Marshal 

func (m *TensorProto) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*TensorProto) XXX_Merge 

func (m *TensorProto) XXX_Merge(src proto.Message)

func (*TensorProto) XXX_Size 

func (m *TensorProto) XXX_Size() int

func (*TensorProto) XXX_Unmarshal 

func (m *TensorProto) XXX_Unmarshal(b []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,proto3" json:"dim,omitempty"`
	// If true, the number of dimensions in the shape is unknown.
	//
	// If true, "dim.size()" must be 0.
	UnknownRank          bool     `protobuf:"varint,3,opt,name=unknown_rank,json=unknownRank,proto3" json:"unknown_rank,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

Dimensions of a tensor.

func (*TensorShapeProto) Descriptor 

func (*TensorShapeProto) Descriptor() ([]byte, []int)

func (*TensorShapeProto) GetDim 

func (m *TensorShapeProto) GetDim() []*TensorShapeProto_Dim

func (*TensorShapeProto) GetUnknownRank 

func (m *TensorShapeProto) GetUnknownRank() bool

func (*TensorShapeProto) ProtoMessage 

func (*TensorShapeProto) ProtoMessage()

func (*TensorShapeProto) Reset 

func (m *TensorShapeProto) Reset()

func (*TensorShapeProto) String 

func (m *TensorShapeProto) String() string

func (*TensorShapeProto) XXX_DiscardUnknown 

func (m *TensorShapeProto) XXX_DiscardUnknown()

func (*TensorShapeProto) XXX_Marshal 

func (m *TensorShapeProto) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*TensorShapeProto) XXX_Merge 

func (m *TensorShapeProto) XXX_Merge(src proto.Message)

func (*TensorShapeProto) XXX_Size 

func (m *TensorShapeProto) XXX_Size() int

func (*TensorShapeProto) XXX_Unmarshal 

func (m *TensorShapeProto) XXX_Unmarshal(b []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"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

One dimension of the tensor.

func (*TensorShapeProto_Dim) Descriptor 

func (*TensorShapeProto_Dim) Descriptor() ([]byte, []int)

func (*TensorShapeProto_Dim) GetName 

func (m *TensorShapeProto_Dim) GetName() string

func (*TensorShapeProto_Dim) GetSize 

func (m *TensorShapeProto_Dim) GetSize() int64

func (*TensorShapeProto_Dim) ProtoMessage 

func (*TensorShapeProto_Dim) ProtoMessage()

func (*TensorShapeProto_Dim) Reset 

func (m *TensorShapeProto_Dim) Reset()

func (*TensorShapeProto_Dim) String 

func (m *TensorShapeProto_Dim) String() string

func (*TensorShapeProto_Dim) XXX_DiscardUnknown 

func (m *TensorShapeProto_Dim) XXX_DiscardUnknown()

func (*TensorShapeProto_Dim) XXX_Marshal 

func (m *TensorShapeProto_Dim) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*TensorShapeProto_Dim) XXX_Merge 

func (m *TensorShapeProto_Dim) XXX_Merge(src proto.Message)

func (*TensorShapeProto_Dim) XXX_Size 

func (m *TensorShapeProto_Dim) XXX_Size() int

func (*TensorShapeProto_Dim) XXX_Unmarshal 

func (m *TensorShapeProto_Dim) XXX_Unmarshal(b []byte) error

type VariableAggregation 

type VariableAggregation int32

Indicates how a distributed variable will be aggregated. 

const (
	// `NONE`: This is the default, giving an error if you use a
	// variable-update operation with multiple replicas.
	VariableAggregation_VARIABLE_AGGREGATION_NONE VariableAggregation = 0
	// `SUM`: Add the updates across replicas.
	VariableAggregation_VARIABLE_AGGREGATION_SUM VariableAggregation = 1
	// `MEAN`: Take the arithmetic mean ("average") of the updates across
	// replicas.
	VariableAggregation_VARIABLE_AGGREGATION_MEAN VariableAggregation = 2
	// `ONLY_FIRST_REPLICA`: This is for when every replica is performing the same
	// update, but we only want to perform the update once. Used, e.g., for the
	// global step counter.
	VariableAggregation_VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA VariableAggregation = 3
)

func (VariableAggregation) EnumDescriptor 

func (VariableAggregation) EnumDescriptor() ([]byte, []int)

func (VariableAggregation) String 

func (x VariableAggregation) String() string

type VariableDef 

type VariableDef struct {
	// Name of the variable tensor.
	VariableName string `protobuf:"bytes,1,opt,name=variable_name,json=variableName,proto3" json:"variable_name,omitempty"`
	// Name of the tensor holding the variable's initial value.
	InitialValueName string `protobuf:"bytes,6,opt,name=initial_value_name,json=initialValueName,proto3" json:"initial_value_name,omitempty"`
	// Name of the initializer op.
	InitializerName string `protobuf:"bytes,2,opt,name=initializer_name,json=initializerName,proto3" json:"initializer_name,omitempty"`
	// Name of the snapshot tensor.
	SnapshotName string `protobuf:"bytes,3,opt,name=snapshot_name,json=snapshotName,proto3" json:"snapshot_name,omitempty"`
	// Support for saving variables as slices of a larger variable.
	SaveSliceInfoDef *SaveSliceInfoDef `protobuf:"bytes,4,opt,name=save_slice_info_def,json=saveSliceInfoDef,proto3" json:"save_slice_info_def,omitempty"`
	// Whether to represent this as a ResourceVariable.
	IsResource bool `protobuf:"varint,5,opt,name=is_resource,json=isResource,proto3" json:"is_resource,omitempty"`
	// Whether this variable should be trained.
	Trainable bool `protobuf:"varint,7,opt,name=trainable,proto3" json:"trainable,omitempty"`
	// Indicates when a distributed variable will be synced.
	Synchronization VariableSynchronization `protobuf:"varint,8,opt,name=synchronization,proto3,enum=tensorflow.VariableSynchronization" json:"synchronization,omitempty"`
	// Indicates how a distributed variable will be aggregated.
	Aggregation          VariableAggregation `protobuf:"varint,9,opt,name=aggregation,proto3,enum=tensorflow.VariableAggregation" json:"aggregation,omitempty"`
	XXX_NoUnkeyedLiteral struct{}            `json:"-"`
	XXX_unrecognized     []byte              `json:"-"`
	XXX_sizecache        int32               `json:"-"`
}

Protocol buffer representing a Variable.

func (*VariableDef) Descriptor 

func (*VariableDef) Descriptor() ([]byte, []int)

func (*VariableDef) GetAggregation 

func (m *VariableDef) GetAggregation() VariableAggregation

func (*VariableDef) GetInitialValueName 

func (m *VariableDef) GetInitialValueName() string

func (*VariableDef) GetInitializerName 

func (m *VariableDef) GetInitializerName() string

func (*VariableDef) GetIsResource 

func (m *VariableDef) GetIsResource() bool

func (*VariableDef) GetSaveSliceInfoDef 

func (m *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef

func (*VariableDef) GetSnapshotName 

func (m *VariableDef) GetSnapshotName() string

func (*VariableDef) GetSynchronization 

func (m *VariableDef) GetSynchronization() VariableSynchronization

func (*VariableDef) GetTrainable 

func (m *VariableDef) GetTrainable() bool

func (*VariableDef) GetVariableName 

func (m *VariableDef) GetVariableName() string

func (*VariableDef) ProtoMessage 

func (*VariableDef) ProtoMessage()

func (*VariableDef) Reset 

func (m *VariableDef) Reset()

func (*VariableDef) String 

func (m *VariableDef) String() string

func (*VariableDef) XXX_DiscardUnknown 

func (m *VariableDef) XXX_DiscardUnknown()

func (*VariableDef) XXX_Marshal 

func (m *VariableDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*VariableDef) XXX_Merge 

func (m *VariableDef) XXX_Merge(src proto.Message)

func (*VariableDef) XXX_Size 

func (m *VariableDef) XXX_Size() int

func (*VariableDef) XXX_Unmarshal 

func (m *VariableDef) XXX_Unmarshal(b []byte) error

type VariableSynchronization 

type VariableSynchronization int32

Indicates when a distributed variable will be synced. 

const (
	// `AUTO`: Indicates that the synchronization will be determined by the
	// current `DistributionStrategy` (eg. With `MirroredStrategy` this would be
	// `ON_WRITE`).
	VariableSynchronization_VARIABLE_SYNCHRONIZATION_AUTO VariableSynchronization = 0
	// `NONE`: Indicates that there will only be one copy of the variable, so
	// there is no need to sync.
	VariableSynchronization_VARIABLE_SYNCHRONIZATION_NONE VariableSynchronization = 1
	// `ON_WRITE`: Indicates that the variable will be updated across devices
	// every time it is written.
	VariableSynchronization_VARIABLE_SYNCHRONIZATION_ON_WRITE VariableSynchronization = 2
	// `ON_READ`: Indicates that the variable will be aggregated across devices
	// when it is read (eg. when checkpointing or when evaluating an op that uses
	// the variable).
	VariableSynchronization_VARIABLE_SYNCHRONIZATION_ON_READ VariableSynchronization = 3
)

func (VariableSynchronization) EnumDescriptor 

func (VariableSynchronization) EnumDescriptor() ([]byte, []int)

func (VariableSynchronization) String 

func (x VariableSynchronization) String() string

type VariantTensorDataProto 

type VariantTensorDataProto struct {
	// Name of the type of objects being serialized.
	TypeName string `protobuf:"bytes,1,opt,name=type_name,json=typeName,proto3" json:"type_name,omitempty"`
	// Portions of the object that are not Tensors.
	Metadata []byte `protobuf:"bytes,2,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// Tensors contained within objects being serialized.
	Tensors              []*TensorProto `protobuf:"bytes,3,rep,name=tensors,proto3" json:"tensors,omitempty"`
	XXX_NoUnkeyedLiteral struct{}       `json:"-"`
	XXX_unrecognized     []byte         `json:"-"`
	XXX_sizecache        int32          `json:"-"`
}

Protocol buffer representing the serialization format of DT_VARIANT tensors.

func (*VariantTensorDataProto) Descriptor 

func (*VariantTensorDataProto) Descriptor() ([]byte, []int)

func (*VariantTensorDataProto) GetMetadata 

func (m *VariantTensorDataProto) GetMetadata() []byte

func (*VariantTensorDataProto) GetTensors 

func (m *VariantTensorDataProto) GetTensors() []*TensorProto

func (*VariantTensorDataProto) GetTypeName 

func (m *VariantTensorDataProto) GetTypeName() string

func (*VariantTensorDataProto) ProtoMessage 

func (*VariantTensorDataProto) ProtoMessage()

func (*VariantTensorDataProto) Reset 

func (m *VariantTensorDataProto) Reset()

func (*VariantTensorDataProto) String 

func (m *VariantTensorDataProto) String() string

func (*VariantTensorDataProto) XXX_DiscardUnknown 

func (m *VariantTensorDataProto) XXX_DiscardUnknown()

func (*VariantTensorDataProto) XXX_Marshal 

func (m *VariantTensorDataProto) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*VariantTensorDataProto) XXX_Merge 

func (m *VariantTensorDataProto) XXX_Merge(src proto.Message)

func (*VariantTensorDataProto) XXX_Size 

func (m *VariantTensorDataProto) XXX_Size() int

func (*VariantTensorDataProto) XXX_Unmarshal 

func (m *VariantTensorDataProto) XXX_Unmarshal(b []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,proto3" json:"bad_consumers,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

Version information for a piece of serialized data

There are different types of versions for each type of data (GraphDef, etc.), but they all have the same common shape described here.

Each consumer has "consumer" and "min_producer" versions (specified elsewhere). A consumer is allowed to consume this data if 

producer >= min_producer
consumer >= min_consumer
consumer not in bad_consumers

func (*VersionDef) Descriptor 

func (*VersionDef) Descriptor() ([]byte, []int)

func (*VersionDef) GetBadConsumers 

func (m *VersionDef) GetBadConsumers() []int32

func (*VersionDef) GetMinConsumer 

func (m *VersionDef) GetMinConsumer() int32

func (*VersionDef) GetProducer 

func (m *VersionDef) GetProducer() int32

func (*VersionDef) ProtoMessage 

func (*VersionDef) ProtoMessage()

func (*VersionDef) Reset 

func (m *VersionDef) Reset()

func (*VersionDef) String 

func (m *VersionDef) String() string

func (*VersionDef) XXX_DiscardUnknown 

func (m *VersionDef) XXX_DiscardUnknown()

func (*VersionDef) XXX_Marshal 

func (m *VersionDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*VersionDef) XXX_Merge 

func (m *VersionDef) XXX_Merge(src proto.Message)

func (*VersionDef) XXX_Size 

func (m *VersionDef) XXX_Size() int

func (*VersionDef) XXX_Unmarshal 

func (m *VersionDef) XXX_Unmarshal(b []byte) error

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.