elif mode == 'replicate':
ret = torch._C._nn.replication_pad1d(input, pad)
elif mode == 'circular':
- ret = pad_circular(input, pad)
+ ret = _pad_circular(input, pad)
else:
ret = input # TODO: remove this when jit raise supports control flow
raise NotImplementedError
elif mode == 'replicate':
ret = torch._C._nn.replication_pad2d(input, pad)
elif mode == 'circular':
- ret = pad_circular(input, pad)
+ ret = _pad_circular(input, pad)
else:
ret = input # TODO: remove this when jit raise supports control flow
raise NotImplementedError
elif mode == 'replicate':
ret = torch._C._nn.replication_pad3d(input, pad)
elif mode == 'circular':
- ret = pad_circular(input, pad)
+ ret = _pad_circular(input, pad)
else:
ret = input # TODO: remove this when jit raise supports control flow
raise NotImplementedError
@weak_script
-def pad_circular(input, padding):
+def _pad_circular(input, padding):
# type: (Tensor, List[int]) -> Tensor
"""
Arguments