Only used for testing
| 248 | } |
| 249 | // Only used for testing |
| 250 | void CopyOutTo(void* memory) const override |
| 251 | { |
| 252 | const_cast<armnn::ClImportTensorHandle*>(this)->Map(true); |
| 253 | switch(this->GetDataType()) |
| 254 | { |
| 255 | case arm_compute::DataType::F32: |
| 256 | armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), |
| 257 | static_cast<float*>(memory)); |
| 258 | break; |
| 259 | case arm_compute::DataType::U8: |
| 260 | case arm_compute::DataType::QASYMM8: |
| 261 | armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), |
| 262 | static_cast<uint8_t*>(memory)); |
| 263 | break; |
| 264 | case arm_compute::DataType::QSYMM8_PER_CHANNEL: |
| 265 | case arm_compute::DataType::QASYMM8_SIGNED: |
| 266 | armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), |
| 267 | static_cast<int8_t*>(memory)); |
| 268 | break; |
| 269 | case arm_compute::DataType::F16: |
| 270 | armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), |
| 271 | static_cast<armnn::Half*>(memory)); |
| 272 | break; |
| 273 | case arm_compute::DataType::S16: |
| 274 | case arm_compute::DataType::QSYMM16: |
| 275 | armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), |
| 276 | static_cast<int16_t*>(memory)); |
| 277 | break; |
| 278 | case arm_compute::DataType::S32: |
| 279 | armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), |
| 280 | static_cast<int32_t*>(memory)); |
| 281 | break; |
| 282 | default: |
| 283 | { |
| 284 | throw armnn::UnimplementedException(); |
| 285 | } |
| 286 | } |
| 287 | const_cast<armnn::ClImportTensorHandle*>(this)->Unmap(); |
| 288 | } |
| 289 | |
| 290 | // Only used for testing |
| 291 | void CopyInFrom(const void* memory) override |
nothing calls this directly
no test coverage detected