Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dhlo_xla_runtime_pipeline.cc56 mlir::SmallVector<int64_t, 2> reduction_2d_tile_sizes = {4, 4}; in AddLinalgTransformations() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_jitrt_pipeline.cc80 mlir::SmallVector<int64_t, 2> reduction_2d_tile_sizes = {4, 4}; in AddLinalgTransformations() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_tile_reduction.cc408 llvm::ArrayRef<int64_t> reduction_2d_tile_sizes) { in CreateTileReductionPass()