Home
last modified time | relevance | path

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

/external/armnn/src/backends/tosaCommon/test/
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
DAvgPool2DIgnoreValueChecker.hpp14 std::vector<std::vector<int32_t>> intermediateShape, in VerifyAvgPool2DIgnoreValue()
/external/armnn/src/backends/tosaCommon/operatorMappings/
DAvgPool2DIgnoreValueOperator.cpp92 std::vector<int32_t> intermediateShape; in ConvertAvgPool2DIgnoreValueToTosaOperator() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Drestrict_max_rank.cc151 auto intermediateShape = llvm::to_vector<4>(inputTy.getShape()); in matchAndRewrite() local