caffe2pb

package
v0.0.0-...-c380afe Latest Latest
Warning

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

 Go to latest Published: Nov 18, 2018 License: MIT Imports: 3 Imported by: 0

Documentation

Overview

Package caffe2pb is a generated protocol buffer package.

It is generated from these files: 

caffe2/proto/hsm.proto
caffe2/proto/predictor_consts.proto
caffe2/proto/caffe2.proto
caffe2/proto/prof_dag.proto
caffe2/proto/metanet.proto
caffe2/proto/caffe2_legacy.proto

It has these top-level messages: 

NodeProto
TreeProto
HierarchyProto
PathProto
PathNodeProto
PredictorConsts
ExternalDataProto
TensorProto
QTensorProto
TensorProtos
TensorShape
TensorShapes
Argument
DeviceOption
OperatorDef
NetDef
ExecutionStep
PlanDef
BlobProto
DBReaderProto
TwoNumberStatsProto
BlobProfile
ProfDAGProto
ProfDAGProtos
ModelInfo
BlobsMap
NetsMap
PlansMap
StringMap
MetaNetDef
CaffeDatum

Index

Constants

View Source 
const Default_CaffeDatum_Encoded bool = false
View Source 
const Default_DeviceOption_DeviceType int32 = 0
View Source 
const Default_ExternalDataProto_Offset int64 = 0
View Source 
const Default_ModelInfo_PredictorType string = "SINGLE_PREDICTOR"
View Source 
const Default_OperatorDef_IsGradientOp bool = false
View Source 
const Default_PredictorConsts_GLOBAL_INIT_NET_TYPE string = "GLOBAL_INIT_NET_TYPE"
View Source 
const Default_PredictorConsts_INPUTS_BLOB_TYPE string = "INPUTS_BLOB_TYPE"
View Source 
const Default_PredictorConsts_META_NET_DEF string = "!!META_NET_DEF"
View Source 
const Default_PredictorConsts_MULTI_PREDICTOR string = "MULTI_PREDICTOR"
View Source 
const Default_PredictorConsts_OUTPUTS_BLOB_TYPE string = "OUTPUTS_BLOB_TYPE"
View Source 
const Default_PredictorConsts_PARAMETERS_BLOB_TYPE string = "PARAMETERS_BLOB_TYPE"
View Source 
const Default_PredictorConsts_PREDICTOR_DBREADER string = "!!PREDICTOR_DBREADER"
View Source 
const Default_PredictorConsts_PREDICT_INIT_NET_TYPE string = "PREDICT_INIT_NET_TYPE"
View Source 
const Default_PredictorConsts_PREDICT_NET_TYPE string = "PREDICT_NET_TYPE"
View Source 
const Default_PredictorConsts_SINGLE_PREDICTOR string = "SINGLE_PREDICTOR"
View Source 
const Default_PredictorConsts_TRAIN_INIT_PLAN_TYPE string = "TRAIN_INIT_PLAN_TYPE"
View Source 
const Default_PredictorConsts_TRAIN_PLAN_TYPE string = "TRAIN_PLAN_TYPE"
View Source 
const Default_TensorShape_UnknownShape bool = false

Variables

View Source 
var DeviceTypeProto_name = map[int32]string{
	0:     "PROTO_CPU",
	1:     "PROTO_CUDA",
	2:     "PROTO_MKLDNN",
	3:     "PROTO_OPENGL",
	4:     "PROTO_OPENCL",
	5:     "PROTO_IDEEP",
	6:     "PROTO_HIP",
	7:     "PROTO_FPGA",
	8:     "PROTO_COMPILE_TIME_MAX_DEVICE_TYPES",
	20901: "PROTO_ONLY_FOR_TEST",
}
View Source 
var DeviceTypeProto_value = map[string]int32{
	"PROTO_CPU":                           0,
	"PROTO_CUDA":                          1,
	"PROTO_MKLDNN":                        2,
	"PROTO_OPENGL":                        3,
	"PROTO_OPENCL":                        4,
	"PROTO_IDEEP":                         5,
	"PROTO_HIP":                           6,
	"PROTO_FPGA":                          7,
	"PROTO_COMPILE_TIME_MAX_DEVICE_TYPES": 8,
	"PROTO_ONLY_FOR_TEST":                 20901,
}
View Source 
var ExternalDataProto_SourceType_name = map[int32]string{
	0: "INLINE_CONTAINER",
	1: "SIMPLE_FILE",
}
View Source 
var ExternalDataProto_SourceType_value = map[string]int32{
	"INLINE_CONTAINER": 0,
	"SIMPLE_FILE":      1,
}
View Source 
var LegacyPadding_name = map[int32]string{
	0: "NOTSET",
	1: "VALID",
	2: "SAME",
	3: "CAFFE_LEGACY_POOLING",
}
View Source 
var LegacyPadding_value = map[string]int32{
	"NOTSET":               0,
	"VALID":                1,
	"SAME":                 2,
	"CAFFE_LEGACY_POOLING": 3,
}
View Source 
var TensorProto_DataType_name = map[int32]string{
	0:  "UNDEFINED",
	1:  "FLOAT",
	2:  "INT32",
	3:  "BYTE",
	4:  "STRING",
	5:  "BOOL",
	6:  "UINT8",
	7:  "INT8",
	8:  "UINT16",
	9:  "INT16",
	10: "INT64",
	12: "FLOAT16",
	13: "DOUBLE",
}
View Source 
var TensorProto_DataType_value = map[string]int32{
	"UNDEFINED": 0,
	"FLOAT":     1,
	"INT32":     2,
	"BYTE":      3,
	"STRING":    4,
	"BOOL":      5,
	"UINT8":     6,
	"INT8":      7,
	"UINT16":    8,
	"INT16":     9,
	"INT64":     10,
	"FLOAT16":   12,
	"DOUBLE":    13,
}
View Source 
var TensorProto_StorageType_name = map[int32]string{
	1: "TYPED",
	2: "RAW",
	3: "EXTERNAL",
	4: "NO_CONTENT",
}
View Source 
var TensorProto_StorageType_value = map[string]int32{
	"TYPED":      1,
	"RAW":        2,
	"EXTERNAL":   3,
	"NO_CONTENT": 4,
}

Functions

This section is empty.

Types

type Argument 

type Argument struct {
	Name             *string        `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	F                *float32       `protobuf:"fixed32,2,opt,name=f" json:"f,omitempty"`
	I                *int64         `protobuf:"varint,3,opt,name=i" json:"i,omitempty"`
	S                []byte         `protobuf:"bytes,4,opt,name=s" json:"s,omitempty"`
	T                *TensorProto   `protobuf:"bytes,10,opt,name=t" json:"t,omitempty"`
	N                *NetDef        `protobuf:"bytes,8,opt,name=n" json:"n,omitempty"`
	Floats           []float32      `protobuf:"fixed32,5,rep,name=floats" json:"floats,omitempty"`
	Ints             []int64        `protobuf:"varint,6,rep,name=ints" json:"ints,omitempty"`
	Strings          [][]byte       `protobuf:"bytes,7,rep,name=strings" json:"strings,omitempty"`
	Tensors          []*TensorProto `protobuf:"bytes,11,rep,name=tensors" json:"tensors,omitempty"`
	Nets             []*NetDef      `protobuf:"bytes,9,rep,name=nets" json:"nets,omitempty"`
	XXX_unrecognized []byte         `json:"-"`
}

A named argument containing either singular float, integer and string values, or repeated float, int and string arrays.

func (*Argument) Descriptor 

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

func (*Argument) GetF 

func (m *Argument) GetF() float32

func (*Argument) GetFloats 

func (m *Argument) GetFloats() []float32

func (*Argument) GetI 

func (m *Argument) GetI() int64

func (*Argument) GetInts 

func (m *Argument) GetInts() []int64

func (*Argument) GetN 

func (m *Argument) GetN() *NetDef

func (*Argument) GetName 

func (m *Argument) GetName() string

func (*Argument) GetNets 

func (m *Argument) GetNets() []*NetDef

func (*Argument) GetS 

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

func (*Argument) GetStrings 

func (m *Argument) GetStrings() [][]byte

func (*Argument) GetT 

func (m *Argument) GetT() *TensorProto

func (*Argument) GetTensors 

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

func (*Argument) ProtoMessage 

func (*Argument) ProtoMessage()

func (*Argument) Reset 

func (m *Argument) Reset()

func (*Argument) String 

func (m *Argument) String() string

type BlobProfile 

type BlobProfile struct {
	// Name of the blob (corresponds to OperatorDef.output).
	Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	// Profiling statistics.
	BytesUsed        *TwoNumberStatsProto `protobuf:"bytes,3,opt,name=bytes_used,json=bytesUsed" json:"bytes_used,omitempty"`
	XXX_unrecognized []byte               `json:"-"`
}

Blob profiling information. Profile for a blob is created every time a node outputs to the blob.

func (*BlobProfile) Descriptor 

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

func (*BlobProfile) GetBytesUsed 

func (m *BlobProfile) GetBytesUsed() *TwoNumberStatsProto

func (*BlobProfile) GetName 

func (m *BlobProfile) GetName() string

func (*BlobProfile) ProtoMessage 

func (*BlobProfile) ProtoMessage()

func (*BlobProfile) Reset 

func (m *BlobProfile) Reset()

func (*BlobProfile) String 

func (m *BlobProfile) String() string

type BlobProto 

type BlobProto struct {
	Name    *string       `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	Type    *string       `protobuf:"bytes,2,opt,name=type" json:"type,omitempty"`
	Tensor  *TensorProto  `protobuf:"bytes,3,opt,name=tensor" json:"tensor,omitempty"`
	Content []byte        `protobuf:"bytes,4,opt,name=content" json:"content,omitempty"`
	Qtensor *QTensorProto `protobuf:"bytes,5,opt,name=qtensor" json:"qtensor,omitempty"`
	// If blob is not Tensor and is divided into chunks, content_num_chunks
	// contains number of chunks, into which blob was divided.
	ContentNumChunks *int32 `protobuf:"varint,6,opt,name=content_num_chunks,json=contentNumChunks" json:"content_num_chunks,omitempty"`
	ContentChunkId   *int32 `protobuf:"varint,7,opt,name=content_chunk_id,json=contentChunkId" json:"content_chunk_id,omitempty"`
	XXX_unrecognized []byte `json:"-"`
}

