Function kn_graph::cpu::cpu_eval_graph_exec
source · pub fn cpu_eval_graph_exec(
graph: &Graph,
batch_size: usize,
inputs: &[Tensor],
keep_all: bool
) -> ExecutionInfoExpand description
Evaluate the given graph on the CPU, with the given batch size and inputs, returning the full execution state including profiling information.
Prefer using cpu_eval_graph if only the output are necessary.
keep_all controls whether all intermediate tensors are kept in memory,
or freed as soon as they are no longer necessary.