autotune_maps

package
v2.14.1 Latest Latest
Warning

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

Go to latest
Published: Dec 8, 2023 License: BSD-3-Clause Imports: 7 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source
var File_tensorflow_core_util_autotune_maps_autotune_map_proto protoreflect.FileDescriptor
View Source
var File_tensorflow_core_util_autotune_maps_conv_parameters_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type AutotuneMapsProto

type AutotuneMapsProto struct {
	ConvMap      *ConvMapProto `protobuf:"bytes,2,opt,name=conv_map,json=convMap,proto3" json:"conv_map,omitempty"`
	FusedConvMap *ConvMapProto `protobuf:"bytes,3,opt,name=fused_conv_map,json=fusedConvMap,proto3" json:"fused_conv_map,omitempty"`
	// contains filtered or unexported fields
}

TODO(b/189530096): Support autotune maps for more ops.

func (*AutotuneMapsProto) Descriptor deprecated

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

Deprecated: Use AutotuneMapsProto.ProtoReflect.Descriptor instead.

func (*AutotuneMapsProto) GetConvMap

func (x *AutotuneMapsProto) GetConvMap() *ConvMapProto

func (*AutotuneMapsProto) GetFusedConvMap

func (x *AutotuneMapsProto) GetFusedConvMap() *ConvMapProto

func (*AutotuneMapsProto) ProtoMessage

func (*AutotuneMapsProto) ProtoMessage()

func (*AutotuneMapsProto) ProtoReflect

func (x *AutotuneMapsProto) ProtoReflect() protoreflect.Message

func (*AutotuneMapsProto) Reset

func (x *AutotuneMapsProto) Reset()

func (*AutotuneMapsProto) String

func (x *AutotuneMapsProto) String() string

type ConvMapProto

type ConvMapProto struct {
	KvPairs []*ConvMapProto_Entry `protobuf:"bytes,1,rep,name=kv_pairs,json=kvPairs,proto3" json:"kv_pairs,omitempty"`
	// contains filtered or unexported fields
}

func (*ConvMapProto) Descriptor deprecated

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

Deprecated: Use ConvMapProto.ProtoReflect.Descriptor instead.

func (*ConvMapProto) GetKvPairs

func (x *ConvMapProto) GetKvPairs() []*ConvMapProto_Entry

func (*ConvMapProto) ProtoMessage

func (*ConvMapProto) ProtoMessage()

func (*ConvMapProto) ProtoReflect

func (x *ConvMapProto) ProtoReflect() protoreflect.Message

func (*ConvMapProto) Reset

func (x *ConvMapProto) Reset()

func (*ConvMapProto) String

func (x *ConvMapProto) String() string

type ConvMapProto_Entry

type ConvMapProto_Entry struct {
	Key   *ConvParametersProto           `protobuf:"bytes,1,opt,name=key,proto3" json:"key,omitempty"`
	Value *protobuf.AlgorithmConfigProto `protobuf:"bytes,2,opt,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

func (*ConvMapProto_Entry) Descriptor deprecated

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

Deprecated: Use ConvMapProto_Entry.ProtoReflect.Descriptor instead.

func (*ConvMapProto_Entry) GetKey

func (*ConvMapProto_Entry) GetValue

func (*ConvMapProto_Entry) ProtoMessage

func (*ConvMapProto_Entry) ProtoMessage()

func (*ConvMapProto_Entry) ProtoReflect

func (x *ConvMapProto_Entry) ProtoReflect() protoreflect.Message

func (*ConvMapProto_Entry) Reset

func (x *ConvMapProto_Entry) Reset()

func (*ConvMapProto_Entry) String

func (x *ConvMapProto_Entry) String() string

type ConvParametersProto

type ConvParametersProto struct {
	Batch     int64   `protobuf:"varint,1,opt,name=batch,proto3" json:"batch,omitempty"`
	InDepths  int64   `protobuf:"varint,2,opt,name=in_depths,json=inDepths,proto3" json:"in_depths,omitempty"`
	OutDepths int64   `protobuf:"varint,3,opt,name=out_depths,json=outDepths,proto3" json:"out_depths,omitempty"`
	In        []int64 `protobuf:"varint,4,rep,packed,name=in,proto3" json:"in,omitempty"`
	// data_format corresponds to type TensorFormat in
	// third_party/tensorflow/core/util/tensor_format.h.
	DataFormat int32              `protobuf:"varint,5,opt,name=data_format,json=dataFormat,proto3" json:"data_format,omitempty"`
	Filter     []int64            `protobuf:"varint,6,rep,packed,name=filter,proto3" json:"filter,omitempty"`
	Dilation   []int64            `protobuf:"varint,7,rep,packed,name=dilation,proto3" json:"dilation,omitempty"`
	Stride     []int64            `protobuf:"varint,8,rep,packed,name=stride,proto3" json:"stride,omitempty"`
	Padding    []int64            `protobuf:"varint,9,rep,packed,name=padding,proto3" json:"padding,omitempty"`
	Dtype      framework.DataType `protobuf:"varint,10,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	GroupCount int32              `protobuf:"varint,11,opt,name=group_count,json=groupCount,proto3" json:"group_count,omitempty"`
	// A string uniquely identifying a particular GPU model, e.g. V100 vs RTX
	// 2080.
	DeviceIdentifier string                      `protobuf:"bytes,12,opt,name=device_identifier,json=deviceIdentifier,proto3" json:"device_identifier,omitempty"`
	Fusion           *ConvParametersProto_Fusion `protobuf:"bytes,13,opt,name=fusion,proto3" json:"fusion,omitempty"`
	// The version number of ConvParameters class. Offline autotune results
	// whose version number is different from the runtime's version number
	// (defined in ConvParameters::kVersion) will be rejected and ignored by
	// LoadSerializedAutotuneMaps. This ensures that we will not load out-of-date
	// autotune results.
	Version int32 `protobuf:"varint,14,opt,name=version,proto3" json:"version,omitempty"`
	// contains filtered or unexported fields
}

