Struct async_tensorrt::ffi::network::Tensor

pub struct Tensor<'parent> { /* private fields */ }

A tensor in a NetworkDefinition.

TensorRT documenation

Implementations

impl<'parent> Tensor<'parent>

pub fn name(&self) -> String

Get the tensor name.

TensorRT documentation

pub fn set_name(&mut self, name: &str)

Set the tensor name.

TensorRT documentation

Arguments

• name - Name to set.

pub fn get_dimensions(&self) -> Vec<i32>

Get the dimensions of a tensor.

TensorRT documentation

pub fn as_ptr(&self) -> *const c_void

Get internal readonly pointer.

pub fn as_mut_ptr(&mut self) -> *mut c_void

Get internal mutable pointer.

Trait Implementations

impl<'parent> Send for Tensor<'parent>

Implements Send for Tensor.

Safety

The TensorRT API is thread-safe with regards to all operations on Tensor.

impl<'parent> Sync for Tensor<'parent>

Implements Sync for Tensor.

Safety

The TensorRT API is thread-safe with regards to all operations on Tensor.