Searched refs:function_map (Results 1 – 2 of 2) sorted by relevance
51 mlir::FuncOp>* function_map,82 mlir::FuncOp>* function_map, in HloFunctionImporter() argument87 function_map_(function_map) { in HloFunctionImporter()
110 std::unordered_map<const HloComputation*, FuncOp>* function_map, in ImportAsFunc() argument112 HloFunctionImporter importer(module, function_map, builder); in ImportAsFunc()