OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
3803d1c901
pytorch/caffe2/onnx/torch_ops
History
Yinghai Lu 34db39d87a Add onnxifi support to SparseLengthsWeightedSum (#14210) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14210

We left `SparseLengthsWeightedSum` as benchmark is not testing it due to fp16 filler issue. It was flushed out by unit tests. Hence we add the support here.

Reviewed By: bddppq

Differential Revision: D13132320

fbshipit-source-id: b21c30c185c9e1fbf3980641bc3cdc39e85af2e1
2018-11-21 15:47:24 -08:00
..
CMakeLists.txt Add "ai.onnx.pytorch" onnx domain (#12157) 2018-09-28 09:57:06 -07:00
constants.h Add "ai.onnx.pytorch" onnx domain (#12157) 2018-09-28 09:57:06 -07:00
defs.cc Add onnxifi support to SparseLengthsWeightedSum (#14210) 2018-11-21 15:47:24 -08:00
operator_sets.h Add onnxifi support to SparseLengthsWeightedSum (#14210) 2018-11-21 15:47:24 -08:00
schema.cc Add "ai.onnx.pytorch" onnx domain (#12157) 2018-09-28 09:57:06 -07:00
schema.h Add "ai.onnx.pytorch" onnx domain (#12157) 2018-09-28 09:57:06 -07:00