Skip to main content

eval_exec_segmented

Function eval_exec_segmented 

Source
pub fn eval_exec_segmented<B: TensorBackend>(
    backend: &mut B,
    program: &ExecProgram,
    inputs: Vec<Tensor>,
) -> Result<Vec<Tensor>>
Expand description

Evaluate an ExecProgram via segment-based dispatch.

ยงExamples

use tenferro::segment::eval_exec_segmented;
use tenferro::exec::ExecProgram;
use tenferro::CpuBackend;

let _eval: fn(&mut CpuBackend, &ExecProgram, Vec<tenferro::Tensor>) -> tenferro::error::Result<Vec<tenferro::Tensor>> =
    eval_exec_segmented::<CpuBackend>;