trt8510 not support topk with input is int or int64

[lix19937]

follow is passed.

import torch

class MyModel(torch.nn.Module):
    def __init__(self):
        super(MyModel, self).__init__()
    def forward(self, x):
        x, _ = x.topk(k=10, dim=0, largest=False)
        return x

model = MyModel()
x = torch.ones(120, 1, 128, 128, dtype=torch.float32)

with torch.no_grad():
    torch.onnx.export(
        model,
        (x),
        'topk.onnx',
        opset_version=17,
        )
    

[lix19937]

follow is wrong.

import torch

class MyModel(torch.nn.Module):
    def __init__(self):
        super(MyModel, self).__init__()
    def forward(self, x):
        x, _ = x.topk(k=10, dim=0, largest=False)
        return x

model = MyModel()
x = torch.ones(120, 1, 128, 128, dtype=torch.int32)

with torch.no_grad():
    torch.onnx.export(
        model,
        (x),
        'topk.onnx',
        opset_version=17,
        )