Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
onnx-diagnostic 0.2.1 documentation
Logo
onnx-diagnostic 0.2.1 documentation

Contents

  • API of onnx_diagnostic
    • onnx_diagnostic.export
      • onnx_diagnostic.export.dynamic_shapes
    • onnx_diagnostic.reference
      • onnx_diagnostic.reference.ops
        • onnx_diagnostic.reference.ops.op_add_add_mul_mul
        • onnx_diagnostic.reference.ops.op_average_pool_grad
        • onnx_diagnostic.reference.ops.op_cast_like
        • onnx_diagnostic.reference.ops.op_complex
        • onnx_diagnostic.reference.ops.op_concat
        • onnx_diagnostic.reference.ops.op_constant_of_shape
        • onnx_diagnostic.reference.ops.op_fused_matmul
        • onnx_diagnostic.reference.ops.op_gather_grad
        • onnx_diagnostic.reference.ops.op_memcpy_host
        • onnx_diagnostic.reference.ops.op_mul_sigmoid
        • onnx_diagnostic.reference.ops.op_negxplus1
        • onnx_diagnostic.reference.ops.op_quick_gelu
        • onnx_diagnostic.reference.ops.op_replace_zero
        • onnx_diagnostic.reference.ops.op_rotary
        • onnx_diagnostic.reference.ops.op_qlinear_average_pool
        • onnx_diagnostic.reference.ops.op_qlinear_conv
        • onnx_diagnostic.reference.ops.op_scatter_elements
        • onnx_diagnostic.reference.ops.op_scatternd_of_shape
        • onnx_diagnostic.reference.ops.op_simplified_layer_normalization
        • onnx_diagnostic.reference.ops.op_skip_layer_normalization
        • onnx_diagnostic.reference.ops.op_slice
        • onnx_diagnostic.reference.ops.op_transpose_cast
        • onnx_diagnostic.reference.ops.op_tri_matrix
      • onnx_diagnostic.reference.evaluator
      • onnx_diagnostic.reference.quantized_tensor
      • onnx_diagnostic.reference.ort_evaluator
    • onnx_diagnostic.torch_export_patches
      • onnx_diagnostic.torch_export_patches.patches
        • onnx_diagnostic.torch_export_patches.patches.patch_torch
        • onnx_diagnostic.torch_export_patches.patches.patch_transformers
    • onnx_diagnostic.torch_models
      • onnx_diagnostic.torch_models.llms
    • onnx_diagnostic.args
    • onnx_diagnostic.cache_helpers
    • onnx_diagnostic.ext_test_case
    • onnx_diagnostic.helpers
    • onnx_diagnostic.onnx_tools
    • onnx_diagnostic.ort_session
    • onnx_diagnostic.torch_test_helper
  • Examples Gallery
    • Export Tiny-LLM with patches
    • Export a model with a control flow (If)
    • Export with DynamicCache and dynamic shapes
    • Find where a model is failing by running submodels
    • Intermediate results with (ONNX) ReferenceEvaluator
    • Intermediate results with onnxruntime
    • Steel method forward to guess the dynamic shapes (with Tiny-LLM)
    • Use DYNAMIC or AUTO when exporting if dynamic shapes has constraints

More

  • Change Logs
  • License
Back to top
View this page

onnx_diagnostic.reference.ops.op_complex¶

class onnx_diagnostic.reference.ops.op_complex.ComplexModule(onnx_node: NodeProto, run_params: dict[str, Any], schema: Any = None)[source]¶
class onnx_diagnostic.reference.ops.op_complex.ToComplex(onnx_node: NodeProto, run_params: dict[str, Any], schema: Any = None)[source]¶
Next
onnx_diagnostic.reference.ops.op_concat
Previous
onnx_diagnostic.reference.ops.op_cast_like
Copyright © 2023-2024
Made with Sphinx and @pradyunsg's Furo
On this page
  • onnx_diagnostic.reference.ops.op_complex
    • ComplexModule
    • ToComplex