summaryrefslogtreecommitdiff
path: root/torch/onnx
AgeCommit message (Expand)AuthorFilesLines
2019-04-23dispatch max_pools with no indices, expose max_pools to torch namespace (#19449)Wanchao Liang1-98/+57
2019-04-22Add onnx support for _unique2 operatorVitaly Fedyunin1-0/+7
2019-04-18Strip doc_string from exported ONNX models (#18882)Lara Haidar-Ahmad1-5/+11
2019-04-18Add constant folding to ONNX graph during export (Resubmission) (#18698)Spandan Tiwari2-15/+30
2019-04-16Fix python lint (#19331)Junjie Bai1-1/+1
2019-04-16Adding pin_memory kwarg to zeros, ones, empty, ... tensor constructors (#18952)Vitaly Fedyunin1-11/+30
2019-04-13get propagate_shape logic out of module.h (#19137)Zachary DeVito1-4/+5
2019-04-12Fix onnx ints (#19102)Lu Fang1-13/+24
2019-04-11Skip Slice if it's no op (#19155)Lu Fang1-6/+14
2019-04-09EmbeddingBag CPU forward with per_sample_weights. (#18735)Richard Zou1-2/+6
2019-04-09add an utility function to check whether it's in the middle of onnx export or...Lu Fang2-44/+71
2019-04-08Export C10 operator in PyTorch Model (#18210)Lu Fang1-1/+1
2019-04-05Revert "Adding pin_memory kwarg to zeros, ones, empty,... (#18854)Vitaly Fedyunin1-26/+18
2019-04-04ONNX Export All Cases of SoftmaxLara1-7/+15
2019-04-03add an assertion to check the param num (#18145)Lu Fang1-0/+4
2019-04-02Fix flake8 issuesIgor Fedan1-1/+1
2019-04-02Adding pin_memory kwarg to zeros, ones, empty, ... tensor constructors. (#18455)Vitaly Fedyunin1-18/+26
2019-03-30Turn on F401: Unused import warning. (#18598)Edward Yang3-12/+1
2019-03-29Refactoring serialization of ONNX initializers to be name-based (Resubmission...Spandan Tiwari1-8/+13
2019-03-28Resubmit PR-18512: Improved onnx export for 3 onnx ops (#18571)bddppq1-1/+9
2019-03-28Revert D14635130: Improved onnx export for 3 onnx ops.Junjie Bai1-9/+1
2019-03-28Improved onnx export for 3 onnx ops. (#18512)Benoit Steiner1-1/+9
2019-03-27Rename isTensor api -> isCompleteTensor (#18437)eellison1-4/+4
2019-03-26Revert D14605905: [pytorch][PR] Add return_counts to torch.uniqueSoumith Chintala1-4/+3
2019-03-26Minor fix for onnx ConstantOfShape export (#18199)BowenBao1-6/+6
2019-03-25Support dim=None for argmax and argmin (#18264)Xiang Gao1-4/+14
2019-03-25Add return_counts to torch.unique (#18391)Xiang Gao1-3/+4
2019-03-20Fix ONNX symbolic for argmin and argmax (#18261)Lu Fang1-2/+2
2019-03-20Retain the parameter names in ONNX exporter (#17551)Lu Fang1-10/+21
2019-03-20Fix typo in docstringAlexandr Morev1-1/+1
2019-03-15ONNX Export IsNan opLara Haidar-Ahmad1-0/+7
2019-03-14Update nonzero onnx export (#18047)BowenBao1-1/+1
2019-03-08Revert D14361993: [pytorch][PR] [Onnx] - refactoring serialization of ONNX in...Lu Fang1-11/+7
2019-03-07Set the default ONNX opset to the latest stable opset (i.e., 9) (#17736)Lu Fang2-4/+8
2019-03-07Add module attributes (#17309)David Riazati1-1/+1
2019-03-07- refactoring serialization of ONNX initializers to be name-based (#17420)Spandan Tiwari1-7/+11
2019-03-07ONNX Export for Max and Average Pooling in CEIL_MODELara Haidar-Ahmad1-17/+66
2019-03-06ONNX Export Narrow opLara Haidar1-0/+5
2019-03-06Improve ONNX symbolic for logsoftmax and softmax (#17672)Lu Fang1-7/+14
2019-03-06ONNX Export Argmin and Argmax opsLara Haidar-Ahmad1-0/+10
2019-03-01Rename prim::Undefined to prim::AutogradZero (#17611)Wanchao Liang1-10/+9
2019-02-27disallow shape analysis with resize ops (#17518)Elias Ellison1-6/+6
2019-02-27ONNX Export Adaptive PoolingLara Haidar1-8/+41
2019-02-26Bump up the ONNX default opset version to 10 (#17419)Lu Fang1-7/+12
2019-02-22add the support for stable ONNX opsets in exporter (#16068)Lu Fang2-13/+38
2019-02-19ONNX Export Maxpool IndicesLara Haidar1-15/+52
2019-02-19move prim::None to prim::Constant (again) (#17186)eellison2-14/+16
2019-02-15Revert D14109636: [pytorch][PR] move prim::None to a case in prim::ConstantElias Ellison2-16/+14
2019-02-15move prim::None to a case in prim::Constant (#16160)Elias Ellison2-14/+16
2019-02-15Extend support for exporting reshape to onnx. (#16971)BowenBao1-0/+9