TopKV2

Inputs:

  • intput: float32
  • k: int

Outputs:

  • output: float32
  • indices: int

Backend:

    • CPU
    • Metal
    • Vulkan
    • OpenCL

Tensorflow op:

  • TopKV2