Searched refs:RewriteFusionOperand (Results 1 – 2 of 2) sorted by relevance
212 xla::StatusOr<Value> RewriteFusionOperand(const xla::HloInstruction* root,
446 StatusOr<Value> LhloDialectEmitter::RewriteFusionOperand( in RewriteFusionOperand() function in mlir::LhloDialectEmitter454 auto v, RewriteFusionOperand(root, shape.tuple_shapes(i), shape_index, in RewriteFusionOperand()491 auto arg, RewriteFusionOperand(operand, operand->shape(), &shape_index, in EmitFusionOp()