Searched refs:func_builder (Results 1 – 2 of 2) sorted by relevance
86 mlir::OpBuilder* func_builder) { in GenerateInstructionLocation() argument89 return mlir::NameLoc::get(func_builder->getIdentifier(instruction->name()), in GenerateInstructionLocation()90 func_builder->getContext()); in GenerateInstructionLocation()94 func_builder->getIdentifier(op_name), func_builder->getContext()); in GenerateInstructionLocation()103 func_builder->getContext())}, in GenerateInstructionLocation()104 func_builder->getContext()); in GenerateInstructionLocation()228 HloInstruction* instruction, mlir::OpBuilder* func_builder) { in ImportInstructionImpl() argument232 mlir::Location loc = GenerateInstructionLocation(instruction, func_builder); in ImportInstructionImpl()244 func_builder->create<mlir::mhlo::ConstOp>(loc, attr.ValueOrDie()); in ImportInstructionImpl()251 return func_builder in ImportInstructionImpl()[all …]
111 mlir::OpBuilder* func_builder);113 HloInstruction* instruction, mlir::OpBuilder* func_builder);