Home
last modified time | relevance | path

Searched refs:nodes_to_replace (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/
Dutils_test.cc85 params1.nodes_to_replace = TfLiteIntArrayCreate(1);
86 params1.nodes_to_replace->data[0] = 1;
90 params2.nodes_to_replace = TfLiteIntArrayCreate(4);
91 params2.nodes_to_replace->data[0] = 0;
92 params2.nodes_to_replace->data[1] = 3;
93 params2.nodes_to_replace->data[2] = 7;
94 params2.nodes_to_replace->data[3] = 8;
98 params3.nodes_to_replace = TfLiteIntArrayCreate(3);
99 params3.nodes_to_replace->data[0] = 2;
100 params3.nodes_to_replace->data[1] = 4;
[all …]
Dutils.cc82 return left->nodes_to_replace->size > in GetFirstNLargestPartitions()
83 right->nodes_to_replace->size; in GetFirstNLargestPartitions()
91 if (p->nodes_to_replace->size < min_nodes_per_partition) { in GetFirstNLargestPartitions()
105 auto nodes = p->nodes_to_replace; in GetNodesOfFirstNLargestPartitionsImpl()
169 for (int i = 0; i < first_partition->nodes_to_replace->size; ++i) { in GetNodesOfFirstNLargestPartitionsImpl()
170 const int node_id = first_partition->nodes_to_replace->data[i]; in GetNodesOfFirstNLargestPartitionsImpl()
207 auto nodes = first_n_partitions[i]->nodes_to_replace; in GetNodesOfFirstNLargestPartitionsImpl()
Ddelegate_test.cc63 ASSERT_EQ(params->nodes_to_replace->size, 3); in TEST_F()
64 EXPECT_EQ(params->nodes_to_replace->data[0], 0); in TEST_F()
65 EXPECT_EQ(params->nodes_to_replace->data[1], 1); in TEST_F()
66 EXPECT_EQ(params->nodes_to_replace->data[2], 2); in TEST_F()
Ddelegate_test_util.cc192 const auto* nodes_in_subset = params_array[idx].nodes_to_replace; in SimpleDelegate()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc137 TfLiteIntArrayFree(params.nodes_to_replace); in ~DelegatedInterpreter()
315 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
318 EXPECT_EQ(nodes_to_replace->size, 1); in TEST()
321 params->nodes_to_replace = TfLiteIntArrayCreate(1); in TEST()
322 params->nodes_to_replace->data[0] = 2; in TEST()
374 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
377 EXPECT_EQ(nodes_to_replace->size, 1); in TEST()
380 params->nodes_to_replace = TfLiteIntArrayCreate(1); in TEST()
381 params->nodes_to_replace->data[0] = 2; in TEST()
437 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
[all …]
Dmodel_builder.cc2601 for (int i = 0; i < delegate_params->nodes_to_replace->size; ++i) { in BuildModel()
2605 context, delegate_params->nodes_to_replace->data[i], &tflite_node, in BuildModel()
2636 context, delegate_params->nodes_to_replace->data[tflite_nodes[i]], in BuildModel()
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc77 ASSERT_EQ(params->nodes_to_replace->size, 3); in TEST_F()
78 EXPECT_EQ(params->nodes_to_replace->data[0], 0); in TEST_F()
79 EXPECT_EQ(params->nodes_to_replace->data[1], 1); in TEST_F()
80 EXPECT_EQ(params->nodes_to_replace->data[2], 2); in TEST_F()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc273 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate) { in ReplaceNodeSubsetsWithDelegateKernels() argument
275 ->ReplaceNodeSubsetsWithDelegateKernels(registration, nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels()
335 params->nodes_to_replace = reinterpret_cast<TfLiteIntArray*>(allocation); in CreateDelegateParams()
336 CopyVectorToTfLiteIntArray(node_subset.nodes, params->nodes_to_replace); in CreateDelegateParams()
358 params->nodes_to_replace = TfLiteIntArrayCreate(node_subset.nodes.size()); in PopulatePreviewDelegateParams()
359 CopyVectorToTfLiteIntArray(node_subset.nodes, params->nodes_to_replace); in PopulatePreviewDelegateParams()
374 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels() argument
377 if (!nodes_to_replace->size) { in ReplaceNodeSubsetsWithDelegateKernels()
388 PartitionGraphIntoIndependentNodeSubsets(&info, nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels()
394 nodes_to_replace->size, in ReplaceNodeSubsetsWithDelegateKernels()
[all …]
Dsubgraph.h493 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate);
501 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace,
535 const TfLiteIntArray* nodes_to_replace,
541 struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace,
/external/tensorflow/tensorflow/lite/c/
Dcommon.h616 TfLiteIntArray* nodes_to_replace; member
672 const TfLiteIntArray* nodes_to_replace, struct TfLiteDelegate* delegate);
754 struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace,
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc397 op_data_->nodes.reserve(params->nodes_to_replace->size); in Init()
399 CHECK(params->nodes_to_replace); in Init()
401 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
/external/tensorflow/tensorflow/lite/g3doc/performance/
Dimplementing_delegate.md144 inputs_.resize(params->nodes_to_replace->size);
145 outputs_.resize(params->nodes_to_replace->size);
146 builtin_code_.resize(params->nodes_to_replace->size);
147 for (int i = 0; i < params->nodes_to_replace->size; ++i) {
148 const int node_index = params->nodes_to_replace->data[i];
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc80 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc3619 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
3670 token_parts[1] = GetHash(params->nodes_to_replace); in Init()
4795 const int cache_key = delegate_params->nodes_to_replace->data[0]; in CacheDelegateKernel()
4801 const int cache_key = delegate_params->nodes_to_replace->data[0]; in MaybeGetCachedDelegateKernel()
4982 partition_params.nodes_to_replace->size); in GetNodesSupportedByAccelerator()
5016 partition_params.nodes_to_replace->data[0]) != in LimitDelegatedPartitions()
5025 return left.nodes_to_replace->size > in LimitDelegatedPartitions()
5026 right.nodes_to_replace->size; in LimitDelegatedPartitions()
5035 partition_params.nodes_to_replace->data, in LimitDelegatedPartitions()
5036 partition_params.nodes_to_replace->data + in LimitDelegatedPartitions()
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_delegate_kernel.mm124 for (int node_index : TfLiteIntArrayView(delegate_params->nodes_to_replace)) {
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc989 TfLiteIntArray nodes_to_replace; in TEST() local
990 nodes_to_replace.size = 0; in TEST()
992 context, delegate_registration, &nodes_to_replace, nullptr), in TEST()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc144 for (int i = 0; i < params->nodes_to_replace->size; i++) { in Create()
145 const int node_index = params->nodes_to_replace->data[i]; in Create()
255 for (int i = 0; i < params->nodes_to_replace->size; i++) { in Create()
256 const int node_index = params->nodes_to_replace->data[i]; in Create()