Searched defs:axis_u (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEStackLayer.cpp | 55 …const unsigned int axis_u = wrap_around(axis, static_cast<int>(input[0]->info()->num_dimensions() … in configure() local
71 const unsigned int axis_u = wrap_around(axis, static_cast<int>(rank + 1)); in validate() local
|
| H A D | NEUnstack.cpp | 77 const unsigned int axis_u = wrap_axis(axis, input->info()); in configure() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLStackLayer.cpp | 62 …const unsigned int axis_u = wrap_around(axis, static_cast<int>(input[0]->info()->num_dimensions() … in configure() local
78 const unsigned int axis_u = wrap_around(axis, static_cast<int>(rank + 1)); in validate() local
|
| H A D | CLUnstack.cpp | 83 const unsigned int axis_u = wrap_axis(axis, input->info()); in configure() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | UnstackFixture.h | 67 …const unsigned int axis_u = wrap_around(axis, static_cast<int>(input_shape.num_dimen… in compute_target() local
95 …const unsigned int axis_u = wrap_around(axis, static_cast<int>(input_shape.num_dimensi… in compute_reference() local
|
| H A D | StackLayerFixture.h | 125 … const unsigned int axis_u = wrap_around(axis, static_cast<int>(shape_src.num_dimensions() + 1)); in compute_reference() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Unstack.cpp | 90 …const unsigned int axis_u = wrap_around(axis, static_cast<int>(input_tensor.shape().num_dimensions… in unstack() local
|