Searched refs:output_buffer (Results 1 – 7 of 7) sorted by relevance
44 let mut output_buffer = vec![0f32; 1001]; in main() localVariable49 &mut output_buffer[..] as *mut [f32] as *mut u8, in main()50 (output_buffer.len() * 4).try_into().unwrap(), in main()56 &sort_results(&output_buffer)[..5] in main()
26 let mut output_buffer = vec![0f32; 1001]; in main() localVariable27 context.get_output(0, &mut output_buffer[..]).unwrap(); in main()31 &sort_results(&output_buffer)[..5] in main()
30 let mut output_buffer = vec![0f32; 1000]; in main() localVariable31 context.get_output(0, &mut output_buffer[..])?; in main()32 let result = softmax(output_buffer); in main()
15 let output_buffer = wit::classify(graph, ("input", tensor))?; in main() localVariable16 let result = softmax(output_buffer); in main()
15 let output_buffer = witx::classify(graph, tensor)?; in main() localVariable16 let result = softmax(output_buffer); in main()
41 let mut output_buffer = vec![0f32; 1000]; in main() localVariable42 context.get_output(0, &mut output_buffer[..]).unwrap(); in main()45 let result = postprocess(output_buffer); in main()
130 let mut output_buffer = vec![0u8; 1001 * std::mem::size_of::<f32>()]; in classify() localVariable131 let num_bytes = context.get_output(0, &mut output_buffer)?; in classify()133 let output: Vec<f32> = output_buffer[..num_bytes] in classify()