caffe2

package module
v0.3.16-0...-affcf57 Latest Latest
Warning

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

 Go to latest Published: Apr 5, 2019 License: NCSA Imports: 18 Imported by: 0

README

MLModelScope Caffe2 Agent

Build Status Build Status Go Report Card License

Documentation

Index

Constants

View Source 
const Default_DeviceOption_DeviceType int32 = 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 (
	ErrInvalidLengthCaffe2 = fmt.Errorf("proto: negative length found during unmarshaling")
	ErrIntOverflowCaffe2   = fmt.Errorf("proto: integer overflow")
)
View Source 
var (
	ErrInvalidLengthHsm = fmt.Errorf("proto: negative length found during unmarshaling")
	ErrIntOverflowHsm   = fmt.Errorf("proto: integer overflow")
)
View Source 
var (
	ErrInvalidLengthMetanet = fmt.Errorf("proto: negative length found during unmarshaling")
	ErrIntOverflowMetanet   = fmt.Errorf("proto: integer overflow")
)
View Source 
var (
	ErrInvalidLengthPredictorConsts = fmt.Errorf("proto: negative length found during unmarshaling")
	ErrIntOverflowPredictorConsts   = fmt.Errorf("proto: integer overflow")
)
View Source 
var (
	ErrInvalidLengthProfDag = fmt.Errorf("proto: negative length found during unmarshaling")
	ErrIntOverflowProfDag   = fmt.Errorf("proto: integer overflow")
)
View Source 
var (
	Version   = "0.3.15"
	BuildDate = "undefined"
	GitCommit = "undefined"
)

Version ...

View Source 
var DeviceType_name = map[int32]string{
	0:        "CPU",
	1:        "CUDA",
	2:        "MKLDNN",
	20901701: "ONLY_FOR_TEST",
}
View Source 
var DeviceType_value = map[string]int32{
	"CPU":           0,
	"CUDA":          1,
	"MKLDNN":        2,
	"ONLY_FOR_TEST": 20901701,
}
View Source 
var FrameworkManifest = dlframework.FrameworkManifest{
	Name:    "Caffe2",
	Version: "0.8.1",
	Container: map[string]*dlframework.ContainerHardware{
		"amd64": {
			Cpu: "raiproject/carml-caffe2:amd64-cpu",
			Gpu: "raiproject/carml-caffe2:amd64-gpu",
		},
		"ppc64le": {
			Cpu: "raiproject/carml-caffe2:ppc64le-gpu",
			Gpu: "raiproject/carml-caffe2:ppc64le-gpu",
		},
	},
}

FrameworkManifest ...

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,
}

Functions

func Asset 

func Asset(name string) ([]byte, error)

Asset loads and returns the asset for the given name. It returns an error if the asset could not be found or could not be loaded.

func AssetDir 

func AssetDir(name string) ([]string, error)

AssetDir returns the file names below a certain directory embedded in the file by go-bindata. For example if you run go-bindata on data/... and data contains the following hierarchy: 

data/
  foo.txt
  img/
    a.png
    b.png

then AssetDir("data") would return []string{"foo.txt", "img"} AssetDir("data/img") would return []string{"a.png", "b.png"} AssetDir("foo.txt") and AssetDir("notexist") would return an error AssetDir("") will return []string{"data"}.

func AssetInfo 

func AssetInfo(name string) (os.FileInfo, error)

AssetInfo loads and returns the asset info for the given name. It returns an error if the asset could not be found or could not be loaded.

func AssetNames 

func AssetNames() []string

AssetNames returns the names of the assets.

func MustAsset 

func MustAsset(name string) []byte

MustAsset is like Asset but panics when Asset would return an error. It simplifies safe initialization of global variables.

func Register 

func Register()

func RestoreAsset 

func RestoreAsset(dir, name string) error

RestoreAsset restores an asset under the given directory

func RestoreAssets 

func RestoreAssets(dir, name string) error

RestoreAssets restores an asset under the given directory recursively

Types

type Argument 

type Argument struct {
	Name    string    `protobuf:"bytes,1,opt,name=name" json:"name"`
	F       float32   `protobuf:"fixed32,2,opt,name=f" json:"f"`
	I       int64     `protobuf:"varint,3,opt,name=i" json:"i"`
	S       []byte    `protobuf:"bytes,4,opt,name=s" json:"s"`
	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"`
}

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

func (m *Argument) GetName() string

func (*Argument) GetS 

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

func (*Argument) GetStrings 

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

func (*Argument) Marshal 

func (m *Argument) Marshal() (dAtA []byte, err error)

func (*Argument) MarshalTo 

func (m *Argument) MarshalTo(dAtA []byte) (int, error)

func (*Argument) ProtoMessage 

func (*Argument) ProtoMessage()

func (*Argument) Reset 

func (m *Argument) Reset()

func (*Argument) Size 

func (m *Argument) Size() (n int)

func (*Argument) String 

func (m *Argument) String() string

func (*Argument) Unmarshal 

func (m *Argument) Unmarshal(dAtA []byte) error

func (*Argument) XXX_DiscardUnknown 

func (m *Argument) XXX_DiscardUnknown()

func (*Argument) XXX_Marshal 

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

