autotune_maps

package
v0.0.0-...-3233e8b 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: 6 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 *stream_executor.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 (x *ConvMapProto_Entry) GetKey() *ConvParametersProto

func (*ConvMapProto_Entry) GetValue 

func (x *ConvMapProto_Entry) GetValue() *stream_executor.AlgorithmConfigProto

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 (x *ConvParametersProto) GetFusion() *ConvParametersProto_Fusion

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 stream_executor.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() stream_executor.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 (x *ConvParametersProto_Fusion) ProtoReflect() protoreflect.Message

func (*ConvParametersProto_Fusion) Reset 

func (x *ConvParametersProto_Fusion) Reset()

func (*ConvParametersProto_Fusion) String 

func (x *ConvParametersProto_Fusion) 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.