pytorch
e535fb3f - [ONNX] Enable true_divide scripting export with ONNX shape inference (#43991)

Commit
4 years ago
[ONNX] Enable true_divide scripting export with ONNX shape inference (#43991) Summary: Fixes the `true_divide` symbolic to cast tensors correctly. The logic depends on knowing input types at export time, which is a known gap for exporting scripted modules. On that end we are improving exporter by enabling ONNX shape inference https://github.com/pytorch/pytorch/issues/40628, and starting to increase coverage for scripting support. Pull Request resolved: https://github.com/pytorch/pytorch/pull/43991 Reviewed By: mruberry Differential Revision: D23674614 Pulled By: bzinodev fbshipit-source-id: 1b1b85340eef641f664a14c4888781389c886a8b
Author
Parents
Loading