return false;
    }
    Operation* user = operand.getOwner();
    if (!IsWeightOnlyQuantizableOp(*user)) {
      return false;
    }
    Method method = GetQuantizationMethodOrDefault(user);
    return HasValidWeightOnlyPtqMethod(method.weight_only_ptq(), rank);
  }

  void rewrite(Operation* op, PatternRewriter& rewriter) const override {

      FindFirstOpFromMainFunc<TF::XlaCallModuleOp>(*module_op);
  ASSERT_TRUE(succeeded(xla_call_module_op));

  // Test that `GetQuantizationMethodOrDefault` returns a valid `Method`
  // corresponding to `"no_quantization {}"`.
  const Method method = GetQuantizationMethodOrDefault(*xla_call_module_op);
  EXPECT_TRUE(method.has_no_quantization());
}

TEST_F(
    LiftAsFunctionCallTest,

// `Method` iff the attribute doesn't exist or the attribute contains an invalid
// textproto for `Method`. `op` must be non-null.
::stablehlo::quantization::Method GetQuantizationMethodOrDefault(
    absl::Nonnull<Operation*> op);

// Creates a function to wrap the section between arguments and results.
// The generated function call op type will be decided by the given call_op_type

    return absl::InternalError(
        absl::StrCat("Failed to parse Method from textproto: ", method_txtpb));
  }

  return quantization_method;
}

Method GetQuantizationMethodOrDefault(absl::Nonnull<Operation*> op) {
  absl::StatusOr<Method> method = GetQuantizationMethod(op);
  if (method.status().code() == absl::StatusCode::kInternal) {

    // TODO: b/331145946 - Each quantization method should be valid
    // (GetQuantizationMethodOrDefault swallows invalid method attribute). Check
    // the validity in `match()`. Use accessors to achieve this.
    const Method quantization_method =
        GetQuantizationMethodOrDefault(xla_call_module_op);

    ReplaceQuantizedXlaCallModuleOpWithQuantizedCallOp(