Documentation
¶
Overview ¶
Package aot is a generated protocol buffer package.
It is generated from these files:
tensorflow/compiler/aot/tfcompile.proto
It has these top-level messages:
TensorId Feed Fetch Config
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Config ¶
type Config struct { // Each feed is a positional input argument for the generated function. The // order of each entry matches the order of each input argument. Feed []*Feed `protobuf:"bytes,1,rep,name=feed" json:"feed,omitempty"` // Each fetch is a positional output argument for the generated function. The // order of each entry matches the order of each output argument. Fetch []*Fetch `protobuf:"bytes,2,rep,name=fetch" json:"fetch,omitempty"` }
Config represents configuration information for tfcompile.
func (*Config) Descriptor ¶
func (*Config) ProtoMessage ¶
func (*Config) ProtoMessage()
type Feed ¶
type Feed struct { Id *TensorId `protobuf:"bytes,1,opt,name=id" json:"id,omitempty"` Shape *tensorflow.TensorShapeProto `protobuf:"bytes,2,opt,name=shape" json:"shape,omitempty"` Name string `protobuf:"bytes,3,opt,name=name" json:"name,omitempty"` }
Feed represents a single feed tensor in the graph, which corresponds to an input argument for the generated function.
func (*Feed) Descriptor ¶
func (*Feed) GetShape ¶
func (m *Feed) GetShape() *tensorflow.TensorShapeProto
func (*Feed) ProtoMessage ¶
func (*Feed) ProtoMessage()
type Fetch ¶
type Fetch struct { Id *TensorId `protobuf:"bytes,1,opt,name=id" json:"id,omitempty"` Name string `protobuf:"bytes,2,opt,name=name" json:"name,omitempty"` }
Fetch represents a single fetch tensor in the graph, which corresponds to an output argument for the generated function.
func (*Fetch) Descriptor ¶
func (*Fetch) ProtoMessage ¶
func (*Fetch) ProtoMessage()
type TensorId ¶
type TensorId struct { NodeName string `protobuf:"bytes,1,opt,name=node_name,json=nodeName" json:"node_name,omitempty"` OutputIndex int64 `protobuf:"varint,2,opt,name=output_index,json=outputIndex" json:"output_index,omitempty"` }
TensorId identifies a tensor in a TensorFlow graph, by specifying the output index of a particular node in the graph. If the output of the named node feeds into other node(s), this corresponds to one or more edges. Otherwise it doesn't correspond to any existing edges at all, e.g. for output nodes.
func (*TensorId) Descriptor ¶
func (*TensorId) GetNodeName ¶
func (*TensorId) GetOutputIndex ¶
func (*TensorId) ProtoMessage ¶
func (*TensorId) ProtoMessage()