Arg<TF_Int32Tensor, [{The TPU device to use. Should be >= 0 and less than the number
of TPU cores in the task on which the node is placed.}]>:$device_ordinal,

    DefaultValuedOptionalAttr<StrArrayAttr, "{}">:$combiners
  );

  let results = (outs);

  TF_DerivedOperandSizeAttr N = TF_DerivedOperandSizeAttr<0>;
  TF_DerivedOperandTypeAttr T1 = TF_DerivedOperandTypeAttr<0>;