| /external/tensorflow/tensorflow/lite/delegates/ |
| D | utils_test.cc | 86 params1.nodes_to_replace = TfLiteIntArrayCreate(1);
87 params1.nodes_to_replace->data[0] = 1;
91 params2.nodes_to_replace = TfLiteIntArrayCreate(4);
92 params2.nodes_to_replace->data[0] = 0;
93 params2.nodes_to_replace->data[1] = 3;
94 params2.nodes_to_replace->data[2] = 7;
95 params2.nodes_to_replace->data[3] = 8;
99 params3.nodes_to_replace = TfLiteIntArrayCreate(3);
100 params3.nodes_to_replace->data[0] = 2;
101 params3.nodes_to_replace->data[1] = 4;
[all …]
|
| D | utils.cc | 85 return left->nodes_to_replace->size > in GetFirstNLargestPartitions()
86 right->nodes_to_replace->size; in GetFirstNLargestPartitions()
94 if (p->nodes_to_replace->size < min_nodes_per_partition) { in GetFirstNLargestPartitions()
108 auto nodes = p->nodes_to_replace; in GetNodesOfFirstNLargestPartitionsImpl()
189 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()
1023 TfLiteIntArray* nodes_to_replace = TfLiteIntArrayCreate(1); in SetUp() local
1024 nodes_to_replace->data[0] = execution_plan->data[1]; in SetUp()
1026 context, DelegateRegistration(), nodes_to_replace, delegate); in SetUp()
1027 TfLiteIntArrayFree(nodes_to_replace); in SetUp()
|
| D | serialization.cc | 268 auto* nodes = delegate_params->nodes_to_replace; in GetEntryImpl()
|
| D | delegate_test_util.cc | 213 const auto* nodes_in_subset = params_array[idx].nodes_to_replace; in SimpleDelegate()
|
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate_test.cc | 78 ASSERT_EQ(params->nodes_to_replace->size, 3); in TEST_F()
79 EXPECT_EQ(params->nodes_to_replace->data[0], 0); in TEST_F()
80 EXPECT_EQ(params->nodes_to_replace->data[1], 1); in TEST_F()
81 EXPECT_EQ(params->nodes_to_replace->data[2], 2); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 143 TfLiteIntArrayFree(params.nodes_to_replace); in ~DelegatedInterpreter()
330 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
333 EXPECT_EQ(nodes_to_replace->size, 1); in TEST()
336 params->nodes_to_replace = TfLiteIntArrayCreate(1); in TEST()
337 params->nodes_to_replace->data[0] = 2; in TEST()
387 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
390 EXPECT_EQ(nodes_to_replace->size, 1); in TEST()
393 params->nodes_to_replace = TfLiteIntArrayCreate(1); in TEST()
394 params->nodes_to_replace->data[0] = 2; in TEST()
450 [](struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace, in TEST()
[all …]
|
| D | model_builder.cc | 3300 for (int i = 0; i < delegate_params->nodes_to_replace->size; ++i) { in BuildModelEnforceIO()
3304 context, delegate_params->nodes_to_replace->data[i], &tflite_node, in BuildModelEnforceIO()
3337 context, delegate_params->nodes_to_replace->data[tflite_nodes[i]], in BuildModelEnforceIO()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 302 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate) { in ReplaceNodeSubsetsWithDelegateKernels() argument
304 ->ReplaceNodeSubsetsWithDelegateKernels(registration, nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels()
364 params->nodes_to_replace = reinterpret_cast<TfLiteIntArray*>(allocation); in CreateDelegateParams()
365 CopyVectorToTfLiteIntArray(node_subset.nodes, params->nodes_to_replace); in CreateDelegateParams()
387 params->nodes_to_replace = TfLiteIntArrayCreate(node_subset.nodes.size()); in PopulatePreviewDelegateParams()
388 CopyVectorToTfLiteIntArray(node_subset.nodes, params->nodes_to_replace); in PopulatePreviewDelegateParams()
403 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels() argument
406 if (!nodes_to_replace->size) { in ReplaceNodeSubsetsWithDelegateKernels()
417 PartitionGraphIntoIndependentNodeSubsets(&info, nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels()
426 nodes_to_replace->size, in ReplaceNodeSubsetsWithDelegateKernels()
[all …]
|
| D | subgraph.h | 578 const TfLiteIntArray* nodes_to_replace, TfLiteDelegate* delegate);
586 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace,
620 const TfLiteIntArray* nodes_to_replace,
626 struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace,
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 656 TfLiteIntArray* nodes_to_replace; member
735 const TfLiteIntArray* nodes_to_replace, struct TfLiteDelegate* delegate);
817 struct TfLiteContext* context, const TfLiteIntArray* nodes_to_replace,
|
| /external/tensorflow/tensorflow/lite/ |
| D | optional_debug_tools.cc | 462 for (int nid : TfLiteIntArrayView(params->nodes_to_replace)) { in PrintInterpreterState()
549 delegate, delegate_params->nodes_to_replace->size); in PrintInterpreterState()
550 PrintTfLiteIntVector(delegate_params->nodes_to_replace, in PrintInterpreterState()
|
| D | interpreter_test.cc | 995 TfLiteIntArray nodes_to_replace; in TEST() local
996 nodes_to_replace.size = 0; in TEST()
998 context, delegate_registration, &nodes_to_replace, nullptr), in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 467 op_data_->nodes.reserve(params->nodes_to_replace->size); in Init()
469 CHECK(params->nodes_to_replace); in Init()
480 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/opaque/src/ |
| D | armnn_delegate.cpp | 424 for (int i = 0; i < parameters->nodes_to_replace->size; ++i) in Create()
426 const int nodeIndex = parameters->nodes_to_replace->data[i]; in Create()
|
| /external/armnn/delegate/classic/src/ |
| D | armnn_delegate.cpp | 356 for (int i = 0; i < parameters->nodes_to_replace->size; ++i) in Create()
358 const int nodeIndex = parameters->nodes_to_replace->data[i]; in Create()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 82 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
|
| /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/nnapi/ |
| D | nnapi_delegate.cc | 4467 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
6366 const int cache_key = delegate_params->nodes_to_replace->data[0]; in CacheDelegateKernel()
6372 const int cache_key = delegate_params->nodes_to_replace->data[0]; in MaybeGetCachedDelegateKernel()
6589 partition_params.nodes_to_replace->size); in GetNodesSupportedByAccelerator()
6623 partition_params.nodes_to_replace->data[0]) != in LimitDelegatedPartitions()
6632 return left.nodes_to_replace->size > in LimitDelegatedPartitions()
6633 right.nodes_to_replace->size; in LimitDelegatedPartitions()
6642 partition_params.nodes_to_replace->data, in LimitDelegatedPartitions()
6643 partition_params.nodes_to_replace->data + in LimitDelegatedPartitions()
6644 partition_params.nodes_to_replace->size); in LimitDelegatedPartitions()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 295 for (int i = 0; i < params->nodes_to_replace->size; i++) { in Create()
296 const int node_index = params->nodes_to_replace->data[i]; in Create()
663 for (int i = 0; i < params->nodes_to_replace->size; i++) { in Create()
664 const int node_index = params->nodes_to_replace->data[i]; in Create()
|