pub fn make_tensor_for_node_output(
ctx: &EvalContext<'_>,
slot: usize,
dt: DatumType,
shape: &[usize],
) -> TractResult<DeviceTensor>Expand description
Like make_tensor_for_node but for one output slot of a multi-output
node: the memory schema reserves one arena region per device output.