Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "c6a19fb23ec41ebb563eeb66db55414f823c4b23"
Unverified Commit 64c7f298 authored by Ningxin Zheng's avatar Ningxin Zheng Committed by GitHub
Browse files

Add the support for tanh. (#2816) 


Signed-off-by: default avatarNingxin <Ningxin.Zheng@microsoft.com>
parent bf5e7792
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
src/sdk/pynni/nni/compression/torch/speedup/infer_shape.py
View file @ 64c7f298
...@@ -222,6 +222,10 @@ infer_from_inshape = { ...@@ -222,6 +222,10 @@ infer_from_inshape = {
'ReLU': lambda module_masks, mask: relu_inshape(module_masks, mask), 'ReLU': lambda module_masks, mask: relu_inshape(module_masks, mask),
'ReLU6': lambda module_masks, mask: relu_inshape(module_masks, mask), 'ReLU6': lambda module_masks, mask: relu_inshape(module_masks, mask),
'aten::relu': lambda module_masks, mask: relu_inshape(module_masks, mask), 'aten::relu': lambda module_masks, mask: relu_inshape(module_masks, mask),
'aten::tanh': lambda module_masks, mask: relu_inshape(module_masks, mask),
'aten::tanh_': lambda module_masks, mask: relu_inshape(module_masks, mask),
'aten::hardtanh': lambda module_masks, mask: relu_inshape(module_masks, mask),
'aten::hardtanh_': lambda module_masks, mask: relu_inshape(module_masks, mask),
'aten::relu_': lambda module_masks, mask: relu_inshape(module_masks, mask), 'aten::relu_': lambda module_masks, mask: relu_inshape(module_masks, mask),
'Conv2d': lambda module_masks, mask: conv2d_inshape(module_masks, mask), 'Conv2d': lambda module_masks, mask: conv2d_inshape(module_masks, mask),
'MaxPool2d': lambda module_masks, mask: maxpool2d_inshape(module_masks, mask), 'MaxPool2d': lambda module_masks, mask: maxpool2d_inshape(module_masks, mask),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment