Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "b39cbc7fdc618c084bbb2adda2e7b068b9f43869"
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 = {
'ReLU': 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::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),
'Conv2d': lambda module_masks, mask: conv2d_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