Searched refs:axis_value (Results 1 – 2 of 2) 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…
28 axis_value=0,
37 axis_value=1,
46 axis_value=0,
59 axis_value=1,
75 axis_value=2,
|
| /packages/modules/NeuralNetworks/runtime/test/specs/AIDL_V3/ |
| D | reverse.mod.py | 19 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)
41 axis_value=0,
50 axis_value=0,
59 axis_value=1,
69 axis_value=2,
|