func (*Argument) XXX_Merge 

func (dst *Argument) XXX_Merge(src proto.Message)

func (*Argument) XXX_Size 

func (m *Argument) XXX_Size() int

func (*Argument) XXX_Unmarshal 

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

type BlobProto 

type BlobProto struct {
	Name    string        `protobuf:"bytes,1,opt,name=name" json:"name"`
	Type    string        `protobuf:"bytes,2,opt,name=type" json:"type"`
	Tensor  *TensorProto  `protobuf:"bytes,3,opt,name=tensor" json:"tensor,omitempty"`
	Content []byte        `protobuf:"bytes,4,opt,name=content" json:"content"`
	Qtensor *QTensorProto `protobuf:"bytes,5,opt,name=qtensor" json:"qtensor,omitempty"`
}

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

func (m *BlobProto) Marshal() (dAtA []byte, err error)

func (*BlobProto) MarshalTo 

func (m *BlobProto) MarshalTo(dAtA []byte) (int, error)

func (*BlobProto) ProtoMessage 

func (*BlobProto) ProtoMessage()

func (*BlobProto) Reset 

func (m *BlobProto) Reset()

func (*BlobProto) Size 

func (m *BlobProto) Size() (n int)

func (*BlobProto) String 

func (m *BlobProto) String() string

func (*BlobProto) Unmarshal 

func (m *BlobProto) Unmarshal(dAtA []byte) error

func (*BlobProto) XXX_DiscardUnknown 

func (m *BlobProto) XXX_DiscardUnknown()

func (*BlobProto) XXX_Marshal 

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

func (*BlobProto) XXX_Merge 

func (dst *BlobProto) XXX_Merge(src proto.Message)

func (*BlobProto) XXX_Size 

func (m *BlobProto) XXX_Size() int

func (*BlobProto) XXX_Unmarshal 

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

type BlobsMap 

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

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

func (m *BlobsMap) Marshal() (dAtA []byte, err error)

func (*BlobsMap) MarshalTo 

func (m *BlobsMap) MarshalTo(dAtA []byte) (int, error)

func (*BlobsMap) ProtoMessage 

func (*BlobsMap) ProtoMessage()

func (*BlobsMap) Reset 

func (m *BlobsMap) Reset()

func (*BlobsMap) Size 

func (m *BlobsMap) Size() (n int)

func (*BlobsMap) String 

func (m *BlobsMap) String() string

func (*BlobsMap) Unmarshal 

func (m *BlobsMap) Unmarshal(dAtA []byte) error

func (*BlobsMap) XXX_DiscardUnknown 

func (m *BlobsMap) XXX_DiscardUnknown()

func (*BlobsMap) XXX_Marshal 

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

func (*BlobsMap) XXX_Merge 

func (dst *BlobsMap) XXX_Merge(src proto.Message)

func (*BlobsMap) XXX_Size 

func (m *BlobsMap) XXX_Size() int

func (*BlobsMap) XXX_Unmarshal 

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

type DBReaderProto 

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

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

func (m *DBReaderProto) Marshal() (dAtA []byte, err error)

func (*DBReaderProto) MarshalTo 

func (m *DBReaderProto) MarshalTo(dAtA []byte) (int, error)

func (*DBReaderProto) ProtoMessage 

func (*DBReaderProto) ProtoMessage()

func (*DBReaderProto) Reset 

func (m *DBReaderProto) Reset()

func (*DBReaderProto) Size 

func (m *DBReaderProto) Size() (n int)

func (*DBReaderProto) String 

func (m *DBReaderProto) String() string

func (*DBReaderProto) Unmarshal 

func (m *DBReaderProto) Unmarshal(dAtA []byte) error

func (*DBReaderProto) XXX_DiscardUnknown 

func (m *DBReaderProto) XXX_DiscardUnknown()

func (*DBReaderProto) XXX_Marshal 

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

func (*DBReaderProto) XXX_Merge 

func (dst *DBReaderProto) XXX_Merge(src proto.Message)

func (*DBReaderProto) XXX_Size 

func (m *DBReaderProto) XXX_Size() int

func (*DBReaderProto) XXX_Unmarshal 

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

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"`
	// [CUDA specific] the cuda gpu id.
	CudaGpuId int32 `protobuf:"varint,2,opt,name=cuda_gpu_id,json=cudaGpuId" json:"cuda_gpu_id"`
	// [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"`
}

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.

func (*DeviceOption) Descriptor 

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

func (*DeviceOption) GetCudaGpuId 

func (m *DeviceOption) GetCudaGpuId() int32

func (*DeviceOption) GetDeviceType 

func (m *DeviceOption) GetDeviceType() int32

func (*DeviceOption) GetRandomSeed 

func (m *DeviceOption) GetRandomSeed() uint32

func (*DeviceOption) Marshal 

func (m *DeviceOption) Marshal() (dAtA []byte, err error)

func (*DeviceOption) MarshalTo 

func (m *DeviceOption) MarshalTo(dAtA []byte) (int, error)

func (*DeviceOption) ProtoMessage 

func (*DeviceOption) ProtoMessage()

func (*DeviceOption) Reset 

func (m *DeviceOption) Reset()

