]>;

// Return true if the product of dimension values of a subsection of the tensor
// is equal to the non-contracting dimension after a reshape
class BroadcastDimsProductEqual<int agg_start_idx> : Constraint<CPred<
  "TFL::BroadcastDimsProductEqual($0, $1, "# agg_start_idx #")">>;

// Returns true if the dimensions of a subsection of two tensors is equal

  }

  void runOnOperation() override;
};

// Return true if the product of dimension values of a subsection of the tensor
// is equal to the non-contracting dimension after a reshape
bool BroadcastDimsProductEqual(Value input, Value output,
                               size_t agg_start_idx) {
  ArrayRef<int64_t> input_shape =
      mlir::cast<ShapedType>(input.getType()).getShape();
  ArrayRef<int64_t> output_shape =