Home
last modified time | relevance | path

Searched defs:broadcastOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dshape_simplification.cc158 auto broadcastOp = op.getTensor().getDefiningOp<BroadcastOp>(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc645 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy broadcastOp, in getIndexingMaps()
683 mhlo::BroadcastInDimOp broadcastOp, Builder* b) { in getIndexingMaps()