t2cudnn

package

v0.0.0-...-b0880df Latest Latest Go to latest Published: Aug 20, 2023 License: Apache-2.0 Imports: 3 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

This section is empty.

This section is empty.

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

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

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

func Dtype2DataType(t tensor.Dtype) cudnn.DataType

Dtype2DataType converts a tensor.Dtype to a cudnnDataType.

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