func (*DeviceOption) Size 

func (m *DeviceOption) Size() (n int)

func (*DeviceOption) String 

func (m *DeviceOption) String() string

func (*DeviceOption) Unmarshal 

func (m *DeviceOption) Unmarshal(dAtA []byte) error

func (*DeviceOption) XXX_DiscardUnknown 

func (m *DeviceOption) XXX_DiscardUnknown()

func (*DeviceOption) XXX_Marshal 

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

func (*DeviceOption) XXX_Merge 

func (dst *DeviceOption) XXX_Merge(src proto.Message)

func (*DeviceOption) XXX_Size 

func (m *DeviceOption) XXX_Size() int

func (*DeviceOption) XXX_Unmarshal 

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

type DeviceType 

type DeviceType int32

DeviceType that Caffe2 currently supports. Note: if you add a device type, make sure you add the corresponding device line in core/blob_serialization.cc. 

const (
	DeviceType_CPU           DeviceType = 0
	DeviceType_CUDA          DeviceType = 1
	DeviceType_MKLDNN        DeviceType = 2
	DeviceType_ONLY_FOR_TEST DeviceType = 20901701
)

func (DeviceType) Enum 

func (x DeviceType) Enum() *DeviceType

func (DeviceType) EnumDescriptor 

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

func (DeviceType) String 

func (x DeviceType) String() string

func (*DeviceType) UnmarshalJSON 

func (x *DeviceType) 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"`
	// 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"`
	// 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"` // Deprecated: Do not use.
	// DEPRECATED. Use `run_every_ms`.
	ReportNet      string `protobuf:"bytes,7,opt,name=report_net,json=reportNet" json:"report_net"`
	ReportInterval int32  `protobuf:"varint,8,opt,name=report_interval,json=reportInterval" json:"report_interval"`
	// 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"`
	// 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"`
	// 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"`
	// 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"`
}

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) GetCriteriaNetwork deprecated 

func (m *ExecutionStep) GetCriteriaNetwork() string

Deprecated: Do not use.

func (*ExecutionStep) GetName 

func (m *ExecutionStep) GetName() string

func (*ExecutionStep) GetNetwork 

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

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

func (m *ExecutionStep) Marshal() (dAtA []byte, err error)

func (*ExecutionStep) MarshalTo 

func (m *ExecutionStep) MarshalTo(dAtA []byte) (int, error)

func (*ExecutionStep) ProtoMessage 

func (*ExecutionStep) ProtoMessage()

func (*ExecutionStep) Reset 

func (m *ExecutionStep) Reset()

func (*ExecutionStep) Size 

func (m *ExecutionStep) Size() (n int)

func (*ExecutionStep) String 

func (m *ExecutionStep) String() string

func (*ExecutionStep) Unmarshal 

func (m *ExecutionStep) Unmarshal(dAtA []byte) error

func (*ExecutionStep) XXX_DiscardUnknown 

func (m *ExecutionStep) XXX_DiscardUnknown()

func (*ExecutionStep) XXX_Marshal 

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

func (*ExecutionStep) XXX_Merge 

func (dst *ExecutionStep) XXX_Merge(src proto.Message)

func (*ExecutionStep) XXX_Size 

func (m *ExecutionStep) XXX_Size() int

func (*ExecutionStep) XXX_Unmarshal 

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

type HierarchyProto 

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

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

func (m *HierarchyProto) Marshal() (dAtA []byte, err error)

func (*HierarchyProto) MarshalTo 

func (m *HierarchyProto) MarshalTo(dAtA []byte) (int, error)

func (*HierarchyProto) ProtoMessage 

func (*HierarchyProto) ProtoMessage()

func (*HierarchyProto) Reset 

func (m *HierarchyProto) Reset()

func (*HierarchyProto) Size 

func (m *HierarchyProto) Size() (n int)

func (*HierarchyProto) String 

func (m *HierarchyProto) String() string

func (*HierarchyProto) Unmarshal 

func (m *HierarchyProto) Unmarshal(dAtA []byte) error

func (*HierarchyProto) XXX_DiscardUnknown 

func (m *HierarchyProto) XXX_DiscardUnknown()

func (*HierarchyProto) XXX_Marshal 

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

func (*HierarchyProto) XXX_Merge 

func (dst *HierarchyProto) XXX_Merge(src proto.Message)

func (*HierarchyProto) XXX_Size 

func (m *HierarchyProto) XXX_Size() int

func (*HierarchyProto) XXX_Unmarshal 

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

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"`
}

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

func (m *MetaNetDef) Marshal() (dAtA []byte, err error)

func (*MetaNetDef) MarshalTo 

func (m *MetaNetDef) MarshalTo(dAtA []byte) (int, error)

func (*MetaNetDef) ProtoMessage 

func (*MetaNetDef) ProtoMessage()

func (*MetaNetDef) Reset 

func (m *MetaNetDef) Reset()

func (*MetaNetDef) Size 

func (m *MetaNetDef) Size() (n int)

func (*MetaNetDef) String 

func (m *MetaNetDef) String() string

func (*MetaNetDef) Unmarshal 

func (m *MetaNetDef) Unmarshal(dAtA []byte) error

