t2cudnn

package
v0.0.0-...-b0880df Latest Latest
Warning

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

Go to latest
Published: Aug 20, 2023 License: Apache-2.0 Imports: 3 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Describe

func Describe(t Tensor) (*cudnn.TensorDescriptor, error)

Describe extracts the metadata from a tensor.Dense and returns a cuDNN TensorDescriptor

func DescribeAsFilter

func DescribeAsFilter(t Tensor, format cudnn.TensorFormat) (*cudnn.Filter, error)

func Dtype2DataType

func Dtype2DataType(t tensor.Dtype) cudnn.DataType

Dtype2DataType converts a tensor.Dtype to a cudnnDataType.

Types

type Tensor

type Tensor interface {
	Shape() tensor.Shape
	Strides() []int
	Dtype() tensor.Dtype
}

Jump to

Keyboard shortcuts

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