I64Attr:$instance_key,
    TF_AnyStrAttrOf<["Min", "Max", "Mul", "Add"]>:$merge_op,
    TF_AnyStrAttrOf<["Id", "Div"]>:$final_op,
    I64ArrayAttr:$subdiv_offsets,
    DefaultValuedOptionalAttr<I64ArrayAttr, "{}">:$wait_for,
    DefaultValuedOptionalAttr<StrAttr, "\"auto\"">:$communication_hint,
    DefaultValuedOptionalAttr<F32Attr, "0.0f">:$timeout_seconds
  );

  let results = (outs
    TF_FpOrI32OrI64Tensor:$data
  );