Protobuf format for blobs that are not Tensors. We use a key to store the type of the blob. For example for a serialized DBProto, the type should be "DBReader" and the content should be a serialized DBProto object.

func (*BlobProto) Descriptor 

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

func (*BlobProto) GetContent 

func (m *BlobProto) GetContent() []byte

func (*BlobProto) GetContentChunkId 

func (m *BlobProto) GetContentChunkId() int32

func (*BlobProto) GetContentNumChunks 

func (m *BlobProto) GetContentNumChunks() int32

func (*BlobProto) GetName 

func (m *BlobProto) GetName() string

func (*BlobProto) GetQtensor 

func (m *BlobProto) GetQtensor() *QTensorProto

func (*BlobProto) GetTensor 

func (m *BlobProto) GetTensor() *TensorProto

func (*BlobProto) GetType 

func (m *BlobProto) GetType() string

func (*BlobProto) ProtoMessage 

func (*BlobProto) ProtoMessage()

func (*BlobProto) Reset 

func (m *BlobProto) Reset()

func (*BlobProto) String 

func (m *BlobProto) String() string

type BlobsMap 

type BlobsMap struct {
	Key              *string  `protobuf:"bytes,1,req,name=key" json:"key,omitempty"`
	Value            []string `protobuf:"bytes,2,rep,name=value" json:"value,omitempty"`
	XXX_unrecognized []byte   `json:"-"`
}

func (*BlobsMap) Descriptor 

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

func (*BlobsMap) GetKey 

func (m *BlobsMap) GetKey() string

func (*BlobsMap) GetValue 

func (m *BlobsMap) GetValue() []string

func (*BlobsMap) ProtoMessage 

func (*BlobsMap) ProtoMessage()

func (*BlobsMap) Reset 

func (m *BlobsMap) Reset()

func (*BlobsMap) String 

func (m *BlobsMap) String() string

type CaffeDatum 

type CaffeDatum struct {
	Channels *int32 `protobuf:"varint,1,opt,name=channels" json:"channels,omitempty"`
	Height   *int32 `protobuf:"varint,2,opt,name=height" json:"height,omitempty"`
	Width    *int32 `protobuf:"varint,3,opt,name=width" json:"width,omitempty"`
	// the actual image data, in bytes
	Data  []byte `protobuf:"bytes,4,opt,name=data" json:"data,omitempty"`
	Label *int32 `protobuf:"varint,5,opt,name=label" json:"label,omitempty"`
	// Optionally, the datum could also hold float data.
	FloatData []float32 `protobuf:"fixed32,6,rep,name=float_data,json=floatData" json:"float_data,omitempty"`
	// If true data contains an encoded image that need to be decoded
	Encoded          *bool  `protobuf:"varint,7,opt,name=encoded,def=0" json:"encoded,omitempty"`
	XXX_unrecognized []byte `json:"-"`
}

Original Caffe1 Datum copy: this is used in image input op to allow us to load caffe1 serialized datum without having to regenerate the database.

func (*CaffeDatum) Descriptor 

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

func (*CaffeDatum) GetChannels 

func (m *CaffeDatum) GetChannels() int32

func (*CaffeDatum) GetData 

func (m *CaffeDatum) GetData() []byte

func (*CaffeDatum) GetEncoded 

func (m *CaffeDatum) GetEncoded() bool

func (*CaffeDatum) GetFloatData 

func (m *CaffeDatum) GetFloatData() []float32

func (*CaffeDatum) GetHeight 

func (m *CaffeDatum) GetHeight() int32

func (*CaffeDatum) GetLabel 

func (m *CaffeDatum) GetLabel() int32

func (*CaffeDatum) GetWidth 

func (m *CaffeDatum) GetWidth() int32

func (*CaffeDatum) ProtoMessage 

func (*CaffeDatum) ProtoMessage()

func (*CaffeDatum) Reset 

func (m *CaffeDatum) Reset()

func (*CaffeDatum) String 

func (m *CaffeDatum) String() string

type DBReaderProto 

type DBReaderProto struct {
	// The name for the DB object in the workspace.
	Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	// The source of the DB
	Source *string `protobuf:"bytes,2,opt,name=source" json:"source,omitempty"`
	// The type of the DB
	DbType *string `protobuf:"bytes,3,opt,name=db_type,json=dbType" json:"db_type,omitempty"`
	// The current key of the DB if the DB supports seeking.
	Key              *string `protobuf:"bytes,4,opt,name=key" json:"key,omitempty"`
	XXX_unrecognized []byte  `json:"-"`
}

Protobuf format to serialize DBReader.

func (*DBReaderProto) Descriptor 

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

func (*DBReaderProto) GetDbType 

func (m *DBReaderProto) GetDbType() string

func (*DBReaderProto) GetKey 

func (m *DBReaderProto) GetKey() string

func (*DBReaderProto) GetName 

func (m *DBReaderProto) GetName() string

func (*DBReaderProto) GetSource 

func (m *DBReaderProto) GetSource() string

func (*DBReaderProto) ProtoMessage 

func (*DBReaderProto) ProtoMessage()

func (*DBReaderProto) Reset 

func (m *DBReaderProto) Reset()

func (*DBReaderProto) String 

func (m *DBReaderProto) String() string

type DeviceOption 

type DeviceOption struct {
	// [general] Options that need to be carried out before running the execution.
	// optional DeviceType device_type = 1 [ default = CPU ];
	DeviceType *int32 `protobuf:"varint,1,opt,name=device_type,json=deviceType,def=0" json:"device_type,omitempty"`
	// [general] Used together with device_type to identify the exact device
	DeviceId *int32 `protobuf:"varint,2,opt,name=device_id,json=deviceId" json:"device_id,omitempty"`
	// [general] The random seed to start the device random number generator with.
	RandomSeed *uint32 `protobuf:"varint,3,opt,name=random_seed,json=randomSeed" json:"random_seed,omitempty"`
	// [general] What node this op should execute on.
	// Used for net transformation purposes. Must be empty at execution time.
	NodeName *string `protobuf:"bytes,4,opt,name=node_name,json=nodeName" json:"node_name,omitempty"`
	// [CPU and Linux specific] NUMA node id
	NumaNodeId *int32 `protobuf:"varint,5,opt,name=numa_node_id,json=numaNodeId" json:"numa_node_id,omitempty"`
	// [general] Extra information passed, not used at execution time currently.
	ExtraInfo        []string `protobuf:"bytes,6,rep,name=extra_info,json=extraInfo" json:"extra_info,omitempty"`
	XXX_unrecognized []byte   `json:"-"`
}

Device-specific options. We do not distinguish DeviceOption protos for different DeviceTypes, so currently all devices share the same DeviceOption proto. Fields that are specific to a device type is ignored if the type does not match. Note: if you add fields to the DeviceOption, make sure you add the corresponding changes to IsSameDevice() function in utils/proto_utils.{h,cc}.

func (*DeviceOption) Descriptor 

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

func (*DeviceOption) GetDeviceId 

func (m *DeviceOption) GetDeviceId() int32

func (*DeviceOption) GetDeviceType 

func (m *DeviceOption) GetDeviceType() int32

func (*DeviceOption) GetExtraInfo 

func (m *DeviceOption) GetExtraInfo() []string

func (*DeviceOption) GetNodeName 

func (m *DeviceOption) GetNodeName() string

func (*DeviceOption) GetNumaNodeId 

func (m *DeviceOption) GetNumaNodeId() int32

func (*DeviceOption) GetRandomSeed 

