| 359 | |
| 360 | template <typename T> |
| 361 | __global__ void FillWithVectorKernel(const T* values, GpuMatView<T> output) { |
| 362 | const size_t row = blockIdx.y * blockDim.y + threadIdx.y; |
| 363 | const size_t col = blockIdx.x * blockDim.x + threadIdx.x; |
| 364 | if (row < output.GetHeight() && col < output.GetWidth()) { |
| 365 | for (size_t slice = 0; slice < output.GetDepth(); ++slice) { |
| 366 | output.Set(row, col, slice, values[slice]); |
| 367 | } |
| 368 | } |
| 369 | } |
| 370 | |
| 371 | template <typename T> |
| 372 | __global__ void FillWithRandomNumbersKernel( |