tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/convolution_pattern.cc
// Specialization for convolution forward fused side input
Specialization for convolution forward fused side input
Specialization for convolution forward fused side input
Status SetConvKind(lmhlo_gpu::ConvForwardFusedSideInputOp op,
                   xla::gpu::GpuConvDescriptor& descriptor) {
  descriptor.kind = xla::gpu::CudnnConvKind::kForwardActivation;
  auto activation_mode_or =
      xla::ConvertConvActivationMode(op.activation_mode());
  if (!activation_mode_or.ok()) {
    return activation_mode_or.status();
  }
  auto activation_mode = activation_mode_or.ValueOrDie();
  descriptor.backend_config.set_activation_mode(
      static_cast<int64_t>(activation_mode));
  descriptor.backend_config.set_side_input_scale(
      op.side_input_scale().convertToDouble());
  return Status::OK();
}
Status SetConvKind ( lmhlo_gpu :: ConvForwardFusedSideInputOp op , xla :: gpu :: GpuConvDescriptor & descriptor ) { descriptor . kind = xla :: gpu :: CudnnConvKind :: kForwardActivation ; auto activation_mode_or = xla :: ConvertConvActivationMode ( op . activation_mode ( ) ) ; if ( ! activation_mode_or . ok ( ) ) { return activation_mode_or . status ( ) ; } auto activation_mode = activation_mode_or . ValueOrDie ( ) ; descriptor . backend_config . set_activation_mode ( static_cast < int64_t > ( activation_mode ) ) ; descriptor . backend_config . set_side_input_scale ( op . side_input_scale ( ) . convertToDouble ( ) ) ; return Status :: OK ( ) ; }
Specialization for convolution forward fused side input
Specialization for convolution forward fused side input