Home
last modified time | relevance | path

Searched defs:intermediateShape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/tosaCommon/test/
H A DOneToManyMappingTests.cpp30 std::vector<std::vector<int32_t>> intermediateShape = {{ 1, 1, 6, 6 }}; variable
71 std::vector<std::vector<int32_t>> intermediateShape = {{ 1, 1, 6, 6 }}; variable
H A DAvgPool2DIgnoreValueChecker.hpp14 std::vector<std::vector<int32_t>> intermediateShape, in VerifyAvgPool2DIgnoreValue()
/aosp_15_r20/external/armnn/src/backends/tosaCommon/operatorMappings/
H A DAvgPool2DIgnoreValueOperator.cpp92 std::vector<int32_t> intermediateShape; in ConvertAvgPool2DIgnoreValueToTosaOperator() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Drestrict_max_rank.cc151 auto intermediateShape = llvm::to_vector<4>(inputTy.getShape()); in matchAndRewrite() local