func (m *DeviceOption) GetRandomSeed() uint32

func (*DeviceOption) ProtoMessage 

func (*DeviceOption) ProtoMessage()

func (*DeviceOption) Reset 

func (m *DeviceOption) Reset()

func (*DeviceOption) String 

func (m *DeviceOption) String() string

type DeviceTypeProto 

type DeviceTypeProto int32

DeviceType that Caffe2 currently supports. Note: if you add a device type, make sure you add the corresponding device line in the DeviceTypeName() function in caffe2/utils/proto_utils.cc and update c10/DeviceType.h 

const (
	DeviceTypeProto_PROTO_CPU    DeviceTypeProto = 0
	DeviceTypeProto_PROTO_CUDA   DeviceTypeProto = 1
	DeviceTypeProto_PROTO_MKLDNN DeviceTypeProto = 2
	DeviceTypeProto_PROTO_OPENGL DeviceTypeProto = 3
	DeviceTypeProto_PROTO_OPENCL DeviceTypeProto = 4
	DeviceTypeProto_PROTO_IDEEP  DeviceTypeProto = 5
	DeviceTypeProto_PROTO_HIP    DeviceTypeProto = 6
	DeviceTypeProto_PROTO_FPGA   DeviceTypeProto = 7
	// Change the following number if you add more devices in the code.
	DeviceTypeProto_PROTO_COMPILE_TIME_MAX_DEVICE_TYPES DeviceTypeProto = 8
	DeviceTypeProto_PROTO_ONLY_FOR_TEST                 DeviceTypeProto = 20901
)

func (DeviceTypeProto) Enum 

func (x DeviceTypeProto) Enum() *DeviceTypeProto

func (DeviceTypeProto) EnumDescriptor 

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

func (DeviceTypeProto) String 

func (x DeviceTypeProto) String() string

func (*DeviceTypeProto) UnmarshalJSON 

func (x *DeviceTypeProto) UnmarshalJSON(data []byte) error

type ExecutionStep 

type ExecutionStep struct {
	// ExecutionStep should either contain a set of substeps, or a set of
	// network names to run in this execution step. They should NOT both be set
	// at the same time.
	Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	// An execution step could be recursive, in which it involves a set of
	// substeps.
	Substep []*ExecutionStep `protobuf:"bytes,2,rep,name=substep" json:"substep,omitempty"`
	// Alternatively, an execution step could involve one or more networks.
	// Note that you cannot have both substeps and networks. Choose one.
	// Note that an execution step refers networks by their name. The actual
	// network definition of the same name should be included in the network field
	// of the plan. The reason is that a network object might hold internal states
	// (think of a data layer), so we want to have the same network object that
	// multiple steps could ask to run.
	Network []string `protobuf:"bytes,3,rep,name=network" json:"network,omitempty"`
	// Number of iterations to run this step. The substeps or the networks
	// specified will be run sequentially, and one sequential run is considered
	// one iteration. If this is not set, the number of iterations is assumed to
	// be 1.
	NumIter *int64 `protobuf:"varint,4,opt,name=num_iter,json=numIter" json:"num_iter,omitempty"`
	// Criteria network specifies a single output (TensorCPU<bool>) of
	// size (1), is run on every iteration by the executor, and
	// execution terminates when the output[0] is `false`.
	CriteriaNetwork *string `protobuf:"bytes,5,opt,name=criteria_network,json=criteriaNetwork" json:"criteria_network,omitempty"`
	// DEPRECATED. Use `run_every_ms`.
	ReportNet      *string `protobuf:"bytes,7,opt,name=report_net,json=reportNet" json:"report_net,omitempty"`
	ReportInterval *int32  `protobuf:"varint,8,opt,name=report_interval,json=reportInterval" json:"report_interval,omitempty"`
	// If provided, execute this step at every time interval (in millisecs)
	// while its sibiling execution steps execute in parallel. This step is
	// guaranteed to run at least once after all non-interval siblings finished.
	RunEveryMs *int64 `protobuf:"varint,11,opt,name=run_every_ms,json=runEveryMs" json:"run_every_ms,omitempty"`
	// If false or not set, execute sub-steps serially.
	// If true, execute all substeps concurrently, each one in a separte thread.
	ConcurrentSubsteps *bool `protobuf:"varint,6,opt,name=concurrent_substeps,json=concurrentSubsteps" json:"concurrent_substeps,omitempty"`
	// Name of a scalar boolean tensor.
	// ES checks this blob AFTER every substeps/subnets.
	// If specified, and the value is true, then ES will skip the rest and return
	// immediately.
	// This means that the report_net and the first step will always be called.
	// Use cases:
	// 1) the first substep stops the rest if data condition not met
	// 2) the first substep decide which of the rest of the steps should be run.
	// 3) external control
	//
	// ** It is the user's responsibility to not to put this blob in race conditions.
	// ** For example when setting this blob in concurrent substeps
	ShouldStopBlob *string `protobuf:"bytes,9,opt,name=should_stop_blob,json=shouldStopBlob" json:"should_stop_blob,omitempty"`
	// if only_once is true, this step will only be executed once. this ONLY takes
	// effect when using should_stop_blob
	OnlyOnce *bool `protobuf:"varint,10,opt,name=only_once,json=onlyOnce" json:"only_once,omitempty"`
	// Whether to create a child workspace for this step.
	// If yes, the workflow and nets are re-created every time this step is run.
	CreateWorkspace *bool `protobuf:"varint,12,opt,name=create_workspace,json=createWorkspace" json:"create_workspace,omitempty"`
	// How many copies of the children execution steps to run concurrently.
	NumConcurrentInstances *int32 `protobuf:"varint,13,opt,name=num_concurrent_instances,json=numConcurrentInstances" json:"num_concurrent_instances,omitempty"`
	XXX_unrecognized       []byte `json:"-"`
}

ExecutionStep is actually a sort-of-hacky way we simulate iteration right now.

func (*ExecutionStep) Descriptor 

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

func (*ExecutionStep) GetConcurrentSubsteps 

func (m *ExecutionStep) GetConcurrentSubsteps() bool

func (*ExecutionStep) GetCreateWorkspace 

func (m *ExecutionStep) GetCreateWorkspace() bool

func (*ExecutionStep) GetCriteriaNetwork 

func (m *ExecutionStep) GetCriteriaNetwork() string

func (*ExecutionStep) GetName 

func (m *ExecutionStep) GetName() string

func (*ExecutionStep) GetNetwork 

func (m *ExecutionStep) GetNetwork() []string

func (*ExecutionStep) GetNumConcurrentInstances 

func (m *ExecutionStep) GetNumConcurrentInstances() int32

func (*ExecutionStep) GetNumIter 

func (m *ExecutionStep) GetNumIter() int64

func (*ExecutionStep) GetOnlyOnce 

func (m *ExecutionStep) GetOnlyOnce() bool

func (*ExecutionStep) GetReportInterval 

func (m *ExecutionStep) GetReportInterval() int32

func (*ExecutionStep) GetReportNet 

func (m *ExecutionStep) GetReportNet() string

func (*ExecutionStep) GetRunEveryMs 

func (m *ExecutionStep) GetRunEveryMs() int64

func (*ExecutionStep) GetShouldStopBlob 

func (m *ExecutionStep) GetShouldStopBlob() string

func (*ExecutionStep) GetSubstep 

func (m *ExecutionStep) GetSubstep() []*ExecutionStep

func (*ExecutionStep) ProtoMessage 

func (*ExecutionStep) ProtoMessage()

func (*ExecutionStep) Reset 

func (m *ExecutionStep) Reset()

func (*ExecutionStep) String 

func (m *ExecutionStep) String() string

type ExternalDataProto 

type ExternalDataProto struct {
	SourceType *ExternalDataProto_SourceType `` /* 132-byte string literal not displayed */
	// used together with type
	RecordId *string `protobuf:"bytes,2,opt,name=record_id,json=recordId" json:"record_id,omitempty"`
	// the size of the entire record (in bytes)
	RecordSize *uint64 `protobuf:"varint,5,opt,name=record_size,json=recordSize" json:"record_size,omitempty"`
	// the offset of the starting point, the content may be shared between
	// multiple tensors
	Offset *int64 `protobuf:"varint,3,opt,name=offset,def=0" json:"offset,omitempty"`
	// the strides of the content
	Strides          []int64 `protobuf:"varint,4,rep,name=strides" json:"strides,omitempty"`
	XXX_unrecognized []byte  `json:"-"`
}

ExternalDataProto stores the pointer to the content of TensorProto the content are stored in the raw format as little endian

func (*ExternalDataProto) Descriptor 

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

func (*ExternalDataProto) GetOffset 

func (m *ExternalDataProto) GetOffset() int64

func (*ExternalDataProto) GetRecordId 

func (m *ExternalDataProto) GetRecordId() string

func (*ExternalDataProto) GetRecordSize 

func (m *ExternalDataProto) GetRecordSize() uint64

func (*ExternalDataProto) GetSourceType 

