-
Notifications
You must be signed in to change notification settings - Fork 5.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
c260163
commit df80243
Showing
758 changed files
with
103,986 additions
and
322,905 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
ivy/compiler/_cache/*.pkl filter=lfs diff=lfs merge=lfs -text |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
23 changes: 23 additions & 0 deletions
23
...piler/_cache/Translated_Outputs/Translated_MaxPool2d_output/run_0/Translated_MaxPool2d.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,23 @@ | ||
import ivy.functional.frontends.torch.nn.functional as F | ||
|
||
import typing | ||
|
||
from .Translated__MaxPoolNd import Translated__MaxPoolNd | ||
|
||
|
||
class Translated_MaxPool2d(Translated__MaxPoolNd): | ||
kernel_size: typing.Any | ||
stride: typing.Any | ||
padding: typing.Any | ||
dilation: typing.Any | ||
|
||
def forward(self, input): | ||
return F.max_pool2d( | ||
input, | ||
self.kernel_size, | ||
self.stride, | ||
self.padding, | ||
self.dilation, | ||
ceil_mode=self.ceil_mode, | ||
return_indices=self.return_indices, | ||
) |
38 changes: 38 additions & 0 deletions
38
...iler/_cache/Translated_Outputs/Translated_MaxPool2d_output/run_0/Translated__MaxPoolNd.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,38 @@ | ||
import ivy.functional.frontends.torch.nn as nn | ||
|
||
import typing | ||
|
||
|
||
class Translated__MaxPoolNd(nn.Module): | ||
__constants__ = [ | ||
"kernel_size", | ||
"stride", | ||
"padding", | ||
"dilation", | ||
"return_indices", | ||
"ceil_mode", | ||
] | ||
return_indices: typing.Any | ||
ceil_mode: typing.Any | ||
|
||
def __init__( | ||
self, | ||
kernel_size, | ||
stride=None, | ||
padding=0, | ||
dilation=1, | ||
return_indices=False, | ||
ceil_mode=False, | ||
): | ||
super().__init__() | ||
self.kernel_size = kernel_size | ||
self.stride = stride if stride is not None else kernel_size | ||
self.padding = padding | ||
self.dilation = dilation | ||
self.return_indices = return_indices | ||
self.ceil_mode = ceil_mode | ||
|
||
def extra_repr(self): | ||
return "kernel_size={kernel_size}, stride={stride}, padding={padding}, dilation={dilation}, ceil_mode={ceil_mode}".format( | ||
**self.__dict__ | ||
) |
File renamed without changes.
6 changes: 3 additions & 3 deletions
6
ivy/compiler/_cache/Translated_Outputs/ivy_BatchNorm2d_output/run_0/ivy_BatchNorm2d.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,8 +1,8 @@ | ||
from .ivy__BatchNorm import ivy__BatchNorm | ||
from .ivy__helpers import ivy_dim | ||
from .ivy__helpers import ivy_dim_frnt_ | ||
|
||
|
||
class ivy_BatchNorm2d(ivy__BatchNorm): | ||
def _check_input_dim(self, input): | ||
if ivy_dim(input) != 4: | ||
raise ValueError(f"expected 4D input (got {ivy_dim(input)}D input)") | ||
if ivy_dim_frnt_(input) != 4: | ||
raise ValueError(f"expected 4D input (got {ivy_dim_frnt_(input)}D input)") |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.