func (*MetaNetDef) XXX_DiscardUnknown 

func (m *MetaNetDef) XXX_DiscardUnknown()

func (*MetaNetDef) XXX_Marshal 

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

func (*MetaNetDef) XXX_Merge 

func (dst *MetaNetDef) XXX_Merge(src proto.Message)

func (*MetaNetDef) XXX_Size 

func (m *MetaNetDef) XXX_Size() int

func (*MetaNetDef) XXX_Unmarshal 

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

type ModelInfo 

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

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

func (m *ModelInfo) Marshal() (dAtA []byte, err error)

func (*ModelInfo) MarshalTo 

func (m *ModelInfo) MarshalTo(dAtA []byte) (int, error)

func (*ModelInfo) ProtoMessage 

func (*ModelInfo) ProtoMessage()

func (*ModelInfo) Reset 

func (m *ModelInfo) Reset()

func (*ModelInfo) Size 

func (m *ModelInfo) Size() (n int)

func (*ModelInfo) String 

func (m *ModelInfo) String() string

func (*ModelInfo) Unmarshal 

func (m *ModelInfo) Unmarshal(dAtA []byte) error

func (*ModelInfo) XXX_DiscardUnknown 

func (m *ModelInfo) XXX_DiscardUnknown()

func (*ModelInfo) XXX_Marshal 

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

func (*ModelInfo) XXX_Merge 

func (dst *ModelInfo) XXX_Merge(src proto.Message)

func (*ModelInfo) XXX_Size 

func (m *ModelInfo) XXX_Size() int

func (*ModelInfo) XXX_Unmarshal 

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

type NetDef 

type NetDef struct {
	Name string `protobuf:"bytes,1,opt,name=name" json:"name"`
	// 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"`
	// 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"` // Deprecated: Do not use.
	// 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"`
}

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 deprecated 

func (m *NetDef) GetNumWorkers() int32

Deprecated: Do not use.

func (*NetDef) GetOp 

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

func (*NetDef) GetType 

func (m *NetDef) GetType() string

func (*NetDef) Marshal 

func (m *NetDef) Marshal() (dAtA []byte, err error)

func (*NetDef) MarshalTo 

func (m *NetDef) MarshalTo(dAtA []byte) (int, error)

func (*NetDef) ProtoMessage 

func (*NetDef) ProtoMessage()

func (*NetDef) Reset 

func (m *NetDef) Reset()

func (*NetDef) Size 

func (m *NetDef) Size() (n int)

func (*NetDef) String 

func (m *NetDef) String() string

func (*NetDef) Unmarshal 

func (m *NetDef) Unmarshal(dAtA []byte) error

func (*NetDef) XXX_DiscardUnknown 

func (m *NetDef) XXX_DiscardUnknown()

func (*NetDef) XXX_Marshal 

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

func (*NetDef) XXX_Merge 

func (dst *NetDef) XXX_Merge(src proto.Message)

func (*NetDef) XXX_Size 

func (m *NetDef) XXX_Size() int

func (*NetDef) XXX_Unmarshal 

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

type NetsMap 

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

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

func (m *NetsMap) Marshal() (dAtA []byte, err error)

func (*NetsMap) MarshalTo 

func (m *NetsMap) MarshalTo(dAtA []byte) (int, error)

func (*NetsMap) ProtoMessage 

func (*NetsMap) ProtoMessage()

func (*NetsMap) Reset 

func (m *NetsMap) Reset()

func (*NetsMap) Size 

func (m *NetsMap) Size() (n int)

func (*NetsMap) String 

func (m *NetsMap) String() string

func (*NetsMap) Unmarshal 

func (m *NetsMap) Unmarshal(dAtA []byte) error

func (*NetsMap) XXX_DiscardUnknown 

func (m *NetsMap) XXX_DiscardUnknown()

func (*NetsMap) XXX_Marshal 

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

func (*NetsMap) XXX_Merge 

func (dst *NetsMap) XXX_Merge(src proto.Message)

func (*NetsMap) XXX_Size 

func (m *NetsMap) XXX_Size() int

func (*NetsMap) XXX_Unmarshal 

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

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"`
	Name    string    `protobuf:"bytes,4,opt,name=name" json:"name"`
	Scores  []float32 `protobuf:"fixed32,5,rep,name=scores" json:"scores,omitempty"`
}

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

func (m *NodeProto) Marshal() (dAtA []byte, err error)

func (*NodeProto) MarshalTo 

func (m *NodeProto) MarshalTo(dAtA []byte) (int, error)

func (*NodeProto) ProtoMessage 

func (*NodeProto) ProtoMessage()

func (*NodeProto) Reset 

func (m *NodeProto) Reset()

func (*NodeProto) Size 

func (m *NodeProto) Size() (n int)

func (*NodeProto) String 

func (m *NodeProto) String() string

func (*NodeProto) Unmarshal 

func (m *NodeProto) Unmarshal(dAtA []byte) error

func (*NodeProto) XXX_DiscardUnknown 

func (m *NodeProto) XXX_DiscardUnknown()

func (*NodeProto) XXX_Marshal 

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

func (*NodeProto) XXX_Merge 

func (dst *NodeProto) XXX_Merge(src proto.Message)

