Home
last modified time | relevance | path

Searched defs:tensorDimensions (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/armnn/src/armnn/layers/
H A DArgMinMaxLayer.cpp52 std::vector<unsigned int> tensorDimensions(1, 1); in InferOutputShapes() local
58 std::vector<unsigned int> tensorDimensions(inputNumDimensions - 1, 0); in InferOutputShapes() local
H A DBatchMatMulLayer.cpp63 std::vector<unsigned int> tensorDimensions(outputNumDimensions, 0); in InferOutputShapes() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp4912 const uint32_t tensorDimensions[3] = {4, 5, 6}; in reverseTest() local
4948 const uint32_t tensorDimensions[3] = {4, 5, 6}; in reverseTestBadQuantization() local
4995 const uint32_t tensorDimensions[3] = {4, 5, 6}; in reverseTestBadRank() local