Searched defs:axis_u (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEStackLayer.cpp | 53 …const unsigned int axis_u = wrap_around(axis, static_cast<int>(input[0]->info()->num_dimensions() … in configure() local
69 const unsigned int axis_u = wrap_around(axis, static_cast<int>(rank + 1)); in validate() local
|
| D | NEUnstack.cpp | 75 const unsigned int axis_u = wrap_axis(axis, input->info()); in configure() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLStackLayer.cpp | 60 …const unsigned int axis_u = wrap_around(axis, static_cast<int>(input[0]->info()->num_dimensions() … in configure() local
76 const unsigned int axis_u = wrap_around(axis, static_cast<int>(rank + 1)); in validate() local
|
| D | CLUnstack.cpp | 80 const unsigned int axis_u = wrap_axis(axis, input->info()); in configure() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| 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
|
| D | StackLayerFixture.h | 125 … const unsigned int axis_u = wrap_around(axis, static_cast<int>(shape_src.num_dimensions() + 1)); in compute_reference() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Unstack.cpp | 90 …const unsigned int axis_u = wrap_around(axis, static_cast<int>(input_tensor.shape().num_dimensions… in unstack() local
|