Skip to content

Improve ONNX symbolic for logsoftmax and softmax#17672

Closed
houseroad wants to merge 1 commit intopytorch:masterfrom
houseroad:export-D14313987
Closed

Improve ONNX symbolic for logsoftmax and softmax#17672
houseroad wants to merge 1 commit intopytorch:masterfrom
houseroad:export-D14313987

Conversation

@houseroad
Copy link
Member

Summary: support dtype in the onnx symboklic

Differential Revision: D14313987

Summary:
Pull Request resolved: pytorch#17672

support dtype in the onnx symbolic

Differential Revision: D14313987

fbshipit-source-id: 41719c92d5749e3adfb45f6886a4ab6bbb479b47
@houseroad houseroad deleted the export-D14313987 branch March 7, 2019 01:01
petrex pushed a commit to petrex/pytorch that referenced this pull request Mar 7, 2019
* upstream/master: (24 commits)
  Automatic update of fbcode/onnx to 96c58ceeacf0f2b73d752e413e4fd78787a12da3 (pytorch#17676)
  Set the default ONNX opset to the latest stable opset (i.e., 9) (pytorch#17736)
  Add module attributes (pytorch#17309)
  - refactoring serialization of ONNX initializers to be name-based (pytorch#17420)
  ONNX Export for Max and Average Pooling in CEIL_MODE
  use flake8-mypy (pytorch#17721)
  use fp16<->fp32 intrinsic (pytorch#17496)
  Implement a Caffe2 standalone LSTM operator (pytorch#17726)
  caffe2:libtorch_cuda depends on caffe2:caffe2_gpu (pytorch#17729)
  add tensor and cost inference functions (pytorch#17684)
  ONNX Export Narrow op
  Keep the dim_type of hinted shape as BATCH if possible (pytorch#17734)
  fix different round behavior on CPU and GPU pytorch#16498 (pytorch#17443)
  Warn about memory overlaps on expanded tensors (pytorch#17576)
  fix exp fam. formula
  refactor caffe2 operator constructors - 10/9 (pytorch#17659)
  Improve ONNX symbolic for logsoftmax and softmax (pytorch#17672)
  Enable using CMD when building cpp extensions on Windows
  Do not rename net boundary inputs/outputs during ssaRewrite. (pytorch#17545)
  Reapply D14078519 (pytorch#17596)
  ...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants