Home
last modified time | relevance | path

Searched refs:custom_op_to_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()
90 if (custom_writer != custom_op_to_writer_.end() && in ExportOperators()
285 if (custom_op_to_writer_.find(custom_name) != custom_op_to_writer_.end()) { in RegisterCustomWriter()
288 custom_op_to_writer_.insert(std::make_pair(custom_name, custom_writer)); in RegisterCustomWriter()
Dwriter_lib.h126 std::unordered_map<std::string, CustomWriter> custom_op_to_writer_; variable