| 134 | } |
| 135 | |
| 136 | Layout TensorImplDft::get_layout() const { |
| 137 | if (is_host()) { |
| 138 | return to_lite_layout(m_host_tensor->layout()); |
| 139 | } else { |
| 140 | return to_lite_layout(m_dev_tensor->layout()); |
| 141 | } |
| 142 | } |
| 143 | |
| 144 | void* TensorImplDft::get_memory_ptr() const { |
| 145 | if (m_get_memory_callback) { |