pytorch/torch/backends/_nnapi
Akshit Khurana bb3e1f30a8 [Pytorch NNAPI] Add compilation_preference & relax_f32_to_f16 APIs (#78758)
Summary:
compilation_preference is one of:

ANEURALNETWORKS_PREFER_LOW_POWER = 0
ANEURALNETWORKS_PREFER_FAST_SINGLE_ANSWER = 1
ANEURALNETWORKS_PREFER_SUSTAINED_SPEED = 2

relax_f32_to_f16 calls Model_relaxComputationFloat32toFloat16

Test Plan:
Tested on device with nnapi models

* Works with existing exported models
* Works with new exported models with options

Differential Revision: D36433236

Pull Request resolved: https://github.com/pytorch/pytorch/pull/78758
Approved by: https://github.com/kimishpatel
2022-06-06 20:57:34 +00:00
..
__init__.py PyTorch NNAPI integration prototype (#46780) 2020-11-05 21:31:01 -08:00
prepare.py [Pytorch NNAPI] Add compilation_preference & relax_f32_to_f16 APIs (#78758) 2022-06-06 20:57:34 +00:00
serializer.py Resolve int[]? arguments to new OptionalIntArrayRef class 2022-03-26 01:45:50 +00:00