#include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc"

const char kICIWeightDistributionMlirBridgeMarker[] =
    "ici_weight_distribution_mlir_bridge_marker";

struct XlaBroadcast : public impl::XlaBroadcastPassBase<XlaBroadcast> {
  void runOnOperation() override;
};

// Returns true iff the broadcast val can be substituted with an XlaAllReduce.