lrn_op.cc | 76 void launch(OpKernelContext* context, OpKernel* kernel, const Tensor& in, function in struct:tensorflow::LaunchLRN 170 void launch(OpKernelContext* context, OpKernel* kernel, const Tensor& in, function in struct:tensorflow::LaunchLRN 215 errors::Internal("NormalizeWithDimensions launch failed")); 272 launcher.launch(context, this, in, output); 313 void launch(OpKernelContext* context, OpKernel* kernel, function in struct:tensorflow::LaunchLRNGrad 387 void launch(OpKernelContext* context, OpKernel* kernel, function in struct:tensorflow::LaunchLRNGrad 436 errors::Internal("NormalizeBackwardWithDimensions launch failed")); 495 launcher.launch(context, this, in_grads, in_image, out_image, output);
|