Searched refs:input_tensors (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/specs/AIDL_V2/ |
| D | pack.mod.py | 16 def test(name, axis_value, input_tensors, output_tensor, inputs_data, output_data): argument
17 …model = Model().Operation("PACK", Int32Scalar("axis", axis_value), *input_tensors).To(output_tenso…
19 …quant8_asymm_dict = dict(zip([*input_tensors, output_tensor], [quant8_asymm_type] * (len(input_ten…
22 …uant8_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…
29 input_tensors=[Input("in0", ("TENSOR_FLOAT32", [2]))],
38 input_tensors=[Input("in0", ("TENSOR_FLOAT32", [2]))],
47 input_tensors=[Input("in0", ("TENSOR_FLOAT32", [3,4])),
60 input_tensors=[Input("in0", ("TENSOR_FLOAT32", [3,4])),
76 input_tensors=[Input("in0", ("TENSOR_FLOAT32", [3,4])),
|