Struct GraphInstructionView

Source

pub struct GraphInstructionView<'a> { /* private fields */ }

Expand description

Read-only lowering view over one execution instruction.

§Examples

use tenferro_runtime::{GraphCompiler, TracedTensor};

let x = TracedTensor::from_vec_col_major(vec![1], vec![2.0_f64]).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&x.neg()).unwrap();
let inst = program.lowering_view().instructions().next().unwrap();
assert_eq!(inst.output_slots().len(), 1);

Implementations§

Source §

impl<'a> GraphInstructionView<'a>

Source

pub fn op(&self) -> GraphOpView<'a>

Return the operation view for this instruction.

§Examples

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

let x = TracedTensor::from_vec_col_major(vec![1], vec![2.0_f64]).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&x.neg()).unwrap();
let inst = program.lowering_view().instructions().next().unwrap();
assert!(matches!(inst.op(), GraphOpView::Negate));

Source

pub fn op_name(&self) -> &'static str

Return a stable operation name for diagnostics.

§Examples

use tenferro_runtime::{GraphCompiler, TracedTensor};

let x = TracedTensor::from_vec_col_major(vec![1], vec![2.0_f64]).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&x.neg()).unwrap();
let inst = program.lowering_view().instructions().next().unwrap();
assert_eq!(inst.op_name(), "Negate");

Source

pub fn input_slots(&self) -> &'a [usize]

Return the input slots consumed by this instruction.

§Examples

use tenferro_runtime::{GraphCompiler, TracedTensor};

let x = TracedTensor::from_vec_col_major(vec![1], vec![2.0_f64]).unwrap();
let y = (&x + &x).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&y).unwrap();
let inst = program.lowering_view().instructions().next().unwrap();
assert_eq!(inst.input_slots().len(), 2);

Source

pub fn output_slots(&self) -> &'a [usize]

Return the output slots written by this instruction.

§Examples

use tenferro_runtime::{GraphCompiler, TracedTensor};

let x = TracedTensor::from_vec_col_major(vec![1], vec![2.0_f64]).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&x.neg()).unwrap();
let inst = program.lowering_view().instructions().next().unwrap();
assert_eq!(inst.output_slots().len(), 1);

Source

pub fn dtype(&self) -> DType

Return the dtype of this instruction’s output.

§Examples

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

let x = TracedTensor::from_vec_col_major(vec![1], vec![2.0_f64]).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&x.neg()).unwrap();
let inst = program.lowering_view().instructions().next().unwrap();
assert_eq!(inst.dtype(), DType::F64);

Source

pub fn static_output_shape( &self, output_index: usize, input_shapes: &[&[usize]], ) -> Result<Vec<usize>, GraphProgramLoweringShapeError>

Resolve an exact static output shape for this instruction.

input_shapes must be ordered the same way as Self::input_slots.

§Examples

use tenferro_runtime::{GraphCompiler, TracedTensor};

let x = TracedTensor::from_vec_col_major(vec![1], vec![2.0_f64]).unwrap();
let mut compiler = GraphCompiler::new();
let program = compiler.compile(&x.neg()).unwrap();
let inst = program.lowering_view().instructions().next().unwrap();
assert_eq!(inst.static_output_shape(0, &[&[1]]).unwrap(), vec![1]);