Struct melior::dialect::ods::gpu::CreateDnTensorOperation

pub struct CreateDnTensorOperation<'c> { /* private fields */ }

A create_dn_tensor operation. Create dense tensor operation.

The gpu.create_dn_tensor operation initializes a dense tensor from the given values buffer and sizes. The buffer must already be copied from the host to the device prior to using this operation. The operation returns a handle to the dense tensor descriptor.

If the async keyword is present, the op is executed asynchronously (i.e. it does not block until the execution has finished on the device). In that case, it returns a !gpu.async.token in addition to the environment.

Example:

%dmat, %token = gpu.create_dn_tensor async [%dep] %mem, %dims : index, index into memref<?xf64>

Implementations

impl<'c> CreateDnTensorOperation<'c>

pub fn name() -> &'static str

Returns a name.

pub fn as_operation(&self) -> &Operation<'c>

Returns a generic operation.

pub fn builder( context: &'c Context, location: Location<'c> ) -> CreateDnTensorOperationBuilder<'c, Unset, Unset, Unset, Unset>

Creates a builder.

pub fn dn_tensor(&self) -> Result<OperationResult<'c, '_>, Error>

pub fn async_token(&self) -> Result<OperationResult<'c, '_>, Error>

pub fn async_dependencies( &self ) -> Result<impl Iterator<Item = Value<'c, '_>>, Error>

pub fn memref(&self) -> Result<Value<'c, '_>, Error>

pub fn dims(&self) -> Result<impl Iterator<Item = Value<'c, '_>>, Error>

Trait Implementations

impl<'c> From<CreateDnTensorOperation<'c>> for Operation<'c>

fn from(operation: CreateDnTensorOperation<'c>) -> Self

Converts to this type from the input type.

impl<'c> TryFrom<Operation<'c>> for CreateDnTensorOperation<'c>

type Error = Error

The type returned in the event of a conversion error.

fn try_from(operation: Operation<'c>) -> Result<Self, Self::Error>

Performs the conversion.