Struct GraphProgramInput 

pub struct GraphProgramInput { /* private fields */ }

A single ordered input required by a GraphProgram.

Examples

use tenferro_runtime::{GraphCompiler, TracedTensor};

let x = TracedTensor::from_vec_col_major(vec![2], vec![1.0_f64, 2.0]).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&x.neg()).unwrap();
let input = &program.input_specs()[0];
assert_eq!(input.shape(), &[2]);

Implementations

impl GraphProgramInput

pub fn dtype(&self) -> DType

Return the dtype expected for this input.

Examples

use tenferro_runtime::{DType, GraphCompiler, TracedTensor};

let x = TracedTensor::input_symbolic_shape(DType::F64, 1).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler
    .compile_with_input_specs(&x, &[(&x, DType::F64, &[2])])
    .unwrap();
assert_eq!(program.input_specs()[0].dtype(), DType::F64);

pub fn shape(&self) -> &[usize]

Return the concrete shape expected for this input.

Examples

use tenferro_runtime::{GraphCompiler, TracedTensor};

let x = TracedTensor::from_vec_col_major(vec![2], vec![1.0_f64, 2.0]).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&x).unwrap();
assert_eq!(program.input_specs()[0].shape(), &[2]);

Trait Implementations

impl Clone for GraphProgramInput

fn clone(&self) -> GraphProgramInput

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphProgramInput

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

Formats the value using the given formatter.