Searched refs:partition_params_array (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 327 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
340 *partition_params_array = interpreter_fp16_add_op->delegate_params(); in TEST()
384 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
397 *partition_params_array = in TEST()
447 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
450 *partition_params_array = nullptr; in TEST()
561 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
571 *partition_params_array = interpreter_fp32->delegate_params(); in TEST()
728 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST() argument
747 *partition_params_array = interpreter2_fp32->delegate_params(); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | utils_test.cc | 150 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in MockPreviewDelegatePartitioning() argument
152 *partition_params_array = mock->delegate_params(); in MockPreviewDelegatePartitioning()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 536 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in PreviewDelegatePartitioning() argument
540 if (!partition_params_array || !num_partitions) return kTfLiteError; in PreviewDelegatePartitioning()
541 *partition_params_array = nullptr; in PreviewDelegatePartitioning()
564 *partition_params_array = partitioning_preview_cache_.data(); in PreviewDelegatePartitioning()
570 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in PreviewDelegatePartitioning() argument
572 ->PreviewDelegatePartitioning(nodes_to_replace, partition_params_array, in PreviewDelegatePartitioning()
1464 TfLiteDelegateParams** partition_params_array, in SwitchToKernelContext()
|
| D | subgraph.h | 567 TfLiteDelegateParams** partition_params_array, int* num_partitions);
573 TfLiteDelegateParams** partition_params_array, int* num_partitions);
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.h | 374 std::vector<TfLiteDelegateParams> partition_params_array,
|
| D | nnapi_delegate.cc | 5594 std::vector<TfLiteDelegateParams> partition_params_array, in LimitDelegatedPartitions() argument
5596 int num_partitions = partition_params_array.size(); in LimitDelegatedPartitions()
5602 partition_params_array.begin(), partition_params_array.end(), in LimitDelegatedPartitions()
5610 std::sort(partition_params_array.begin(), partition_params_array.end(), in LimitDelegatedPartitions()
5621 const TfLiteDelegateParams& partition_params = partition_params_array[i]; in LimitDelegatedPartitions()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 797 TfLiteDelegateParams** partition_params_array, int* num_partitions);
|