some fixes (#1281)

This commit is contained in:
Awni Hannun
2024-07-23 11:49:05 -07:00
committed by GitHub
parent 6768c6a54a
commit e2aa6ec8ae
4 changed files with 14 additions and 5 deletions

View File

@@ -2833,7 +2833,7 @@ void init_ops(nb::module_& m) {
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig(
"def pad(a: array, pad_with: Union[int, Tuple[int], Tuple[int, int], List[Tuple[int, int]]], constant_values: Union[scalar, array] = 0, *, stream: Union[None, Stream, Device] = None) -> array"),
"def pad(a: array, pad_width: Union[int, Tuple[int], Tuple[int, int], List[Tuple[int, int]]], constant_values: Union[scalar, array] = 0, *, stream: Union[None, Stream, Device] = None) -> array"),
R"pbdoc(
Pad an array with a constant value