Home
last modified time | relevance | path

Searched refs:node_and_reg (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dtest_util.mm46 const auto* node_and_reg = interpreter->node_and_registration(node_index);
47 XCTAssertTrue(node_and_reg != nullptr);
48 XCTAssertTrue(node_and_reg->second.custom_name != nullptr);
50 node_and_reg->second.custom_name ==
60 const auto* node_and_reg = interpreter->node_and_registration(node_idx);
61 XCTAssertTrue(node_and_reg != nullptr);
62 if (node_and_reg->second.custom_name != nullptr) {
64 node_and_reg->second.custom_name !=
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dhexagon_delegate_op_model.h68 const auto* node_and_reg = interpreter_->node_and_registration(node); in ApplyDelegateAndInvoke() local
69 ASSERT_TRUE(node_and_reg != nullptr); in ApplyDelegateAndInvoke()
70 ASSERT_TRUE(node_and_reg->second.custom_name != nullptr); in ApplyDelegateAndInvoke()
71 ASSERT_STREQ(kDelegateName, node_and_reg->second.custom_name); in ApplyDelegateAndInvoke()
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc71 const auto* node_and_reg = interpreter_->node_and_registration(node); in TEST_F() local
72 EXPECT_STREQ("DummyDelegate", node_and_reg->second.custom_name); in TEST_F()
73 EXPECT_EQ(1, node_and_reg->second.version); in TEST_F()
76 node_and_reg->first.builtin_data); in TEST_F()
/external/tensorflow/tensorflow/lite/
Doptional_debug_tools.cc127 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg = in PrintInterpreterState() local
129 const TfLiteNode& node = node_and_reg->first; in PrintInterpreterState()
130 const TfLiteRegistration& reg = node_and_reg->second; in PrintInterpreterState()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test.cc57 const auto* node_and_reg = interpreter_->node_and_registration(node); in TEST_F() local
58 EXPECT_EQ(node_and_reg->second.custom_name, in TEST_F()
62 node_and_reg->first.builtin_data); in TEST_F()
266 const auto* node_and_reg = interpreter_->node_and_registration(3); in TEST_F() local
267 ASSERT_EQ(node_and_reg->second.custom_name, in TEST_F()
361 const auto* node_and_reg = interpreter_->node_and_registration(3); in TEST_F() local
362 ASSERT_EQ(node_and_reg->second.custom_name, in TEST_F()
784 const auto* node_and_reg = in TEST_F() local
786 EXPECT_EQ(node_and_reg->second.custom_name, in TEST_F()
801 const auto* node_and_reg = in TEST_F() local
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc766 auto& node_and_reg = nodes_and_registration_.back(); in AddNodeWithParameters() local
767 TfLiteNode& node = node_and_reg.first; in AddNodeWithParameters()
803 node_and_reg.second = *registration; in AddNodeWithParameters()
1180 auto& node_and_reg = nodes_and_registration_[node_index]; in GetNodeAndRegistration() local
1181 *node = &node_and_reg.first; in GetNodeAndRegistration()
1182 *registration = &node_and_reg.second; in GetNodeAndRegistration()
1416 auto& node_and_reg = nodes_and_registration_[node_index]; in UndoAllDelegates() local
1417 const TfLiteNode& node = node_and_reg.first; in UndoAllDelegates()
1418 const TfLiteRegistration& reg = node_and_reg.second; in UndoAllDelegates()
1434 auto& node_and_reg = nodes_and_registration_[node_index]; in UndoAllDelegates() local
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.cc292 const auto* node_and_reg = interpreter->node_and_registration(entry.first); in GetNodeOpInfoMapAndContext() local
293 op_info.registration = &node_and_reg->second; in GetNodeOpInfoMapAndContext()
294 node_ptr_opinfo_map->insert({&node_and_reg->first, op_info}); in GetNodeOpInfoMapAndContext()