Lines Matching refs:nn
52 using namespace ::android::nn::sl_wrapper;
58 constexpr std::optional<::android::nn::wrapper::ExecutePriority> convertToNDKPriority( in convertToNDKPriority()
62 return ::android::nn::wrapper::ExecutePriority::LOW; in convertToNDKPriority()
64 return ::android::nn::wrapper::ExecutePriority::MEDIUM; in convertToNDKPriority()
66 return ::android::nn::wrapper::ExecutePriority::HIGH; in convertToNDKPriority()
72 constexpr std::optional<::android::nn::wrapper::ExecutePreference> convertToNDKPreference( in convertToNDKPreference()
76 return ::android::nn::wrapper::ExecutePreference::PREFER_LOW_POWER; in convertToNDKPreference()
78 return ::android::nn::wrapper::ExecutePreference::PREFER_FAST_SINGLE_ANSWER; in convertToNDKPreference()
80 return ::android::nn::wrapper::ExecutePreference::PREFER_SUSTAINED_SPEED; in convertToNDKPreference()
196 ShimBuffer(const NnApiSupportLibrary* nnApi, const ::android::nn::Dimensions initialDimensions, in ShimBuffer()
197 const ::android::nn::OperandType type, in ShimBuffer()
198 std::shared_ptr<::android::nn::sl_wrapper::Memory> memory, in ShimBuffer()
206 bool tensorHasUnspecifiedDimensions(::android::nn::OperandType type, in tensorHasUnspecifiedDimensions()
207 const ::android::nn::Dimensions& dimensions) { in tensorHasUnspecifiedDimensions()
208 if (!::android::nn::isExtension(type)) { in tensorHasUnspecifiedDimensions()
217 bool validateDimensions(const ::android::nn::Dimensions& dimensions) { in validateDimensions()
243 const auto combined = ::android::nn::combineDimensions(kInitialDimensions, dimensions); in validateDimensions()
259 const auto unsignedDimensions = ::android::nn::toUnsigned(dimensions); in copyFrom()
311 const ::android::nn::Dimensions kInitialDimensions;
312 const ::android::nn::OperandType kType;
315 std::shared_ptr<::android::nn::sl_wrapper::Memory> mMemory;
334 auto unsignedDimensions = ::android::nn::toUnsigned(desc.dimensions).value(); in allocate()
352 std::optional<::android::nn::OperandType> type; in allocate()
353 std::vector<uint32_t> dimensions = ::android::nn::toUnsigned(desc.dimensions).value(); in allocate()
384 auto operandType = static_cast<::android::nn::OperandType>(op.operandType.type); in allocate()
421 auto operandType = static_cast<::android::nn::OperandType>(op.operandType.type); in allocate()
430 auto typeSize = ::android::nn::getNonExtensionSize(*type, dimensions); in allocate()
439 std::make_shared<::android::nn::sl_wrapper::Memory>(mNnapi.get(), slDesc, *typeSize); in allocate()
563 auto compilation = ::android::nn::sl_wrapper::Compilation::createForDevice( in prepareModelCommon()
598 const auto prefix = ::android::nn::getExtensionPrefix(uToken); in prepareModelCommon()
599 const auto attributeCodeWithinExtension = ::android::nn::getTypeWithinExtension(uToken); in prepareModelCommon()