Thomas J. Fan a7f4f80903 ENH Adds dtype to nn.functional.one_hot (#58090) ... Summary: Fixes https://github.com/pytorch/pytorch/issues/33046 Related to https://github.com/pytorch/pytorch/issues/53785 Pull Request resolved: https://github.com/pytorch/pytorch/pull/58090 Reviewed By: zou3519 Differential Revision: D28640893 Pulled By: jbschlosser fbshipit-source-id: 3686579517ccc75beaa74f0f6d167f5e40a83fd2		2021-05-24 13:48:25 -07:00
..
expect	Fix some tensor operators to return NotImplemented for invalid inputs (#58216)	2021-05-19 13:09:57 -07:00
model_defs	[ONNX] enable word_language_model GRU and LSTM scripting (#54310) (#56170)	2021-04-20 23:00:48 -07:00
debug_embed_params.py
export_onnx_tests_filter.py
export_onnx_tests_generator.py
pytorch_helper.py
test_caffe2_common.py
test_custom_ops.py	[ONNX] Support registering custom export for prim::PythonOp from torch.autograd.Function (#55630) (#57600)	2021-05-13 13:42:49 -07:00
test_models.py	[ONNX] Support registering custom export for prim::PythonOp from torch.autograd.Function (#55630) (#57600)	2021-05-13 13:42:49 -07:00
test_models_onnxruntime.py	Add lint for unqualified noqa (#56272)	2021-04-19 13:16:18 -07:00
test_onnx_common.py
test_onnx_opset.py	Back out "[ONNX] Process const folding progressively when converts to ONNX (#54569)"	2021-05-14 13:45:49 -07:00
test_operators.py	torch.linalg.det (#53119)	2021-04-05 08:45:27 -07:00
test_pytorch_common.py	Un-ignore F403 in .flake8 (#55838)	2021-04-13 09:24:07 -07:00
test_pytorch_helper.py
test_pytorch_onnx_caffe2.py	Back out "[ONNX] Process const folding progressively when converts to ONNX (#54569)"	2021-05-14 13:45:49 -07:00
test_pytorch_onnx_caffe2_quantized.py
test_pytorch_onnx_onnxruntime.py	ENH Adds dtype to nn.functional.one_hot (#58090)	2021-05-24 13:48:25 -07:00
test_pytorch_onnx_onnxruntime_cuda.py	Add lint for unqualified noqa (#56272)	2021-04-19 13:16:18 -07:00
test_pytorch_onnx_shape_inference.py	[ONNX] Special post process for onnx::Cast and onnx::ConstantOfShape shape type inference (#55962) (#57597)	2021-05-13 13:42:44 -07:00
test_utility_funs.py	Back out "[ONNX] Process const folding progressively when converts to ONNX (#54569)"	2021-05-14 13:45:49 -07:00
test_verify.py
verify.py	Back out "[ONNX] Process const folding progressively when converts to ONNX (#54569)"	2021-05-14 13:45:49 -07:00