Home
last modified time | relevance | path

Searched refs:partition_params_array (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder_test.cc331 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
344 *partition_params_array = interpreter_fp16_add_op->delegate_params(); in TEST()
388 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
401 *partition_params_array = in TEST()
451 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
454 *partition_params_array = nullptr; in TEST()
565 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
575 *partition_params_array = interpreter_fp32->delegate_params(); in TEST()
732 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
751 *partition_params_array = interpreter2_fp32->delegate_params(); in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc554 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in PreviewDelegatePartitioning() argument
558 if (!partition_params_array || !num_partitions) return kTfLiteError; in PreviewDelegatePartitioning()
559 *partition_params_array = nullptr; in PreviewDelegatePartitioning()
582 *partition_params_array = partitioning_preview_cache_.data(); in PreviewDelegatePartitioning()
588 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in PreviewDelegatePartitioning() argument
590 ->PreviewDelegatePartitioning(nodes_to_replace, partition_params_array, in PreviewDelegatePartitioning()
1650 TfLiteDelegateParams** partition_params_array, in SwitchToKernelContext()
H A Dsubgraph.h621 TfLiteDelegateParams** partition_params_array, int* num_partitions);
627 TfLiteDelegateParams** partition_params_array, int* num_partitions);
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dutils_test.cc151 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in MockPreviewDelegatePartitioning() argument
153 *partition_params_array = mock->delegate_params(); in MockPreviewDelegatePartitioning()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.h427 std::vector<TfLiteDelegateParams> partition_params_array,
H A Dnnapi_delegate.cc6612 std::vector<TfLiteDelegateParams> partition_params_array, in LimitDelegatedPartitions() argument
6614 int num_partitions = partition_params_array.size(); in LimitDelegatedPartitions()
6620 partition_params_array.begin(), partition_params_array.end(), in LimitDelegatedPartitions()
6628 std::sort(partition_params_array.begin(), partition_params_array.end(), in LimitDelegatedPartitions()
6639 const TfLiteDelegateParams& partition_params = partition_params_array[i]; in LimitDelegatedPartitions()
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon.h818 TfLiteDelegateParams** partition_params_array, int* num_partitions);
/aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/
H A Dcommon.h866 TfLiteDelegateParams** partition_params_array, int* num_partitions);