let hasVerifier = 1;
}

def TF_ApproxTopKOp : TF_Op<"ApproxTopK", [Pure]> {
  let summary = [{
Returns min/max k values and their indices of the input operand in an approximate manner.
  }];

  let description = [{
See https://arxiv.org/abs/2206.14286 for the algorithm details.
This op is only optimized on TPU currently.
  }];

  let arguments = (ins