This is the underlying data structure of class ConvParameters, which are used as the keys in cuDNN autotuning maps for retrieving corresponding cuDNN algorithms. This is used as a serialization format for saving/loading autotuning databases.

func (*ConvParametersProto) Descriptor deprecated

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

Deprecated: Use ConvParametersProto.ProtoReflect.Descriptor instead.

func (*ConvParametersProto) GetBatch

func (x *ConvParametersProto) GetBatch() int64

func (*ConvParametersProto) GetDataFormat

func (x *ConvParametersProto) GetDataFormat() int32

func (*ConvParametersProto) GetDeviceIdentifier

func (x *ConvParametersProto) GetDeviceIdentifier() string

func (*ConvParametersProto) GetDilation

func (x *ConvParametersProto) GetDilation() []int64

func (*ConvParametersProto) GetDtype

func (x *ConvParametersProto) GetDtype() framework.DataType

func (*ConvParametersProto) GetFilter

func (x *ConvParametersProto) GetFilter() []int64

func (*ConvParametersProto) GetFusion

func (*ConvParametersProto) GetGroupCount

func (x *ConvParametersProto) GetGroupCount() int32

func (*ConvParametersProto) GetIn

func (x *ConvParametersProto) GetIn() []int64

func (*ConvParametersProto) GetInDepths

func (x *ConvParametersProto) GetInDepths() int64

func (*ConvParametersProto) GetOutDepths

func (x *ConvParametersProto) GetOutDepths() int64

func (*ConvParametersProto) GetPadding

func (x *ConvParametersProto) GetPadding() []int64

func (*ConvParametersProto) GetStride

func (x *ConvParametersProto) GetStride() []int64

func (*ConvParametersProto) GetVersion

func (x *ConvParametersProto) GetVersion() int32

func (*ConvParametersProto) ProtoMessage

func (*ConvParametersProto) ProtoMessage()

func (*ConvParametersProto) ProtoReflect

func (x *ConvParametersProto) ProtoReflect() protoreflect.Message

func (*ConvParametersProto) Reset

func (x *ConvParametersProto) Reset()

func (*ConvParametersProto) String

func (x *ConvParametersProto) String() string

type ConvParametersProto_Fusion

type ConvParametersProto_Fusion struct {

	// If true, this proto corresponds to a FusedConvBiasActivation operation
	// implemented in the contrib library, otherwise this proto corresponds to
	// the FusedConv operation implemented in the core library.
	// Compared with FusedConv, FusedConvBiasActivation supports more types of
	// activation function (including no activation) as well as the side_input.
	// For now they have same type of keys in autotune maps, but the semantics
	// of some fields (like padding) are different. So we add this field to
	// distinguish them.
	// TODO(b/177365158) Remove this field once these two operations are merged.
	IsContrib      bool                    `protobuf:"varint,1,opt,name=is_contrib,json=isContrib,proto3" json:"is_contrib,omitempty"`
	ActivationMode protobuf.ActivationMode `` /* 144-byte string literal not displayed */
	ConvScale      float64                 `protobuf:"fixed64,3,opt,name=conv_scale,json=convScale,proto3" json:"conv_scale,omitempty"`
	SideInputScale float64                 `protobuf:"fixed64,4,opt,name=side_input_scale,json=sideInputScale,proto3" json:"side_input_scale,omitempty"`
	// contains filtered or unexported fields
}

This stores the information for fused convolution operations where an activation and a side input might follow the convolution.

func (*ConvParametersProto_Fusion) Descriptor deprecated

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

Deprecated: Use ConvParametersProto_Fusion.ProtoReflect.Descriptor instead.

func (*ConvParametersProto_Fusion) GetActivationMode

func (x *ConvParametersProto_Fusion) GetActivationMode() protobuf.ActivationMode

func (*ConvParametersProto_Fusion) GetConvScale

func (x *ConvParametersProto_Fusion) GetConvScale() float64

