Searched defs:LaunchConv2DOp (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops.cc | 251 struct LaunchConv2DOp<CPUDevice, T> { struct
252 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
322 struct LaunchConv2DOp<GPUDevice, int32> { struct
323 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
771 template struct LaunchConv2DOp<CPUDevice, Eigen::bfloat16>; variable
772 template struct LaunchConv2DOp<CPUDevice, Eigen::half>; variable
773 template struct LaunchConv2DOp<CPUDevice, float>; variable
774 template struct LaunchConv2DOp<CPUDevice, double>; variable
1226 template struct LaunchConv2DOp<GPUDevice, float>; variable
1227 template struct LaunchConv2DOp<GPUDevice, Eigen::half>; variable
[all …]
|
| H A D | conv_ops.h | 35 struct LaunchConv2DOp { struct
|