pytorch
a77d2be6 - Codegen arange.start_out (#67029)

Commit
3 years ago
Codegen arange.start_out (#67029) * Add codegen support for arange.start_out - does not add full support for factory functions (with TensorOption style args), but does add codegen support for handling expanding TensorOption args into their components. Needs more work in order to support ops like `arange` - supports arange.start_out: needs changes to how shape inference functions are named
Author
Parents
Loading