Searched refs:node_to_opinfo (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 164 const std::unordered_map<const TfLiteNode*, OperatorInfo>& node_to_opinfo, in CreateCalibrator() argument
173 node_to_opinfo, std::move(logging_op_resolver), reporter); in CreateCalibrator()
176 for (const auto& entry : node_to_opinfo) { in CreateCalibrator()
301 node_to_opinfo, in GetNodeOpInfoMapAndContext() argument
313 interpreter->execution_plan().size() <= node_to_opinfo.size()); in GetNodeOpInfoMapAndContext()
314 for (const auto& entry : node_to_opinfo) { in GetNodeOpInfoMapAndContext()
391 absl::flat_hash_map<std::tuple<int, int>, OperatorInfo> node_to_opinfo; in BuildLoggingInterpreter() local
449 node_to_opinfo[key] = op_info; in BuildLoggingInterpreter()
471 node_to_opinfo, interpreter->get(), &node_ptr_opinfo_map, &context)); in BuildLoggingInterpreter()
|