Searched defs:LaunchConv2DBackpropFilterOp (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_filter_ops.cc | 111 struct LaunchConv2DBackpropFilterOp<CPUDevice, T> { struct
112 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
650 template struct LaunchConv2DBackpropFilterOp<CPUDevice, Eigen::bfloat16>; variable
651 template struct LaunchConv2DBackpropFilterOp<CPUDevice, Eigen::half>; variable
652 template struct LaunchConv2DBackpropFilterOp<CPUDevice, float>; variable
653 template struct LaunchConv2DBackpropFilterOp<CPUDevice, double>; variable
1058 template struct LaunchConv2DBackpropFilterOp<GPUDevice, float>; variable
1059 template struct LaunchConv2DBackpropFilterOp<GPUDevice, Eigen::half>; variable
1060 template struct LaunchConv2DBackpropFilterOp<GPUDevice, double>; variable
|
| H A D | conv_grad_ops.h | 183 struct LaunchConv2DBackpropFilterOp { struct
|