// TODO(b/154065712): Remove propagate_caller_callee_constants once using
  // SCCP pass instead.
  bool propagate_caller_callee_constants_;

  // XlaCallModule loader, which is used to deserialize the StableHLO module in
  // each `XlaCallModule` op. Uses its own MLIRContext since the loader needs to
  // load additional dialects, which is not allowed for the main context since
  // shape inference may be called from a pass.

      # This is to verify the invocation of StableHLO quantizer works. More
      # thorough functional tests are in StableHLO quantizer directory.
      self.assertTrue(self._contains_op(output_graphdef, 'XlaCallModule'))
    else:
      self.assertTrue(self._contains_quantized_function_call(output_graphdef))
    self.assertFalse(self._contains_op(output_graphdef, 'FusedBatchNormV3'))

  @parameterized.named_parameters(

void XlaCallModuleOp::getEffects(
    SmallVectorImpl<SideEffects::EffectInstance<MemoryEffects::Effect>>
        &effects) {
  if (!getFunctionList().empty()) {
    // The StableHLO module embedded in XlaCallModule contains
    // `stablehlo.custom_call` calling TF host callback functions.
    // `stablehlo.custom_call` will be lowered to `stablehlo.send` and
    // `stablehlo.recv`.

    // InferTypeOpInterface:
    static bool isCompatibleReturnTypes(TypeRange l, TypeRange r) {
      return ArraysAreCastCompatible(l, r);
    }
  }];
}

def TF_XlaCallModuleOp : TF_Op<"XlaCallModule", [DeclareOpInterfaceMethods<MemoryEffectsOpInterface>, DeclareOpInterfaceMethods<SymbolUserOpInterface>]> {
  let summary = "Invokes a StableHLO module.";

  let description = [{