Searched refs:RunGpuConvImpl (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 221 Status RunGpuConvImpl(const GpuConvParams& params, in RunGpuConvImpl() function
516 return RunGpuConvImpl<Eigen::half, Eigen::half, Eigen::half>( in RunGpuConv()
519 return RunGpuConvImpl<float, float, float>(params, scratch_allocator, in RunGpuConv()
522 return RunGpuConvImpl<double, double, double>(params, scratch_allocator, in RunGpuConv()
528 return RunGpuConvImpl<int8, float, float>(params, scratch_allocator, in RunGpuConv()
531 return RunGpuConvImpl<int8, float, int8>(params, scratch_allocator, in RunGpuConv()
|