func.return %0 : tensor<4x128xf32>
  }
  func.func @_func(%arg0: tensor<4x128xf32>) -> tensor<4x128xf32> {
    func.return %arg0 : tensor<4x128xf32>
  }

func.func @testAddV2IdentityBroadcastTensor(%arg0: tensor<4x1xf32>, %arg1: tensor<4x2xf32>) -> (tensor<4x2xf32>, tensor<4x2xf32>, tensor<4x2xf32>, tensor<4x2xf32>) {
  %0 = "tf.Const"() {value = dense<0.0> : tensor<1x2xf32>} : () -> tensor<1x2xf32>

  // Operand and identity shapes are broadcastable. However, we cannot fold
  // because the operand does not match the result shape.
  %1 = "tf.AddV2"(%arg0, %0) : (tensor<4x1xf32>, tensor<1x2xf32>) -> tensor<4x2xf32>

}

// -----

// 'input_to_output_weights' input for lstm op has unmatched rank with `input`.