diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-11 15:36:21 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-11 15:45:27 -0700 |
| commit | 6305a6d83552ba6a472cd72398b60d9241467f1f (patch) | |
| tree | d02fb52415cefa562db46816c230e7522a702c90 /tensorflow/contrib/lite/nnapi_delegate.cc | |
| parent | b1f29d4c618d6bfa85130917848cd7eb89bf0f3b (diff) | |
Add an experimental API to allow half precision for FP32 calculation.
PiperOrigin-RevId: 212535448
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
| -rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index 817486e898..698de3dd39 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -757,6 +757,11 @@ TfLiteStatus NNAPIDelegate::BuildGraph(Interpreter* interpreter) { reinterpret_cast<const uint32_t*>(augmented_inputs.data()), static_cast<uint32_t>(augmented_outputs.size()), reinterpret_cast<const uint32_t*>(augmented_outputs.data()))); + + if (GetAndroidSdkVersionCached() >= 28) { + CHECK_NN(ANeuralNetworksModel_relaxComputationFloat32toFloat16( + nn_model_, interpreter->GetAllowFp16PrecisionForFp32())); + } CHECK_NN(ANeuralNetworksModel_finish(nn_model_)); } if (!nn_compiled_model_) { |