Lines Matching refs:gpu_device
116 const Eigen::GpuDevice& gpu_device,
126 To32Bit(*output).slice(offset, size).device(gpu_device) =
129 output->slice(offset, size).device(gpu_device) = *inputs_flat[i];
137 void ConcatGPUImpl(const Eigen::GpuDevice& gpu_device,
143 output->dimension(0), gpu_device);
148 gpu_device.stream()>>>(input_ptrs, split_size, output->dimension(0),
151 IntType smem_max = gpu_device.sharedMemPerBlock();
161 gpu_device.stream()>>>(input_ptrs, output_scan,
167 gpu_device.stream()>>>(input_ptrs, output_scan,
175 const Eigen::GpuDevice& gpu_device, \
182 const Eigen::GpuDevice& gpu_device, \
