diff options
author | surmeh01 <surabhi.mehta@arm.com> | 2018-07-05 12:06:04 +0100 |
---|---|---|
committer | surmeh01 <surabhi.mehta@arm.com> | 2018-07-05 12:06:04 +0100 |
commit | deb3bdbe028a59da0759dd7a560387d03a11d322 (patch) | |
tree | 869b7ee10d8f1f19a0861e0b552bb453330adf0a /test/DriverTestHelpers.hpp | |
parent | 49b9e100bfbb3b8da01472a0ff48b2bd92944e01 (diff) | |
download | android-nn-driver-deb3bdbe028a59da0759dd7a560387d03a11d322.tar.gz |
Release 18.05.02
Diffstat (limited to 'test/DriverTestHelpers.hpp')
-rw-r--r-- | test/DriverTestHelpers.hpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/test/DriverTestHelpers.hpp b/test/DriverTestHelpers.hpp index e90f7ecf..57541a35 100644 --- a/test/DriverTestHelpers.hpp +++ b/test/DriverTestHelpers.hpp @@ -72,9 +72,9 @@ android::sp<IMemory> AddPoolAndGetData(uint32_t size, Request& request); void AddPoolAndSetData(uint32_t size, Request& request, const float* data); -void AddOperand(Model& model, const Operand& op); +void AddOperand(V1_0::Model& model, const Operand& op); -void AddIntOperand(Model& model, int32_t value); +void AddIntOperand(V1_0::Model& model, int32_t value); template<typename T> OperandType TypeToOperandType(); @@ -86,7 +86,7 @@ template<> OperandType TypeToOperandType<int32_t>(); template<typename T> -void AddTensorOperand(Model& model, hidl_vec<uint32_t> dimensions, T* values) +void AddTensorOperand(V1_0::Model& model, hidl_vec<uint32_t> dimensions, T* values) { uint32_t totalElements = 1; for (uint32_t dim : dimensions) @@ -113,14 +113,14 @@ void AddTensorOperand(Model& model, hidl_vec<uint32_t> dimensions, T* values) AddOperand(model, op); } -void AddInputOperand(Model& model, hidl_vec<uint32_t> dimensions); +void AddInputOperand(V1_0::Model& model, hidl_vec<uint32_t> dimensions); -void AddOutputOperand(Model& model, hidl_vec<uint32_t> dimensions); +void AddOutputOperand(V1_0::Model& model, hidl_vec<uint32_t> dimensions); -android::sp<IPreparedModel> PrepareModel(const Model& model, +android::sp<IPreparedModel> PrepareModel(const V1_0::Model& model, armnn_driver::ArmnnDriver& driver); -android::sp<IPreparedModel> PrepareModelWithStatus(const Model& model, +android::sp<IPreparedModel> PrepareModelWithStatus(const V1_0::Model& model, armnn_driver::ArmnnDriver& driver, ErrorStatus & prepareStatus, ErrorStatus expectedStatus=ErrorStatus::NONE); |