Fixed Eigen-version Conv2DBackpropInput kernel, marked with "eigen_tensor".
LaunchConv2DBackpropInputOp, a template functor class used in Conv2DFastBackpropInputOp class was specialized in a wrong place, in conv_grad_filter_ops.cc (ODR violation). This caused wrong functor to be called inside the op kernel class, because the correct functor definition was overriden by a wrong specialization. The functor class that should have been specialized in conv_grad_filter_ops.cc was LaunchConv2DBackpropFilterOp. PiperOrigin-RevId: 182438025
Loading
Please sign in to comment