Searched refs:custom_writer (Results 1 – 2 of 2) sorted by relevance
128 auto custom_writer = custom_op_to_writer_.find(registration.custom_name); in ExportOperators() local129 if (custom_writer != custom_op_to_writer_.end() && in ExportOperators()130 custom_writer->second) { in ExportOperators()132 custom_writer->second(fbb, subgraph_, op_index, &custom_options, in ExportOperators()319 const std::string& custom_name, CustomWriter custom_writer) { in RegisterCustomWriter() argument323 custom_op_to_writer_.insert(std::make_pair(custom_name, custom_writer)); in RegisterCustomWriter()
106 CustomWriter custom_writer);