stream_executor

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 ActivationMode_name = map[int32]string{
	0: "kNone",
	1: "kSigmoid",
	2: "kRelu",
	3: "kRelu6",
	4: "kReluX",
	5: "kTanh",
	6: "kBandPass",
}
View Source 
var ActivationMode_value = map[string]int32{
	"kNone":     0,
	"kSigmoid":  1,
	"kRelu":     2,
	"kRelu6":    3,
	"kReluX":    4,
	"kTanh":     5,
	"kBandPass": 6,
}
View Source 
var AlgorithmProto_MathType_name = map[int32]string{
	0: "DEFAULT_MATH",
	1: "TENSOR_OP_MATH",
}
View Source 
var AlgorithmProto_MathType_value = map[string]int32{
	"DEFAULT_MATH":   0,
	"TENSOR_OP_MATH": 1,
}
View Source 
var ConvolutionKind_name = map[int32]string{
	0: "INVALID",
	1: "FORWARD",
	2: "BACKWARD_FILTER",
	3: "BACKWARD_DATA",
	4: "FORWARD_BIAS_ACTIVATION",
}
View Source 
var ConvolutionKind_value = map[string]int32{
	"INVALID":                 0,
	"FORWARD":                 1,
	"BACKWARD_FILTER":         2,
	"BACKWARD_DATA":           3,
	"FORWARD_BIAS_ACTIVATION": 4,
}
View Source 
var ConvolutionMode_name = map[int32]string{
	0: "CROSS_CORRELATION",
	1: "CONVOLUTION",
}
View Source 
var ConvolutionMode_value = map[string]int32{
	"CROSS_CORRELATION": 0,
	"CONVOLUTION":       1,
}
View Source 
var DataLayout_name = map[int32]string{
	0: "kYXDepthBatch",
	1: "kYXBatchDepth",
	2: "kBatchYXDepth",
	3: "kBatchDepthYX",
	4: "kBatchDepthYX4",
}
View Source 
var DataLayout_value = map[string]int32{
	"kYXDepthBatch":  0,
	"kYXBatchDepth":  1,
	"kBatchYXDepth":  2,
	"kBatchDepthYX":  3,
	"kBatchDepthYX4": 4,
}
View Source 
var DataType_name = map[int32]string{
	0: "kFloat",
	1: "kDouble",
	2: "kHalf",
	3: "kInt8",
	4: "kInt32",
	5: "kComplexFloat",
	6: "kComplexDouble",
}
View Source 
var DataType_value = map[string]int32{
	"kFloat":         0,
	"kDouble":        1,
	"kHalf":          2,
	"kInt8":          3,
	"kInt32":         4,
	"kComplexFloat":  5,
	"kComplexDouble": 6,
}
View Source 
var FilterLayout_name = map[int32]string{
	0: "kOutputInputYX",
	1: "kOutputYXInput",
	2: "kOutputInputYX4",
	3: "kInputYXOutput",
	4: "kYXInputOutput",
}
View Source 
var FilterLayout_value = map[string]int32{
	"kOutputInputYX":  0,
	"kOutputYXInput":  1,
	"kOutputInputYX4": 2,
	"kInputYXOutput":  3,
	"kYXInputOutput":  4,
}

Functions

This section is empty.

Types

type ActivationMode 

type ActivationMode int32

Describes a kind of non-linearity (threshold-like mathematical function). 

const (
	ActivationMode_kNone    ActivationMode = 0
	ActivationMode_kSigmoid ActivationMode = 1
	// Rectified linear activation: f(x) = x < 0 ? 0 : x
	ActivationMode_kRelu ActivationMode = 2
	// Rectified linear activation; where upper maximum is 6.0.
	ActivationMode_kRelu6 ActivationMode = 3
	// Rectified linear activation; where upper maximum specified by
	// BatchDescriptor::value_max().
	ActivationMode_kReluX ActivationMode = 4
	ActivationMode_kTanh  ActivationMode = 5
	// Like ReluX; but passes all values in the range [-X,X].
	ActivationMode_kBandPass ActivationMode = 6
)

