// accumulation over the given input type.
Type GetSumAccumulationType(Type input_type) {
  MLIRContext *ctx = input_type.getContext();
  if (input_type.isBF16() || input_type.isF16()) return FloatType::getF32(ctx);
  if (input_type.isSignlessInteger(8) || input_type.isSignlessInteger(16))
    return IntegerType::get(ctx, 32);
  return input_type;
}

// Returns axis in HLO format from TF elements attr with exactly one element or

	69,  // 186: v1alpha1.IntOrString.strVal:type_name -> google.protobuf.StringValue
	187, // [187:187] is the sub-list for method output_type
	187, // [187:187] is the sub-list for method input_type
	187, // [187:187] is the sub-list for extension type_name
	187, // [187:187] is the sub-list for extension extendee
	0,   // [0:187] is the sub-list for field type_name
}