aot

package
v0.0.0-...-d1695c4 Latest Latest
Warning

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

Go to latest
Published: Sep 29, 2021 License: Apache-2.0 Imports: 4 Imported by: 0

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

func (*Config) GetFeed

func (m *Config) GetFeed() []*Feed

func (*Config) GetFetch

func (m *Config) GetFetch() []*Fetch

func (*Config) ProtoMessage

func (*Config) ProtoMessage()

func (*Config) Reset

func (m *Config) Reset()

func (*Config) String

func (m *Config) String() string

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

func (*Feed) GetId

func (m *Feed) GetId() *TensorId

func (*Feed) GetName

func (m *Feed) GetName() string

func (*Feed) GetShape

func (m *Feed) GetShape() *tensorflow.TensorShapeProto

func (*Feed) ProtoMessage

func (*Feed) ProtoMessage()

func (*Feed) Reset

func (m *Feed) Reset()

func (*Feed) String

func (m *Feed) String() string

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

func (*Fetch) GetId

func (m *Fetch) GetId() *TensorId

func (*Fetch) GetName

func (m *Fetch) GetName() string

func (*Fetch) ProtoMessage

func (*Fetch) ProtoMessage()

func (*Fetch) Reset

func (m *Fetch) Reset()

func (*Fetch) String

func (m *Fetch) String() string

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

func (*TensorId) GetNodeName

func (m *TensorId) GetNodeName() string

func (*TensorId) GetOutputIndex

func (m *TensorId) GetOutputIndex() int64

func (*TensorId) ProtoMessage

func (*TensorId) ProtoMessage()

func (*TensorId) Reset

func (m *TensorId) Reset()

func (*TensorId) String

func (m *TensorId) String() string

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL