Searched refs:function_map_ (Results 1 – 4 of 4) sorted by relevance
47 *module.entry_computation(), module_, &function_map_, &builder_); in Import()51 *computation, module_, &function_map_, &builder_)); in Import()
87 function_map_(function_map) { in HloFunctionImporter()163 std::unordered_map<const xla::HloComputation*, mlir::FuncOp>* function_map_; variable
57 std::unordered_map<const xla::HloComputation*, mlir::FuncOp> function_map_; variable
126 auto& imported = (*function_map_)[&computation]; in ImportAsFunc()