| /external/tensorflow/tensorflow/lite/delegates/ |
| D | utils_test.cc | 85 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 …]
|
| D | utils.cc | 84 return left->nodes_to_replace->size > in GetFirstNLargestPartitions()
85 right->nodes_to_replace->size; in GetFirstNLargestPartitions()
93 if (p->nodes_to_replace->size < min_nodes_per_partition) { in GetFirstNLargestPartitions()
107 auto nodes = p->nodes_to_replace; in GetNodesOfFirstNLargestPartitionsImpl()
188 auto nodes = first_n_partitions[i]->nodes_to_replace; in GetNodesOfFirstNLargestPartitionsImpl()
|
| D | serialization_test.cc | 68 auto* nodes_to_replace = TfLiteIntArrayCreate(num_nodes); in GenerateTfLiteDelegateParams() local
71 owned_arrays_.push_back(nodes_to_replace); in GenerateTfLiteDelegateParams()
75 nodes_to_replace->data[i] = i; in GenerateTfLiteDelegateParams()
87 params.nodes_to_replace = nodes_to_replace; in GenerateTfLiteDelegateParams()
|
| D | delegate_test.cc | 65 ASSERT_EQ(params->nodes_to_replace->size, 3); in TEST_F()
66 EXPECT_EQ(params->nodes_to_replace->data[0], 0); in TEST_F()
67 EXPECT_EQ(params->nodes_to_replace->data[1], 1); in TEST_F()
68 EXPECT_EQ(params->nodes_to_replace->data[2], 2); in TEST_F()
|
| D | serialization.cc | 272 auto* nodes = delegate_params->nodes_to_replace; in GetEntryImpl()
|
| D | delegate_test_util.cc | 208 const auto* nodes_in_subset = params_array[idx].nodes_to_replace; in SimpleDelegate()
|
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate_test.cc | 77 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/delegates/gpu/common/ |
| D | model_builder_test.cc | 139 TfLiteIntArrayFree(params.nodes_to_replace); in ~DelegatedInterpreter()
326 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
329 EXPECT_EQ(nodes_to_replace->size, 1); in TEST()
332 params->nodes_to_replace = TfLiteIntArrayCreate(1); in TEST()
333 params->nodes_to_replace->data[0] = 2; in TEST()
383 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
386 EXPECT_EQ(nodes_to_replace->size, 1); in TEST()
389 params->nodes_to_replace = TfLiteIntArrayCreate(1); in TEST()
390 params->nodes_to_replace->data[0] = 2; in TEST()
446 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
[all …]
|
| D | model_builder.cc | 2806 for (int i = 0; i < delegate_params->nodes_to_replace->size; ++i) { in BuildModelEnforceIO()
2810 context, delegate_params->nodes_to_replace->data[i], &tflite_node, in BuildModelEnforceIO()
2843 context, delegate_params->nodes_to_replace->data[tflite_nodes[i]], in BuildModelEnforceIO()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 273 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()
398 nodes_to_replace->size, in ReplaceNodeSubsetsWithDelegateKernels()
[all …]
|
| D | subgraph.h | 524 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate);
532 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace,
566 const TfLiteIntArray* nodes_to_replace,
572 struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace,
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 635 TfLiteIntArray* nodes_to_replace; member
714 const TfLiteIntArray* nodes_to_replace, struct TfLiteDelegate* delegate);
796 struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace,
|
| /external/tensorflow/tensorflow/lite/ |
| D | optional_debug_tools.cc | 414 for (int nid : TfLiteIntArrayView(params->nodes_to_replace)) { in PrintInterpreterState()
501 delegate, delegate_params->nodes_to_replace->size); in PrintInterpreterState()
502 PrintTfLiteIntVector(delegate_params->nodes_to_replace, in PrintInterpreterState()
|
| D | interpreter_test.cc | 988 TfLiteIntArray nodes_to_replace; in TEST() local
989 nodes_to_replace.size = 0; in TEST()
991 context, delegate_registration, &nodes_to_replace, nullptr), in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 403 op_data_->nodes.reserve(params->nodes_to_replace->size); in Init()
405 CHECK(params->nodes_to_replace); in Init()
407 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
|
| /external/tensorflow/tensorflow/lite/g3doc/performance/ |
| D | implementing_delegate.md | 144 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/armnn/delegate/src/ |
| D | armnn_delegate.cpp | 295 for (int i = 0; i < parameters->nodes_to_replace->size; ++i) in Create()
297 const int nodeIndex = parameters->nodes_to_replace->data[i]; in Create()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 80 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 3869 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
5363 const int cache_key = delegate_params->nodes_to_replace->data[0]; in CacheDelegateKernel()
5369 const int cache_key = delegate_params->nodes_to_replace->data[0]; in MaybeGetCachedDelegateKernel()
5571 partition_params.nodes_to_replace->size); in GetNodesSupportedByAccelerator()
5605 partition_params.nodes_to_replace->data[0]) != in LimitDelegatedPartitions()
5614 return left.nodes_to_replace->size > in LimitDelegatedPartitions()
5615 right.nodes_to_replace->size; in LimitDelegatedPartitions()
5624 partition_params.nodes_to_replace->data, in LimitDelegatedPartitions()
5625 partition_params.nodes_to_replace->data + in LimitDelegatedPartitions()
5626 partition_params.nodes_to_replace->size); in LimitDelegatedPartitions()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 124 for (int node_index : TfLiteIntArrayView(delegate_params->nodes_to_replace)) {
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 151 for (int i = 0; i < params->nodes_to_replace->size; i++) { in Create()
152 const int node_index = params->nodes_to_replace->data[i]; in Create()
498 for (int i = 0; i < params->nodes_to_replace->size; i++) { in Create()
499 const int node_index = params->nodes_to_replace->data[i]; in Create()
|