| 56 | |
| 57 | template <typename T> |
| 58 | void apply() const { |
| 59 | // TODO(Yancey1989): support other place |
| 60 | phi::CPUPlace cpu; |
| 61 | std::memcpy(dst_->mutable_data<T>(cpu) + dst_offset_, |
| 62 | src_.data<T>() + src_offset_, |
| 63 | size_ * sizeof(T)); |
| 64 | } |
| 65 | |
| 66 | DenseTensor* dst_; |
| 67 | int64_t dst_offset_; |
no outgoing calls
no test coverage detected