Home
last modified time | relevance | path

Searched refs:TfLiteDelegateParams (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/
Dutils_test.cc84 TfLiteDelegateParams params1({nullptr});
89 TfLiteDelegateParams params2({nullptr});
97 TfLiteDelegateParams params3({nullptr});
104 TfLiteDelegateParams params4({nullptr});
128 TfLiteDelegateParams* delegate_params() { return &delegate_params_.front(); } in delegate_params()
150 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in MockPreviewDelegatePartitioning()
167 std::vector<TfLiteDelegateParams> delegate_params_;
177 const std::vector<TfLiteDelegateParams*>& partitions) { in GetNodesToReplaceFromPartitions()
Dutils.cc55 TfLiteDelegateParams* partition_params_array_ = nullptr; in Partition()
71 std::vector<TfLiteDelegateParams*>
78 std::vector<TfLiteDelegateParams*> sorted_partitions(partitions_); in GetFirstNLargestPartitions()
80 [](TfLiteDelegateParams* left, TfLiteDelegateParams* right) { in GetFirstNLargestPartitions()
86 std::vector<TfLiteDelegateParams*> results; in GetFirstNLargestPartitions()
Dutils.h77 std::vector<TfLiteDelegateParams*> GetFirstNLargestPartitions(
109 std::vector<TfLiteDelegateParams*> partitions_;
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.h259 void CacheDelegateKernel(const TfLiteDelegateParams* delegate_params,
264 const TfLiteDelegateParams* delegate_params);
307 TfLiteDelegateParams** params_array, int* nnapi_errno);
320 std::vector<TfLiteDelegateParams> partition_params_array,
Dnnapi_delegate_disabled.cc62 const TfLiteDelegateParams* delegate_params, in CacheDelegateKernel()
66 const TfLiteDelegateParams* delegate_params) { in MaybeGetCachedDelegateKernel()
Dnnapi_delegate_kernel.h284 TfLiteStatus Init(TfLiteContext* context, const TfLiteDelegateParams* params,
Dnnapi_delegate.cc3617 const TfLiteDelegateParams* params, in Init()
4793 const TfLiteDelegateParams* delegate_params, in CacheDelegateKernel()
4800 const TfLiteDelegateParams* delegate_params) { in MaybeGetCachedDelegateKernel()
4955 TfLiteDelegateParams** params_array, int* nnapi_errno) { in GetNodesSupportedByAccelerator()
4969 TfLiteDelegateParams params_with_delegate = partition_params; in GetNodesSupportedByAccelerator()
5005 std::vector<TfLiteDelegateParams> partition_params_array, in LimitDelegatedPartitions()
5014 [nodes_to_delegate](const TfLiteDelegateParams& partition_params) { in LimitDelegatedPartitions()
5022 [](const TfLiteDelegateParams& left, in LimitDelegatedPartitions()
5023 const TfLiteDelegateParams& right) -> bool { in LimitDelegatedPartitions()
5032 const TfLiteDelegateParams& partition_params = partition_params_array[i]; in LimitDelegatedPartitions()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.h48 TfLiteContext* context, const TfLiteDelegateParams* delegate_params,
62 TfLiteContext* context, const TfLiteDelegateParams* delegate_params,
Dmodel_builder_test.cc169 TfLiteDelegateParams* add_delegate_params() { in add_delegate_params()
170 delegate_params_.push_back(TfLiteDelegateParams()); in add_delegate_params()
173 TfLiteDelegateParams* delegate_params() { return &delegate_params_.front(); } in delegate_params()
185 std::vector<TfLiteDelegateParams> delegate_params_;
316 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
375 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
438 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
552 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
719 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
939 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_delegate_kernel.h34 TfLiteStatus Init(TfLiteContext* context, const TfLiteDelegateParams* params);
47 const TfLiteDelegateParams* params);
Dcoreml_delegate_kernel.mm62 const TfLiteDelegateParams* delegate_params) {
117 const TfLiteDelegateParams* delegate_params) {
/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/
Dtflite_model_reader.cc38 const TfLiteDelegateParams* delegate_params) { in Init()
53 reinterpret_cast<const TfLiteDelegateParams*>(buffer))) { in DelegatePrepare()
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate.cc42 const TfLiteDelegateParams* params = in GetDelegateKernelRegistration()
43 reinterpret_cast<const TfLiteDelegateParams*>(buffer); in GetDelegateKernelRegistration()
Dsimple_delegate_test.cc75 const TfLiteDelegateParams* params = static_cast<const TfLiteDelegateParams*>( in TEST_F()
Dsimple_delegate.h50 const TfLiteDelegateParams* params) = 0;
/external/tensorflow/tensorflow/lite/c/
Dcommon.h614 typedef struct TfLiteDelegateParams { struct
619 } TfLiteDelegateParams; argument
755 TfLiteDelegateParams** partition_params_array, int* num_partitions);
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc308 TfLiteDelegateParams* CreateDelegateParams(TfLiteDelegate* delegate, in CreateDelegateParams()
311 int allocation_size = sizeof(TfLiteDelegateParams); in CreateDelegateParams()
330 TfLiteDelegateParams* params = in CreateDelegateParams()
331 reinterpret_cast<TfLiteDelegateParams*>(allocation); in CreateDelegateParams()
333 allocation += sizeof(TfLiteDelegateParams); in CreateDelegateParams()
353 TfLiteDelegateParams* params) { in PopulatePreviewDelegateParams()
414 TfLiteDelegateParams* params = in ReplaceNodeSubsetsWithDelegateKernels()
499 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in PreviewDelegatePartitioning()
533 TfLiteDelegateParams** partition_params_array, int* num_partitions) { in PreviewDelegatePartitioning()
1378 TfLiteDelegateParams** partition_params_array, in SwitchToKernelContext()
Dsubgraph.h536 TfLiteDelegateParams** partition_params_array, int* num_partitions);
542 TfLiteDelegateParams** partition_params_array, int* num_partitions);
708 std::vector<TfLiteDelegateParams> partitioning_preview_cache_;
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.h33 const TfLiteDelegateParams* params) override;
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.h49 const TfLiteDelegateParams* params) override;
Dhexagon_delegate_kernel.cc47 const TfLiteDelegateParams* params) { in Init()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Ddelegate.cc122 const TfLiteDelegateParams* delegate_params) { in Prepare()
258 const TfLiteDelegateParams* delegate_params, in InitializeGraph()
371 reinterpret_cast<const TfLiteDelegateParams*>(buffer); in DelegatePrepare()
Dgl_delegate.cc132 const TfLiteDelegateParams* delegate_params) { in Prepare()
386 reinterpret_cast<const TfLiteDelegateParams*>(buffer); in DelegatePrepare()
/external/tensorflow/tensorflow/lite/delegates/utils/dummy_delegate/
Ddummy_delegate.cc31 const TfLiteDelegateParams* params) override { in Init()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dgpu_api_delegate.cc91 const TfLiteDelegateParams* delegate_params) { in Prepare()
317 reinterpret_cast<const TfLiteDelegateParams*>(buffer); in DelegatePrepare()

12