Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_tile_transpose.cc82 auto indexing_maps = generic_op.getIndexingMapsArray(); in IsTransposeGenericOp() local
134 auto indexing_maps = generic_op.getIndexingMapsArray(); in runOnOperation() local
H A Dtf_jitrt_tile_reduction.cc167 auto indexing_maps = linalg_op.getIndexingMapsArray(); in matchAndRewrite() local
213 SmallVector<mlir::AffineMap, 2> indexing_maps( in matchAndRewrite() local
H A Dtf_jitrt_detensorize_linalg.cc62 mlir::SmallVector<AffineMap, 3> indexing_maps = op.getIndexingMapsArray(); in matchAndRewrite() local