| 126 | } |
| 127 | |
| 128 | void TensorImplDft::set_mge_tensor_compnode(const mgb::CompNode& comp_node) { |
| 129 | if (is_host()) { |
| 130 | m_host_tensor->comp_node(comp_node, true); |
| 131 | } else { |
| 132 | m_dev_tensor->comp_node(comp_node, true); |
| 133 | } |
| 134 | } |
| 135 | |
| 136 | Layout TensorImplDft::get_layout() const { |
| 137 | if (is_host()) { |
no test coverage detected