Searched refs:node_to_opinfo (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 133 const std::unordered_map<const TfLiteNode*, OperatorInfo>& node_to_opinfo, in CreateCalibrator() argument
142 node_to_opinfo, std::move(logging_op_resolver)); in CreateCalibrator()
218 const std::unordered_map<int, OperatorInfo>& node_to_opinfo, in GetNodeOpInfoMapAndContext() argument
224 NodeInfoDelegateObserver delegate_observer(node_to_opinfo, in GetNodeOpInfoMapAndContext()
284 std::unordered_map<int, OperatorInfo> node_to_opinfo; in BuildLoggingInterpreter() local
311 node_to_opinfo[i] = op_info; in BuildLoggingInterpreter()
330 GetNodeOpInfoMapAndContext(node_to_opinfo, interpreter->get(), in BuildLoggingInterpreter()
|
| D | node_info_delegate_test.cc | 115 std::unordered_map<const TfLiteNode*, OperatorInfo> node_to_opinfo; in TEST() local
116 NodeInfoDelegateObserver observer(index_to_opinfo, &node_to_opinfo); in TEST()
129 EXPECT_EQ(index_to_opinfo.size(), node_to_opinfo.size()); in TEST()
130 EXPECT_EQ(interpreter->nodes_size(), node_to_opinfo.size()); in TEST()
132 for (const auto& node_and_opinfo : node_to_opinfo) { in TEST()
|