pub struct GenerateOperation<'c> { /* private fields */ }
Expand description
A generate
operation. Creates a dynamically sized tensor from elements.
This operation creates a dynamically sized tensor with elements of any type. It expects one index operand per dynamic extent of the result tensor.
The body region defines the tensor’s elements. It takes index operands as
its region arguments that span the index space. The element at the given
position is yielded with the yield
operation (see YieldOp
). There is
no defined ordering to the invocations of the body. It is conceptually
a “parallel map” operation.
Example:
%tnsr = tensor.generate %m, %n {
^bb0(%i : index, %j : index, %k : index):
...
yield %elem : f32
} : tensor<?x3x?f32>
Implementations§
source§impl<'c> GenerateOperation<'c>
impl<'c> GenerateOperation<'c>
sourcepub fn as_operation(&self) -> &Operation<'c>
pub fn as_operation(&self) -> &Operation<'c>
Returns a generic operation.
sourcepub fn builder(
context: &'c Context,
location: Location<'c>
) -> GenerateOperationBuilder<'c, Unset, Unset, Unset>
pub fn builder( context: &'c Context, location: Location<'c> ) -> GenerateOperationBuilder<'c, Unset, Unset, Unset>
Creates a builder.
pub fn result(&self) -> Result<OperationResult<'c, '_>, Error>
pub fn dynamic_extents(&self) -> impl Iterator<Item = Value<'c, '_>>
pub fn body(&self) -> Result<RegionRef<'c, '_>, Error>
Trait Implementations§
source§impl<'c> From<GenerateOperation<'c>> for Operation<'c>
impl<'c> From<GenerateOperation<'c>> for Operation<'c>
source§fn from(operation: GenerateOperation<'c>) -> Self
fn from(operation: GenerateOperation<'c>) -> Self
Converts to this type from the input type.
Auto Trait Implementations§
impl<'c> RefUnwindSafe for GenerateOperation<'c>
impl<'c> !Send for GenerateOperation<'c>
impl<'c> !Sync for GenerateOperation<'c>
impl<'c> Unpin for GenerateOperation<'c>
impl<'c> UnwindSafe for GenerateOperation<'c>
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more