func (*NodeProto) XXX_Size 

func (m *NodeProto) XXX_Size() int

func (*NodeProto) XXX_Unmarshal 

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

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"`
	// the operator type. This is needed to create the object from the operator
	// registry.
	Type string      `protobuf:"bytes,4,opt,name=type" json:"type"`
	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"`
	// 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"`
}

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

func (m *OperatorDef) GetDeviceOption() *DeviceOption

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

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

func (*OperatorDef) GetType 

func (m *OperatorDef) GetType() string

func (*OperatorDef) Marshal 

func (m *OperatorDef) Marshal() (dAtA []byte, err error)

func (*OperatorDef) MarshalTo 

func (m *OperatorDef) MarshalTo(dAtA []byte) (int, error)

func (*OperatorDef) ProtoMessage 

func (*OperatorDef) ProtoMessage()

func (*OperatorDef) Reset 

func (m *OperatorDef) Reset()

func (*OperatorDef) Size 

func (m *OperatorDef) Size() (n int)

func (*OperatorDef) String 

func (m *OperatorDef) String() string

func (*OperatorDef) Unmarshal 

func (m *OperatorDef) Unmarshal(dAtA []byte) error

func (*OperatorDef) XXX_DiscardUnknown 

func (m *OperatorDef) XXX_DiscardUnknown()

func (*OperatorDef) XXX_Marshal 

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

func (*OperatorDef) XXX_Merge 

func (dst *OperatorDef) XXX_Merge(src proto.Message)

func (*OperatorDef) XXX_Size 

func (m *OperatorDef) XXX_Size() int

func (*OperatorDef) XXX_Unmarshal 

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

type PathNodeProto 

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

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

func (m *PathNodeProto) Marshal() (dAtA []byte, err error)

func (*PathNodeProto) MarshalTo 

func (m *PathNodeProto) MarshalTo(dAtA []byte) (int, error)

func (*PathNodeProto) ProtoMessage 

func (*PathNodeProto) ProtoMessage()

func (*PathNodeProto) Reset 

func (m *PathNodeProto) Reset()

func (*PathNodeProto) Size 

func (m *PathNodeProto) Size() (n int)

func (*PathNodeProto) String 

func (m *PathNodeProto) String() string

func (*PathNodeProto) Unmarshal 

func (m *PathNodeProto) Unmarshal(dAtA []byte) error

func (*PathNodeProto) XXX_DiscardUnknown 

func (m *PathNodeProto) XXX_DiscardUnknown()

func (*PathNodeProto) XXX_Marshal 

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

func (*PathNodeProto) XXX_Merge 

func (dst *PathNodeProto) XXX_Merge(src proto.Message)

func (*PathNodeProto) XXX_Size 

func (m *PathNodeProto) XXX_Size() int

func (*PathNodeProto) XXX_Unmarshal 

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

type PathProto 

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

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

func (m *PathProto) Marshal() (dAtA []byte, err error)

func (*PathProto) MarshalTo 

func (m *PathProto) MarshalTo(dAtA []byte) (int, error)

func (*PathProto) ProtoMessage 

func (*PathProto) ProtoMessage()

func (*PathProto) Reset 

func (m *PathProto) Reset()

func (*PathProto) Size 

func (m *PathProto) Size() (n int)

func (*PathProto) String 

func (m *PathProto) String() string

func (*PathProto) Unmarshal 

func (m *PathProto) Unmarshal(dAtA []byte) error

func (*PathProto) XXX_DiscardUnknown 

func (m *PathProto) XXX_DiscardUnknown()

func (*PathProto) XXX_Marshal 

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

func (*PathProto) XXX_Merge 

func (dst *PathProto) XXX_Merge(src proto.Message)

func (*PathProto) XXX_Size 

func (m *PathProto) XXX_Size() int

func (*PathProto) XXX_Unmarshal 

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

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"`
	// 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"`
}

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

func (m *PlanDef) Marshal() (dAtA []byte, err error)

func (*PlanDef) MarshalTo 

func (m *PlanDef) MarshalTo(dAtA []byte) (int, error)

func (*PlanDef) ProtoMessage 

func (*PlanDef) ProtoMessage()

func (*PlanDef) Reset 

func (m *PlanDef) Reset()

func (*PlanDef) Size 

func (m *PlanDef) Size() (n int)

func (*PlanDef) String 

func (m *PlanDef) String() string

func (*PlanDef) Unmarshal 

func (m *PlanDef) Unmarshal(dAtA []byte) error

func (*PlanDef) XXX_DiscardUnknown 

func (m *PlanDef) XXX_DiscardUnknown()

func (*PlanDef) XXX_Marshal 

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

func (*PlanDef) XXX_Merge 

func (dst *PlanDef) XXX_Merge(src proto.Message)

func (*PlanDef) XXX_Size 

func (m *PlanDef) XXX_Size() int

func (*PlanDef) XXX_Unmarshal 

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

type PlansMap 

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

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

func (m *PlansMap) Marshal() (dAtA []byte, err error)

func (*PlansMap) MarshalTo 

func (m *PlansMap) MarshalTo(dAtA []byte) (int, error)