func (m *ExternalDataProto) GetSourceType() ExternalDataProto_SourceType

func (*ExternalDataProto) GetStrides 

func (m *ExternalDataProto) GetStrides() []int64

func (*ExternalDataProto) ProtoMessage 

func (*ExternalDataProto) ProtoMessage()

func (*ExternalDataProto) Reset 

func (m *ExternalDataProto) Reset()

func (*ExternalDataProto) String 

func (m *ExternalDataProto) String() string

type ExternalDataProto_SourceType 

type ExternalDataProto_SourceType int32

type of the external storage type, can be the following: 

const (
	// the container defined in torch/csrc/jit/serialization.h is used,
	// and record_id is the tag to help the runtime identify the data
	// this type of storage is set as DEFAULT and recommended for external
	// data storage
	ExternalDataProto_INLINE_CONTAINER ExternalDataProto_SourceType = 0
	// use external file to store the data, and record_id is the POSIX relative path
	// to the file. this (simple) file is only for the data, and the data is stored
	// as little endian in the file
	ExternalDataProto_SIMPLE_FILE ExternalDataProto_SourceType = 1
)
const Default_ExternalDataProto_SourceType ExternalDataProto_SourceType = ExternalDataProto_INLINE_CONTAINER

func (ExternalDataProto_SourceType) Enum 

func (x ExternalDataProto_SourceType) Enum() *ExternalDataProto_SourceType

func (ExternalDataProto_SourceType) EnumDescriptor 

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

func (ExternalDataProto_SourceType) String 

func (x ExternalDataProto_SourceType) String() string

func (*ExternalDataProto_SourceType) UnmarshalJSON 

func (x *ExternalDataProto_SourceType) UnmarshalJSON(data []byte) error

type HierarchyProto 

type HierarchyProto struct {
	Size             *int32       `protobuf:"varint,1,opt,name=size" json:"size,omitempty"`
	Paths            []*PathProto `protobuf:"bytes,2,rep,name=paths" json:"paths,omitempty"`
	XXX_unrecognized []byte       `json:"-"`
}

Internal Protobuf format which represents the path in the tree hierarchy for each word in the vocabulary.

func (*HierarchyProto) Descriptor 

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

func (*HierarchyProto) GetPaths 

func (m *HierarchyProto) GetPaths() []*PathProto

func (*HierarchyProto) GetSize 

func (m *HierarchyProto) GetSize() int32

func (*HierarchyProto) ProtoMessage 

func (*HierarchyProto) ProtoMessage()

func (*HierarchyProto) Reset 

func (m *HierarchyProto) Reset()

func (*HierarchyProto) String 

func (m *HierarchyProto) String() string

type LegacyPadding 

type LegacyPadding int32
const (
	LegacyPadding_NOTSET LegacyPadding = 0
	// VALID and SAME are two strategies adopted in Google DistBelief: it forces
	// the input shape as follows. For SAME, the output is:
	//   R_out = ceil(float(R) / float(S))
	//   C_out = ceil(float(C) / float(S))
	// where R and C are row and column, S is the stride, and K is the kernel.
	// The number of padded pixels is then computed as
	//   Pr = ((R_out - 1) * S + K - R)
	//   Pc = ((C_out - 1) * S + K - C)
	// When Pr and Pc are even numbers, both sides (left and right, or top and
	// bottom) get half each. When Pr and Pc are odd numbers, the right and the
	// bottom gets the one additional padding pixel.
	// For VALID, padding values of 0 are always used.
	LegacyPadding_VALID LegacyPadding = 1
	LegacyPadding_SAME  LegacyPadding = 2
	// CAFFE_LEGACY_POOLING is a flag that notifies the code to use the old Caffe
	// padding strategy.
	// Basically, in caffe2, after padding the convolution and pooling use the
	// same computation strategy: half-windows at the right and bottom are
	// discarded. In Caffe, convolution follows this strategy but if there are
	// some pixels in the half-windows, the pooling layer will actually put one
	// additional output. If you set LegacyPadding to this, we will compute the
	// equivalent padding strategy in caffe2 so that the output size is
	// backward compatible with Caffe.
	// THIS IS NOW DEPRECATED. ANY non-conventional use has to be manually
	// converted.
	LegacyPadding_CAFFE_LEGACY_POOLING LegacyPadding = 3
)

func (LegacyPadding) Enum 

func (x LegacyPadding) Enum() *LegacyPadding

func (LegacyPadding) EnumDescriptor 

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

func (LegacyPadding) String 

func (x LegacyPadding) String() string

func (*LegacyPadding) UnmarshalJSON 

func (x *LegacyPadding) UnmarshalJSON(data []byte) error

type MetaNetDef 

type MetaNetDef struct {
	Blobs []*BlobsMap `protobuf:"bytes,1,rep,name=blobs" json:"blobs,omitempty"`
	// Text-format serialized NetDefs.
	Nets []*NetsMap `protobuf:"bytes,2,rep,name=nets" json:"nets,omitempty"`
	// Info about where the model comes from. Possible use cases:
	// 1) sanity check or diagnose
	// 2) provide info for evaluation.
	ModelInfo               *ModelInfo   `protobuf:"bytes,3,opt,name=modelInfo" json:"modelInfo,omitempty"`
	Plans                   []*PlansMap  `protobuf:"bytes,4,rep,name=plans" json:"plans,omitempty"`
	ApplicationSpecificInfo []*StringMap `protobuf:"bytes,5,rep,name=applicationSpecificInfo" json:"applicationSpecificInfo,omitempty"`
	XXX_unrecognized        []byte       `json:"-"`
}

func (*MetaNetDef) Descriptor 

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

func (*MetaNetDef) GetApplicationSpecificInfo 

func (m *MetaNetDef) GetApplicationSpecificInfo() []*StringMap

func (*MetaNetDef) GetBlobs 

func (m *MetaNetDef) GetBlobs() []*BlobsMap

func (*MetaNetDef) GetModelInfo 

func (m *MetaNetDef) GetModelInfo() *ModelInfo

func (*MetaNetDef) GetNets 

func (m *MetaNetDef) GetNets() []*NetsMap

func (*MetaNetDef) GetPlans 

func (m *MetaNetDef) GetPlans() []*PlansMap

func (*MetaNetDef) ProtoMessage 

func (*MetaNetDef) ProtoMessage()

func (*MetaNetDef) Reset 

func (m *MetaNetDef) Reset()

func (*MetaNetDef) String 

func (m *MetaNetDef) String() string

type ModelInfo 

type ModelInfo struct {
	Project          *string `protobuf:"bytes,1,opt,name=project" json:"project,omitempty"`
	ModelClass       *string `protobuf:"bytes,2,opt,name=modelClass" json:"modelClass,omitempty"`
	Version          *string `protobuf:"bytes,3,opt,name=version" json:"version,omitempty"`
	PredictorType    *string `protobuf:"bytes,4,opt,name=predictorType,def=SINGLE_PREDICTOR" json:"predictorType,omitempty"`
	ModelId          *string `protobuf:"bytes,5,opt,name=modelId" json:"modelId,omitempty"`
	XXX_unrecognized []byte  `json:"-"`
}

func (*ModelInfo) Descriptor 

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

func (*ModelInfo) GetModelClass 

func (m *ModelInfo) GetModelClass() string

func (*ModelInfo) GetModelId 

func (m *ModelInfo) GetModelId() string

func (*ModelInfo) GetPredictorType 

func (m *ModelInfo) GetPredictorType() string

func (*ModelInfo) GetProject 

func (m *ModelInfo) GetProject() string

func (*ModelInfo) GetVersion 

func (m *ModelInfo) GetVersion() string

func (*ModelInfo) ProtoMessage 

func (*ModelInfo) ProtoMessage()

func (*ModelInfo) Reset 

func (m *ModelInfo) Reset()

func (*ModelInfo) String 

func (m *ModelInfo) String() string

type NetDef 

type NetDef struct {
	Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	// Operators that the network contains.
	// Note: this is not named "operator" because that is a reserved word in C++.
	Op []*OperatorDef `protobuf:"bytes,2,rep,name=op" json:"op,omitempty"`
	// The type of network that the net should be run with. This routes the
	// network instantiation to different execution modes. The default mode,
	// "simple", runs the operators in a sequential way as the original Caffe
	// implementation does.
	Type *string `protobuf:"bytes,3,opt,name=type" json:"type,omitempty"`
	// the number of workers, if the operators in the network is to be carried out
	// in parallel.
	// Note: This is to be deprecated. Using the arg field with "num_workers" as
	// key.
	NumWorkers *int32 `protobuf:"varint,4,opt,name=num_workers,json=numWorkers" json:"num_workers,omitempty"`
	// The device option for the network. If a network has a specific device
	// option and one of its operators does not have it set, we will copy over the
	// device option to the operator. This allows us to basically avoid putting
	// device options at every operator.
	DeviceOption *DeviceOption `protobuf:"bytes,5,opt,name=device_option,json=deviceOption" json:"device_option,omitempty"`
	Arg          []*Argument   `protobuf:"bytes,6,rep,name=arg" json:"arg,omitempty"`
	// Two optional fields to declare external input and output of a net.
	// If these two are set, when a net is created, we will sanity check for
	// every op whether its input is declared (either as an external input,
	// or as an intermediate blob created by one of the ops), and sanity check
	// if all blobs in external_output are produced.
	//
	// In cases of memory optimization, declaring external_input and
	// external_output also ensures that storage of these blobs are persistent:
	// for any blob in external_input and external_output, after a network run
	// finishes, their content are actually the right content. Any intermediate
	// blobs' contents may be overwritten.
	ExternalInput    []string `protobuf:"bytes,7,rep,name=external_input,json=externalInput" json:"external_input,omitempty"`
	ExternalOutput   []string `protobuf:"bytes,8,rep,name=external_output,json=externalOutput" json:"external_output,omitempty"`
	XXX_unrecognized []byte   `json:"-"`
}

