Searched defs:LaunchLRN (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op.cc | 84 struct LaunchLRN<CPUDevice, T> { struct
85 LaunchLRN(int depth_radius, T bias, T alpha, T beta) in LaunchLRN() argument
88 void launch(OpKernelContext* context, OpKernel* kernel, const Tensor& in, in launch()
128 typedef typename Eigen::Tensor<T, 1, Eigen::RowMajor>::DimensionPair DimPair;
130 void SingleThreadedLRN(const Tensor& in, const int batch, const int rows, in SingleThreadedLRN()
169 int depth_radius_;
170 T bias_;
171 T alpha_;
172 T beta_;
178 struct LaunchLRN<GPUDevice, T> { struct
[all …]
|