Searched refs:LhloDialectEmitter (Results 1 – 3 of 3) sorted by relevance
210 Status LhloDialectEmitter::CreateOperands( in CreateOperands()227 OpType LhloDialectEmitter::CreateOpWithoutAttrs(const HloInstruction* instr, in CreateOpWithoutAttrs()235 StatusOr<OpType> LhloDialectEmitter::CreateOpWithoutAttrs( in CreateOpWithoutAttrs()244 StatusOr<mlir::Operation*> LhloDialectEmitter::EmitOp( in EmitOp()412 Status LhloDialectEmitter::DefaultAction(const HloInstruction* instr) { in DefaultAction()416 StatusOr<lmhlo::SortOp> LhloDialectEmitter::EmitSortOp( in EmitSortOp()446 StatusOr<Value> LhloDialectEmitter::RewriteFusionOperand( in RewriteFusionOperand()474 StatusOr<lmhlo::FusionOp> LhloDialectEmitter::EmitFusionOp( in EmitFusionOp()544 LhloDialectEmitter::GetScatterDimensionNumbers(const HloInstruction* instr) { in GetScatterDimensionNumbers()558 StatusOr<lmhlo::ScatterOp> LhloDialectEmitter::EmitScatterOp( in EmitScatterOp()[all …]
38 class LhloDialectEmitter : public xla::ConstDfsHloVisitorWithDefault {44 LhloDialectEmitter(const xla::BufferAssignment& assignment, in LhloDialectEmitter() function
750 absl::optional<mlir::LhloDialectEmitter> lhlo_scratch_emitter_;