Home
last modified time | relevance | path

Searched refs:delegateOptions (Results 1 – 7 of 7) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_delegate_kernel.cpp79 NnrtDelegate::Options delegateOptions; in Init() local
117 NnrtDelegate::Options delegateOptions; in Prepare() local
120 TF_LITE_ENSURE_STATUS(SetNnOptions(context, delegateOptions)); in Prepare()
229 const NnrtDelegate::Options& delegateOptions) in AddOpsAndTensors() argument
238 .delegateOptions = delegateOptions in AddOpsAndTensors()
397 auto performance = delegateOptions.executionPerformance; in SetNnOptions()
405 std::string cacheDir = delegateOptions.cacheDir; in SetNnOptions()
406 std::string modelToken = delegateOptions.modelToken; in SetNnOptions()
407 uint32_t version = delegateOptions.version; in SetNnOptions()
408 if (!cacheDir.empty() && (!IsUseTargetDevice(delegateOptions) || in SetNnOptions()
[all …]
H A Dnnrt_delegate.cpp242 NnrtDelegate::Options delegateOptions; in CheckDeviceValid() local
243 TF_LITE_ENSURE_STATUS(NnrtDelegate::GetOptions(delegate, delegateOptions)); in CheckDeviceValid()
245 if (tflite::IsUseTargetDevice(delegateOptions)) { in CheckDeviceValid()
292 NnrtDelegate::Options delegateOptions; in DoPrepare() local
293 TF_LITE_ENSURE_STATUS(NnrtDelegate::GetOptions(delegate, delegateOptions)); in DoPrepare()
296 delegateOptions.maxNumberDelegatedPartitions, partitionParamsArray, nodesToDelegate)); in DoPrepare()
H A Dnnrt_utils.cpp88 bool IsUseTargetDevice(NnrtDelegate::Options delegateOptions, bool excludeNnrtReference) in IsUseTargetDevice() argument
90 const std::string& deviceName = delegateOptions.acceleratorName; in IsUseTargetDevice()
106 NnrtDelegate::Options delegateOptions; in GetTargetDevice() local
107 TF_LITE_ENSURE_STATUS(NnrtDelegate::GetOptions(delegate, delegateOptions)); in GetTargetDevice()
108 const std::string& deviceName = delegateOptions.acceleratorName; in GetTargetDevice()
H A Dnnrt_delegate_kernel.h67 const NnrtDelegate::Options& delegateOptions);
75 TfLiteStatus SetNnOptions(TfLiteContext* context, const NnrtDelegate::Options& delegateOptions);
H A Dnnrt_utils.h82 NnrtDelegate::Options delegateOptions, bool excludeNnrtReference = false);
H A Dnnrt_op_builder.h51 NnrtDelegate::Options delegateOptions; member
H A Dnnrt_op_builder.cpp40 m_allowDynamicDimensions(opBuilderArgs.delegateOptions.allowDynamicDimensions) in NnrtOpBuilder()