I32Attr:$calibration_method,
    // The number of histogram bins.
    I32Attr:$num_bins,
    // Min percentile to be included in the selected range, only used in the
    // `HISTOGRAM_PERCENTILE` method.
    F32Attr:$min_percentile,
    // Max percentile to be included in the selected range, only used in the
    // `HISTOGRAM_PERCENTILE` method.
    F32Attr:$max_percentile
  );

  }];

  let arguments = (ins
    TF_Tensor:$max_iterations,
    Variadic<TF_Tensor>:$tensor_list_or_flow_in,
    Variadic<TF_Tensor>:$invariant_args,
    FlatSymbolRefAttr:$body_fn,
    I32Attr:$num_tensor_list_or_flow_in
  );

  let results = (outs
     Variadic<TF_Tensor>:$result
  );