Searched refs:LaunchConv2DOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 245 extern template struct LaunchConv2DOp<CPUDevice, Eigen::half>;
246 extern template struct LaunchConv2DOp<CPUDevice, float>;
247 extern template struct LaunchConv2DOp<CPUDevice, double>;
252 extern template struct LaunchConv2DOp<GPUDevice, Eigen::half>;
253 extern template struct LaunchConv2DOp<GPUDevice, float>;
254 extern template struct LaunchConv2DOp<GPUDevice, double>;
445 LaunchConv2DOp<Device, T> launcher_;
|
| D | conv_ops.h | 35 struct LaunchConv2DOp { struct
46 struct LaunchConv2DOp<Eigen::GpuDevice, T> {
|
| D | conv_ops.cc | 123 struct LaunchConv2DOp<CPUDevice, T> { struct
528 LaunchConv2DOp<Device, T> launcher_;
547 template struct LaunchConv2DOp<CPUDevice, Eigen::half>; variable
548 template struct LaunchConv2DOp<CPUDevice, float>; variable
549 template struct LaunchConv2DOp<CPUDevice, double>; variable
578 void LaunchConv2DOp<GPUDevice, T>::operator()( in operator ()()
970 template struct LaunchConv2DOp<GPUDevice, float>; variable
971 template struct LaunchConv2DOp<GPUDevice, Eigen::half>; variable
972 template struct LaunchConv2DOp<GPUDevice, double>; variable
|