Network definition.

func (*NetDef) Descriptor 

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

func (*NetDef) GetArg 

func (m *NetDef) GetArg() []*Argument

func (*NetDef) GetDeviceOption 

func (m *NetDef) GetDeviceOption() *DeviceOption

func (*NetDef) GetExternalInput 

func (m *NetDef) GetExternalInput() []string

func (*NetDef) GetExternalOutput 

func (m *NetDef) GetExternalOutput() []string

func (*NetDef) GetName 

func (m *NetDef) GetName() string

func (*NetDef) GetNumWorkers 

func (m *NetDef) GetNumWorkers() int32

func (*NetDef) GetOp 

func (m *NetDef) GetOp() []*OperatorDef

func (*NetDef) GetType 

func (m *NetDef) GetType() string

func (*NetDef) ProtoMessage 

func (*NetDef) ProtoMessage()

func (*NetDef) Reset 

func (m *NetDef) Reset()

func (*NetDef) String 

func (m *NetDef) String() string

type NetsMap 

type NetsMap struct {
	Key              *string `protobuf:"bytes,1,req,name=key" json:"key,omitempty"`
	Value            *NetDef `protobuf:"bytes,2,req,name=value" json:"value,omitempty"`
	XXX_unrecognized []byte  `json:"-"`
}

func (*NetsMap) Descriptor 

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

func (*NetsMap) GetKey 

func (m *NetsMap) GetKey() string

func (*NetsMap) GetValue 

func (m *NetsMap) GetValue() *NetDef

func (*NetsMap) ProtoMessage 

func (*NetsMap) ProtoMessage()

func (*NetsMap) Reset 

func (m *NetsMap) Reset()

func (*NetsMap) String 

func (m *NetsMap) String() string

type NodeProto 

type NodeProto struct {
	// Links to non-terminal children nodes
	Children []*NodeProto `protobuf:"bytes,1,rep,name=children" json:"children,omitempty"`
	// Links to terminal (leaf) nodes
	WordIds          []int32   `protobuf:"varint,2,rep,name=word_ids,json=wordIds" json:"word_ids,omitempty"`
	Offset           *int32    `protobuf:"varint,3,opt,name=offset" json:"offset,omitempty"`
	Name             *string   `protobuf:"bytes,4,opt,name=name" json:"name,omitempty"`
	Scores           []float32 `protobuf:"fixed32,5,rep,name=scores" json:"scores,omitempty"`
	XXX_unrecognized []byte    `json:"-"`
}

Each node in the hierarchy contains links to either leaf nodes or more non-terminal nodes

func (*NodeProto) Descriptor 

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

func (*NodeProto) GetChildren 

func (m *NodeProto) GetChildren() []*NodeProto

func (*NodeProto) GetName 

func (m *NodeProto) GetName() string

func (*NodeProto) GetOffset 

func (m *NodeProto) GetOffset() int32

func (*NodeProto) GetScores 

func (m *NodeProto) GetScores() []float32

func (*NodeProto) GetWordIds 

func (m *NodeProto) GetWordIds() []int32

func (*NodeProto) ProtoMessage 

func (*NodeProto) ProtoMessage()

func (*NodeProto) Reset 

func (m *NodeProto) Reset()

func (*NodeProto) String 

func (m *NodeProto) String() string

type OperatorDef 

type OperatorDef struct {
	Input  []string `protobuf:"bytes,1,rep,name=input" json:"input,omitempty"`
	Output []string `protobuf:"bytes,2,rep,name=output" json:"output,omitempty"`
	Name   *string  `protobuf:"bytes,3,opt,name=name" json:"name,omitempty"`
	// the operator type. This is needed to create the object from the operator
	// registry.
	Type *string `protobuf:"bytes,4,opt,name=type" json:"type,omitempty"`
	// arg is for the argument defined in operator schema
	Arg []*Argument `protobuf:"bytes,5,rep,name=arg" json:"arg,omitempty"`
	// The device option that the operator should run under.
	DeviceOption *DeviceOption `protobuf:"bytes,6,opt,name=device_option,json=deviceOption" json:"device_option,omitempty"`
	// Optionally, one can specify an engine when there are multiple
	// implementations available simultaneously for one device type.
	// If one specifies an engine but that engine does not exist in the compiled
	// Caffe2 binary, Caffe2 will fall back to the default engine of that device
	// type.
	Engine *string `protobuf:"bytes,7,opt,name=engine" json:"engine,omitempty"`
	// Additional 'fake' inputs used for expressing control dependencies
	// in the operator graph. This can be used to ensure that an
	// operator does not run until another operator is ready, for e.g.
	// scheduling control. These are not passed as actual inputs to the
	// Operator implementation, and are only used by the Net class for
	// scheduling purposes.
	ControlInput []string `protobuf:"bytes,8,rep,name=control_input,json=controlInput" json:"control_input,omitempty"`
	// is_gradient_op argument is only used as a hint in shape inference
	// and has no runtime significance
	IsGradientOp *bool `protobuf:"varint,9,opt,name=is_gradient_op,json=isGradientOp,def=0" json:"is_gradient_op,omitempty"`
	// debug information associated with the construction of the operator.
	// This is an optional string with no assumed characteristics as
	// operators can be constructed in any language.
	DebugInfo *string `protobuf:"bytes,10,opt,name=debug_info,json=debugInfo" json:"debug_info,omitempty"`
	// the domain of the operator to help runtime distinguish which operator
	// library this OperatorDef refers to. For example, both caffe2 and aten
	// has `Add` operator, with domain, we can easily decide which operator
	// to execute. to support multiple operator libs, we use domain to
	// distinguish which operator lib we refer to:
	//   - "caffe2" means this uses Caffe2 operator library
	//   - "aten" means this uses ATen operator library
	//   - "c10" is for the fused library
	//   - if the domain is missing or empty, we use "caffe2", this is for
	//     legacy models, new serializer should always export an OperatorDef
	//     with domain and op_version
	Domain *string `protobuf:"bytes,11,opt,name=domain" json:"domain,omitempty"`
	// each operator is has its own version number.
	// operator version information
	// each time, we change the API or semantics of the operator,
	// we bump the version for the operator.
	// the runtime system should check the op_version of each OperatorDef
	// and decide it should reject or accept the model
	OpVersion        *int64 `protobuf:"varint,12,opt,name=op_version,json=opVersion" json:"op_version,omitempty"`
	XXX_unrecognized []byte `json:"-"`
}

Operator Definition.

func (*OperatorDef) Descriptor 

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

func (*OperatorDef) GetArg 

func (m *OperatorDef) GetArg() []*Argument

func (*OperatorDef) GetControlInput 

func (m *OperatorDef) GetControlInput() []string

func (*OperatorDef) GetDebugInfo 

func (m *OperatorDef) GetDebugInfo() string

func (*OperatorDef) GetDeviceOption 

func (m *OperatorDef) GetDeviceOption() *DeviceOption

func (*OperatorDef) GetDomain 

func (m *OperatorDef) GetDomain() string

func (*OperatorDef) GetEngine 

func (m *OperatorDef) GetEngine() string

func (*OperatorDef) GetInput 

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

func (*OperatorDef) GetIsGradientOp 

func (m *OperatorDef) GetIsGradientOp() bool

func (*OperatorDef) GetName 

func (m *OperatorDef) GetName() string

func (*OperatorDef) GetOpVersion 

func (m *OperatorDef) GetOpVersion() int64

func (*OperatorDef) GetOutput 

func (m *OperatorDef) GetOutput() []string

func (*OperatorDef) GetType 

func (m *OperatorDef) GetType() string

func (*OperatorDef) ProtoMessage 

func (*OperatorDef) ProtoMessage()

func (*OperatorDef) Reset 

func (m *OperatorDef) Reset()

func (*OperatorDef) String 

func (m *OperatorDef) String() string

type PathNodeProto 

