Struct InputSpec 

pub struct InputSpec {
    pub shape: Vec<usize>,
    pub dtype: DType,
}

Shape + dtype descriptor for a single JIT input. Used by jit_wrapper!-generated prepare() calls to allocate zero-initialized placeholder buffers internally — callers no longer construct fake Tensor::zeros(..).realize() placeholders.

Fields

shape: Vec<usize>

dtype: DType

Implementations

impl InputSpec

pub fn new(shape: &[usize], dtype: DType) -> Self

pub fn f32(shape: &[usize]) -> Self

pub fn i32(shape: &[usize]) -> Self

pub fn i64(shape: &[usize]) -> Self

Trait Implementations

impl Clone for InputSpec

fn clone(&self) -> InputSpec

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for InputSpec

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.