Searched defs:axis0 (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| D | concat_float_1.mod.py | 21 axis0 = Int32Scalar("axis0", 0) variable
|
| D | concat_float_2.mod.py | 27 axis0 = Int32Scalar("axis0", 0) variable
|
| D | concat_quant8_2.mod.py | 27 axis0 = Int32Scalar("axis0", 0) variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | concat_float16_1.mod.py | 21 axis0 = Int32Scalar("axis0", 0) variable
|
| D | concat_float16_2.mod.py | 27 axis0 = Int32Scalar("axis0", 0) variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| D | concat_float_1_relaxed.mod.py | 21 axis0 = Int32Scalar("axis0", 0) variable
|
| D | concat_float_2_relaxed.mod.py | 27 axis0 = Int32Scalar("axis0", 0) variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | concat_quant8_signed.mod.py | 86 axis0 = Int32Scalar("axis0", 0) variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.cpp | 3925 Tensor swapaxes(const Tensor& self, int64_t axis0, int64_t axis1) { in swapaxes()
3929 Tensor& swapaxes_(Tensor& self, int64_t axis0, int64_t axis1) { in swapaxes_()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | generated_ops.cpp | 5168 const auto axis0 = p_node->Input(1).toInt(); in __anon5d9c3eb91f602() local
|