Skip to content

Commit 92522eb

Browse files
authored
[torchlib] Add memory_format to ones_like (#2049)
It was missing
1 parent 7d43f92 commit 92522eb

File tree

1 file changed

+1
-0
lines changed
  • onnxscript/function_libs/torch_lib/ops

1 file changed

+1
-0
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6476,6 +6476,7 @@ def aten_ones_like(
64766476
layout: str = "",
64776477
device: str = "",
64786478
pin_memory: bool = False,
6479+
memory_format: str = "",
64796480
) -> TTensor:
64806481
"""ones_like(Tensor self, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor
64816482

0 commit comments

Comments
 (0)