message: [XLA] Algebraic simplifier incorrectly transformed convolutions into bitcasts
diff: @@ -1505,9 +1505,9 @@ Status AlgebraicSimplifierVisitor::HandleConvolution(
  // We cannot insert bitcasts if the layouts will not be compatible.
  // TODO(b/33178038): Consider inserting a transpose if a bitcast would be
  // invalid.
  - if (!valid_bitcast_callback_(input_shape, lhs->shape()) ||
  - !valid_bitcast_callback_(new_filter_shape, rhs->shape()) ||
  - !valid_bitcast_callback_(convolution_shape, dot_output_shape)) {
  + if (!valid_bitcast_callback_(input_shape, new_input_shape) ||
  + !valid_bitcast_callback_(filter_shape, new_filter_shape) ||
  + !valid_bitcast_callback_(dot_output_shape, convolution_shape)) {
  return Status::OK();
  }
  