| 152 | /* ========================= SetMeshIndexing =========================== */ |
| 153 | template <typename T> |
| 154 | void exec_set_mesh_indexing( |
| 155 | const TensorND& data, const TensorND& value, |
| 156 | const SetMeshIndexing::IndexDesc& desc) { |
| 157 | auto iter = tensor_iter<T>(value).begin(); |
| 158 | size_t ndim = value.layout.ndim; |
| 159 | auto ptr = data.ptr<T>(); |
| 160 | for (size_t idx = 0; idx < value.layout.total_nr_elems(); ++idx) { |
| 161 | int index[TensorShape::MAX_NDIM]; |
| 162 | std::copy(iter.idx(), iter.idx() + ndim, index); |
| 163 | size_t data_idx = get_index(data, value, desc, index); |
| 164 | ptr[data_idx] = *iter; |
| 165 | ++iter; |
| 166 | } |
| 167 | } |
| 168 | |
| 169 | void SetMeshIndexingImpl::exec( |
| 170 | _megdnn_tensor_inout data, _megdnn_tensor_in value, const IndexDesc& desc, |
nothing calls this directly
no test coverage detected