Files
pytorch/caffe2
Antonio Kim 02c4d877b4 Codegen Non-Native IR Nodes (#76535)
Add codegen infrastructure to generate IR nodes for non-native ops.

The proposed change is to add a `non_native` key to the `{backend}_native_functions.yaml` file that contains schema definitions similar to what is found in `native_functions.yaml`. e.g.
```
non_native:
    ...
    - func: expand(Tensor input, int[] size, bool is_scalar_expand) -> Tensor
    ...
```
these definitions are parsed into a `LazyIrSchema` that can be used for generating IR nodes using `GenLazyIR`.

Fixes #74628

CC: @wconstab @desertfire @henrytwo

Pull Request resolved: https://github.com/pytorch/pytorch/pull/76535
Approved by: https://github.com/wconstab
2022-05-24 19:29:23 +00:00
..
2021-12-10 04:26:23 -08:00
2021-12-07 16:07:50 -08:00
2022-04-11 21:55:59 +00:00
2022-04-05 00:08:05 +00:00
2022-04-11 21:55:59 +00:00
2021-12-07 16:53:27 -08:00
2022-04-26 23:54:22 +00:00
2021-12-07 16:53:27 -08:00
2020-07-20 10:19:08 -07:00
2018-10-04 02:09:56 -07:00

Caffe2

Jenkins Build Status

Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.

Questions and Feedback

Please use GitHub issues (https://github.com/pytorch/pytorch/issues) to ask questions, report bugs, and request new features.

Further Resources on Caffe2.ai