| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| H A D | logging_op_resolver_test.cc | 69 BuiltinOpsSet ops_to_replace = { in TEST() local
74 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
103 BuiltinOpsSet ops_to_replace = { in TEST() local
108 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
131 BuiltinOpsSet ops_to_replace = { in TEST() local
135 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
155 CustomOpsSet ops_to_replace = { in TEST() local
159 LoggingOpResolver resolver(BuiltinOpsSet(), ops_to_replace, base_resolver, in TEST()
176 CustomOpsSet ops_to_replace = { in TEST() local
181 LoggingOpResolver(BuiltinOpsSet(), ops_to_replace, base_resolver, in TEST()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/ |
| H A D | utils.cc | 106 std::vector<int> ops_to_replace; in GetNodesOfFirstNLargestPartitionsImpl() local
109 ops_to_replace.insert(ops_to_replace.end(), nodes->data, in GetNodesOfFirstNLargestPartitionsImpl()
112 return ops_to_replace; in GetNodesOfFirstNLargestPartitionsImpl()
168 std::vector<int> ops_to_replace; in GetNodesOfFirstNLargestPartitionsImpl() local
177 ops_to_replace.push_back(node_id); in GetNodesOfFirstNLargestPartitionsImpl()
187 if (first_n_partitions.empty()) return ops_to_replace; in GetNodesOfFirstNLargestPartitionsImpl()
190 ops_to_replace.insert(ops_to_replace.end(), nodes->data, in GetNodesOfFirstNLargestPartitionsImpl()
196 RemapFp16InputTensors(ops_to_replace); in GetNodesOfFirstNLargestPartitionsImpl()
197 return ops_to_replace; in GetNodesOfFirstNLargestPartitionsImpl()
|
| H A D | delegate_test_util.cc | 492 std::vector<int> ops_to_replace = in FP16Delegate() local
495 nodes_to_separate = ConvertVectorToTfLiteIntArray(ops_to_replace); in FP16Delegate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | model_builder_test.cc | 349 TfLiteIntArray* ops_to_replace = GetOpsToReplace(context); in TEST() local
352 EXPECT_EQ(ops_to_replace->size, 3); in TEST()
361 TfLiteIntArrayFree(ops_to_replace); in TEST()
407 TfLiteIntArray* ops_to_replace = GetOpsToReplace(context); in TEST() local
410 EXPECT_EQ(ops_to_replace->size, 1); in TEST()
413 context->GetNodeAndRegistration(context, ops_to_replace->data[0], &node, in TEST()
419 TfLiteIntArrayFree(ops_to_replace); in TEST()
459 TfLiteIntArray* ops_to_replace = GetOpsToReplace(context); in TEST() local
462 EXPECT_EQ(ops_to_replace->size, 0); in TEST()
473 TfLiteIntArrayFree(ops_to_replace); in TEST()
[all …]
|
| H A D | model_builder.cc | 3196 std::vector<int> ops_to_replace = in GetOpsToReplace() local
3201 partition_helper.num_total_nodes() > ops_to_replace.size()) { in GetOpsToReplace()
3206 if (!ops_to_replace.empty()) { in GetOpsToReplace()
3208 &error_message, ops_to_replace.size(), in GetOpsToReplace()
3210 partition_helper.num_total_nodes() - ops_to_replace.size()); in GetOpsToReplace()
3219 return ConvertVectorToTfLiteIntArray(ops_to_replace); in GetOpsToReplace()
3427 TfLiteIntArray* ops_to_replace = GetOpsToReplace( in DelegatePrepare() local
3430 context, registration, ops_to_replace, delegate); in DelegatePrepare()
3431 TfLiteIntArrayFree(ops_to_replace); in DelegatePrepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| H A D | gpu_api_delegate.cc | 358 TfLiteIntArray* ops_to_replace = GetOpsToReplace(context); in DelegatePrepare() local
360 context, kRegistration, ops_to_replace, delegate); in DelegatePrepare()
361 TfLiteIntArrayFree(ops_to_replace); in DelegatePrepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/ |
| H A D | gl_delegate.cc | 414 TfLiteIntArray* ops_to_replace = GetOpsToReplace(context); in DelegatePrepare() local
416 context, kRegistration, ops_to_replace, delegate); in DelegatePrepare()
417 TfLiteIntArrayFree(ops_to_replace); in DelegatePrepare()
|
| H A D | delegate.cc | 573 TfLiteIntArray* ops_to_replace = in DelegatePrepare() local
577 context, kRegistration, ops_to_replace, delegate); in DelegatePrepare()
580 TfLiteIntArrayFree(ops_to_replace); in DelegatePrepare()
|
| H A D | metal_delegate.mm | 701 TfLiteIntArray* ops_to_replace =
704 ops_to_replace, delegate);
705 TfLiteIntArrayFree(ops_to_replace);
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/benchmarking/ |
| H A D | main.mm | 399 TfLiteIntArray* ops_to_replace = GetOpsToReplace(context);
401 context, kRegistration, ops_to_replace, delegate);
402 TfLiteIntArrayFree(ops_to_replace);
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | xnnpack_delegate.cc | 5164 TfLiteIntArray* ops_to_replace = in DelegatePrepare() local
5167 if (ops_to_replace == nullptr) { in DelegatePrepare()
5172 context, kSubgraphRegistration, ops_to_replace, delegate); in DelegatePrepare()
5173 TfLiteIntArrayFree(ops_to_replace); in DelegatePrepare()
|