Home
last modified time | relevance | path

Searched refs:custom_writer (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib.cc89 auto custom_writer = custom_op_to_writer_.find(registration.custom_name); in ExportOperators() local
90 if (custom_writer != custom_op_to_writer_.end() && in ExportOperators()
91 custom_writer->second) { in ExportOperators()
93 custom_writer->second(fbb, interpreter_, op_index, &custom_options, in ExportOperators()
284 const std::string& custom_name, CustomWriter custom_writer) { in RegisterCustomWriter() argument
288 custom_op_to_writer_.insert(std::make_pair(custom_name, custom_writer)); in RegisterCustomWriter()
Dwriter_lib.h64 CustomWriter custom_writer);