| 142 | } |
| 143 | |
| 144 | void* TensorImplDft::get_memory_ptr() const { |
| 145 | if (m_get_memory_callback) { |
| 146 | m_get_memory_callback(const_cast<TensorImplDft*>(this)); |
| 147 | } |
| 148 | if (is_host()) { |
| 149 | return static_cast<void*>(m_host_tensor->raw_ptr()); |
| 150 | } else { |
| 151 | return static_cast<void*>(m_dev_tensor->raw_ptr()); |
| 152 | } |
| 153 | } |
| 154 | |
| 155 | void* TensorImplDft::get_memory_ptr(const std::vector<size_t>& idx) const { |
| 156 | if (m_get_memory_callback) { |