Searched defs:LaunchLRNGrad (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op.cc | 387 struct LaunchLRNGrad<CPUDevice, T> { struct
388 LaunchLRNGrad(int depth_radius, T bias, T alpha, T beta) in LaunchLRNGrad() function
395 void launch(OpKernelContext* context, OpKernel* kernel, in launch()
460 int depth_radius_;
461 T bias_;
462 T alpha_;
463 T beta_;
464 T alpha_beta_2_;
470 struct LaunchLRNGrad<GPUDevice, T> { struct
471 LaunchLRNGrad(int depth_radius, T bias, T alpha, T beta) in LaunchLRNGrad() argument
[all …]
|