Searched refs:output_memory (Results 1 – 4 of 4) sorted by relevance
399 let mut output_memory = [0u8; 5]; in single_file_passthrough() localVariable400 let output_volatile_memory = VolatileSlice::new(&mut output_memory[..]); in single_file_passthrough()404 assert_eq!(input_memory, output_memory); in single_file_passthrough()466 let mut output_memory = [0u8; 200]; in triple_file_passthrough() localVariable467 let output_volatile_memory = VolatileSlice::new(&mut output_memory[..]); in triple_file_passthrough()471 assert!(input_memory.iter().eq(output_memory.iter())); in triple_file_passthrough()502 let mut output_memory = [0u8; 300]; in triple_file_punch_hole() localVariable503 let output_volatile_memory = VolatileSlice::new(&mut output_memory[..]); in triple_file_punch_hole()511 assert!(input_memory.iter().eq(output_memory.iter())); in triple_file_punch_hole()547 let mut output_memory = [0u8; 300]; in triple_file_write_zeroes() localVariable[all …]
69 dnnl_memory_t output_memory = nullptr; in DNNLSoftArgMax() local71 &output_memory, &memory_descriptor, engine, y.data()) != dnnl_success) in DNNLSoftArgMax()105 {DNNL_ARG_DST, output_memory}, in DNNLSoftArgMax()155 if (dnnl_memory_destroy(output_memory) != dnnl_success) { in DNNLSoftArgMax()
108 repeated int64 output_memory = 1; field
148 map<int32, Memory> output_memory = 11; field