func (*ConvParametersProto_Fusion) GetIsContrib

func (x *ConvParametersProto_Fusion) GetIsContrib() bool

func (*ConvParametersProto_Fusion) GetSideInputScale

func (x *ConvParametersProto_Fusion) GetSideInputScale() float64

func (*ConvParametersProto_Fusion) ProtoMessage

func (*ConvParametersProto_Fusion) ProtoMessage()

func (*ConvParametersProto_Fusion) ProtoReflect

func (*ConvParametersProto_Fusion) Reset

func (x *ConvParametersProto_Fusion) Reset()

func (*ConvParametersProto_Fusion) String

func (x *ConvParametersProto_Fusion) String() string

type MatmulParametersProto added in v2.11.0

type MatmulParametersProto struct {
	AbDtype          framework.DataType      `protobuf:"varint,1,opt,name=ab_dtype,json=abDtype,proto3,enum=tensorflow.DataType" json:"ab_dtype,omitempty"`
	CDtype           framework.DataType      `protobuf:"varint,2,opt,name=c_dtype,json=cDtype,proto3,enum=tensorflow.DataType" json:"c_dtype,omitempty"`
	TransA           bool                    `protobuf:"varint,3,opt,name=trans_a,json=transA,proto3" json:"trans_a,omitempty"`
	TransB           bool                    `protobuf:"varint,4,opt,name=trans_b,json=transB,proto3" json:"trans_b,omitempty"`
	M                uint64                  `protobuf:"varint,5,opt,name=m,proto3" json:"m,omitempty"`
	N                uint64                  `protobuf:"varint,6,opt,name=n,proto3" json:"n,omitempty"`
	K                uint64                  `protobuf:"varint,7,opt,name=k,proto3" json:"k,omitempty"`
	Lda              int64                   `protobuf:"varint,8,opt,name=lda,proto3" json:"lda,omitempty"`
	Ldb              int64                   `protobuf:"varint,9,opt,name=ldb,proto3" json:"ldb,omitempty"`
	Ldc              int64                   `protobuf:"varint,10,opt,name=ldc,proto3" json:"ldc,omitempty"`
	ActivationMode   protobuf.ActivationMode `` /* 145-byte string literal not displayed */
	DeviceIdentifier string                  `protobuf:"bytes,12,opt,name=device_identifier,json=deviceIdentifier,proto3" json:"device_identifier,omitempty"`
	Version          int32                   `protobuf:"varint,14,opt,name=version,proto3" json:"version,omitempty"`
	// contains filtered or unexported fields
}

func (*MatmulParametersProto) Descriptor deprecated added in v2.11.0

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

Deprecated: Use MatmulParametersProto.ProtoReflect.Descriptor instead.

func (*MatmulParametersProto) GetAbDtype added in v2.11.0

func (x *MatmulParametersProto) GetAbDtype() framework.DataType

func (*MatmulParametersProto) GetActivationMode added in v2.11.0

func (x *MatmulParametersProto) GetActivationMode() protobuf.ActivationMode

func (*MatmulParametersProto) GetCDtype added in v2.11.0

func (x *MatmulParametersProto) GetCDtype() framework.DataType

func (*MatmulParametersProto) GetDeviceIdentifier added in v2.11.0

func (x *MatmulParametersProto) GetDeviceIdentifier() string

func (*MatmulParametersProto) GetK added in v2.11.0

func (x *MatmulParametersProto) GetK() uint64

func (*MatmulParametersProto) GetLda added in v2.11.0

func (x *MatmulParametersProto) GetLda() int64

func (*MatmulParametersProto) GetLdb added in v2.11.0

func (x *MatmulParametersProto) GetLdb() int64

func (*MatmulParametersProto) GetLdc added in v2.11.0

func (x *MatmulParametersProto) GetLdc() int64

func (*MatmulParametersProto) GetM added in v2.11.0

func (x *MatmulParametersProto) GetM() uint64

func (*MatmulParametersProto) GetN added in v2.11.0

func (x *MatmulParametersProto) GetN() uint64

func (*MatmulParametersProto) GetTransA added in v2.11.0

func (x *MatmulParametersProto) GetTransA() bool

func (*MatmulParametersProto) GetTransB added in v2.11.0

func (x *MatmulParametersProto) GetTransB() bool

func (*MatmulParametersProto) GetVersion added in v2.11.0

func (x *MatmulParametersProto) GetVersion() int32

func (*MatmulParametersProto) ProtoMessage added in v2.11.0

func (*MatmulParametersProto) ProtoMessage()

func (*MatmulParametersProto) ProtoReflect added in v2.11.0

func (x *MatmulParametersProto) ProtoReflect() protoreflect.Message

func (*MatmulParametersProto) Reset added in v2.11.0

func (x *MatmulParametersProto) Reset()

func (*MatmulParametersProto) String added in v2.11.0

func (x *MatmulParametersProto) String() string

Jump to

Keyboard shortcuts

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