variable_go_proto

package
v1.12.2-0...-1b6f133 Latest Latest
Warning

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

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

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var VariableAggregation_name = map[int32]string{
	0: "VARIABLE_AGGREGATION_NONE",
	1: "VARIABLE_AGGREGATION_SUM",
	2: "VARIABLE_AGGREGATION_MEAN",
	3: "VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA",
}
View Source 
var VariableAggregation_value = map[string]int32{
	"VARIABLE_AGGREGATION_NONE":               0,
	"VARIABLE_AGGREGATION_SUM":                1,
	"VARIABLE_AGGREGATION_MEAN":               2,
	"VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA": 3,
}
View Source 
var VariableSynchronization_name = map[int32]string{
	0: "VARIABLE_SYNCHRONIZATION_AUTO",
	1: "VARIABLE_SYNCHRONIZATION_NONE",
	2: "VARIABLE_SYNCHRONIZATION_ON_WRITE",
	3: "VARIABLE_SYNCHRONIZATION_ON_READ",
}
View Source 
var VariableSynchronization_value = map[string]int32{
	"VARIABLE_SYNCHRONIZATION_AUTO":     0,
	"VARIABLE_SYNCHRONIZATION_NONE":     1,
	"VARIABLE_SYNCHRONIZATION_ON_WRITE": 2,
	"VARIABLE_SYNCHRONIZATION_ON_READ":  3,
}

Functions

This section is empty.

Types

type SaveSliceInfoDef 

type SaveSliceInfoDef struct {
	// Name of the full variable of which this is a slice.
	FullName string `protobuf:"bytes,1,opt,name=full_name,json=fullName,proto3" json:"full_name,omitempty"`
	// Shape of the full variable.
	FullShape []int64 `protobuf:"varint,2,rep,packed,name=full_shape,json=fullShape,proto3" json:"full_shape,omitempty"`
	// Offset of this variable into the full variable.
	VarOffset []int64 `protobuf:"varint,3,rep,packed,name=var_offset,json=varOffset,proto3" json:"var_offset,omitempty"`
	// Shape of this variable.
	VarShape             []int64  `protobuf:"varint,4,rep,packed,name=var_shape,json=varShape,proto3" json:"var_shape,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

func (*SaveSliceInfoDef) Descriptor 

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

func (*SaveSliceInfoDef) GetFullName 

func (m *SaveSliceInfoDef) GetFullName() string

func (*SaveSliceInfoDef) GetFullShape 

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

func (*SaveSliceInfoDef) GetVarOffset 

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

func (*SaveSliceInfoDef) GetVarShape 

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

func (*SaveSliceInfoDef) ProtoMessage 

func (*SaveSliceInfoDef) ProtoMessage()

func (*SaveSliceInfoDef) Reset 

func (m *SaveSliceInfoDef) Reset()

func (*SaveSliceInfoDef) String 

func (m *SaveSliceInfoDef) String() string

func (*SaveSliceInfoDef) XXX_DiscardUnknown 

func (m *SaveSliceInfoDef) XXX_DiscardUnknown()

func (*SaveSliceInfoDef) XXX_Marshal 

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

func (*SaveSliceInfoDef) XXX_Merge 

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

func (*SaveSliceInfoDef) XXX_Size 

func (m *SaveSliceInfoDef) XXX_Size() int

func (*SaveSliceInfoDef) XXX_Unmarshal 

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

type VariableAggregation 

type VariableAggregation int32

Indicates how a distributed variable will be aggregated. 

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

func (VariableAggregation) EnumDescriptor 

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

func (VariableAggregation) String 

func (x VariableAggregation) String() string

type VariableDef 

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

Protocol buffer representing a Variable.

func (*VariableDef) Descriptor 

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

func (*VariableDef) GetAggregation 

func (m *VariableDef) GetAggregation() VariableAggregation

func (*VariableDef) GetInitialValueName 

func (m *VariableDef) GetInitialValueName() string

func (*VariableDef) GetInitializerName 

func (m *VariableDef) GetInitializerName() string

func (*VariableDef) GetIsResource 

func (m *VariableDef) GetIsResource() bool

func (*VariableDef) GetSaveSliceInfoDef 

func (m *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef

func (*VariableDef) GetSnapshotName 

func (m *VariableDef) GetSnapshotName() string

func (*VariableDef) GetSynchronization 

func (m *VariableDef) GetSynchronization() VariableSynchronization

func (*VariableDef) GetTrainable 

func (m *VariableDef) GetTrainable() bool

func (*VariableDef) GetVariableName 

func (m *VariableDef) GetVariableName() string

func (*VariableDef) ProtoMessage 

func (*VariableDef) ProtoMessage()

func (*VariableDef) Reset 

func (m *VariableDef) Reset()

func (*VariableDef) String 

func (m *VariableDef) String() string

func (*VariableDef) XXX_DiscardUnknown 

func (m *VariableDef) XXX_DiscardUnknown()

func (*VariableDef) XXX_Marshal 

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

func (*VariableDef) XXX_Merge 

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

func (*VariableDef) XXX_Size 

func (m *VariableDef) XXX_Size() int

func (*VariableDef) XXX_Unmarshal 

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

type VariableSynchronization 

type VariableSynchronization int32

Indicates when a distributed variable will be synced. 

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

func (VariableSynchronization) EnumDescriptor 

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

func (VariableSynchronization) String 

func (x VariableSynchronization) String() string

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.