Home
last modified time | relevance | path

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

/external/armnn/src/armnn/layers/
DArgMinMaxLayer.cpp52 std::vector<unsigned int> tensorDimensions(1, 1); in InferOutputShapes() local
58 std::vector<unsigned int> tensorDimensions(inputNumDimensions - 1, 0); in InferOutputShapes() local
DBatchMatMulLayer.cpp63 std::vector<unsigned int> tensorDimensions(outputNumDimensions, 0); in InferOutputShapes() local