pytorch
34eb6c85 - [Caffe2] ScriptModuleOp support pass_inputs_as_tensor_list (#56813)

Commit
3 years ago
[Caffe2] ScriptModuleOp support pass_inputs_as_tensor_list (#56813) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56813 When the arg `pass_inputs_as_tensor_list` is True, the input tensors are wrapped into a TensorList and passes in as a single param. Test Plan: buck test //caffe2/caffe2/python:workspace_test -- TestScriptModule Reviewed By: dzhulgakov Differential Revision: D27972928 fbshipit-source-id: 5a199649445b0306f3134086c85bd55da45e1a0b
Author
Parents
Loading