Searched refs:custom_op_to_writer_ (Results 1 – 2 of 2) sorted by relevance
128 auto custom_writer = custom_op_to_writer_.find(registration.custom_name); in ExportOperators()129 if (custom_writer != custom_op_to_writer_.end() && in ExportOperators()320 if (custom_op_to_writer_.find(custom_name) != custom_op_to_writer_.end()) { in RegisterCustomWriter()323 custom_op_to_writer_.insert(std::make_pair(custom_name, custom_writer)); in RegisterCustomWriter()
198 std::unordered_map<std::string, CustomWriter> custom_op_to_writer_; variable