Searched refs:nodes_to_replace (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 190 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate) { in ReplaceNodeSubsetsWithDelegateKernels() argument
192 ->ReplaceNodeSubsetsWithDelegateKernels(registration, nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels()
252 params->nodes_to_replace = reinterpret_cast<TfLiteIntArray*>(allocation); in CreateDelegateParams()
253 CopyVectorToTfLiteIntArray(node_subset.nodes, params->nodes_to_replace); in CreateDelegateParams()
271 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels() argument
280 PartitionGraphIntoIndependentNodeSubsets(&info, nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels()
986 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate) { in SwitchToKernelContext() argument
|
| D | subgraph.h | 383 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate);
391 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace,
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_internal.h | 444 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate);
597 TfLiteIntArray* nodes_to_replace; member
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 395 op_data->nodes.reserve(params->nodes_to_replace->size); in Init()
397 CHECK(params->nodes_to_replace); in Init()
399 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 760 TfLiteIntArray nodes_to_replace; in TEST() local
761 nodes_to_replace.size = 0; in TEST()
763 context, delegate_registration, &nodes_to_replace, nullptr), in TEST()
1205 ASSERT_EQ(params->nodes_to_replace->size, 3); in TEST_F()
1206 EXPECT_EQ(params->nodes_to_replace->data[0], 0); in TEST_F()
1207 EXPECT_EQ(params->nodes_to_replace->data[1], 1); in TEST_F()
1208 EXPECT_EQ(params->nodes_to_replace->data[2], 2); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 987 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
|