func (ActivationMode) EnumDescriptor 

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

func (ActivationMode) String 

func (x ActivationMode) String() string

type AlgorithmProto 

type AlgorithmProto struct {
	AlgoId               int64                   `protobuf:"varint,1,opt,name=algo_id,json=algoId,proto3" json:"algo_id,omitempty"`
	MathType             AlgorithmProto_MathType `` /* 135-byte string literal not displayed */
	XXX_NoUnkeyedLiteral struct{}                `json:"-"`
	XXX_unrecognized     []byte                  `json:"-"`
	XXX_sizecache        int32                   `json:"-"`
}

Generic algorithm representation.

func (*AlgorithmProto) Descriptor 

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

func (*AlgorithmProto) GetAlgoId 

func (m *AlgorithmProto) GetAlgoId() int64

func (*AlgorithmProto) GetMathType 

func (m *AlgorithmProto) GetMathType() AlgorithmProto_MathType

func (*AlgorithmProto) ProtoMessage 

func (*AlgorithmProto) ProtoMessage()

func (*AlgorithmProto) Reset 

func (m *AlgorithmProto) Reset()

func (*AlgorithmProto) String 

func (m *AlgorithmProto) String() string

func (*AlgorithmProto) XXX_DiscardUnknown 

func (m *AlgorithmProto) XXX_DiscardUnknown()

func (*AlgorithmProto) XXX_Marshal 

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

func (*AlgorithmProto) XXX_Merge 

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

func (*AlgorithmProto) XXX_Size 

func (m *AlgorithmProto) XXX_Size() int

func (*AlgorithmProto) XXX_Unmarshal 

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

type AlgorithmProto_MathType 

type AlgorithmProto_MathType int32
const (
	AlgorithmProto_DEFAULT_MATH AlgorithmProto_MathType = 0
	// The GPU may operate 4x4 matrix FMA.
	// See cuDNN's documentation for CUDNN_TENSOR_OP_MATH.
	AlgorithmProto_TENSOR_OP_MATH AlgorithmProto_MathType = 1
)

func (AlgorithmProto_MathType) EnumDescriptor 

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

func (AlgorithmProto_MathType) String 

func (x AlgorithmProto_MathType) String() string

type ConvolutionDescriptorProto 

