pytorch
b3e18024 - Static runtime support for fb::expand_dims (#57282)

Commit
3 years ago
Static runtime support for fb::expand_dims (#57282) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57282 Added support for fb::expand_dims for SR. Test Plan: buck test caffe2/torch/fb/sparsenn:gpu_test -- test_expand_dims buck test caffe2/benchmarks/static_runtime/fb:test_fb_operators Reviewed By: hlu1 Differential Revision: D28043049 fbshipit-source-id: 01f59db7b507f027b220f044d6ff23602adbdb06
Author
Edvard Ghazaryan
Parents
Loading