Searched refs:LaunchConv2DOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 250 extern template struct LaunchConv2DOp<CPUDevice, Eigen::half>;
251 extern template struct LaunchConv2DOp<CPUDevice, float>;
252 extern template struct LaunchConv2DOp<CPUDevice, double>;
257 extern template struct LaunchConv2DOp<GPUDevice, Eigen::half>;
258 extern template struct LaunchConv2DOp<GPUDevice, float>;
259 extern template struct LaunchConv2DOp<GPUDevice, double>;
485 LaunchConv2DOp<Device, T> launcher_;
|
| D | conv_ops.h | 35 struct LaunchConv2DOp { struct
46 struct LaunchConv2DOp<Eigen::GpuDevice, T> {
|
| D | conv_ops.cc | 144 struct LaunchConv2DOp<CPUDevice, T> { struct
181 struct LaunchConv2DOp<GPUDevice, int32> { struct
594 LaunchConv2DOp<Device, T> launcher_;
614 template struct LaunchConv2DOp<CPUDevice, Eigen::half>; variable
615 template struct LaunchConv2DOp<CPUDevice, float>; variable
616 template struct LaunchConv2DOp<CPUDevice, double>; variable
646 void LaunchConv2DOp<GPUDevice, T>::operator()( in operator ()()
1199 template struct LaunchConv2DOp<GPUDevice, float>; variable
1200 template struct LaunchConv2DOp<GPUDevice, Eigen::half>; variable
1201 template struct LaunchConv2DOp<GPUDevice, double>; variable
|