func (*PlansMap) ProtoMessage 

func (*PlansMap) ProtoMessage()

func (*PlansMap) Reset 

func (m *PlansMap) Reset()

func (*PlansMap) Size 

func (m *PlansMap) Size() (n int)

func (*PlansMap) String 

func (m *PlansMap) String() string

func (*PlansMap) Unmarshal 

func (m *PlansMap) Unmarshal(dAtA []byte) error

func (*PlansMap) XXX_DiscardUnknown 

func (m *PlansMap) XXX_DiscardUnknown()

func (*PlansMap) XXX_Marshal 

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

func (*PlansMap) XXX_Merge 

func (dst *PlansMap) XXX_Merge(src proto.Message)

func (*PlansMap) XXX_Size 

func (m *PlansMap) XXX_Size() int

func (*PlansMap) XXX_Unmarshal 

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

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"`
}

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

func (m *PredictorConsts) Marshal() (dAtA []byte, err error)

func (*PredictorConsts) MarshalTo 

func (m *PredictorConsts) MarshalTo(dAtA []byte) (int, error)

func (*PredictorConsts) ProtoMessage 

func (*PredictorConsts) ProtoMessage()

func (*PredictorConsts) Reset 

func (m *PredictorConsts) Reset()

func (*PredictorConsts) Size 

func (m *PredictorConsts) Size() (n int)

func (*PredictorConsts) String 

func (m *PredictorConsts) String() string

func (*PredictorConsts) Unmarshal 

func (m *PredictorConsts) Unmarshal(dAtA []byte) error

func (*PredictorConsts) XXX_DiscardUnknown 

func (m *PredictorConsts) XXX_DiscardUnknown()

func (*PredictorConsts) XXX_Marshal 

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

func (*PredictorConsts) XXX_Merge 

func (dst *PredictorConsts) XXX_Merge(src proto.Message)

func (*PredictorConsts) XXX_Size 

func (m *PredictorConsts) XXX_Size() int

func (*PredictorConsts) XXX_Unmarshal 

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

type ProfDAGProto 

type ProfDAGProto struct {
	// The name for the operator
	Name string `protobuf:"bytes,1,req,name=name" json:"name"`
	// The mean execution time
	Mean float32 `protobuf:"fixed32,2,req,name=mean" json:"mean"`
	// The standard deviation
	Stddev float32 `protobuf:"fixed32,3,req,name=stddev" json:"stddev"`
}

Protobuf format to serialize profiler data

func (*ProfDAGProto) Descriptor 

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

func (*ProfDAGProto) GetMean 

func (m *ProfDAGProto) GetMean() float32

func (*ProfDAGProto) GetName 

func (m *ProfDAGProto) GetName() string

func (*ProfDAGProto) GetStddev 

func (m *ProfDAGProto) GetStddev() float32

func (*ProfDAGProto) Marshal 

func (m *ProfDAGProto) Marshal() (dAtA []byte, err error)

func (*ProfDAGProto) MarshalTo 

func (m *ProfDAGProto) MarshalTo(dAtA []byte) (int, error)

func (*ProfDAGProto) ProtoMessage 

func (*ProfDAGProto) ProtoMessage()

func (*ProfDAGProto) Reset 

func (m *ProfDAGProto) Reset()

func (*ProfDAGProto) Size 

func (m *ProfDAGProto) Size() (n int)

func (*ProfDAGProto) String 

func (m *ProfDAGProto) String() string

func (*ProfDAGProto) Unmarshal 

func (m *ProfDAGProto) Unmarshal(dAtA []byte) error

func (*ProfDAGProto) XXX_DiscardUnknown 

func (m *ProfDAGProto) XXX_DiscardUnknown()

func (*ProfDAGProto) XXX_Marshal 

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

func (*ProfDAGProto) XXX_Merge 

func (dst *ProfDAGProto) XXX_Merge(src proto.Message)

func (*ProfDAGProto) XXX_Size 

func (m *ProfDAGProto) XXX_Size() int

func (*ProfDAGProto) XXX_Unmarshal 

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

type ProfDAGProtos 

type ProfDAGProtos struct {
	Stats []*ProfDAGProto `protobuf:"bytes,1,rep,name=stats" json:"stats,omitempty"`
}

func (*ProfDAGProtos) Descriptor 

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

func (*ProfDAGProtos) GetStats 

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

func (*ProfDAGProtos) Marshal 

func (m *ProfDAGProtos) Marshal() (dAtA []byte, err error)

func (*ProfDAGProtos) MarshalTo 

func (m *ProfDAGProtos) MarshalTo(dAtA []byte) (int, error)

func (*ProfDAGProtos) ProtoMessage 

func (*ProfDAGProtos) ProtoMessage()

func (*ProfDAGProtos) Reset 

func (m *ProfDAGProtos) Reset()

func (*ProfDAGProtos) Size 

func (m *ProfDAGProtos) Size() (n int)

func (*ProfDAGProtos) String 

func (m *ProfDAGProtos) String() string

func (*ProfDAGProtos) Unmarshal 

func (m *ProfDAGProtos) Unmarshal(dAtA []byte) error

func (*ProfDAGProtos) XXX_DiscardUnknown 

