Searched refs:input_tensor (Results 1 – 3 of 3) sorted by relevance
| /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)
23 input_tensor: quant8_asymm_type,
28 input_tensor: quant8_asymm_signed_type,
32 input_tensor: input_data,
42 input_tensor=Input("in", ("TENSOR_FLOAT32", [3])),
51 input_tensor=Input("in", ("TENSOR_FLOAT32", [2,3,4])),
60 input_tensor=Input("in", ("TENSOR_FLOAT32", [2,3,4])),
70 input_tensor=Input("in", ("TENSOR_FLOAT32", [2,3,4])),
|
| D | mirror_pad.mod.py | 43 input_tensor = Input("in", ("TENSOR_FLOAT32", input_dims))
47 … model = Model().Operation("MIRROR_PAD", input_tensor, padding_tensor, mode).To(output_tensor)
60 input_tensor: quant8_asymm_type,
65 input_tensor: quant8_asymm_signed_type,
70 input_tensor: input_data,
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 2660 * [-rank(input_tensor), rank(input_tensor)).
|