Home
last modified time | relevance | path

Searched refs:spatialDimMapping (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc1913 llvm::SmallVector<int64_t> spatialDimMapping(rank - 2); in matchAndRewrite() local
1914 std::iota(spatialDimMapping.begin(), spatialDimMapping.end(), 1); in matchAndRewrite()
1916 op.lhs_dilationAttr(), spatialDimMapping, in matchAndRewrite()
2317 llvm::SmallVector<int64_t> spatialDimMapping(spatialRank); in matchAndRewrite() local
2318 std::iota(spatialDimMapping.begin(), spatialDimMapping.end(), 1); in matchAndRewrite()
2320 op.lhs_dilationAttr(), spatialDimMapping, in matchAndRewrite()