type PathNodeProto struct {
	// Parameter matrix offset for this node
	Index *int32 `protobuf:"varint,1,opt,name=index" json:"index,omitempty"`
	// Number of children
	Length *int32 `protobuf:"varint,2,opt,name=length" json:"length,omitempty"`
	// Index of the next node in the path
	Target           *int32 `protobuf:"varint,3,opt,name=target" json:"target,omitempty"`
	XXX_unrecognized []byte `json:"-"`
}

Represents a node in the path from the root node all the way down to the word (leaf).

func (*PathNodeProto) Descriptor 

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

func (*PathNodeProto) GetIndex 

func (m *PathNodeProto) GetIndex() int32

func (*PathNodeProto) GetLength 

func (m *PathNodeProto) GetLength() int32

func (*PathNodeProto) GetTarget 

func (m *PathNodeProto) GetTarget() int32

func (*PathNodeProto) ProtoMessage 

func (*PathNodeProto) ProtoMessage()

func (*PathNodeProto) Reset 

func (m *PathNodeProto) Reset()

func (*PathNodeProto) String 

func (m *PathNodeProto) String() string

type PathProto 

type PathProto struct {
	WordId           *int32           `protobuf:"varint,1,opt,name=word_id,json=wordId" json:"word_id,omitempty"`
	PathNodes        []*PathNodeProto `protobuf:"bytes,2,rep,name=path_nodes,json=pathNodes" json:"path_nodes,omitempty"`
	XXX_unrecognized []byte           `json:"-"`
}

Each PathProto belongs to a word and is an array of nodes in the path from the root to the leaf (which is the word itself) in the tree.

func (*PathProto) Descriptor 

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

func (*PathProto) GetPathNodes 

func (m *PathProto) GetPathNodes() []*PathNodeProto

func (*PathProto) GetWordId 

func (m *PathProto) GetWordId() int32

func (*PathProto) ProtoMessage 

func (*PathProto) ProtoMessage()

func (*PathProto) Reset 

func (m *PathProto) Reset()

func (*PathProto) String 

func (m *PathProto) String() string

type PlanDef 

type PlanDef struct {
	// All the networks that are used in this execution. Note that networks should
	// be ordered in the way they are executed, i.e. for a layer in a network, all
	// its input blobs should already have been initialized by the layers or
	// networks defined before it.
	Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	// The networks that are going to be used in this plan.
	Network          []*NetDef        `protobuf:"bytes,2,rep,name=network" json:"network,omitempty"`
	ExecutionStep    []*ExecutionStep `protobuf:"bytes,3,rep,name=execution_step,json=executionStep" json:"execution_step,omitempty"`
	XXX_unrecognized []byte           `json:"-"`
}

func (*PlanDef) Descriptor 

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

func (*PlanDef) GetExecutionStep 

func (m *PlanDef) GetExecutionStep() []*ExecutionStep

func (*PlanDef) GetName 

func (m *PlanDef) GetName() string

func (*PlanDef) GetNetwork 

func (m *PlanDef) GetNetwork() []*NetDef

func (*PlanDef) ProtoMessage 

func (*PlanDef) ProtoMessage()

func (*PlanDef) Reset 

func (m *PlanDef) Reset()

func (*PlanDef) String 

func (m *PlanDef) String() string

type PlansMap 

type PlansMap struct {
	Key              *string  `protobuf:"bytes,1,req,name=key" json:"key,omitempty"`
	Value            *PlanDef `protobuf:"bytes,2,req,name=value" json:"value,omitempty"`
	XXX_unrecognized []byte   `json:"-"`
}

func (*PlansMap) Descriptor 

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

func (*PlansMap) GetKey 

func (m *PlansMap) GetKey() string

func (*PlansMap) GetValue 

func (m *PlansMap) GetValue() *PlanDef

func (*PlansMap) ProtoMessage 

func (*PlansMap) ProtoMessage()

func (*PlansMap) Reset 

func (m *PlansMap) Reset()

func (*PlansMap) String 

func (m *PlansMap) String() string

type PredictorConsts 

type PredictorConsts struct {
	// Important - to ensure ordered traversal of the DB, these must be
	// set in the given (lexicographic) order in the input DBReader.
	META_NET_DEF       *string `protobuf:"bytes,1,opt,name=META_NET_DEF,json=METANETDEF,def=!!META_NET_DEF" json:"META_NET_DEF,omitempty"`
	PREDICTOR_DBREADER *string `` /* 130-byte string literal not displayed */
	// Blob types used in MetaNetDef blobs
	PARAMETERS_BLOB_TYPE *string `` /* 135-byte string literal not displayed */
	INPUTS_BLOB_TYPE     *string `protobuf:"bytes,4,opt,name=INPUTS_BLOB_TYPE,json=INPUTSBLOBTYPE,def=INPUTS_BLOB_TYPE" json:"INPUTS_BLOB_TYPE,omitempty"`
	OUTPUTS_BLOB_TYPE    *string `protobuf:"bytes,5,opt,name=OUTPUTS_BLOB_TYPE,json=OUTPUTSBLOBTYPE,def=OUTPUTS_BLOB_TYPE" json:"OUTPUTS_BLOB_TYPE,omitempty"`
	// Net types used in MetaNetDef nets
	GLOBAL_INIT_NET_TYPE  *string `` /* 134-byte string literal not displayed */
	PREDICT_INIT_NET_TYPE *string `` /* 138-byte string literal not displayed */
	PREDICT_NET_TYPE      *string `protobuf:"bytes,8,opt,name=PREDICT_NET_TYPE,json=PREDICTNETTYPE,def=PREDICT_NET_TYPE" json:"PREDICT_NET_TYPE,omitempty"`
	SINGLE_PREDICTOR      *string `protobuf:"bytes,9,opt,name=SINGLE_PREDICTOR,json=SINGLEPREDICTOR,def=SINGLE_PREDICTOR" json:"SINGLE_PREDICTOR,omitempty"`
	MULTI_PREDICTOR       *string `protobuf:"bytes,10,opt,name=MULTI_PREDICTOR,json=MULTIPREDICTOR,def=MULTI_PREDICTOR" json:"MULTI_PREDICTOR,omitempty"`
	TRAIN_INIT_PLAN_TYPE  *string `` /* 135-byte string literal not displayed */
	TRAIN_PLAN_TYPE       *string `protobuf:"bytes,12,opt,name=TRAIN_PLAN_TYPE,json=TRAINPLANTYPE,def=TRAIN_PLAN_TYPE" json:"TRAIN_PLAN_TYPE,omitempty"`
	XXX_unrecognized      []byte  `json:"-"`
}

func (*PredictorConsts) Descriptor 

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

func (*PredictorConsts) GetGLOBAL_INIT_NET_TYPE 

func (m *PredictorConsts) GetGLOBAL_INIT_NET_TYPE() string

func (*PredictorConsts) GetINPUTS_BLOB_TYPE 

func (m *PredictorConsts) GetINPUTS_BLOB_TYPE() string

func (*PredictorConsts) GetMETA_NET_DEF 

func (m *PredictorConsts) GetMETA_NET_DEF() string

func (*PredictorConsts) GetMULTI_PREDICTOR 

func (m *PredictorConsts) GetMULTI_PREDICTOR() string

func (*PredictorConsts) GetOUTPUTS_BLOB_TYPE 

func (m *PredictorConsts) GetOUTPUTS_BLOB_TYPE() string

func (*PredictorConsts) GetPARAMETERS_BLOB_TYPE 

func (m *PredictorConsts) GetPARAMETERS_BLOB_TYPE() string

func (*PredictorConsts) GetPREDICTOR_DBREADER 

func (m *PredictorConsts) GetPREDICTOR_DBREADER() string

func (*PredictorConsts) GetPREDICT_INIT_NET_TYPE 

func (m *PredictorConsts) GetPREDICT_INIT_NET_TYPE() string

func (*PredictorConsts) GetPREDICT_NET_TYPE 

func (m *PredictorConsts) GetPREDICT_NET_TYPE() string

func (*PredictorConsts) GetSINGLE_PREDICTOR 

func (m *PredictorConsts) GetSINGLE_PREDICTOR() string

func (*PredictorConsts) GetTRAIN_INIT_PLAN_TYPE 

func (m *PredictorConsts) GetTRAIN_INIT_PLAN_TYPE() string

func (*PredictorConsts) GetTRAIN_PLAN_TYPE 

func (m *PredictorConsts) GetTRAIN_PLAN_TYPE() string

func (*PredictorConsts) ProtoMessage 

func (*PredictorConsts) ProtoMessage()

func (*PredictorConsts) Reset 

func (m *PredictorConsts) Reset()

func (*PredictorConsts) String 

func (m *PredictorConsts) String() string

type ProfDAGProto 