func (m *ProfDAGProtos) XXX_DiscardUnknown()

func (*ProfDAGProtos) XXX_Marshal 

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

func (*ProfDAGProtos) XXX_Merge 

func (dst *ProfDAGProtos) XXX_Merge(src proto.Message)

func (*ProfDAGProtos) XXX_Size 

func (m *ProfDAGProtos) XXX_Size() int

func (*ProfDAGProtos) XXX_Unmarshal 

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

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"`
	Scale     float64 `protobuf:"fixed64,3,req,name=scale" json:"scale"`
	Bias      float64 `protobuf:"fixed64,4,req,name=bias" json:"bias"`
	IsSigned  bool    `protobuf:"varint,5,req,name=is_signed,json=isSigned" json:"is_signed"`
	Data      []int32 `protobuf:"varint,6,rep,packed,name=data" json:"data,omitempty"`
	Name      string  `protobuf:"bytes,7,opt,name=name" json:"name"`
}

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

func (m *QTensorProto) Marshal() (dAtA []byte, err error)

func (*QTensorProto) MarshalTo 

func (m *QTensorProto) MarshalTo(dAtA []byte) (int, error)

func (*QTensorProto) ProtoMessage 

func (*QTensorProto) ProtoMessage()

func (*QTensorProto) Reset 

func (m *QTensorProto) Reset()

func (*QTensorProto) Size 

func (m *QTensorProto) Size() (n int)

func (*QTensorProto) String 

func (m *QTensorProto) String() string

func (*QTensorProto) Unmarshal 

func (m *QTensorProto) Unmarshal(dAtA []byte) error

func (*QTensorProto) XXX_DiscardUnknown 

func (m *QTensorProto) XXX_DiscardUnknown()

func (*QTensorProto) XXX_Marshal 

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

func (*QTensorProto) XXX_Merge 

func (dst *QTensorProto) XXX_Merge(src proto.Message)

func (*QTensorProto) XXX_Size 

func (m *QTensorProto) XXX_Size() int

func (*QTensorProto) XXX_Unmarshal 

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

type StringMap 

type StringMap struct {
	Key   string `protobuf:"bytes,1,req,name=key" json:"key"`
	Value string `protobuf:"bytes,2,req,name=value" json:"value"`
}

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

func (m *StringMap) Marshal() (dAtA []byte, err error)

func (*StringMap) MarshalTo 

func (m *StringMap) MarshalTo(dAtA []byte) (int, error)

func (*StringMap) ProtoMessage 

func (*StringMap) ProtoMessage()

func (*StringMap) Reset 

func (m *StringMap) Reset()

func (*StringMap) Size 

func (m *StringMap) Size() (n int)

func (*StringMap) String 

func (m *StringMap) String() string

func (*StringMap) Unmarshal 

func (m *StringMap) Unmarshal(dAtA []byte) error

func (*StringMap) XXX_DiscardUnknown 

func (m *StringMap) XXX_DiscardUnknown()

func (*StringMap) XXX_Marshal 

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

func (*StringMap) XXX_Merge 

func (dst *StringMap) XXX_Merge(src proto.Message)

func (*StringMap) XXX_Size 

func (m *StringMap) XXX_Size() int

func (*StringMap) XXX_Unmarshal 

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

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"`
	// 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"`
	// 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"`
	// Optionally, a name for the tensor.
	Name string `protobuf:"bytes,7,opt,name=name" json:"name"`
	// 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"`
}

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

func (m *TensorProto) GetSegment() *TensorProto_Segment

func (*TensorProto) GetStringData 

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

func (*TensorProto) Marshal 

func (m *TensorProto) Marshal() (dAtA []byte, err error)

func (*TensorProto) MarshalTo 

func (m *TensorProto) MarshalTo(dAtA []byte) (int, error)

func (*TensorProto) ProtoMessage 

func (*TensorProto) ProtoMessage()

func (*TensorProto) Reset 

func (m *TensorProto) Reset()

func (*TensorProto) Size 

func (m *TensorProto) Size() (n int)

func (*TensorProto) String 

func (m *TensorProto) String() string

func (*TensorProto) Unmarshal 

func (m *TensorProto) Unmarshal(dAtA []byte) error

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 (dst *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 TensorProto_DataType 

type TensorProto_DataType int32
const (
	TensorProto_UNDEFINED TensorProto_DataType = 0
	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_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"`
	End   int64 `protobuf:"varint,2,req,name=end" json:"end"`
}

When loading from chunks this is going to indicate where to put data in the full array. When not used full data must 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) Marshal 

func (m *TensorProto_Segment) Marshal() (dAtA []byte, err error)

func (*TensorProto_Segment) MarshalTo 

func (m *TensorProto_Segment) MarshalTo(dAtA []byte) (int, error)

func (*TensorProto_Segment) ProtoMessage 

func (*TensorProto_Segment) ProtoMessage()

func (*TensorProto_Segment) Reset 

func (m *TensorProto_Segment) Reset()

func (*TensorProto_Segment) Size 

func (m *TensorProto_Segment) Size() (n int)

func (*TensorProto_Segment) String 

func (m *TensorProto_Segment) String() string

func (*TensorProto_Segment) Unmarshal 

