| 47 | } |
| 48 | |
| 49 | uint8_t *OperatorTensor::buffer() const |
| 50 | { |
| 51 | switch (_mem_type) |
| 52 | { |
| 53 | case MemoryType::CPU: |
| 54 | return (uint8_t *)utils::cast::polymorphic_downcast<MemoryRegion *>(_memory->region())->buffer(); |
| 55 | default: |
| 56 | ARM_COMPUTE_ERROR("Memory type not supported."); |
| 57 | } |
| 58 | } |
| 59 | } // namespace experimental |
| 60 | } // namespace arm_compute |