Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dhlo_legalize_to_memref.cc348 auto broadcastInDimOp = cast<mhlo::DynamicBroadcastInDimOp>(op); in bufferize() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc3307 if (auto broadcastInDimOp = llvm::dyn_cast_or_null<BroadcastInDimOp>( in matchAndRewrite() local
7283 auto broadcastInDimOp = op.operand().getDefiningOp<BroadcastInDimOp>(); in eliminateBroadcastInDimTranspose() local