type ProfDAGProto struct {
	// The name for the operator
	Name *string `protobuf:"bytes,1,req,name=name" json:"name,omitempty"`
	// The mean execution time
	Mean *float32 `protobuf:"fixed32,2,req,name=mean" json:"mean,omitempty"`
	// The standard deviation
	Stddev *float32 `protobuf:"fixed32,3,req,name=stddev" json:"stddev,omitempty"`
	// New field to represent the numbers above, and with count.
	ExecutionTime *TwoNumberStatsProto `protobuf:"bytes,4,opt,name=execution_time,json=executionTime" json:"execution_time,omitempty"`
	// Blob profiles that this node outputs.
	OutputProfile    []*BlobProfile `protobuf:"bytes,5,rep,name=output_profile,json=outputProfile" json:"output_profile,omitempty"`
	XXX_unrecognized []byte         `json:"-"`
}

Protobuf format to serialize profiler data.

func (*ProfDAGProto) Descriptor 

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

func (*ProfDAGProto) GetExecutionTime 

func (m *ProfDAGProto) GetExecutionTime() *TwoNumberStatsProto

func (*ProfDAGProto) GetMean 

func (m *ProfDAGProto) GetMean() float32

func (*ProfDAGProto) GetName 

func (m *ProfDAGProto) GetName() string

func (*ProfDAGProto) GetOutputProfile 

func (m *ProfDAGProto) GetOutputProfile() []*BlobProfile

func (*ProfDAGProto) GetStddev 

func (m *ProfDAGProto) GetStddev() float32

func (*ProfDAGProto) ProtoMessage 

func (*ProfDAGProto) ProtoMessage()

func (*ProfDAGProto) Reset 

func (m *ProfDAGProto) Reset()

func (*ProfDAGProto) String 

func (m *ProfDAGProto) String() string

type ProfDAGProtos 

type ProfDAGProtos struct {
	Stats            []*ProfDAGProto `protobuf:"bytes,1,rep,name=stats" json:"stats,omitempty"`
	NetName          *string         `protobuf:"bytes,2,opt,name=net_name,json=netName" json:"net_name,omitempty"`
	XXX_unrecognized []byte          `json:"-"`
}

Operator profiling information.

Note: The indices for elements of 'stats' and the indices of 'output_profile' inside each 'stats' are assumed to match the indices of 'op' elements of a corresponding NetDef and the 'output' indices within each 'op'.

func (*ProfDAGProtos) Descriptor 

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

func (*ProfDAGProtos) GetNetName 

func (m *ProfDAGProtos) GetNetName() string

func (*ProfDAGProtos) GetStats 

func (m *ProfDAGProtos) GetStats() []*ProfDAGProto

func (*ProfDAGProtos) ProtoMessage 

func (*ProfDAGProtos) ProtoMessage()

func (*ProfDAGProtos) Reset 

func (m *ProfDAGProtos) Reset()

func (*ProfDAGProtos) String 

func (m *ProfDAGProtos) String() string

type QTensorProto 

type QTensorProto struct {
	Dims             []int64               `protobuf:"varint,1,rep,name=dims" json:"dims,omitempty"`
	Precision        *int32                `protobuf:"varint,2,req,name=precision" json:"precision,omitempty"`
	Scale            *float64              `protobuf:"fixed64,3,req,name=scale" json:"scale,omitempty"`
	Bias             *float64              `protobuf:"fixed64,4,req,name=bias" json:"bias,omitempty"`
	IsSigned         *bool                 `protobuf:"varint,5,req,name=is_signed,json=isSigned" json:"is_signed,omitempty"`
	Data             []int32               `protobuf:"varint,6,rep,packed,name=data" json:"data,omitempty"`
	Name             *string               `protobuf:"bytes,7,opt,name=name" json:"name,omitempty"`
	DataType         *TensorProto_DataType `protobuf:"varint,8,opt,name=data_type,json=dataType,enum=caffe2.TensorProto_DataType,def=2" json:"data_type,omitempty"`
	XXX_unrecognized []byte                `json:"-"`
}

func (*QTensorProto) Descriptor 

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

func (*QTensorProto) GetBias 

func (m *QTensorProto) GetBias() float64

func (*QTensorProto) GetData 

func (m *QTensorProto) GetData() []int32

func (*QTensorProto) GetDataType 

func (m *QTensorProto) GetDataType() TensorProto_DataType

func (*QTensorProto) GetDims 

func (m *QTensorProto) GetDims() []int64

func (*QTensorProto) GetIsSigned 

func (m *QTensorProto) GetIsSigned() bool

func (*QTensorProto) GetName 

func (m *QTensorProto) GetName() string

func (*QTensorProto) GetPrecision 

func (m *QTensorProto) GetPrecision() int32

func (*QTensorProto) GetScale 

func (m *QTensorProto) GetScale() float64

func (*QTensorProto) ProtoMessage 

func (*QTensorProto) ProtoMessage()

func (*QTensorProto) Reset 

func (m *QTensorProto) Reset()

func (*QTensorProto) String 

func (m *QTensorProto) String() string

type StringMap 

type StringMap struct {
	Key              *string `protobuf:"bytes,1,req,name=key" json:"key,omitempty"`
	Value            *string `protobuf:"bytes,2,req,name=value" json:"value,omitempty"`
	XXX_unrecognized []byte  `json:"-"`
}

func (*StringMap) Descriptor 

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

func (*StringMap) GetKey 

func (m *StringMap) GetKey() string

func (*StringMap) GetValue 

func (m *StringMap) GetValue() string

func (*StringMap) ProtoMessage 

func (*StringMap) ProtoMessage()

func (*StringMap) Reset 

func (m *StringMap) Reset()

func (*StringMap) String 

func (m *StringMap) String() string

type TensorProto 

type TensorProto struct {
	// The dimensions in the tensor.
	Dims        []int64                  `protobuf:"varint,1,rep,name=dims" json:"dims,omitempty"`
	DataType    *TensorProto_DataType    `protobuf:"varint,2,opt,name=data_type,json=dataType,enum=caffe2.TensorProto_DataType,def=1" json:"data_type,omitempty"`
	StorageType *TensorProto_StorageType `` /* 131-byte string literal not displayed */
	// For float
	FloatData []float32 `protobuf:"fixed32,3,rep,packed,name=float_data,json=floatData" json:"float_data,omitempty"`
	// For int32, uint8, int8, uint16, int16, bool, and float16
	// Note about float16: in storage we will basically convert float16 byte-wise
	// to unsigned short and then store them in the int32_data field.
	Int32Data []int32 `protobuf:"varint,4,rep,packed,name=int32_data,json=int32Data" json:"int32_data,omitempty"`
	// For bytes
	ByteData []byte `protobuf:"bytes,5,opt,name=byte_data,json=byteData" json:"byte_data,omitempty"`
	// For strings
	StringData [][]byte `protobuf:"bytes,6,rep,name=string_data,json=stringData" json:"string_data,omitempty"`
	// For double
	DoubleData []float64 `protobuf:"fixed64,9,rep,packed,name=double_data,json=doubleData" json:"double_data,omitempty"`
	// For int64
	Int64Data []int64 `protobuf:"varint,10,rep,packed,name=int64_data,json=int64Data" json:"int64_data,omitempty"`
	// store the raw data, contents are serialized as little-endian
	RawData []byte `protobuf:"bytes,13,opt,name=raw_data,json=rawData" json:"raw_data,omitempty"`
	// store the pointer to the data
	ExternalData *ExternalDataProto `protobuf:"bytes,14,opt,name=external_data,json=externalData" json:"external_data,omitempty"`
	// Optionally, a name for the tensor.
	Name *string `protobuf:"bytes,7,opt,name=name" json:"name,omitempty"`
	// Optionally, a TensorProto can contain the details about the device that
	// it was serialized from. This is useful in cases like snapshotting a whole
	// workspace in a multi-GPU environment.
	DeviceDetail     *DeviceOption        `protobuf:"bytes,8,opt,name=device_detail,json=deviceDetail" json:"device_detail,omitempty"`
	Segment          *TensorProto_Segment `protobuf:"bytes,11,opt,name=segment" json:"segment,omitempty"`
	XXX_unrecognized []byte               `json:"-"`
}

TensorProto stores serialized Tensor objects.

func (*TensorProto) Descriptor 

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

func (*TensorProto) GetByteData 

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

func (*TensorProto) GetDataType 

func (m *TensorProto) GetDataType() TensorProto_DataType

func (*TensorProto) GetDeviceDetail 

func (m *TensorProto) GetDeviceDetail() *DeviceOption

func (*TensorProto) GetDims 

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

func (*TensorProto) GetDoubleData 

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

func (*TensorProto) GetExternalData 

func (m *TensorProto) GetExternalData() *ExternalDataProto

func (*TensorProto) GetFloatData 

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

func (*TensorProto) GetInt32Data 

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

func (*TensorProto) GetInt64Data 

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

func (*TensorProto) GetName 

func (m *TensorProto) GetName() string

func (*TensorProto) GetRawData 

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

func (*TensorProto) GetSegment 

func (m *TensorProto) GetSegment() *TensorProto_Segment

func (*TensorProto) GetStorageType 

func (m *TensorProto) GetStorageType() TensorProto_StorageType

