Searched defs:LaunchConv2DBackpropInputOp (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_input_ops.cc | 35 template struct LaunchConv2DBackpropInputOp<CPUDevice, bfloat16>; variable
36 template struct LaunchConv2DBackpropInputOp<CPUDevice, Eigen::half>; variable
37 template struct LaunchConv2DBackpropInputOp<CPUDevice, float>; variable
38 template struct LaunchConv2DBackpropInputOp<CPUDevice, double>; variable
57 struct LaunchConv2DBackpropInputOp<GPUDevice, int32> { struct
58 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
508 template struct LaunchConv2DBackpropInputOp<GPUDevice, float>; variable
509 template struct LaunchConv2DBackpropInputOp<GPUDevice, Eigen::half>; variable
510 template struct LaunchConv2DBackpropInputOp<GPUDevice, double>; variable
|
| H A D | conv_grad_ops.h | 173 struct LaunchConv2DBackpropInputOp { struct
|