Home
last modified time | relevance | path

Searched refs:output_tensor (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/runtime/test/specs/AIDL_V2/
Dpack.mod.py16 def test(name, axis_value, input_tensors, output_tensor, inputs_data, output_data): argument
17 …odel = Model().Operation("PACK", Int32Scalar("axis", axis_value), *input_tensors).To(output_tensor)
19 …quant8_asymm_dict = dict(zip([*input_tensors, output_tensor], [quant8_asymm_type] * (len(input_ten…
22 …quant8_asymm_signed_dict = dict(zip([*input_tensors, output_tensor], [quant8_asymm_signed_type] * …
24 …Example((dict(zip(input_tensors, inputs_data)), {output_tensor: output_data}), model=model, name=n…
30 output_tensor=Output("out", ("TENSOR_FLOAT32", [1,2])),
39 output_tensor=Output("out", ("TENSOR_FLOAT32", [2,1])),
49 output_tensor=Output("out", ("TENSOR_FLOAT32", [2,3,4])),
62 output_tensor=Output("out", ("TENSOR_FLOAT32", [3,2,4])),
78 output_tensor=Output("out", ("TENSOR_FLOAT32", [3,4,2])),
/packages/modules/NeuralNetworks/runtime/test/specs/AIDL_V3/
Dreverse.mod.py19 def test(name, axis_value, input_tensor, output_tensor, input_data, output_data): argument
20 model = Model().Operation("REVERSE", input_tensor, [axis_value]).To(output_tensor)
24 output_tensor: quant8_asymm_type,
29 output_tensor: quant8_asymm_signed_type,
33 output_tensor: output_data,
43 output_tensor=Output("out", ("TENSOR_FLOAT32", [3])),
52 output_tensor=Output("out", ("TENSOR_FLOAT32", [2,3,4])),
61 output_tensor=Output("out", ("TENSOR_FLOAT32", [2,3,4])),
71 output_tensor=Output("out", ("TENSOR_FLOAT32", [2,3,4])),
Dmirror_pad.mod.py46 output_tensor = Output("out", ("TENSOR_FLOAT32", output_dims))
47 … model = Model().Operation("MIRROR_PAD", input_tensor, padding_tensor, mode).To(output_tensor)
61 output_tensor: quant8_asymm_type,
66 output_tensor: quant8_asymm_signed_type,
71 output_tensor: output_data,
/packages/modules/NeuralNetworks/runtime/
DNeuralNetworksV2.cpp865 std::memcpy(context->outputs[i], context->interpreter->output_tensor(i)->data.raw, in ANeuralNetworksExecution_compute()
1398 if (bufferSize != context->interpreter->output_tensor(index)->bytes) { in ANeuralNetworksExecution_setOutput()