Skip to main content

make_tensor_for_node

tract_gpu::session_handler

Function make_tensor_for_node 

Source 
pub fn make_tensor_for_node(
    session: &TurnState,
    node_id: usize,
    dt: DatumType,
    shape: &[usize],
) -> TractResult<DeviceTensor>