summaryrefslogtreecommitdiff
path: root/torch/onnx
AgeCommit message (Expand)AuthorFilesLines
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
2019-02-13Support nonzero onnx exportBowenBao1-0/+5
2019-02-11Revert D14020906: [pytorch][PR] Extend support for exporting reshape to onnx.Edward Yang1-9/+0
2019-02-10Extend support for exporting reshape to onnx.BowenBao1-0/+9
2019-02-07Remove undefined tensor in jit script (#16379)Wanchao Liang2-10/+10
2019-02-06Rename DynamicType -> TensorType (#16787)Zachary DeVito1-1/+1
2019-01-30fix the linterLu Fang1-1/+1
2019-01-30ONNX Export Flatten operatorLara Haidar-Ahmad1-0/+25
2019-01-30Fix the dropout onnx symbolic, and ensure all exported models in test_operato...Lu Fang1-0/+2
2019-01-26Trace fork and join callsJames Reed1-0/+5
2019-01-22Revert D13540278: [pytorch][PR] Unhide unique from C++, make unique partially...Wanchao Liang1-10/+4
2019-01-21Move away from ConstantFill (#16214)Lu Fang1-4/+6