Actual torchscript from some model looks the following
torchscript Python:
output = torch.zeros([_17, int(num_channels), 7, 7], dtype=6, layout=None, pin_memory=False)
output0 = torch.index_put_(output, _20, _19, False)
torchscript IR:
%output.1 : Float(1000, 256, 7, 7, strides=[12544, 49, 7, 1], requires_grad=0) = aten::zeros(%1990, %1942, %1945, %1943, %1944)
%output.2 : Float(1000, 256, 7, 7, strides=[12544, 49, 7, 1], requires_grad=0) = aten::index_put_(%output.1, %2000, %1999, %1944)