func (*TensorProto) GetStringData 

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

func (*TensorProto) ProtoMessage 

func (*TensorProto) ProtoMessage()

func (*TensorProto) Reset 

func (m *TensorProto) Reset()

func (*TensorProto) String 

func (m *TensorProto) String() string

type TensorProto_DataType 

type TensorProto_DataType int32

Data type 

const (
	TensorProto_UNDEFINED TensorProto_DataType = 0
	// Basic types
	TensorProto_FLOAT  TensorProto_DataType = 1
	TensorProto_INT32  TensorProto_DataType = 2
	TensorProto_BYTE   TensorProto_DataType = 3
	TensorProto_STRING TensorProto_DataType = 4
	// Less-commonly used data types
	TensorProto_BOOL    TensorProto_DataType = 5
	TensorProto_UINT8   TensorProto_DataType = 6
	TensorProto_INT8    TensorProto_DataType = 7
	TensorProto_UINT16  TensorProto_DataType = 8
	TensorProto_INT16   TensorProto_DataType = 9
	TensorProto_INT64   TensorProto_DataType = 10
	TensorProto_FLOAT16 TensorProto_DataType = 12
	TensorProto_DOUBLE  TensorProto_DataType = 13
)
const Default_QTensorProto_DataType TensorProto_DataType = TensorProto_INT32
const Default_TensorProto_DataType TensorProto_DataType = TensorProto_FLOAT
const Default_TensorShape_DataType TensorProto_DataType = TensorProto_FLOAT

func (TensorProto_DataType) Enum 

func (x TensorProto_DataType) Enum() *TensorProto_DataType

func (TensorProto_DataType) EnumDescriptor 

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

func (TensorProto_DataType) String 

func (x TensorProto_DataType) String() string

func (*TensorProto_DataType) UnmarshalJSON 

func (x *TensorProto_DataType) UnmarshalJSON(data []byte) error

type TensorProto_Segment 

type TensorProto_Segment struct {
	Begin            *int64 `protobuf:"varint,1,req,name=begin" json:"begin,omitempty"`
	End              *int64 `protobuf:"varint,2,req,name=end" json:"end,omitempty"`
	XXX_unrecognized []byte `json:"-"`
}

When loading from chunks this is going to indicate where to put data in the full array. When not used full data have to be present

func (*TensorProto_Segment) Descriptor 

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

func (*TensorProto_Segment) GetBegin 

func (m *TensorProto_Segment) GetBegin() int64

func (*TensorProto_Segment) GetEnd 

func (m *TensorProto_Segment) GetEnd() int64

func (*TensorProto_Segment) ProtoMessage 

func (*TensorProto_Segment) ProtoMessage()

func (*TensorProto_Segment) Reset 

func (m *TensorProto_Segment) Reset()

func (*TensorProto_Segment) String 

func (m *TensorProto_Segment) String() string

type TensorProto_StorageType 

type TensorProto_StorageType int32

data storage 

const (
	// the content is stored in typed field, for example, if the data_type is
	// FLOAT, float_data is used to store the content.
	TensorProto_TYPED TensorProto_StorageType = 1
	// the content is serialized in field raw_data as little-endian
	TensorProto_RAW TensorProto_StorageType = 2
	// the pointer to the content is stored in field external_data
	// the content is serialized as little-endian
	TensorProto_EXTERNAL TensorProto_StorageType = 3
	// When StorageType is NO_CONTENT, we use TensorProto to store only type
	// and shape information. Reuse TensorProto to store type and shape
	// because we can just have one proto, not having another ValueInfoProto
	TensorProto_NO_CONTENT TensorProto_StorageType = 4
)
const Default_TensorProto_StorageType TensorProto_StorageType = TensorProto_TYPED

func (TensorProto_StorageType) Enum 

func (x TensorProto_StorageType) Enum() *TensorProto_StorageType

func (TensorProto_StorageType) EnumDescriptor 

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

func (TensorProto_StorageType) String 

func (x TensorProto_StorageType) String() string

func (*TensorProto_StorageType) UnmarshalJSON 

func (x *TensorProto_StorageType) UnmarshalJSON(data []byte) error

type TensorProtos 

type TensorProtos struct {
	Protos           []*TensorProto `protobuf:"bytes,1,rep,name=protos" json:"protos,omitempty"`
	XXX_unrecognized []byte         `json:"-"`
}

TensorProtos stores multiple TensorProto objects in one single proto. This is useful for small tensors; For anything big, consider using a DB for storage.

func (*TensorProtos) Descriptor 

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

func (*TensorProtos) GetProtos 

func (m *TensorProtos) GetProtos() []*TensorProto

func (*TensorProtos) ProtoMessage 

func (*TensorProtos) ProtoMessage()

func (*TensorProtos) Reset 

func (m *TensorProtos) Reset()

func (*TensorProtos) String 

func (m *TensorProtos) String() string

type TensorShape 

type TensorShape struct {
	Dims             []int64               `protobuf:"varint,1,rep,name=dims" json:"dims,omitempty"`
	DataType         *TensorProto_DataType `protobuf:"varint,2,opt,name=data_type,json=dataType,enum=caffe2.TensorProto_DataType,def=1" json:"data_type,omitempty"`
	UnknownDims      []int32               `protobuf:"varint,3,rep,name=unknown_dims,json=unknownDims" json:"unknown_dims,omitempty"`
	UnknownShape     *bool                 `protobuf:"varint,4,opt,name=unknown_shape,json=unknownShape,def=0" json:"unknown_shape,omitempty"`
	Name             *string               `protobuf:"bytes,5,opt,name=name" json:"name,omitempty"`
	XXX_unrecognized []byte                `json:"-"`
}

func (*TensorShape) Descriptor 

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

func (*TensorShape) GetDataType 

func (m *TensorShape) GetDataType() TensorProto_DataType

func (*TensorShape) GetDims 

func (m *TensorShape) GetDims() []int64

func (*TensorShape) GetName 

func (m *TensorShape) GetName() string

func (*TensorShape) GetUnknownDims 

func (m *TensorShape) GetUnknownDims() []int32

func (*TensorShape) GetUnknownShape 

func (m *TensorShape) GetUnknownShape() bool

func (*TensorShape) ProtoMessage 

func (*TensorShape) ProtoMessage()

func (*TensorShape) Reset 

func (m *TensorShape) Reset()

func (*TensorShape) String 

func (m *TensorShape) String() string

type TensorShapes 

type TensorShapes struct {
	Shapes           []*TensorShape `protobuf:"bytes,1,rep,name=shapes" json:"shapes,omitempty"`
	XXX_unrecognized []byte         `json:"-"`
}

func (*TensorShapes) Descriptor 

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

func (*TensorShapes) GetShapes 

func (m *TensorShapes) GetShapes() []*TensorShape

func (*TensorShapes) ProtoMessage 

func (*TensorShapes) ProtoMessage()

func (*TensorShapes) Reset 

func (m *TensorShapes) Reset()

func (*TensorShapes) String 

func (m *TensorShapes) String() string

type TreeProto 

type TreeProto struct {
	RootNode         *NodeProto `protobuf:"bytes,1,opt,name=root_node,json=rootNode" json:"root_node,omitempty"`
	XXX_unrecognized []byte     `json:"-"`
}

Protobuf format to accept hierarchy for hierarchical softmax operator. TreeProto points to the root node.

func (*TreeProto) Descriptor 

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

func (*TreeProto) GetRootNode 

func (m *TreeProto) GetRootNode() *NodeProto

func (*TreeProto) ProtoMessage 

func (*TreeProto) ProtoMessage()

func (*TreeProto) Reset 

func (m *TreeProto) Reset()

func (*TreeProto) String 

func (m *TreeProto) String() string

type TwoNumberStatsProto 

type TwoNumberStatsProto struct {
	Mean             *float32 `protobuf:"fixed32,1,opt,name=mean" json:"mean,omitempty"`
	Stddev           *float32 `protobuf:"fixed32,2,opt,name=stddev" json:"stddev,omitempty"`
	Count            *int64   `protobuf:"varint,3,opt,name=count" json:"count,omitempty"`
	XXX_unrecognized []byte   `json:"-"`
}

A two number summary for a value. It also has count for restoring.

func (*TwoNumberStatsProto) Descriptor 

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

func (*TwoNumberStatsProto) GetCount 

func (m *TwoNumberStatsProto) GetCount() int64

func (*TwoNumberStatsProto) GetMean 

func (m *TwoNumberStatsProto) GetMean() float32

func (*TwoNumberStatsProto) GetStddev 

func (m *TwoNumberStatsProto) GetStddev() float32

func (*TwoNumberStatsProto) ProtoMessage 

func (*TwoNumberStatsProto) ProtoMessage()

func (*TwoNumberStatsProto) Reset 

func (m *TwoNumberStatsProto) Reset()

func (*TwoNumberStatsProto) String 

func (m *TwoNumberStatsProto) String() string

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.