onnxruntime/onnxruntime/python/tools
KeDengMS e1a54c4090
Symbolic shape inference: fix a bug in shape merge (#5519)
* Symbolic shape inference: fix a bug in shape merge

OpType Where:
input0: ['mt_src_tokens_batch', 1, 1, 'mt_src_tokens_len']
input1: []
input2: ['mt_prev_output_tokens_batch', 12, 'mt_prev_output_tokens_len', 'floor(mt_src_tokens_batch*mt_src_tokens_len/mt_prev_output_tokens_batch)'] 1
output: [None, 12, 'mt_prev_output_tokens_len', None]

* Undo unintended TRT change
2020-10-16 17:54:57 -07:00
..
featurizer_ops Update data_frame_tool to latest (#3919) 2020-05-18 21:13:56 -07:00
quantization Update ORT to 1.5.1 in Bert Quantization Notebook (#5396) 2020-10-08 09:55:01 -07:00
tensorrt/perf [Python] Move symbolic_shape_infer from nuphar to tools (#5162) 2020-09-18 09:31:06 -07:00
transformers Optimize openai-gpt/albert model and add fusion test (#5466) 2020-10-13 19:24:14 -07:00
__init__.py Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
onnxruntime_test.py Improve handling of symbolic dimensions in the onnxruntime_test.py script. (#3959) 2020-05-18 16:51:09 +10:00
symbolic_shape_infer.py Symbolic shape inference: fix a bug in shape merge (#5519) 2020-10-16 17:54:57 -07:00