func (m *TensorProto_Segment) Unmarshal(dAtA []byte) error

func (*TensorProto_Segment) XXX_DiscardUnknown 

func (m *TensorProto_Segment) XXX_DiscardUnknown()

func (*TensorProto_Segment) XXX_Marshal 

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

func (*TensorProto_Segment) XXX_Merge 

func (dst *TensorProto_Segment) XXX_Merge(src proto.Message)

func (*TensorProto_Segment) XXX_Size 

func (m *TensorProto_Segment) XXX_Size() int

func (*TensorProto_Segment) XXX_Unmarshal 

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

type TensorProtos 

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

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

func (m *TensorProtos) Marshal() (dAtA []byte, err error)

func (*TensorProtos) MarshalTo 

func (m *TensorProtos) MarshalTo(dAtA []byte) (int, error)

func (*TensorProtos) ProtoMessage 

func (*TensorProtos) ProtoMessage()

func (*TensorProtos) Reset 

func (m *TensorProtos) Reset()

func (*TensorProtos) Size 

func (m *TensorProtos) Size() (n int)

func (*TensorProtos) String 

func (m *TensorProtos) String() string

func (*TensorProtos) Unmarshal 

func (m *TensorProtos) Unmarshal(dAtA []byte) error

func (*TensorProtos) XXX_DiscardUnknown 

func (m *TensorProtos) XXX_DiscardUnknown()

func (*TensorProtos) XXX_Marshal 

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

func (*TensorProtos) XXX_Merge 

func (dst *TensorProtos) XXX_Merge(src proto.Message)

func (*TensorProtos) XXX_Size 

func (m *TensorProtos) XXX_Size() int

func (*TensorProtos) XXX_Unmarshal 

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

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"`
}

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

func (m *TensorShape) Marshal() (dAtA []byte, err error)

func (*TensorShape) MarshalTo 

func (m *TensorShape) MarshalTo(dAtA []byte) (int, error)

func (*TensorShape) ProtoMessage 

func (*TensorShape) ProtoMessage()

func (*TensorShape) Reset 

func (m *TensorShape) Reset()

func (*TensorShape) Size 

func (m *TensorShape) Size() (n int)

func (*TensorShape) String 

func (m *TensorShape) String() string

func (*TensorShape) Unmarshal 

func (m *TensorShape) Unmarshal(dAtA []byte) error

func (*TensorShape) XXX_DiscardUnknown 

func (m *TensorShape) XXX_DiscardUnknown()

func (*TensorShape) XXX_Marshal 

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

func (*TensorShape) XXX_Merge 

func (dst *TensorShape) XXX_Merge(src proto.Message)

func (*TensorShape) XXX_Size 

func (m *TensorShape) XXX_Size() int

func (*TensorShape) XXX_Unmarshal 

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

type TensorShapes 

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

func (*TensorShapes) Descriptor 

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

func (*TensorShapes) GetShapes 

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

func (*TensorShapes) Marshal 

func (m *TensorShapes) Marshal() (dAtA []byte, err error)

func (*TensorShapes) MarshalTo 

func (m *TensorShapes) MarshalTo(dAtA []byte) (int, error)

func (*TensorShapes) ProtoMessage 

func (*TensorShapes) ProtoMessage()

func (*TensorShapes) Reset 

func (m *TensorShapes) Reset()

func (*TensorShapes) Size 

func (m *TensorShapes) Size() (n int)

func (*TensorShapes) String 

func (m *TensorShapes) String() string

func (*TensorShapes) Unmarshal 

func (m *TensorShapes) Unmarshal(dAtA []byte) error

func (*TensorShapes) XXX_DiscardUnknown 

func (m *TensorShapes) XXX_DiscardUnknown()

func (*TensorShapes) XXX_Marshal 

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

func (*TensorShapes) XXX_Merge 

func (dst *TensorShapes) XXX_Merge(src proto.Message)

func (*TensorShapes) XXX_Size 

func (m *TensorShapes) XXX_Size() int

func (*TensorShapes) XXX_Unmarshal 

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

type TreeProto 

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

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

func (m *TreeProto) Marshal() (dAtA []byte, err error)

func (*TreeProto) MarshalTo 

func (m *TreeProto) MarshalTo(dAtA []byte) (int, error)

func (*TreeProto) ProtoMessage 

func (*TreeProto) ProtoMessage()

func (*TreeProto) Reset 

func (m *TreeProto) Reset()

func (*TreeProto) Size 

func (m *TreeProto) Size() (n int)

func (*TreeProto) String 

func (m *TreeProto) String() string

func (*TreeProto) Unmarshal 

func (m *TreeProto) Unmarshal(dAtA []byte) error

func (*TreeProto) XXX_DiscardUnknown 

func (m *TreeProto) XXX_DiscardUnknown()

func (*TreeProto) XXX_Marshal 

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

func (*TreeProto) XXX_Merge 

func (dst *TreeProto) XXX_Merge(src proto.Message)

func (*TreeProto) XXX_Size 

func (m *TreeProto) XXX_Size() int

func (*TreeProto) XXX_Unmarshal 

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

Source Files

View all Source files

Directories

Path Synopsis

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.