Searched defs:axisShape (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | SplitTest.cpp | 21 std::vector<int32_t> axisShape { 1 }; in SplitUint8Test() local
51 std::vector<int32_t> axisShape { 1 }; in SplitFp32Test() local
135 std::vector<int32_t> axisShape { 1 }; in SplitVUint8Test() local
173 std::vector<int32_t> axisShape { 1 }; in SplitVFp32Test() local
|
| H A D | ArgMinMaxTest.cpp | 23 std::vector<int32_t> axisShape { 1 }; in ArgMaxFP32Test() local
55 std::vector<int32_t> axisShape { 1 }; in ArgMinFP32Test() local
87 std::vector<int32_t> axisShape { 1 }; in ArgMaxUint8Test() local
|
| H A D | ArgMinMaxTestHelper.hpp | 133 const std::vector<int32_t>& axisShape, in ArgMinMaxTest()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SimpleMath.cpp | 40 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
54 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
|
| D | Reverse.cpp | 49 const Shape axisShape = context->getInputShape(kInputAxisTensor); in prepare() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 501 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare()
|