type ConvolutionDescriptorProto struct {
	Paddings  []int64 `protobuf:"varint,1,rep,packed,name=paddings,proto3" json:"paddings,omitempty"`
	Strides   []int64 `protobuf:"varint,2,rep,packed,name=strides,proto3" json:"strides,omitempty"`
	Dilations []int64 `protobuf:"varint,3,rep,packed,name=dilations,proto3" json:"dilations,omitempty"`
	// The "accumulator" type. For example, use F32 as an accumulator for F16
	// convolutions.
	// See cuDNN's cudnnConvolutionMode_t.
	ComputeMode DataType `` /* 129-byte string literal not displayed */
	// See cuDNN's group count.
	GroupCount      int32           `protobuf:"varint,5,opt,name=group_count,json=groupCount,proto3" json:"group_count,omitempty"`
	ConvolutionMode ConvolutionMode `` /* 148-byte string literal not displayed */
	// Tensorflow node name, same as in NodeDef, for debugging purposes.
	Name                 string   `protobuf:"bytes,7,opt,name=name,proto3" json:"name,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

Convolution-specific parameters.

func (*ConvolutionDescriptorProto) Descriptor 

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

func (*ConvolutionDescriptorProto) GetComputeMode 

func (m *ConvolutionDescriptorProto) GetComputeMode() DataType

func (*ConvolutionDescriptorProto) GetConvolutionMode 

func (m *ConvolutionDescriptorProto) GetConvolutionMode() ConvolutionMode

func (*ConvolutionDescriptorProto) GetDilations 

func (m *ConvolutionDescriptorProto) GetDilations() []int64

func (*ConvolutionDescriptorProto) GetGroupCount 

func (m *ConvolutionDescriptorProto) GetGroupCount() int32

func (*ConvolutionDescriptorProto) GetName 

func (m *ConvolutionDescriptorProto) GetName() string

func (*ConvolutionDescriptorProto) GetPaddings 

func (m *ConvolutionDescriptorProto) GetPaddings() []int64

func (*ConvolutionDescriptorProto) GetStrides 

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

func (*ConvolutionDescriptorProto) ProtoMessage 

func (*ConvolutionDescriptorProto) ProtoMessage()

func (*ConvolutionDescriptorProto) Reset 

func (m *ConvolutionDescriptorProto) Reset()

func (*ConvolutionDescriptorProto) String 

func (m *ConvolutionDescriptorProto) String() string

func (*ConvolutionDescriptorProto) XXX_DiscardUnknown 

func (m *ConvolutionDescriptorProto) XXX_DiscardUnknown()

func (*ConvolutionDescriptorProto) XXX_Marshal 

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

func (*ConvolutionDescriptorProto) XXX_Merge 

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

func (*ConvolutionDescriptorProto) XXX_Size 

func (m *ConvolutionDescriptorProto) XXX_Size() int

func (*ConvolutionDescriptorProto) XXX_Unmarshal 

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

type ConvolutionKind 

type ConvolutionKind int32
const (
	ConvolutionKind_INVALID                 ConvolutionKind = 0
	ConvolutionKind_FORWARD                 ConvolutionKind = 1
	ConvolutionKind_BACKWARD_FILTER         ConvolutionKind = 2
	ConvolutionKind_BACKWARD_DATA           ConvolutionKind = 3
	ConvolutionKind_FORWARD_BIAS_ACTIVATION ConvolutionKind = 4
)

func (ConvolutionKind) EnumDescriptor 

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

func (ConvolutionKind) String 

func (x ConvolutionKind) String() string

type ConvolutionMode 

type ConvolutionMode int32

Describe the math definition for the conv op. The popular behavior is actually called cross-correlation in math, despite the operation is often referred as convolution. See cuDNN cudnnConvolutionMode_t. 

const (
	ConvolutionMode_CROSS_CORRELATION ConvolutionMode = 0
	ConvolutionMode_CONVOLUTION       ConvolutionMode = 1
)

func (ConvolutionMode) EnumDescriptor 

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

func (ConvolutionMode) String 

func (x ConvolutionMode) String() string

type DataLayout 

type DataLayout int32

Describes how a convolution input or output layer's data is formatted. 

const (
	// Naming convention:
	// Y <-> row or height
	// X <-> column or width
	// Batch <-> batch, or N
	// Depth <-> feature, or channel
	// TODO(timshen): turn them into cuDNN names, e.g. kNCHW.
	DataLayout_kYXDepthBatch  DataLayout = 0
	DataLayout_kYXBatchDepth  DataLayout = 1
	DataLayout_kBatchYXDepth  DataLayout = 2
	DataLayout_kBatchDepthYX  DataLayout = 3
	DataLayout_kBatchDepthYX4 DataLayout = 4
)

func (DataLayout) EnumDescriptor 

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

func (DataLayout) String 

func (x DataLayout) String() string

type DataType 

type DataType int32

Specifies the data type used by an operation. 

const (
	DataType_kFloat         DataType = 0
	DataType_kDouble        DataType = 1
	DataType_kHalf          DataType = 2
	DataType_kInt8          DataType = 3
	DataType_kInt32         DataType = 4
	DataType_kComplexFloat  DataType = 5
	DataType_kComplexDouble DataType = 6
)

func (DataType) EnumDescriptor 

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

func (DataType) String 

func (x DataType) String() string

type FilterLayout 

type FilterLayout int32

Describes how a convolution filter is laid out in the memory. 

const (
	// Naming convention:
	// Y <-> row or height
	// X <-> column or width
	// Output <-> output feature, or N
	// Input <-> input feature, or N
	// TODO(timshen): turn them into cuDNN names, e.g. kNCHW.
	FilterLayout_kOutputInputYX  FilterLayout = 0
	FilterLayout_kOutputYXInput  FilterLayout = 1
	FilterLayout_kOutputInputYX4 FilterLayout = 2
	FilterLayout_kInputYXOutput  FilterLayout = 3
	FilterLayout_kYXInputOutput  FilterLayout = 4
)

func (FilterLayout) EnumDescriptor 

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

func (FilterLayout) String 

func (x FilterLayout) String() string

type TensorDescriptorProto 

type TensorDescriptorProto struct {
	Dimensions []int64  `protobuf:"varint,1,rep,packed,name=dimensions,proto3" json:"dimensions,omitempty"`
	DataType   DataType `protobuf:"varint,2,opt,name=data_type,json=dataType,proto3,enum=stream_executor.dnn.DataType" json:"data_type,omitempty"`
	// Types that are valid to be assigned to LayoutOneof:
	//	*TensorDescriptorProto_DataLayout
	//	*TensorDescriptorProto_FilterLayout
	LayoutOneof          isTensorDescriptorProto_LayoutOneof `protobuf_oneof:"layout_oneof"`
	XXX_NoUnkeyedLiteral struct{}                            `json:"-"`
	XXX_unrecognized     []byte                              `json:"-"`
	XXX_sizecache        int32                               `json:"-"`
}

Generic tensor representation.

func (*TensorDescriptorProto) Descriptor 

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

func (*TensorDescriptorProto) GetDataLayout 

func (m *TensorDescriptorProto) GetDataLayout() DataLayout

func (*TensorDescriptorProto) GetDataType 

func (m *TensorDescriptorProto) GetDataType() DataType

func (*TensorDescriptorProto) GetDimensions 

func (m *TensorDescriptorProto) GetDimensions() []int64

func (*TensorDescriptorProto) GetFilterLayout 

func (m *TensorDescriptorProto) GetFilterLayout() FilterLayout

func (*TensorDescriptorProto) GetLayoutOneof 

func (m *TensorDescriptorProto) GetLayoutOneof() isTensorDescriptorProto_LayoutOneof

func (*TensorDescriptorProto) ProtoMessage 

func (*TensorDescriptorProto) ProtoMessage()

func (*TensorDescriptorProto) Reset 

func (m *TensorDescriptorProto) Reset()

func (*TensorDescriptorProto) String 

func (m *TensorDescriptorProto) String() string

func (*TensorDescriptorProto) XXX_DiscardUnknown 

func (m *TensorDescriptorProto) XXX_DiscardUnknown()

func (*TensorDescriptorProto) XXX_Marshal 

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

func (*TensorDescriptorProto) XXX_Merge 

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

func (*TensorDescriptorProto) XXX_OneofWrappers 

func (*TensorDescriptorProto) XXX_OneofWrappers() []interface{}

XXX_OneofWrappers is for the internal use of the proto package.

func (*TensorDescriptorProto) XXX_Size 

func (m *TensorDescriptorProto) XXX_Size() int

func (*TensorDescriptorProto) XXX_Unmarshal 

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

type TensorDescriptorProto_DataLayout 

type TensorDescriptorProto_DataLayout struct {
	DataLayout DataLayout `protobuf:"varint,3,opt,name=data_layout,json=dataLayout,proto3,enum=stream_executor.dnn.DataLayout,oneof"`
}

type TensorDescriptorProto_FilterLayout 

type TensorDescriptorProto_FilterLayout struct {
	FilterLayout FilterLayout `protobuf:"varint,4,opt,name=filter_layout,json=filterLayout,proto3,enum=stream_executor.dnn.FilterLayout,oneof"`
}

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.