onnxscript 0.7.2.dev20260819__tar.gz → 0.7.2.dev20260821__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (203) hide show
  1. {onnxscript-0.7.2.dev20260819/onnxscript.egg-info → onnxscript-0.7.2.dev20260821}/PKG-INFO +2 -2
  2. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/fft.py +19 -3
  3. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821/onnxscript.egg-info}/PKG-INFO +2 -2
  4. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/LICENSE +0 -0
  5. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/MANIFEST.in +0 -0
  6. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/README.md +0 -0
  7. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/VERSION +0 -0
  8. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/__init__.py +0 -0
  9. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_framework_apis/__init__.py +0 -0
  10. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_framework_apis/torch_2_11.py +0 -0
  11. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_framework_apis/torch_2_5.py +0 -0
  12. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_framework_apis/torch_2_6.py +0 -0
  13. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_framework_apis/torch_2_7.py +0 -0
  14. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_framework_apis/torch_2_8.py +0 -0
  15. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_framework_apis/torch_2_9.py +0 -0
  16. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/__init__.py +0 -0
  17. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/_inference.py +0 -0
  18. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/_inliner.py +0 -0
  19. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/analysis.py +0 -0
  20. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/ast_utils.py +0 -0
  21. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/autocast.py +0 -0
  22. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/builder.py +0 -0
  23. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/converter.py +0 -0
  24. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/deprecation.py +0 -0
  25. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/evaluator.py +0 -0
  26. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/irbuilder.py +0 -0
  27. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/main.py +0 -0
  28. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/param_manipulation.py +0 -0
  29. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/sourceinfo.py +0 -0
  30. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/tape_builder.py +0 -0
  31. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/type_annotation.py +0 -0
  32. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/utils.py +0 -0
  33. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/values.py +0 -0
  34. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/_internal/version_utils.py +0 -0
  35. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/backend/__init__.py +0 -0
  36. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/backend/onnx_backend.py +0 -0
  37. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/backend/onnx_export.py +0 -0
  38. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/evaluator.py +0 -0
  39. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/tools/torch_lib/deduce_type_constraints.py +0 -0
  40. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/tools/torch_lib/generate_aten_signatures.py +0 -0
  41. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/tools/torch_lib/generate_prims_signatures.py +0 -0
  42. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/__init__.py +0 -0
  43. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/_constants.py +0 -0
  44. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/_flags.py +0 -0
  45. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/graph_building/__init__.py +0 -0
  46. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/__init__.py +0 -0
  47. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/common.py +0 -0
  48. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/core.py +0 -0
  49. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/linalg.py +0 -0
  50. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/nested.py +0 -0
  51. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/nn.py +0 -0
  52. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/prims.py +0 -0
  53. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/quantized_decomposed.py +0 -0
  54. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/sparse.py +0 -0
  55. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/special.py +0 -0
  56. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/vision.py +0 -0
  57. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/registration.py +0 -0
  58. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/tensor_typing.py +0 -0
  59. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/ir/__init__.py +0 -0
  60. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/ir/_schemas.py +0 -0
  61. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/ir/convenience.py +0 -0
  62. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/ir/passes/__init__.py +0 -0
  63. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/ir/passes/common/__init__.py +0 -0
  64. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/nn/__init__.py +0 -0
  65. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/nn/_module.py +0 -0
  66. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/nn/_module_list.py +0 -0
  67. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/nn/_parameter.py +0 -0
  68. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/nn/_sequential.py +0 -0
  69. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/__init__.py +0 -0
  70. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset1.py +0 -0
  71. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset10.py +0 -0
  72. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset11.py +0 -0
  73. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset12.py +0 -0
  74. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset13.py +0 -0
  75. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset14.py +0 -0
  76. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset15.py +0 -0
  77. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset16.py +0 -0
  78. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset17.py +0 -0
  79. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset18.py +0 -0
  80. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset19.py +0 -0
  81. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset2.py +0 -0
  82. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset20.py +0 -0
  83. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset21.py +0 -0
  84. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset22.py +0 -0
  85. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset23.py +0 -0
  86. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset24.py +0 -0
  87. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset25.py +0 -0
  88. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset26.py +0 -0
  89. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset27.py +0 -0
  90. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset3.py +0 -0
  91. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset4.py +0 -0
  92. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset5.py +0 -0
  93. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset6.py +0 -0
  94. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset7.py +0 -0
  95. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset8.py +0 -0
  96. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset9.py +0 -0
  97. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml1.py +0 -0
  98. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml2.py +0 -0
  99. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml3.py +0 -0
  100. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml4.py +0 -0
  101. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml5.py +0 -0
  102. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_opset/_impl/opset_ai_onnx_preview1.py +0 -0
  103. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/onnx_types.py +0 -0
  104. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/optimizer/__init__.py +0 -0
  105. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/optimizer/_constant_folding.py +0 -0
  106. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/optimizer/_optimizer.py +0 -0
  107. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/py.typed +0 -0
  108. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/__init__.py +0 -0
  109. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/_basics.py +0 -0
  110. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/_context.py +0 -0
  111. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/_fusion_utils.py +0 -0
  112. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/_ir_utils.py +0 -0
  113. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/_matcher.py +0 -0
  114. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/_pattern_ir.py +0 -0
  115. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/_rewrite_rule.py +0 -0
  116. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_bart_encoder.py +0 -0
  117. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_phi2lm.py +0 -0
  118. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_phi4lm.py +0 -0
  119. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_rotary_embedding_models.py +0 -0
  120. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_smollm_1.py +0 -0
  121. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_smollm_2.py +0 -0
  122. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_test_models.py +0 -0
  123. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_whisper_decoder.py +0 -0
  124. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/models/_whisper_encoder.py +0 -0
  125. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/onnx_fusions/__init__.py +0 -0
  126. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/onnx_fusions/_onnx_fusions.py +0 -0
  127. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/onnxruntime/__init__.py +0 -0
  128. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/onnxruntime/bfloat16_utils/bfloat16_converter.py +0 -0
  129. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/__init__.py +0 -0
  130. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/_core.py +0 -0
  131. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/_test_utils.py +0 -0
  132. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/attention.py +0 -0
  133. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/bias_gelu.py +0 -0
  134. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/cos_sin_cache.py +0 -0
  135. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/erfgelu.py +0 -0
  136. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/fused_matmul_rule_sets.py +0 -0
  137. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/gelu.py +0 -0
  138. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/gqa.py +0 -0
  139. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/gqa_packed_qkv.py +0 -0
  140. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/group_normalization_merge_silu.py +0 -0
  141. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/instance_to_group_normalization.py +0 -0
  142. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/mha.py +0 -0
  143. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/mha_bias.py +0 -0
  144. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/mha_scale.py +0 -0
  145. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/rms_normalization.py +0 -0
  146. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/rotary_embedding.py +0 -0
  147. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/sdpa.py +0 -0
  148. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/sdpa_via_mha.py +0 -0
  149. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/shape_optimization.py +0 -0
  150. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/skip_normalization.py +0 -0
  151. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/ort_fusions/softmax.py +0 -0
  152. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/pattern.py +0 -0
  153. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/__init__.py +0 -0
  154. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/__init__.py +0 -0
  155. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_basic_rules.py +0 -0
  156. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_broadcast_to_matmul.py +0 -0
  157. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_cast_constant_of_shape.py +0 -0
  158. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_collapse_slices.py +0 -0
  159. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_fuse_batchnorm.py +0 -0
  160. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_fuse_conv_affine.py +0 -0
  161. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_fuse_hardswish.py +0 -0
  162. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_fuse_pad_into_conv.py +0 -0
  163. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_fuse_relus_clips.py +0 -0
  164. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_gemm_to_matmul_add.py +0 -0
  165. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_materialize_reshape_shape.py +0 -0
  166. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_matmul_add_to_gemm.py +0 -0
  167. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_min_max_to_clip.py +0 -0
  168. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_no_op.py +0 -0
  169. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_redundant_scatter_nd.py +0 -0
  170. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_remove_expand_before_binary_op.py +0 -0
  171. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/common/_remove_optional_bias.py +0 -0
  172. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/fusion/__init__.py +0 -0
  173. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/fusion/_gqa.py +0 -0
  174. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/fusion/_layer_norm.py +0 -0
  175. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/fusion/_rms_normalization.py +0 -0
  176. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/rules/fusion/_rotary_embedding.py +0 -0
  177. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/rewriter/testing.py +0 -0
  178. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/tensor.py +0 -0
  179. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/testing/__init__.py +0 -0
  180. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/tools/__init__.py +0 -0
  181. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/tools/memory_peak.py +0 -0
  182. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/tools/transformers_models/__init__.py +0 -0
  183. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/tools/transformers_models/llama.py +0 -0
  184. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/tools/transformers_models/mistral.py +0 -0
  185. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/tools/transformers_models/phi.py +0 -0
  186. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/tools/transformers_models/phi3.py +0 -0
  187. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/utils/__init__.py +0 -0
  188. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/utils/evaluation_utils.py +0 -0
  189. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/utils/metadata_merger.py +0 -0
  190. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/utils/replace.py +0 -0
  191. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/utils/timing_utils.py +0 -0
  192. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/utils/utils.py +0 -0
  193. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/values.py +0 -0
  194. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/version_converter/__init__.py +0 -0
  195. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/version_converter/_c_api_utils.py +0 -0
  196. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/version_converter/_version_converter.py +0 -0
  197. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript.egg-info/SOURCES.txt +0 -0
  198. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript.egg-info/dependency_links.txt +0 -0
  199. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript.egg-info/requires.txt +0 -0
  200. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript.egg-info/top_level.txt +0 -0
  201. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/pyproject.toml +0 -0
  202. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/setup.cfg +0 -0
  203. {onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/setup.py +0 -0
{onnxscript-0.7.2.dev20260819/onnxscript.egg-info → onnxscript-0.7.2.dev20260821}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: onnxscript
3
- Version: 0.7.2.dev20260819
3
+ Version: 0.7.2.dev20260821
4
4
  Summary: Naturally author ONNX functions and models using a subset of Python
5
5
  Author-email: Microsoft Corporation <onnx@microsoft.com>
6
6
  License: MIT License
@@ -27,7 +27,7 @@ License: MIT License
27
27
 
28
28
  Project-URL: Homepage, https://microsoft.github.io/onnxscript/
29
29
  Project-URL: Repository, https://github.com/microsoft/onnxscript
30
- Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/59c720fb32b0d3b6de624c1525a68e02d5c35441
30
+ Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/a00b0e0520f18875081d657e0b3e137107ca1c08
31
31
  Classifier: Development Status :: 4 - Beta
32
32
  Classifier: Environment :: Console
33
33
  Classifier: Intended Audience :: Developers
{onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/fft.py RENAMED
@@ -172,17 +172,33 @@ def aten__fft_r2c(
172
172
  transformed = op.Unsqueeze(self, axes=[-1])
173
173
 
174
174
  for idx, dimension in enumerate(reversed(dim)):
175
+ dft_length = op.Squeeze(
176
+ op.Shape(transformed, start=dimension, end=dimension + 1),
177
+ axes=[0],
178
+ )
175
179
  transformed = _fftn_onnx_normalization(
176
180
  transformed,
177
181
  normalization,
178
- op.Shape(transformed, start=dimension, end=dimension + 1),
182
+ dft_length,
179
183
  inverse=False,
180
184
  )
181
185
  if idx > 0:
182
- transformed = op.DFT(transformed, axis=dimension, inverse=False, onesided=False)
186
+ transformed = op.DFT(
187
+ transformed,
188
+ dft_length=dft_length,
189
+ axis=dimension,
190
+ inverse=False,
191
+ onesided=False,
192
+ )
183
193
  else:
184
194
  # Torch computes one-sided FFT on the last dimension only.
185
- transformed = op.DFT(transformed, axis=dimension, inverse=False, onesided=onesided)
195
+ transformed = op.DFT(
196
+ transformed,
197
+ dft_length=dft_length,
198
+ axis=dimension,
199
+ inverse=False,
200
+ onesided=onesided,
201
+ )
186
202
 
187
203
  if unsqueeze_first_dim:
188
204
  transformed = op.Squeeze(transformed, axes=[0])
{onnxscript-0.7.2.dev20260819 → onnxscript-0.7.2.dev20260821/onnxscript.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: onnxscript
3
- Version: 0.7.2.dev20260819
3
+ Version: 0.7.2.dev20260821
4
4
  Summary: Naturally author ONNX functions and models using a subset of Python
5
5
  Author-email: Microsoft Corporation <onnx@microsoft.com>
6
6
  License: MIT License
@@ -27,7 +27,7 @@ License: MIT License
27
27
 
28
28
  Project-URL: Homepage, https://microsoft.github.io/onnxscript/
29
29
  Project-URL: Repository, https://github.com/microsoft/onnxscript
30
- Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/59c720fb32b0d3b6de624c1525a68e02d5c35441
30
+ Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/a00b0e0520f18875081d657e0b3e137107ca1c08
31
31
  Classifier: Development Status :: 4 - Beta
32
32
  Classifier: Environment :: Console
33
33
  Classifier: Intended Audience :: Developers