Index _ | A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | X | Y | Z _ __eq__() (ml_switcheroo.analysis.symbol_table.ModuleType method) (ml_switcheroo.analysis.symbol_table.SymbolType method) (ml_switcheroo.analysis.symbol_table.TensorType method) (ml_switcheroo.analysis.symbol_table.UnionType method) __str__() (ml_switcheroo.analysis.symbol_table.SymbolType method) (ml_switcheroo.analysis.symbol_table.UnionType method) (ml_switcheroo.compiler.frontends.rdna.Comment method) (ml_switcheroo.compiler.frontends.rdna.Directive method) (ml_switcheroo.compiler.frontends.rdna.Immediate method) (ml_switcheroo.compiler.frontends.rdna.Instruction method) (ml_switcheroo.compiler.frontends.rdna.Label method) (ml_switcheroo.compiler.frontends.rdna.LabelRef method) (ml_switcheroo.compiler.frontends.rdna.Memory method) (ml_switcheroo.compiler.frontends.rdna.Modifier method) (ml_switcheroo.compiler.frontends.rdna.nodes.Comment method) (ml_switcheroo.compiler.frontends.rdna.nodes.Directive method) (ml_switcheroo.compiler.frontends.rdna.nodes.Immediate method) (ml_switcheroo.compiler.frontends.rdna.nodes.Instruction method) (ml_switcheroo.compiler.frontends.rdna.nodes.Label method) (ml_switcheroo.compiler.frontends.rdna.nodes.LabelRef method) (ml_switcheroo.compiler.frontends.rdna.nodes.Memory method) (ml_switcheroo.compiler.frontends.rdna.nodes.Modifier method) (ml_switcheroo.compiler.frontends.rdna.nodes.Operand method) (ml_switcheroo.compiler.frontends.rdna.nodes.RdnaNode method) (ml_switcheroo.compiler.frontends.rdna.nodes.SGPR method) (ml_switcheroo.compiler.frontends.rdna.nodes.VGPR method) (ml_switcheroo.compiler.frontends.rdna.Operand method) (ml_switcheroo.compiler.frontends.rdna.RdnaNode method) (ml_switcheroo.compiler.frontends.rdna.SGPR method) (ml_switcheroo.compiler.frontends.rdna.VGPR method) (ml_switcheroo.compiler.frontends.sass.Comment method) (ml_switcheroo.compiler.frontends.sass.Directive method) (ml_switcheroo.compiler.frontends.sass.Immediate method) (ml_switcheroo.compiler.frontends.sass.Instruction method) (ml_switcheroo.compiler.frontends.sass.Label method) (ml_switcheroo.compiler.frontends.sass.Memory method) (ml_switcheroo.compiler.frontends.sass.nodes.Comment method) (ml_switcheroo.compiler.frontends.sass.nodes.Directive method) (ml_switcheroo.compiler.frontends.sass.nodes.Immediate method) (ml_switcheroo.compiler.frontends.sass.nodes.Instruction method) (ml_switcheroo.compiler.frontends.sass.nodes.Label method) (ml_switcheroo.compiler.frontends.sass.nodes.Memory method) (ml_switcheroo.compiler.frontends.sass.nodes.Predicate method) (ml_switcheroo.compiler.frontends.sass.nodes.Register method) (ml_switcheroo.compiler.frontends.sass.nodes.SassNode method) (ml_switcheroo.compiler.frontends.sass.parser.LabelRef method) (ml_switcheroo.compiler.frontends.sass.Predicate method) (ml_switcheroo.compiler.frontends.sass.Register method) (ml_switcheroo.compiler.frontends.sass.SassNode method) __test__ (ml_switcheroo.generated_tests.generator.TestCaseGenerator attribute) __version__ (in module ml_switcheroo) _current_violations (ml_switcheroo.analysis.purity.PurityScanner attribute) _dynamic_impurity_methods (ml_switcheroo.analysis.purity.PurityScanner attribute) _global_rng_methods (ml_switcheroo.analysis.purity.PurityScanner attribute) _IO_FUNCTIONS (ml_switcheroo.analysis.purity.PurityScanner attribute) _MUTATION_METHODS (ml_switcheroo.analysis.purity.PurityScanner attribute) _package_cache (ml_switcheroo.discovery.ApiInspector attribute) (ml_switcheroo.discovery.inspector.ApiInspector attribute) A absolute (ml_switcheroo.compiler.frontends.sass.nodes.Register attribute), [1] (ml_switcheroo.compiler.frontends.sass.Register attribute), [1] ACTIVATION (ml_switcheroo.frameworks.base.StandardCategory attribute) adapt_to_framework() (ml_switcheroo.testing.fuzzer.core.InputFuzzer method) (ml_switcheroo.testing.fuzzer.InputFuzzer method) adapter (ml_switcheroo.core.discovery.SimulatedReflection attribute) add_variant() (ml_switcheroo.discovery.consensus.CandidateStandard method) adjust_shape_rank() (in module ml_switcheroo.testing.fuzzer.utils) alias (ml_switcheroo.core.dsl.ImportReq attribute) (ml_switcheroo.core.import_fixer.ImportReq attribute) (ml_switcheroo.core.import_fixer.resolution.ImportReq attribute) (ml_switcheroo.semantics.schema.FrameworkTraits attribute) alias_map (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) aliases (ml_switcheroo.discovery.harvester.ImportScanner attribute) (ml_switcheroo.discovery.harvester.TargetCallVisitor attribute) align (ml_switcheroo.core.tikz.nodes.TikzTable attribute) (ml_switcheroo.core.tikz.TikzTable attribute) align_signatures() (ml_switcheroo.discovery.consensus.ConsensusEngine method) (ml_switcheroo.discovery.ConsensusEngine method) allocate_scalar_temp() (ml_switcheroo.compiler.backends.rdna.macros.RegisterAllocatorProtocol method) (ml_switcheroo.compiler.backends.rdna.synthesizer.RegisterAllocator method) allocate_ssa() (ml_switcheroo.core.mlir.emitter.SSAContext method) allocate_temp() (ml_switcheroo.compiler.backends.sass.macros.RegisterAllocatorProtocol method) (ml_switcheroo.compiler.backends.sass.synthesizer.RegisterAllocator method) allocate_vector_temp() (ml_switcheroo.compiler.backends.rdna.macros.RegisterAllocatorProtocol method) (ml_switcheroo.compiler.backends.rdna.synthesizer.RegisterAllocator method) allocator (ml_switcheroo.compiler.backends.rdna.RdnaSynthesizer attribute) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaSynthesizer attribute) (ml_switcheroo.compiler.backends.sass.SassSynthesizer attribute) (ml_switcheroo.compiler.backends.sass.synthesizer.SassSynthesizer attribute) allow_partial (ml_switcheroo.core.dsl.PatternDef attribute) allowed_roots (ml_switcheroo.analysis.audit.CoverageScanner attribute) ANALYSIS_WARNING (ml_switcheroo.core.tracer.TraceEventType attribute) analyze_block() (ml_switcheroo.compiler.frontends.rdna.analysis.RdnaAnalyzer static method) (ml_switcheroo.compiler.frontends.rdna.RdnaAnalyzer static method) (ml_switcheroo.compiler.frontends.sass.analysis.SassAnalyzer static method) (ml_switcheroo.compiler.frontends.sass.SassAnalyzer static method) annotation (ml_switcheroo.core.ghost.GhostParam attribute) api (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) api_path (ml_switcheroo.core.ghost.GhostRef attribute) ApiInspector (class in ml_switcheroo.discovery) (class in ml_switcheroo.discovery.inspector) ApiPass (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.passes) (class in ml_switcheroo.core.rewriter.passes.api) ApiTransformer (class in ml_switcheroo.core.rewriter.passes) (class in ml_switcheroo.core.rewriter.passes.api) apply_index_select() (in module transformers) apply_strict_guards() (in module guards) apply_wiring() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) ARG_ALIASES (ml_switcheroo.discovery.consensus.ConsensusEngine attribute) (ml_switcheroo.discovery.ConsensusEngine attribute) arg_mappings (ml_switcheroo.discovery.consensus.CandidateStandard attribute) arg_values (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) ArgInjectorType (in module ml_switcheroo.core.hooks) args (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.core.latex.ComputeNode attribute) (ml_switcheroo.core.latex.nodes.ComputeNode attribute) (ml_switcheroo.core.latex.nodes.StateOpNode attribute) (ml_switcheroo.core.latex.StateOpNode attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) arguments (ml_switcheroo.core.mlir.nodes.BlockNode attribute) arity_penalty (ml_switcheroo.discovery.Scaffolder attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) ARRAY_API (ml_switcheroo.enums.SemanticTier attribute) ArrayApiSpecImporter (class in ml_switcheroo.importers.array_api_reader) ARROW (ml_switcheroo.core.mlir.tokens.TokenKind attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) arrows (ml_switcheroo.core.html.GridBox attribute) (ml_switcheroo.core.html.nodes.GridBox attribute) AST_MUTATION (ml_switcheroo.core.tracer.TraceEventType attribute) AST_SNAPSHOT (ml_switcheroo.core.tracer.TraceEventType attribute) ASTEngine (class in ml_switcheroo.core.engine) attribute_id (ml_switcheroo.core.latex.nodes.StateOpNode attribute) (ml_switcheroo.core.latex.StateOpNode attribute) AttributeMixin (class in ml_switcheroo.core.import_fixer.attributes_mixin) AttributeNode (class in ml_switcheroo.core.mlir.nodes) attributes (ml_switcheroo.core.mlir.nodes.OperationNode attribute) attrs (ml_switcheroo.core.html.parser.GridParser attribute) auto_strip_magic_args (ml_switcheroo.semantics.schema.StructuralTraits attribute) auto_wire (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) AutoWireSpec (class in ml_switcheroo.core.hooks) AuxiliaryPass (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.passes) (class in ml_switcheroo.core.rewriter.passes.auxiliary) AuxiliaryTransformer (class in ml_switcheroo.core.rewriter.passes) (class in ml_switcheroo.core.rewriter.passes.auxiliary) available_frameworks() (in module ml_switcheroo.frameworks) (in module ml_switcheroo.frameworks.base) B base (ml_switcheroo.compiler.frontends.rdna.Memory attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Memory attribute), [1] (ml_switcheroo.compiler.frontends.sass.Memory attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Memory attribute), [1] BaseFrontend (class in ml_switcheroo.compiler.registry) BaseGeneratorMixin (class in ml_switcheroo.core.mlir.gen_base) BaseImportFixer (class in ml_switcheroo.core.import_fixer.base) BatchValidator (class in ml_switcheroo.testing.batch_runner) BLOCK (ml_switcheroo.core.dsl.PluginType attribute) BLOCK_LABEL (ml_switcheroo.core.mlir.tokens.TokenKind attribute) BlockNode (class in ml_switcheroo.core.mlir.nodes) blocks (ml_switcheroo.core.mlir.nodes.RegionNode attribute) body (ml_switcheroo.core.mlir.nodes.ModuleNode attribute) (ml_switcheroo.core.mlir.nodes.TypeNode attribute) body_node (injector_plugin.BodyExtractor attribute) body_text (ml_switcheroo.core.html.GridBox attribute) (ml_switcheroo.core.html.nodes.GridBox attribute) BodyExtractor (class in injector_plugin) build() (ml_switcheroo.utils.doc_context.DocContextBuilder method) build_strategies() (ml_switcheroo.testing.fuzzer.core.InputFuzzer method) (ml_switcheroo.testing.fuzzer.InputFuzzer method) C c_SGPR() (in module ml_switcheroo.compiler.frontends.rdna) (in module ml_switcheroo.compiler.frontends.rdna.nodes) c_VGPR() (in module ml_switcheroo.compiler.frontends.rdna) (in module ml_switcheroo.compiler.frontends.rdna.nodes) CALL (ml_switcheroo.core.dsl.PluginType attribute) CandidateStandard (class in ml_switcheroo.discovery.consensus) capture_eval_state() (in module ml_switcheroo.plugins.state_flag_injection) capture_node_source() (in module ml_switcheroo.utils.node_diff) casts (ml_switcheroo.core.dsl.FrameworkVariant attribute) cell_buffer (ml_switcheroo.importers.sass_reader.SassHtmlParser attribute) check() (ml_switcheroo.testing.linter.StructuralLinter method) check_version_constraints() (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) children (ml_switcheroo.core.html.HtmlDocument attribute) (ml_switcheroo.core.html.nodes.HtmlDocument attribute) (ml_switcheroo.core.latex.ModelContainer attribute) (ml_switcheroo.core.latex.nodes.ModelContainer attribute) (ml_switcheroo.core.tikz.nodes.TikzGraph attribute) (ml_switcheroo.core.tikz.TikzGraph attribute) ClassBodyReplacer (class in ml_switcheroo.compiler.backends.python) clear_definition_cache() (in module ml_switcheroo.frameworks.loader) clear_hooks() (in module ml_switcheroo.core.hooks) cluster() (ml_switcheroo.discovery.consensus.ConsensusEngine method) (ml_switcheroo.discovery.ConsensusEngine method) code (ml_switcheroo.compiler.frontends.python.PythonFrontend attribute) (ml_switcheroo.core.conversion_result.ConversionResult attribute) code_text (ml_switcheroo.core.html.GridBox attribute) (ml_switcheroo.core.html.nodes.GridBox attribute) CodeExtractor (class in ml_switcheroo.utils.code_extractor) col (ml_switcheroo.core.html.GridBox attribute) (ml_switcheroo.core.html.nodes.GridBox attribute) (ml_switcheroo.core.mlir.parser.Token attribute) (ml_switcheroo.core.tikz.parser.TikzLexer attribute) (ml_switcheroo.core.tikz.parser.Token attribute) collect_api() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) COLON (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.core.mlir.tokens.Symbol attribute) COLORS (ml_switcheroo.utils.visualizer.MermaidGenerator attribute) column (ml_switcheroo.compiler.frontends.rdna.Token attribute), [1] (ml_switcheroo.compiler.frontends.rdna.tokens.Token attribute), [1] (ml_switcheroo.compiler.frontends.sass.Token attribute), [1] (ml_switcheroo.compiler.frontends.sass.tokens.Token attribute), [1] COMMA (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) (ml_switcheroo.core.mlir.tokens.Symbol attribute) COMMAND (ml_switcheroo.core.tikz.parser.TokenKind attribute) Comment (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) COMMENT (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) (ml_switcheroo.core.mlir.tokens.TokenKind attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) CompatibilityMatrix (class in ml_switcheroo.cli.matrix) compile() (ml_switcheroo.compiler.backend.CompilerBackend method) (ml_switcheroo.compiler.backends.extras.HtmlBackend method) (ml_switcheroo.compiler.backends.extras.LatexBackend method) (ml_switcheroo.compiler.backends.extras.MlirBackend method) (ml_switcheroo.compiler.backends.extras.StableHloBackend method) (ml_switcheroo.compiler.backends.extras.TikzBackend method) (ml_switcheroo.compiler.backends.python.PythonBackend method) (ml_switcheroo.compiler.backends.PythonBackend method) (ml_switcheroo.compiler.backends.rdna.RdnaBackend method) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaBackend method) (ml_switcheroo.compiler.backends.sass.SassBackend method) (ml_switcheroo.compiler.backends.sass.synthesizer.SassBackend method) (ml_switcheroo.compiler.CompilerBackend method) CompilerBackend (class in ml_switcheroo.compiler) (class in ml_switcheroo.compiler.backend) complexity (ml_switcheroo.core.dsl.OperationDef attribute) compute_permutation() (in module utils) ComputeNode (class in ml_switcheroo.core.latex) (class in ml_switcheroo.core.latex.nodes) config (ml_switcheroo.core.latex.MemoryNode attribute) (ml_switcheroo.core.latex.nodes.MemoryNode attribute) (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer property) (ml_switcheroo.core.rewriter.passes.ApiTransformer property) (ml_switcheroo.core.rewriter.RewriterContext attribute) connector (ml_switcheroo.core.tikz.nodes.TikzEdge attribute) (ml_switcheroo.core.tikz.TikzEdge attribute) ConsensusEngine (class in ml_switcheroo.discovery) (class in ml_switcheroo.discovery.consensus) console (in module ml_switcheroo.utils.console) (ml_switcheroo.cli.matrix.CompatibilityMatrix attribute) (ml_switcheroo.cli.wizard.MappingWizard attribute) (ml_switcheroo.discovery.FrameworkSyncer attribute) (ml_switcheroo.discovery.Scaffolder attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) (ml_switcheroo.discovery.syncer.FrameworkSyncer attribute) consume() (ml_switcheroo.core.mlir.parser.MlirParser method) ContainerType (class in ml_switcheroo.core.dsl) content (ml_switcheroo.core.mlir.nodes.TriviaNode attribute) (ml_switcheroo.core.tikz.nodes.TikzNode attribute) (ml_switcheroo.core.tikz.nodes.TriviaNode attribute) (ml_switcheroo.core.tikz.TikzNode attribute) (ml_switcheroo.core.tikz.TriviaNode attribute) CONTEXT (ml_switcheroo.core.dsl.OpType attribute) context (ml_switcheroo.core.rewriter.passes.api.ApiTransformer attribute) (ml_switcheroo.core.rewriter.passes.ApiTransformer attribute) (ml_switcheroo.core.rewriter.passes.auxiliary.AuxiliaryTransformer attribute) (ml_switcheroo.core.rewriter.passes.AuxiliaryTransformer attribute) (ml_switcheroo.core.rewriter.passes.StructuralTransformer attribute) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer attribute) ConversionResult (class in ml_switcheroo.core.conversion_result) convert() (in module ml_switcheroo) (ml_switcheroo.core.mlir.emitter.PythonToMlirEmitter method) (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) convert_load_state_dict() (in module ml_switcheroo.plugins.state_container) convert_parameters() (in module ml_switcheroo.plugins.state_container) convert_register_buffer() (in module ml_switcheroo.plugins.state_container) convert_register_parameter() (in module ml_switcheroo.plugins.state_container) convert_state_dict() (in module ml_switcheroo.plugins.state_container) convert_to_cst_literal() (in module injector_fw.utils) convert_value_to_cst() (in module ml_switcheroo.core.rewriter.normalization_utils) count (ml_switcheroo.compiler.frontends.rdna.nodes.SGPR attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.VGPR attribute), [1] (ml_switcheroo.compiler.frontends.rdna.SGPR attribute), [1] (ml_switcheroo.compiler.frontends.rdna.VGPR attribute), [1] CoverageScanner (class in ml_switcheroo.analysis.audit) create_dotted_name() (in module ml_switcheroo.core.import_fixer.utils) (in module ml_switcheroo.plugins.utils) create_parser() (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) css_class (ml_switcheroo.core.html.GridBox attribute) (ml_switcheroo.core.html.nodes.GridBox attribute) ctx (ml_switcheroo.core.mlir.emitter.PythonToMlirEmitter attribute) (ml_switcheroo.core.mlir.gen_statements.StatementGeneratorMixin attribute) (ml_switcheroo.core.mlir.generator.MlirToPythonGenerator attribute) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer property) (ml_switcheroo.core.rewriter.passes.ApiTransformer property) current_class (ml_switcheroo.core.html.parser.GridParser attribute) current_op_id (ml_switcheroo.core.hooks.HookContext attribute) current_row_cells (ml_switcheroo.importers.sass_reader.SassHtmlParser attribute) current_scope (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer attribute) current_stmt_errors (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) current_stmt_warnings (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) current_variant (ml_switcheroo.core.hooks.HookContext property) D data (ml_switcheroo.semantics.manager.SemanticsManager attribute) declare() (ml_switcheroo.core.mlir.emitter.SSAContext method) declared_magic_args (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) DECORATOR (ml_switcheroo.core.dsl.OpType attribute) default (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.ghost.GhostParam attribute) DEFAULT_TEST_TEMPLATES (in module ml_switcheroo.generated_tests.templates) deferred_exprs (ml_switcheroo.core.mlir.generator.MlirToPythonGenerator attribute) definitions (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) DEFINITIONS_DIR (in module ml_switcheroo.frameworks.loader) DeleteAction (class in ml_switcheroo.compiler.differ) (class in ml_switcheroo.core.rewriter.patcher) DependencyScanner (class in ml_switcheroo.analysis.dependencies) deprecated (ml_switcheroo.core.dsl.OperationDef attribute) description (ml_switcheroo.core.dsl.OperationDef attribute) (ml_switcheroo.core.tracer.TraceEvent attribute) device_constraints (ml_switcheroo.core.dsl.OperationDef attribute) DialectRegistry (class in ml_switcheroo.core.mlir.dialect) diff() (ml_switcheroo.compiler.differ.GraphDiffer method) diff_nodes() (in module ml_switcheroo.utils.node_diff) differentiable (ml_switcheroo.core.dsl.OperationDef attribute) Directive (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) DIRECTIVE (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) discover() (ml_switcheroo.core.discovery.SimulatedReflection method) DISCOVERED_FILENAMES (in module ml_switcheroo.semantics.file_loader) discovery_heuristics (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) dispatch module dispatch_rules (ml_switcheroo.core.dsl.FrameworkVariant attribute) display_name (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter attribute) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter attribute) (ml_switcheroo.frameworks.jax.JaxCoreAdapter attribute) (ml_switcheroo.frameworks.keras.KerasAdapter attribute) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter attribute) (ml_switcheroo.frameworks.mlir.MlirAdapter attribute) (ml_switcheroo.frameworks.mlx.MLXAdapter attribute) (ml_switcheroo.frameworks.numpy.NumpyAdapter attribute) (ml_switcheroo.frameworks.paxml.PaxmlAdapter attribute) (ml_switcheroo.frameworks.rdna.RdnaAdapter attribute) (ml_switcheroo.frameworks.sass.SassAdapter attribute) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter attribute) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter attribute) (ml_switcheroo.frameworks.tikz.TikzAdapter attribute) (ml_switcheroo.frameworks.torch.TorchAdapter attribute) doc (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) DocContextBuilder (class in ml_switcheroo.utils.doc_context) docstring (ml_switcheroo.core.ghost.GhostRef attribute) documentation_url (ml_switcheroo.core.dsl.OperationDef attribute) dtype (ml_switcheroo.core.dsl.ParameterDef attribute) E edges (ml_switcheroo.compiler.ir.LogicalGraph attribute) (ml_switcheroo.compiler.LogicalGraph attribute) (ml_switcheroo.core.graph.LogicalGraph attribute) (ml_switcheroo.utils.visualizer.MermaidGenerator attribute) effective_source (ml_switcheroo.config.RuntimeConfig property) effective_target (ml_switcheroo.config.RuntimeConfig property) emit() (ml_switcheroo.compiler.backends.rdna.emitter.RdnaEmitter method) (ml_switcheroo.compiler.backends.rdna.RdnaEmitter method) (ml_switcheroo.compiler.backends.sass.emitter.SassEmitter method) (ml_switcheroo.compiler.backends.sass.SassEmitter method) emit_call() (ml_switcheroo.compiler.backends.python_snippet.PythonSnippetEmitter method) emit_expression() (ml_switcheroo.compiler.backends.python_snippet.PythonSnippetEmitter method) emit_init() (ml_switcheroo.compiler.backends.python_snippet.PythonSnippetEmitter method) emitter (ml_switcheroo.compiler.backends.rdna.RdnaBackend attribute) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaBackend attribute) (ml_switcheroo.compiler.backends.sass.SassBackend attribute) (ml_switcheroo.compiler.backends.sass.synthesizer.SassBackend attribute) (ml_switcheroo.core.rewriter.patcher.GraphPatcher attribute) enable_graph_optimization (ml_switcheroo.config.RuntimeConfig attribute) enable_import_fixer (ml_switcheroo.config.RuntimeConfig attribute) END_MARKER (ml_switcheroo.core.escape_hatch.EscapeHatch attribute) end_phase() (ml_switcheroo.core.tracer.TraceLogger method) enforce_purity_analysis (ml_switcheroo.semantics.schema.PluginTraits attribute) ensure_determinism() (in module ml_switcheroo.generated_tests.runtime) ensure_runtime_module() (in module ml_switcheroo.generated_tests.runtime_builder) enter_scope() (ml_switcheroo.core.mlir.emitter.SSAContext method) EOF (ml_switcheroo.core.mlir.tokens.TokenKind attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) EQ (ml_switcheroo.enums.LogicOp attribute) EQUAL (ml_switcheroo.core.mlir.tokens.Symbol attribute) EquivalenceRunner (class in ml_switcheroo.testing.runner) errors (ml_switcheroo.core.conversion_result.ConversionResult attribute) EscapeHatch (class in ml_switcheroo.core.escape_hatch) evaluate_dispatch_rules() (in module dispatch) execute_strategy() (in module strategy) existing_args (ml_switcheroo.core.rewriter.types.SignatureContext attribute) exit_scope() (ml_switcheroo.core.mlir.emitter.SSAContext method) expand_conv2d() (in module ml_switcheroo.compiler.backends.rdna.macros) (in module ml_switcheroo.compiler.backends.sass.macros) expand_linear() (in module ml_switcheroo.compiler.backends.rdna.macros) (in module ml_switcheroo.compiler.backends.sass.macros) expect() (ml_switcheroo.core.mlir.parser.MlirParser method) export() (ml_switcheroo.core.tracer.TraceLogger method) ExpressionGeneratorMixin (class in ml_switcheroo.core.mlir.gen_expressions) extends (ml_switcheroo.semantics.schema.FrameworkTraits attribute) extract_class() (ml_switcheroo.utils.code_extractor.CodeExtractor static method) extract_primitive_key() (in module ml_switcheroo.core.rewriter.normalization_utils) extracted_ops (ml_switcheroo.importers.sass_reader.SassHtmlParser attribute) extractor (ml_switcheroo.testing.harness_generator.HarnessGenerator attribute) EXTRAS (ml_switcheroo.enums.SemanticTier attribute) F files (in module ml_switcheroo.semantics.paths) filter_common() (ml_switcheroo.discovery.consensus.ConsensusEngine method) (ml_switcheroo.discovery.ConsensusEngine method) flax_nnx (in module ml_switcheroo.frameworks.flax_nnx) FlaxNNXAdapter (class in ml_switcheroo.frameworks.flax_nnx) forbidden_roots (ml_switcheroo.testing.linter.StructuralLinter attribute) forward_method (ml_switcheroo.semantics.schema.StructuralTraits attribute) found (injector_fw.core.FrameworkInjector attribute) (injector_plugin.BodyExtractor attribute) (injector_spec.StandardsInjector attribute) (ml_switcheroo.compiler.backends.python.ClassBodyReplacer attribute) (ml_switcheroo.core.scanners.SimpleNameScanner attribute) found_usages (ml_switcheroo.core.scanners.UsageScanner attribute) framework (ml_switcheroo.analysis.symbol_table.TensorType attribute) (ml_switcheroo.compiler.backends.python.PythonBackend attribute) (ml_switcheroo.compiler.backends.python_snippet.PythonSnippetEmitter attribute) (ml_switcheroo.compiler.backends.PythonBackend attribute) (ml_switcheroo.core.discovery.SimulatedReflection attribute) framework_configs (ml_switcheroo.semantics.manager.SemanticsManager attribute) FrameworkAdapter (class in ml_switcheroo.frameworks) (class in ml_switcheroo.frameworks.base) FrameworkAlias (class in ml_switcheroo.semantics.schema) FrameworkInjector (class in injector_fw.core) frameworks (ml_switcheroo.semantics.schema.SemanticsFile attribute) FrameworkSyncer (class in ml_switcheroo.discovery) (class in ml_switcheroo.discovery.syncer) FrameworkTraits (class in ml_switcheroo.semantics.schema) FrameworkVariant (class in ml_switcheroo.core.dsl) from_graph() (ml_switcheroo.compiler.backends.rdna.RdnaSynthesizer method) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaSynthesizer method) (ml_switcheroo.compiler.backends.sass.SassSynthesizer method) (ml_switcheroo.compiler.backends.sass.synthesizer.SassSynthesizer method) func_name (injector_plugin.BodyExtractor attribute) FUNCTION (ml_switcheroo.core.dsl.OpType attribute) functional_execution_method (ml_switcheroo.semantics.schema.StructuralTraits attribute) fuzzer (ml_switcheroo.testing.runner.EquivalenceRunner attribute) G GAP_HEIGHT (ml_switcheroo.compiler.backends.extras.HtmlBackend attribute) generate() (injector_plugin.PluginGenerator method) (ml_switcheroo.cli.handlers.convert_weights.WeightScriptGenerator method) (ml_switcheroo.compiler.backends.python.PythonBackend method) (ml_switcheroo.compiler.backends.PythonBackend method) (ml_switcheroo.core.mlir.generator.MlirToPythonGenerator method) (ml_switcheroo.generated_tests.generator.TestCaseGenerator method) (ml_switcheroo.testing.harness_generator.HarnessGenerator method) (ml_switcheroo.utils.doc_gen.MigrationGuideGenerator method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) generate_array() (in module ml_switcheroo.testing.fuzzer.generators) generate_by_heuristic() (in module ml_switcheroo.testing.fuzzer.heuristics) generate_fake_callable() (in module ml_switcheroo.testing.fuzzer.generators) generate_from_hint() (in module ml_switcheroo.testing.fuzzer.parser) generate_input_value_code() (in module ml_switcheroo.generated_tests.inputs) generate_scalar_float() (in module ml_switcheroo.testing.fuzzer.generators) generate_scalar_int() (in module ml_switcheroo.testing.fuzzer.generators) get() (ml_switcheroo.analysis.symbol_table.Scope method) get_abstract_op() (ml_switcheroo.core.mlir.dialect.DialectRegistry class method) get_adapter() (in module ml_switcheroo.frameworks) (in module ml_switcheroo.frameworks.base) get_all_hook_metadata() (in module ml_switcheroo.core.hooks) get_all_rng_methods() (ml_switcheroo.semantics.manager.SemanticsManager method) get_backend_class() (in module ml_switcheroo.compiler.registry) get_console() (in module ml_switcheroo.utils.console) get_definition() (ml_switcheroo.semantics.manager.SemanticsManager method) get_definition_by_id() (ml_switcheroo.semantics.manager.SemanticsManager method) get_definitions_path() (in module ml_switcheroo.frameworks.loader) get_device_check_syntax() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_device_syntax() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_doc_url() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_example_code() (ml_switcheroo.frameworks.mlir.MlirAdapter class method) get_fallback_base_value() (in module ml_switcheroo.testing.fuzzer.parser) get_framework_aliases() (ml_switcheroo.semantics.manager.SemanticsManager method) get_framework_config() (ml_switcheroo.semantics.manager.SemanticsManager method) get_framework_priority_order() (in module ml_switcheroo.config) get_full_name() (in module ml_switcheroo.core.scanners) get_harness_init_code() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_hook() (in module ml_switcheroo.core.hooks) get_import_map() (ml_switcheroo.semantics.manager.SemanticsManager method) get_import_root() (in module injector_fw.utils) get_json() (ml_switcheroo.cli.matrix.CompatibilityMatrix method) get_known_apis() (ml_switcheroo.semantics.manager.SemanticsManager method) get_patterns() (ml_switcheroo.semantics.manager.SemanticsManager method) get_random_shape() (in module ml_switcheroo.testing.fuzzer.generators) get_register() (ml_switcheroo.compiler.backends.sass.macros.RegisterAllocatorProtocol method) (ml_switcheroo.compiler.backends.sass.synthesizer.RegisterAllocator method) get_result() (ml_switcheroo.core.scanners.UsageScanner method) get_rng_split_syntax() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_root_name() (in module ml_switcheroo.core.import_fixer.utils) get_scalar_register() (ml_switcheroo.compiler.backends.rdna.macros.RegisterAllocatorProtocol method) (ml_switcheroo.compiler.backends.rdna.synthesizer.RegisterAllocator method) get_serialization_imports() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_serialization_syntax() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_shim_code() (in module ml_switcheroo.plugins.data_loader) get_signature() (in module ml_switcheroo.core.import_fixer.utils) get_template() (in module ml_switcheroo.generated_tests.templates) get_tensor_to_numpy_expr() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_test_template() (ml_switcheroo.semantics.manager.SemanticsManager method) get_tiered_examples() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.jax.JaxCoreAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.paxml.PaxmlAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_to_numpy_code() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_tracer() (in module ml_switcheroo.core.tracer) get_type() (ml_switcheroo.analysis.symbol_table.SymbolTable method) get_vector_register() (ml_switcheroo.compiler.backends.rdna.macros.RegisterAllocatorProtocol method) (ml_switcheroo.compiler.backends.rdna.synthesizer.RegisterAllocator method) get_weight_conversion_imports() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_weight_load_code() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) get_weight_save_code() (ml_switcheroo.frameworks.base.FrameworkAdapter method) (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin method) (ml_switcheroo.frameworks.FrameworkAdapter method) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter method) (ml_switcheroo.frameworks.keras.KerasAdapter method) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter method) (ml_switcheroo.frameworks.mlir.MlirAdapter method) (ml_switcheroo.frameworks.mlx.MLXAdapter method) (ml_switcheroo.frameworks.numpy.NumpyAdapter method) (ml_switcheroo.frameworks.rdna.RdnaAdapter method) (ml_switcheroo.frameworks.sass.SassAdapter method) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter method) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter method) (ml_switcheroo.frameworks.tikz.TikzAdapter method) (ml_switcheroo.frameworks.torch.TorchAdapter method) GHOST (ml_switcheroo.frameworks.base.InitMode attribute) GhostInspector (class in ml_switcheroo.core.ghost) GhostParam (class in ml_switcheroo.core.ghost) GhostRef (class in ml_switcheroo.core.ghost) graph (ml_switcheroo.core.graph.GraphExtractor attribute), [1] (ml_switcheroo.core.tikz.analyser.GraphExtractor attribute) (ml_switcheroo.core.tikz.parser.TikzParser attribute) GraphDiffer (class in ml_switcheroo.compiler.differ) GraphExtractor (class in ml_switcheroo.core.graph) (class in ml_switcheroo.core.tikz.analyser) GraphFrontend (class in ml_switcheroo.compiler.registry) GraphOptimizer (class in ml_switcheroo.core.graph_optimizer) GraphPatcher (class in ml_switcheroo.core.rewriter.patcher) GridBox (class in ml_switcheroo.core.html) (class in ml_switcheroo.core.html.nodes) GridParser (class in ml_switcheroo.core.html.parser) GT (ml_switcheroo.enums.LogicOp attribute) GTE (ml_switcheroo.enums.LogicOp attribute) guards module guess_dtype_by_name() (in module ml_switcheroo.testing.fuzzer.heuristics) H handle_audit() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers.audit) handle_ci() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.verify) handle_convert() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers.convert) handle_data() (ml_switcheroo.core.html.parser.GridParser method) (ml_switcheroo.importers.sass_reader.SassHtmlParser method) handle_define() (in module ml_switcheroo.cli.handlers.define) handle_docs() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.dev) handle_endtag() (ml_switcheroo.core.html.parser.GridParser method) (ml_switcheroo.importers.sass_reader.SassHtmlParser method) handle_gen_tests() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.dev) handle_gen_weight_script() (in module ml_switcheroo.cli.commands) handle_harvest() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.learning) handle_import_spec() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.discovery) handle_matrix() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.dev) handle_post_processing() (in module post) handle_pre_checks() (in module pre) handle_scaffold() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.discovery) handle_schema() (in module ml_switcheroo.cli.handlers.meta) handle_snapshot() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.snapshots) handle_starttag() (ml_switcheroo.core.html.parser.GridParser method) (ml_switcheroo.importers.sass_reader.SassHtmlParser method) handle_suggest() (in module ml_switcheroo.cli.handlers.suggest) handle_sync() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.snapshots) handle_sync_standards() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.discovery) handle_wizard() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.cli.handlers) (in module ml_switcheroo.cli.handlers.learning) harness_imports (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) HARNESS_TEMPLATE (in module ml_switcheroo.testing.harness_generator_template) HarnessGenerator (class in ml_switcheroo.testing.harness_generator) harvest_file() (ml_switcheroo.discovery.harvester.SemanticHarvester method) (ml_switcheroo.discovery.SemanticHarvester method) has_arg() (ml_switcheroo.core.ghost.GhostRef method) has_errors (ml_switcheroo.core.conversion_result.ConversionResult property) has_numpy_compatible_arrays (ml_switcheroo.semantics.schema.PluginTraits attribute) has_results (ml_switcheroo.core.mlir.dialect.OpSchema attribute) has_varargs (ml_switcheroo.core.ghost.GhostRef attribute) header_text (ml_switcheroo.core.html.GridBox attribute) (ml_switcheroo.core.html.nodes.GridBox attribute) HELPER_LOGIC (in module injector_plugin) hook_context (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) HookContext (class in ml_switcheroo.core.hooks) HookFunction (in module ml_switcheroo.core.hooks) HtmlBackend (class in ml_switcheroo.compiler.backends.extras) HtmlDocument (class in ml_switcheroo.core.html) (class in ml_switcheroo.core.html.nodes) HtmlDSLAdapter (class in ml_switcheroo.frameworks.html_dsl) HtmlNode (class in ml_switcheroo.core.html) (class in ml_switcheroo.core.html.nodes) HtmlParser (class in ml_switcheroo.core.html.parser) hydrate() (ml_switcheroo.core.ghost.GhostInspector static method) (ml_switcheroo.semantics.registry_loader.RegistryLoader method) I id (ml_switcheroo.compiler.ir.LogicalNode attribute) (ml_switcheroo.compiler.LogicalNode attribute) (ml_switcheroo.core.graph.LogicalNode attribute) (ml_switcheroo.core.tracer.TraceEvent attribute) IDENTIFIER (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) (ml_switcheroo.core.mlir.tokens.TokenKind attribute) if_arg (ml_switcheroo.core.dsl.Rule attribute) IGNORED_SUFFIXES (ml_switcheroo.discovery.consensus.ConsensusEngine attribute) (ml_switcheroo.discovery.ConsensusEngine attribute) Immediate (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) IMMEDIATE (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) implicit_method_roots (ml_switcheroo.semantics.schema.StructuralTraits attribute) IMPORT_ACTION (ml_switcheroo.core.tracer.TraceEventType attribute) import_alias (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) import_namespaces (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) ImportConfig (class in ml_switcheroo.frameworks.base) ImportFixer (class in ml_switcheroo.core.import_fixer) ImportMixin (class in ml_switcheroo.core.import_fixer.imports_mixin) ImportReq (class in ml_switcheroo.core.dsl) (class in ml_switcheroo.core.import_fixer) (class in ml_switcheroo.core.import_fixer.resolution) ImportResolver (class in ml_switcheroo.core.import_fixer) (class in ml_switcheroo.core.import_fixer.resolution) imports (ml_switcheroo.semantics.schema.SemanticsFile attribute) ImportScanner (class in ml_switcheroo.discovery.harvester) impurity_methods (ml_switcheroo.semantics.schema.StructuralTraits attribute) IN (ml_switcheroo.enums.LogicOp attribute) in_box (ml_switcheroo.core.html.parser.GridParser attribute) in_cell (ml_switcheroo.importers.sass_reader.SassHtmlParser attribute) in_code (ml_switcheroo.core.html.parser.GridParser attribute) in_header_txt (ml_switcheroo.core.html.parser.GridParser attribute) in_module_class (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) in_row (ml_switcheroo.importers.sass_reader.SassHtmlParser attribute) in_table (ml_switcheroo.importers.sass_reader.SassHtmlParser attribute) in_tbody (ml_switcheroo.importers.sass_reader.SassHtmlParser attribute) index (ml_switcheroo.compiler.frontends.rdna.nodes.SGPR attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.VGPR attribute), [1] (ml_switcheroo.compiler.frontends.rdna.SGPR attribute), [1] (ml_switcheroo.compiler.frontends.rdna.VGPR attribute), [1] infer_tier_from_priority() (in module ml_switcheroo.semantics.merging) ingest_code() (in module ml_switcheroo.core.ingestion) inherits_from (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter attribute) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter attribute) (ml_switcheroo.frameworks.jax.JaxCoreAdapter attribute) (ml_switcheroo.frameworks.keras.KerasAdapter attribute) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter attribute) (ml_switcheroo.frameworks.mlir.MlirAdapter attribute) (ml_switcheroo.frameworks.mlx.MLXAdapter attribute) (ml_switcheroo.frameworks.numpy.NumpyAdapter attribute) (ml_switcheroo.frameworks.paxml.PaxmlAdapter attribute) (ml_switcheroo.frameworks.rdna.RdnaAdapter attribute) (ml_switcheroo.frameworks.sass.SassAdapter attribute) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter attribute) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter attribute) (ml_switcheroo.frameworks.tikz.TikzAdapter attribute) (ml_switcheroo.frameworks.torch.TorchAdapter attribute) init_method_name (ml_switcheroo.semantics.schema.StructuralTraits attribute) InitializationTracker (class in ml_switcheroo.analysis.lifecycle) InitMode (class in ml_switcheroo.frameworks.base) inject() (injector_fw.core.FrameworkInjector method) (injector_spec.StandardsInjector method) inject_args (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) inject_kwarg() (in module utils) inject_magic_args (ml_switcheroo.semantics.schema.StructuralTraits attribute) inject_permute_call() (in module utils) inject_preamble() (ml_switcheroo.core.hooks.HookContext method) inject_prng_threading() (in module ml_switcheroo.plugins.rng_threading) inject_signature_arg() (ml_switcheroo.core.hooks.HookContext method) inject_training_flag_call() (in module ml_switcheroo.plugins.state_flag_injection) injected_args (ml_switcheroo.core.rewriter.types.SignatureContext attribute) InjectionMixin (class in ml_switcheroo.core.import_fixer.injection_mixin) injector_fw module injector_fw.core module injector_fw.utils module injector_plugin module injector_spec module input_vars (ml_switcheroo.compiler.differ.ReplaceAction attribute) (ml_switcheroo.core.rewriter.patcher.ReplaceAction attribute), [1] InputFuzzer (class in ml_switcheroo.testing.fuzzer) (class in ml_switcheroo.testing.fuzzer.core) InputNode (class in ml_switcheroo.core.latex) (class in ml_switcheroo.core.latex.nodes) inspect() (ml_switcheroo.core.ghost.GhostInspector static method) (ml_switcheroo.discovery.ApiInspector method) (ml_switcheroo.discovery.inspector.ApiInspector method) INSPECTION (ml_switcheroo.core.tracer.TraceEventType attribute) inspector (ml_switcheroo.discovery.Scaffolder attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) Instruction (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) intermediate (ml_switcheroo.config.RuntimeConfig attribute) is_builtin() (in module utils) is_docstring() (in module injector_fw.utils) (in module ml_switcheroo.core.import_fixer.utils) is_framework_module_node() (in module ml_switcheroo.plugins.utils) is_functional_apply() (in module utils) is_future_import() (in module injector_fw.utils) (in module ml_switcheroo.core.import_fixer.utils) is_hex (ml_switcheroo.compiler.frontends.rdna.Immediate attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Immediate attribute), [1] (ml_switcheroo.compiler.frontends.sass.Immediate attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Immediate attribute), [1] is_init (ml_switcheroo.compiler.differ.ReplaceAction attribute) (ml_switcheroo.core.rewriter.patcher.ReplaceAction attribute), [1] (ml_switcheroo.core.rewriter.types.SignatureContext attribute) is_inplace (ml_switcheroo.core.dsl.OperationDef attribute) is_isa_source() (in module ml_switcheroo.compiler.registry) is_isa_target() (in module ml_switcheroo.compiler.registry) is_module_method (ml_switcheroo.core.rewriter.types.SignatureContext attribute) is_pipe_top_level() (in module ml_switcheroo.testing.fuzzer.utils) is_static_arg() (in module ml_switcheroo.generated_tests.templates) is_super_call() (in module utils) IS_TYPE (ml_switcheroo.enums.LogicOp attribute) is_val (ml_switcheroo.core.dsl.Rule attribute) is_variadic (ml_switcheroo.core.dsl.ParameterDef attribute) is_verified() (ml_switcheroo.semantics.manager.SemanticsManager method) J jax (in module ml_switcheroo.frameworks.flax_nnx) (in module ml_switcheroo.frameworks.jax) jax_test_config (ml_switcheroo.frameworks.common.jax_stack.JAXStackMixin property) JaxCoreAdapter (class in ml_switcheroo.frameworks.jax) JAXStackMixin (class in ml_switcheroo.frameworks.common.jax_stack) jit_static_args (ml_switcheroo.semantics.schema.StructuralTraits attribute) json_path (injector_fw.core.FrameworkInjector attribute) K keras (in module ml_switcheroo.frameworks.keras) KerasAdapter (class in ml_switcheroo.frameworks.keras) key (ml_switcheroo.core.tikz.nodes.TikzOption attribute) (ml_switcheroo.core.tikz.TikzOption attribute) KEY_MAPPER_SOURCE (in module ml_switcheroo.plugins.checkpoint_keys) kind (ml_switcheroo.compiler.frontends.rdna.Token attribute), [1] (ml_switcheroo.compiler.frontends.rdna.tokens.Token attribute), [1] (ml_switcheroo.compiler.frontends.sass.Token attribute), [1] (ml_switcheroo.compiler.frontends.sass.tokens.Token attribute), [1] (ml_switcheroo.compiler.ir.LogicalNode attribute) (ml_switcheroo.compiler.LogicalNode attribute) (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.ghost.GhostParam attribute) (ml_switcheroo.core.ghost.GhostRef attribute) (ml_switcheroo.core.graph.LogicalNode attribute) (ml_switcheroo.core.mlir.nodes.TriviaNode attribute) (ml_switcheroo.core.mlir.parser.Token attribute) (ml_switcheroo.core.tikz.parser.Token attribute) KnowledgeBaseLoader (class in ml_switcheroo.semantics.file_loader) known_inference_methods (ml_switcheroo.semantics.schema.StructuralTraits attribute) known_magic_args (ml_switcheroo.semantics.manager.SemanticsManager attribute) kwargs_map (ml_switcheroo.core.dsl.FrameworkVariant attribute) L Label (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) label (ml_switcheroo.core.mlir.nodes.BlockNode attribute) LABEL_DEF (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) LabelRef (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (class in ml_switcheroo.compiler.frontends.sass.parser) LatexBackend (class in ml_switcheroo.compiler.backends.extras) LatexDSLAdapter (class in ml_switcheroo.frameworks.latex_dsl) LatexNode (class in ml_switcheroo.core.latex) (class in ml_switcheroo.core.latex.nodes) LatexParser (class in ml_switcheroo.core.latex.parser) LAYER (ml_switcheroo.frameworks.base.StandardCategory attribute) layer_registry (ml_switcheroo.core.graph.GraphExtractor attribute), [1] (ml_switcheroo.core.tikz.analyser.GraphExtractor attribute) layout_map (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) LBRACE (ml_switcheroo.core.mlir.tokens.Symbol attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) LBRACKET (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.core.mlir.tokens.Symbol attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) leading_trivia (ml_switcheroo.core.mlir.nodes.BlockNode attribute) (ml_switcheroo.core.mlir.nodes.OperationNode attribute) (ml_switcheroo.core.tikz.nodes.TikzEdge attribute) (ml_switcheroo.core.tikz.nodes.TikzNode attribute) (ml_switcheroo.core.tikz.TikzEdge attribute) (ml_switcheroo.core.tikz.TikzNode attribute) leave_Annotation() (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) leave_Arg() (ml_switcheroo.utils.visualizer.MermaidGenerator method) leave_Assign() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.patcher.GraphPatcher method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) leave_Attribute() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.import_fixer.attributes_mixin.AttributeMixin method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) leave_Call() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.patcher.GraphPatcher method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) leave_ClassDef() (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.compiler.backends.python.ClassBodyReplacer method) (ml_switcheroo.core.graph.GraphExtractor method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) leave_Decorator() (ml_switcheroo.core.rewriter.passes.auxiliary.AuxiliaryTransformer method) (ml_switcheroo.core.rewriter.passes.AuxiliaryTransformer method) leave_Expr() (ml_switcheroo.core.rewriter.patcher.GraphPatcher method) leave_For() (ml_switcheroo.core.rewriter.passes.auxiliary.AuxiliaryTransformer method) (ml_switcheroo.core.rewriter.passes.AuxiliaryTransformer method) leave_FunctionDef() (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.graph.GraphExtractor method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) (ml_switcheroo.core.tikz.analyser.GraphExtractor method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) leave_IfExp() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) leave_Import() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.import_fixer.imports_mixin.ImportMixin method) (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) leave_ImportFrom() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.import_fixer.imports_mixin.ImportMixin method) (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) leave_Module() (ml_switcheroo.core.graph.GraphExtractor method) (ml_switcheroo.core.import_fixer.injection_mixin.InjectionMixin method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) (ml_switcheroo.core.tikz.analyser.GraphExtractor method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) leave_Name() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) (ml_switcheroo.plugins.static_unroll.LoopVarReplacer method) leave_SimpleStatementLine() (ml_switcheroo.analysis.purity.PurityScanner method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.auxiliary.AuxiliaryTransformer method) (ml_switcheroo.core.rewriter.passes.AuxiliaryTransformer method) (ml_switcheroo.core.rewriter.patcher.GraphPatcher method) lexer (ml_switcheroo.compiler.frontends.rdna.parser.RdnaParser attribute) (ml_switcheroo.compiler.frontends.rdna.RdnaParser attribute) (ml_switcheroo.compiler.frontends.sass.parser.SassParser attribute) (ml_switcheroo.compiler.frontends.sass.SassParser attribute) (ml_switcheroo.core.tikz.parser.TikzParser attribute) lifecycle_strip_methods (ml_switcheroo.semantics.schema.StructuralTraits attribute) lifecycle_warn_methods (ml_switcheroo.semantics.schema.StructuralTraits attribute) lift() (ml_switcheroo.compiler.frontends.rdna.lifter.RdnaLifter method) (ml_switcheroo.compiler.frontends.rdna.RdnaLifter method) (ml_switcheroo.compiler.frontends.sass.lifter.SassLifter method) (ml_switcheroo.compiler.frontends.sass.SassLifter method) line (ml_switcheroo.compiler.frontends.rdna.Token attribute), [1] (ml_switcheroo.compiler.frontends.rdna.tokens.Token attribute), [1] (ml_switcheroo.compiler.frontends.sass.Token attribute), [1] (ml_switcheroo.compiler.frontends.sass.tokens.Token attribute), [1] (ml_switcheroo.core.mlir.parser.Token attribute) (ml_switcheroo.core.tikz.parser.TikzLexer attribute) (ml_switcheroo.core.tikz.parser.Token attribute) lineno (ml_switcheroo.core.tracer.TraceEvent attribute) LIST (ml_switcheroo.core.dsl.ContainerType attribute) LIVE (ml_switcheroo.frameworks.base.InitMode attribute) load() (ml_switcheroo.config.RuntimeConfig class method) load_definitions() (in module ml_switcheroo.frameworks.loader) load_knowledge_graph() (ml_switcheroo.semantics.file_loader.KnowledgeBaseLoader method) load_plugins() (in module ml_switcheroo.core.hooks) load_snapshot_for_adapter() (in module ml_switcheroo.frameworks.base) load_validation_report() (ml_switcheroo.semantics.manager.SemanticsManager method) log_diff() (in module utils) log_error() (in module ml_switcheroo.utils.console) log_info() (in module ml_switcheroo.utils.console) log_inspection() (ml_switcheroo.core.tracer.TraceLogger method) log_match() (ml_switcheroo.core.tracer.TraceLogger method) log_mutation() (ml_switcheroo.core.tracer.TraceLogger method) log_snapshot() (ml_switcheroo.core.tracer.TraceLogger method) log_success() (in module ml_switcheroo.utils.console) log_warning() (in module ml_switcheroo.utils.console) (ml_switcheroo.core.tracer.TraceLogger method) logger (in module ml_switcheroo.core.discovery) (ml_switcheroo.testing.bisector.SemanticsBisector attribute) LogicalEdge (class in ml_switcheroo.compiler) (class in ml_switcheroo.compiler.ir) (class in ml_switcheroo.core.graph) LogicalGraph (class in ml_switcheroo.compiler) (class in ml_switcheroo.compiler.ir) (class in ml_switcheroo.core.graph) LogicalNode (class in ml_switcheroo.compiler) (class in ml_switcheroo.compiler.ir) (class in ml_switcheroo.core.graph) LogicOp (class in ml_switcheroo.enums) lookup() (ml_switcheroo.core.mlir.emitter.SSAContext method) (ml_switcheroo.core.mlir.naming.NamingContext method) lookup_api() (ml_switcheroo.core.hooks.HookContext method) lookup_signature() (ml_switcheroo.core.hooks.HookContext method) LoopVarReplacer (class in ml_switcheroo.plugins.static_unroll) LOSS (ml_switcheroo.frameworks.base.StandardCategory attribute) LPAREN (ml_switcheroo.core.mlir.tokens.Symbol attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) LT (ml_switcheroo.enums.LogicOp attribute) LTE (ml_switcheroo.enums.LogicOp attribute) M macro_registry (ml_switcheroo.compiler.backends.rdna.RdnaSynthesizer attribute) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaSynthesizer attribute) (ml_switcheroo.compiler.backends.sass.SassSynthesizer attribute) (ml_switcheroo.compiler.backends.sass.synthesizer.SassSynthesizer attribute) macro_template (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) main() (in module ml_switcheroo.cli.__main__) make_broadcastable_shape() (in module ml_switcheroo.testing.fuzzer.generators) mappings (ml_switcheroo.core.import_fixer.resolution.ResolutionPlan attribute) (ml_switcheroo.core.import_fixer.ResolutionPlan attribute) (ml_switcheroo.discovery.harvester.TargetCallVisitor attribute) MappingWizard (class in ml_switcheroo.cli.wizard) mark_failure() (ml_switcheroo.core.escape_hatch.EscapeHatch static method) marker_end (ml_switcheroo.core.html.nodes.SvgArrow attribute) (ml_switcheroo.core.html.SvgArrow attribute) match() (ml_switcheroo.core.mlir.parser.MlirParser method) MATCH_SEMANTICS (ml_switcheroo.core.tracer.TraceEventType attribute) max (ml_switcheroo.core.dsl.ParameterDef attribute) MAX_REGISTERS (in module ml_switcheroo.compiler.backends.sass.synthesizer) MAX_SGPR (in module ml_switcheroo.compiler.backends.rdna.synthesizer) max_version (ml_switcheroo.core.dsl.FrameworkVariant attribute) MAX_VGPR (in module ml_switcheroo.compiler.backends.rdna.synthesizer) Memory (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) MEMORY (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) MemoryNode (class in ml_switcheroo.core.latex) (class in ml_switcheroo.core.latex.nodes) merge_frameworks() (in module ml_switcheroo.semantics.merging) merge_overlay_data() (in module ml_switcheroo.semantics.merging) merge_patterns() (in module ml_switcheroo.semantics.merging) merge_tier_data() (in module ml_switcheroo.semantics.merging) MermaidGenerator (class in ml_switcheroo.utils.visualizer) metadata (ml_switcheroo.compiler.ir.LogicalNode attribute) (ml_switcheroo.compiler.LogicalNode attribute) (ml_switcheroo.core.graph.LogicalNode attribute) (ml_switcheroo.core.hooks.HookContext attribute) (ml_switcheroo.core.tracer.TraceEvent attribute) mgr (ml_switcheroo.semantics.file_loader.KnowledgeBaseLoader attribute) (ml_switcheroo.semantics.registry_loader.RegistryLoader attribute) MigrationGuideGenerator (class in ml_switcheroo.utils.doc_gen) min (ml_switcheroo.core.dsl.ParameterDef attribute) min_version (ml_switcheroo.core.dsl.FrameworkVariant attribute) MISMATCH (ml_switcheroo.core.mlir.tokens.TokenKind attribute) missing_message (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) ml_switcheroo module ml_switcheroo.__main__ module ml_switcheroo.analysis module ml_switcheroo.analysis.audit module ml_switcheroo.analysis.dependencies module ml_switcheroo.analysis.lifecycle module ml_switcheroo.analysis.purity module ml_switcheroo.analysis.symbol_table module ml_switcheroo.cli module ml_switcheroo.cli.__main__ module ml_switcheroo.cli.commands module ml_switcheroo.cli.handlers module ml_switcheroo.cli.handlers.audit module ml_switcheroo.cli.handlers.convert module ml_switcheroo.cli.handlers.convert_weights module ml_switcheroo.cli.handlers.define module ml_switcheroo.cli.handlers.dev module ml_switcheroo.cli.handlers.discovery module ml_switcheroo.cli.handlers.learning module ml_switcheroo.cli.handlers.meta module ml_switcheroo.cli.handlers.snapshots module ml_switcheroo.cli.handlers.suggest module ml_switcheroo.cli.handlers.verify module ml_switcheroo.cli.matrix module ml_switcheroo.cli.wizard module ml_switcheroo.compiler module ml_switcheroo.compiler.backend module ml_switcheroo.compiler.backends module ml_switcheroo.compiler.backends.extras module ml_switcheroo.compiler.backends.python module ml_switcheroo.compiler.backends.python_snippet module ml_switcheroo.compiler.backends.rdna module ml_switcheroo.compiler.backends.rdna.emitter module ml_switcheroo.compiler.backends.rdna.macros module ml_switcheroo.compiler.backends.rdna.synthesizer module ml_switcheroo.compiler.backends.sass module ml_switcheroo.compiler.backends.sass.emitter module ml_switcheroo.compiler.backends.sass.macros module ml_switcheroo.compiler.backends.sass.synthesizer module ml_switcheroo.compiler.differ module ml_switcheroo.compiler.frontends module ml_switcheroo.compiler.frontends.python module ml_switcheroo.compiler.frontends.rdna module ml_switcheroo.compiler.frontends.rdna.analysis module ml_switcheroo.compiler.frontends.rdna.lifter module ml_switcheroo.compiler.frontends.rdna.nodes module ml_switcheroo.compiler.frontends.rdna.parser module ml_switcheroo.compiler.frontends.rdna.tokens module ml_switcheroo.compiler.frontends.sass module ml_switcheroo.compiler.frontends.sass.analysis module ml_switcheroo.compiler.frontends.sass.lifter module ml_switcheroo.compiler.frontends.sass.nodes module ml_switcheroo.compiler.frontends.sass.parser module ml_switcheroo.compiler.frontends.sass.tokens module ml_switcheroo.compiler.ir module ml_switcheroo.compiler.registry module ml_switcheroo.config module ml_switcheroo.core module ml_switcheroo.core.conversion_result module ml_switcheroo.core.discovery module ml_switcheroo.core.dsl module ml_switcheroo.core.engine module ml_switcheroo.core.escape_hatch module ml_switcheroo.core.ghost module ml_switcheroo.core.graph module ml_switcheroo.core.graph_optimizer module ml_switcheroo.core.hooks module ml_switcheroo.core.html module ml_switcheroo.core.html.nodes module ml_switcheroo.core.html.parser module ml_switcheroo.core.import_fixer module ml_switcheroo.core.import_fixer.attributes_mixin module ml_switcheroo.core.import_fixer.base module ml_switcheroo.core.import_fixer.imports_mixin module ml_switcheroo.core.import_fixer.injection_mixin module ml_switcheroo.core.import_fixer.resolution module ml_switcheroo.core.import_fixer.utils module ml_switcheroo.core.ingestion module ml_switcheroo.core.latex module ml_switcheroo.core.latex.nodes module ml_switcheroo.core.latex.parser module ml_switcheroo.core.mlir module ml_switcheroo.core.mlir.dialect module ml_switcheroo.core.mlir.emitter module ml_switcheroo.core.mlir.gen_base module ml_switcheroo.core.mlir.gen_expressions module ml_switcheroo.core.mlir.gen_statements module ml_switcheroo.core.mlir.generator module ml_switcheroo.core.mlir.naming module ml_switcheroo.core.mlir.nodes module ml_switcheroo.core.mlir.parser module ml_switcheroo.core.mlir.stablehlo_emitter module ml_switcheroo.core.mlir.tokens module ml_switcheroo.core.rewriter module ml_switcheroo.core.rewriter.context module ml_switcheroo.core.rewriter.interface module ml_switcheroo.core.rewriter.normalization_utils module ml_switcheroo.core.rewriter.passes module ml_switcheroo.core.rewriter.passes.api module ml_switcheroo.core.rewriter.passes.auxiliary module ml_switcheroo.core.rewriter.passes.structure module ml_switcheroo.core.rewriter.patcher module ml_switcheroo.core.rewriter.pipeline module ml_switcheroo.core.rewriter.types module ml_switcheroo.core.scanners module ml_switcheroo.core.tikz module ml_switcheroo.core.tikz.analyser module ml_switcheroo.core.tikz.nodes module ml_switcheroo.core.tikz.parser module ml_switcheroo.core.tracer module ml_switcheroo.discovery module ml_switcheroo.discovery.consensus module ml_switcheroo.discovery.harvester module ml_switcheroo.discovery.inspector module ml_switcheroo.discovery.scaffolder module ml_switcheroo.discovery.syncer module ml_switcheroo.enums module ml_switcheroo.frameworks module ml_switcheroo.frameworks.base module ml_switcheroo.frameworks.common module ml_switcheroo.frameworks.common.jax_stack module ml_switcheroo.frameworks.flax_nnx module ml_switcheroo.frameworks.html_dsl module ml_switcheroo.frameworks.jax module ml_switcheroo.frameworks.keras module ml_switcheroo.frameworks.latex_dsl module ml_switcheroo.frameworks.loader module ml_switcheroo.frameworks.mlir module ml_switcheroo.frameworks.mlx module ml_switcheroo.frameworks.numpy module ml_switcheroo.frameworks.optax_shim module ml_switcheroo.frameworks.paxml module ml_switcheroo.frameworks.rdna module ml_switcheroo.frameworks.sass module ml_switcheroo.frameworks.stablehlo module ml_switcheroo.frameworks.tensorflow module ml_switcheroo.frameworks.tikz module ml_switcheroo.frameworks.torch module ml_switcheroo.generated_tests module ml_switcheroo.generated_tests.generator module ml_switcheroo.generated_tests.inputs module ml_switcheroo.generated_tests.runtime module ml_switcheroo.generated_tests.runtime_builder module ml_switcheroo.generated_tests.templates module ml_switcheroo.importers module ml_switcheroo.importers.array_api_reader module ml_switcheroo.importers.onnx_reader module ml_switcheroo.importers.sass_reader module ml_switcheroo.importers.stablehlo_reader module ml_switcheroo.plugins module ml_switcheroo.plugins.attention_packing module ml_switcheroo.plugins.batch_norm module ml_switcheroo.plugins.casting module ml_switcheroo.plugins.checkpoint_keys module ml_switcheroo.plugins.clipping module ml_switcheroo.plugins.context_to_function_wrap module ml_switcheroo.plugins.data_loader module ml_switcheroo.plugins.device_allocator module ml_switcheroo.plugins.device_checks module ml_switcheroo.plugins.einsum module ml_switcheroo.plugins.flatten module ml_switcheroo.plugins.gather module ml_switcheroo.plugins.inplace_unroll module ml_switcheroo.plugins.io_handler module ml_switcheroo.plugins.keras_sequential module ml_switcheroo.plugins.loop_unroll module ml_switcheroo.plugins.loss_wrapper module ml_switcheroo.plugins.method_property module ml_switcheroo.plugins.mlx_extras module ml_switcheroo.plugins.mlx_optimizers module ml_switcheroo.plugins.nnx_to_torch_params module ml_switcheroo.plugins.optimizer_step module ml_switcheroo.plugins.padding module ml_switcheroo.plugins.reshape module ml_switcheroo.plugins.rng_threading module ml_switcheroo.plugins.scatter module ml_switcheroo.plugins.schedulers module ml_switcheroo.plugins.shape_packing module ml_switcheroo.plugins.state_container module ml_switcheroo.plugins.state_flag_injection module ml_switcheroo.plugins.static_unroll module ml_switcheroo.plugins.tf_data_loader module ml_switcheroo.plugins.topk module ml_switcheroo.plugins.utils module ml_switcheroo.semantics module ml_switcheroo.semantics.autogen module ml_switcheroo.semantics.file_loader module ml_switcheroo.semantics.manager module ml_switcheroo.semantics.merging module ml_switcheroo.semantics.paths module ml_switcheroo.semantics.registry_loader module ml_switcheroo.semantics.schema module ml_switcheroo.testing module ml_switcheroo.testing.batch_runner module ml_switcheroo.testing.bisector module ml_switcheroo.testing.fuzzer module ml_switcheroo.testing.fuzzer.core module ml_switcheroo.testing.fuzzer.generators module ml_switcheroo.testing.fuzzer.heuristics module ml_switcheroo.testing.fuzzer.parser module ml_switcheroo.testing.fuzzer.strategies module ml_switcheroo.testing.fuzzer.utils module ml_switcheroo.testing.harness_generator module ml_switcheroo.testing.harness_generator_template module ml_switcheroo.testing.linter module ml_switcheroo.testing.runner module ml_switcheroo.utils module ml_switcheroo.utils.code_extractor module ml_switcheroo.utils.console module ml_switcheroo.utils.doc_context module ml_switcheroo.utils.doc_gen module ml_switcheroo.utils.doc_renderer module ml_switcheroo.utils.node_diff module ml_switcheroo.utils.readme_editor module ml_switcheroo.utils.visualizer module MlirAdapter (class in ml_switcheroo.frameworks.mlir) MlirBackend (class in ml_switcheroo.compiler.backends.extras) MlirNode (class in ml_switcheroo.core.mlir.nodes) MlirParser (class in ml_switcheroo.core.mlir.parser) MlirToPythonGenerator (class in ml_switcheroo.core.mlir.generator) MLXAdapter (class in ml_switcheroo.frameworks.mlx) model_config (ml_switcheroo.core.dsl.Rule attribute) (ml_switcheroo.core.hooks.AutoWireSpec attribute) (ml_switcheroo.semantics.schema.FrameworkTraits attribute) (ml_switcheroo.semantics.schema.PluginTraits attribute) (ml_switcheroo.semantics.schema.SemanticsFile attribute) (ml_switcheroo.semantics.schema.StructuralTraits attribute) model_name (ml_switcheroo.core.graph.GraphExtractor attribute) (ml_switcheroo.core.html.HtmlDocument attribute) (ml_switcheroo.core.html.nodes.HtmlDocument attribute) (ml_switcheroo.core.html.parser.GridParser attribute) (ml_switcheroo.core.tikz.analyser.GraphExtractor attribute) ModelContainer (class in ml_switcheroo.core.latex) (class in ml_switcheroo.core.latex.nodes) Modifier (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) MODIFIER (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) module dispatch guards injector_fw injector_fw.core injector_fw.utils injector_plugin injector_spec ml_switcheroo ml_switcheroo.__main__ ml_switcheroo.analysis ml_switcheroo.analysis.audit ml_switcheroo.analysis.dependencies ml_switcheroo.analysis.lifecycle ml_switcheroo.analysis.purity ml_switcheroo.analysis.symbol_table ml_switcheroo.cli ml_switcheroo.cli.__main__ ml_switcheroo.cli.commands ml_switcheroo.cli.handlers ml_switcheroo.cli.handlers.audit ml_switcheroo.cli.handlers.convert ml_switcheroo.cli.handlers.convert_weights ml_switcheroo.cli.handlers.define ml_switcheroo.cli.handlers.dev ml_switcheroo.cli.handlers.discovery ml_switcheroo.cli.handlers.learning ml_switcheroo.cli.handlers.meta ml_switcheroo.cli.handlers.snapshots ml_switcheroo.cli.handlers.suggest ml_switcheroo.cli.handlers.verify ml_switcheroo.cli.matrix ml_switcheroo.cli.wizard ml_switcheroo.compiler ml_switcheroo.compiler.backend ml_switcheroo.compiler.backends ml_switcheroo.compiler.backends.extras ml_switcheroo.compiler.backends.python ml_switcheroo.compiler.backends.python_snippet ml_switcheroo.compiler.backends.rdna ml_switcheroo.compiler.backends.rdna.emitter ml_switcheroo.compiler.backends.rdna.macros ml_switcheroo.compiler.backends.rdna.synthesizer ml_switcheroo.compiler.backends.sass ml_switcheroo.compiler.backends.sass.emitter ml_switcheroo.compiler.backends.sass.macros ml_switcheroo.compiler.backends.sass.synthesizer ml_switcheroo.compiler.differ ml_switcheroo.compiler.frontends ml_switcheroo.compiler.frontends.python ml_switcheroo.compiler.frontends.rdna ml_switcheroo.compiler.frontends.rdna.analysis ml_switcheroo.compiler.frontends.rdna.lifter ml_switcheroo.compiler.frontends.rdna.nodes ml_switcheroo.compiler.frontends.rdna.parser ml_switcheroo.compiler.frontends.rdna.tokens ml_switcheroo.compiler.frontends.sass ml_switcheroo.compiler.frontends.sass.analysis ml_switcheroo.compiler.frontends.sass.lifter ml_switcheroo.compiler.frontends.sass.nodes ml_switcheroo.compiler.frontends.sass.parser ml_switcheroo.compiler.frontends.sass.tokens ml_switcheroo.compiler.ir ml_switcheroo.compiler.registry ml_switcheroo.config ml_switcheroo.core ml_switcheroo.core.conversion_result ml_switcheroo.core.discovery ml_switcheroo.core.dsl ml_switcheroo.core.engine ml_switcheroo.core.escape_hatch ml_switcheroo.core.ghost ml_switcheroo.core.graph ml_switcheroo.core.graph_optimizer ml_switcheroo.core.hooks ml_switcheroo.core.html ml_switcheroo.core.html.nodes ml_switcheroo.core.html.parser ml_switcheroo.core.import_fixer ml_switcheroo.core.import_fixer.attributes_mixin ml_switcheroo.core.import_fixer.base ml_switcheroo.core.import_fixer.imports_mixin ml_switcheroo.core.import_fixer.injection_mixin ml_switcheroo.core.import_fixer.resolution ml_switcheroo.core.import_fixer.utils ml_switcheroo.core.ingestion ml_switcheroo.core.latex ml_switcheroo.core.latex.nodes ml_switcheroo.core.latex.parser ml_switcheroo.core.mlir ml_switcheroo.core.mlir.dialect ml_switcheroo.core.mlir.emitter ml_switcheroo.core.mlir.gen_base ml_switcheroo.core.mlir.gen_expressions ml_switcheroo.core.mlir.gen_statements ml_switcheroo.core.mlir.generator ml_switcheroo.core.mlir.naming ml_switcheroo.core.mlir.nodes ml_switcheroo.core.mlir.parser ml_switcheroo.core.mlir.stablehlo_emitter ml_switcheroo.core.mlir.tokens ml_switcheroo.core.rewriter ml_switcheroo.core.rewriter.context ml_switcheroo.core.rewriter.interface ml_switcheroo.core.rewriter.normalization_utils ml_switcheroo.core.rewriter.passes ml_switcheroo.core.rewriter.passes.api ml_switcheroo.core.rewriter.passes.auxiliary ml_switcheroo.core.rewriter.passes.structure ml_switcheroo.core.rewriter.patcher ml_switcheroo.core.rewriter.pipeline ml_switcheroo.core.rewriter.types ml_switcheroo.core.scanners ml_switcheroo.core.tikz ml_switcheroo.core.tikz.analyser ml_switcheroo.core.tikz.nodes ml_switcheroo.core.tikz.parser ml_switcheroo.core.tracer ml_switcheroo.discovery ml_switcheroo.discovery.consensus ml_switcheroo.discovery.harvester ml_switcheroo.discovery.inspector ml_switcheroo.discovery.scaffolder ml_switcheroo.discovery.syncer ml_switcheroo.enums ml_switcheroo.frameworks ml_switcheroo.frameworks.base ml_switcheroo.frameworks.common ml_switcheroo.frameworks.common.jax_stack ml_switcheroo.frameworks.flax_nnx ml_switcheroo.frameworks.html_dsl ml_switcheroo.frameworks.jax ml_switcheroo.frameworks.keras ml_switcheroo.frameworks.latex_dsl ml_switcheroo.frameworks.loader ml_switcheroo.frameworks.mlir ml_switcheroo.frameworks.mlx ml_switcheroo.frameworks.numpy ml_switcheroo.frameworks.optax_shim ml_switcheroo.frameworks.paxml ml_switcheroo.frameworks.rdna ml_switcheroo.frameworks.sass ml_switcheroo.frameworks.stablehlo ml_switcheroo.frameworks.tensorflow ml_switcheroo.frameworks.tikz ml_switcheroo.frameworks.torch ml_switcheroo.generated_tests ml_switcheroo.generated_tests.generator ml_switcheroo.generated_tests.inputs ml_switcheroo.generated_tests.runtime ml_switcheroo.generated_tests.runtime_builder ml_switcheroo.generated_tests.templates ml_switcheroo.importers ml_switcheroo.importers.array_api_reader ml_switcheroo.importers.onnx_reader ml_switcheroo.importers.sass_reader ml_switcheroo.importers.stablehlo_reader ml_switcheroo.plugins ml_switcheroo.plugins.attention_packing ml_switcheroo.plugins.batch_norm ml_switcheroo.plugins.casting ml_switcheroo.plugins.checkpoint_keys ml_switcheroo.plugins.clipping ml_switcheroo.plugins.context_to_function_wrap ml_switcheroo.plugins.data_loader ml_switcheroo.plugins.device_allocator ml_switcheroo.plugins.device_checks ml_switcheroo.plugins.einsum ml_switcheroo.plugins.flatten ml_switcheroo.plugins.gather ml_switcheroo.plugins.inplace_unroll ml_switcheroo.plugins.io_handler ml_switcheroo.plugins.keras_sequential ml_switcheroo.plugins.loop_unroll ml_switcheroo.plugins.loss_wrapper ml_switcheroo.plugins.method_property ml_switcheroo.plugins.mlx_extras ml_switcheroo.plugins.mlx_optimizers ml_switcheroo.plugins.nnx_to_torch_params ml_switcheroo.plugins.optimizer_step ml_switcheroo.plugins.padding ml_switcheroo.plugins.reshape ml_switcheroo.plugins.rng_threading ml_switcheroo.plugins.scatter ml_switcheroo.plugins.schedulers ml_switcheroo.plugins.shape_packing ml_switcheroo.plugins.state_container ml_switcheroo.plugins.state_flag_injection ml_switcheroo.plugins.static_unroll ml_switcheroo.plugins.tf_data_loader ml_switcheroo.plugins.topk ml_switcheroo.plugins.utils ml_switcheroo.semantics ml_switcheroo.semantics.autogen ml_switcheroo.semantics.file_loader ml_switcheroo.semantics.manager ml_switcheroo.semantics.merging ml_switcheroo.semantics.paths ml_switcheroo.semantics.registry_loader ml_switcheroo.semantics.schema ml_switcheroo.testing ml_switcheroo.testing.batch_runner ml_switcheroo.testing.bisector ml_switcheroo.testing.fuzzer ml_switcheroo.testing.fuzzer.core ml_switcheroo.testing.fuzzer.generators ml_switcheroo.testing.fuzzer.heuristics ml_switcheroo.testing.fuzzer.parser ml_switcheroo.testing.fuzzer.strategies ml_switcheroo.testing.fuzzer.utils ml_switcheroo.testing.harness_generator ml_switcheroo.testing.harness_generator_template ml_switcheroo.testing.linter ml_switcheroo.testing.runner ml_switcheroo.utils ml_switcheroo.utils.code_extractor ml_switcheroo.utils.console ml_switcheroo.utils.doc_context ml_switcheroo.utils.doc_gen ml_switcheroo.utils.doc_renderer ml_switcheroo.utils.node_diff ml_switcheroo.utils.readme_editor ml_switcheroo.utils.visualizer post pre strategy transformers utils module (ml_switcheroo.core.dsl.ImportReq attribute) (ml_switcheroo.core.import_fixer.ImportReq attribute) (ml_switcheroo.core.import_fixer.resolution.ImportReq attribute) (ml_switcheroo.semantics.schema.FrameworkAlias attribute) module_base (ml_switcheroo.semantics.schema.StructuralTraits attribute) module_preamble (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) ModuleNode (class in ml_switcheroo.core.mlir.nodes) ModuleType (class in ml_switcheroo.analysis.symbol_table) N name (ml_switcheroo.analysis.symbol_table.Scope attribute) (ml_switcheroo.analysis.symbol_table.SymbolType attribute) (ml_switcheroo.compiler.frontends.rdna.Directive attribute), [1] (ml_switcheroo.compiler.frontends.rdna.Label attribute), [1] (ml_switcheroo.compiler.frontends.rdna.LabelRef attribute), [1] (ml_switcheroo.compiler.frontends.rdna.Modifier attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Directive attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Label attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.LabelRef attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Modifier attribute), [1] (ml_switcheroo.compiler.frontends.sass.Directive attribute), [1] (ml_switcheroo.compiler.frontends.sass.Label attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Directive attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Label attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Predicate attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Register attribute), [1] (ml_switcheroo.compiler.frontends.sass.parser.LabelRef attribute) (ml_switcheroo.compiler.frontends.sass.Predicate attribute), [1] (ml_switcheroo.compiler.frontends.sass.Register attribute), [1] (ml_switcheroo.compiler.ir.LogicalGraph attribute) (ml_switcheroo.compiler.LogicalGraph attribute) (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.dsl.PatternDef attribute) (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) (ml_switcheroo.core.ghost.GhostParam attribute) (ml_switcheroo.core.ghost.GhostRef attribute) (ml_switcheroo.core.graph.LogicalGraph attribute) (ml_switcheroo.core.latex.InputNode attribute) (ml_switcheroo.core.latex.ModelContainer attribute) (ml_switcheroo.core.latex.nodes.InputNode attribute) (ml_switcheroo.core.latex.nodes.ModelContainer attribute) (ml_switcheroo.core.mlir.dialect.OpSchema attribute) (ml_switcheroo.core.mlir.nodes.AttributeNode attribute) (ml_switcheroo.core.mlir.nodes.OperationNode attribute) (ml_switcheroo.core.mlir.nodes.ValueNode attribute) (ml_switcheroo.discovery.consensus.CandidateStandard attribute) (ml_switcheroo.semantics.schema.FrameworkAlias attribute) name_trivia (ml_switcheroo.core.mlir.nodes.OperationNode attribute) NamingContext (class in ml_switcheroo.core.mlir.naming) negated (ml_switcheroo.compiler.frontends.sass.nodes.Predicate attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Register attribute), [1] (ml_switcheroo.compiler.frontends.sass.Predicate attribute), [1] (ml_switcheroo.compiler.frontends.sass.Register attribute), [1] NEQ (ml_switcheroo.enums.LogicOp attribute) NEURAL (ml_switcheroo.enums.SemanticTier attribute) NEURAL_OPS (ml_switcheroo.enums.SemanticTier attribute) new_forward (ml_switcheroo.compiler.backends.python.ClassBodyReplacer attribute) new_init (ml_switcheroo.compiler.backends.python.ClassBodyReplacer attribute) new_node (ml_switcheroo.compiler.differ.ReplaceAction attribute) (ml_switcheroo.core.rewriter.patcher.ReplaceAction attribute), [1] NEWLINE (ml_switcheroo.core.mlir.tokens.TokenKind attribute) node_id (ml_switcheroo.compiler.differ.PatchAction attribute) (ml_switcheroo.core.latex.ComputeNode attribute) (ml_switcheroo.core.latex.MemoryNode attribute) (ml_switcheroo.core.latex.nodes.ComputeNode attribute) (ml_switcheroo.core.latex.nodes.MemoryNode attribute) (ml_switcheroo.core.latex.nodes.StateOpNode attribute) (ml_switcheroo.core.latex.StateOpNode attribute) (ml_switcheroo.core.rewriter.patcher.PatchAction attribute) (ml_switcheroo.core.tikz.nodes.TikzNode attribute) (ml_switcheroo.core.tikz.TikzNode attribute) node_map (ml_switcheroo.core.graph.GraphExtractor attribute), [1] nodes (ml_switcheroo.compiler.ir.LogicalGraph attribute) (ml_switcheroo.compiler.LogicalGraph attribute) (ml_switcheroo.core.graph.LogicalGraph attribute) (ml_switcheroo.utils.visualizer.MermaidGenerator attribute) nondeterministic (ml_switcheroo.core.dsl.OperationDef attribute) normalize_arg() (ml_switcheroo.discovery.consensus.ConsensusEngine class method) (ml_switcheroo.discovery.ConsensusEngine class method) normalize_einsum() (in module ml_switcheroo.plugins.einsum) normalize_harness_imports() (ml_switcheroo.utils.code_extractor.CodeExtractor static method) normalize_name() (ml_switcheroo.discovery.consensus.ConsensusEngine class method) (ml_switcheroo.discovery.ConsensusEngine class method) NOT_IN (ml_switcheroo.enums.LogicOp attribute) num_regions (ml_switcheroo.core.mlir.dialect.OpSchema attribute) NUMBER (ml_switcheroo.core.mlir.tokens.TokenKind attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) NumpyAdapter (class in ml_switcheroo.frameworks.numpy) O offset (ml_switcheroo.compiler.frontends.rdna.Memory attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Memory attribute), [1] (ml_switcheroo.compiler.frontends.sass.Memory attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Memory attribute), [1] OnnxSpecImporter (class in ml_switcheroo.importers.onnx_reader) op (ml_switcheroo.core.dsl.Rule attribute) op_def (injector_spec.StandardsInjector attribute) op_name (injector_fw.core.FrameworkInjector attribute) op_type (ml_switcheroo.core.dsl.OperationDef attribute) (ml_switcheroo.core.latex.ComputeNode attribute) (ml_switcheroo.core.latex.MemoryNode attribute) (ml_switcheroo.core.latex.nodes.ComputeNode attribute) (ml_switcheroo.core.latex.nodes.MemoryNode attribute) opcode (ml_switcheroo.compiler.frontends.rdna.Instruction attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Instruction attribute), [1] (ml_switcheroo.compiler.frontends.sass.Instruction attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Instruction attribute), [1] Operand (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) operands (ml_switcheroo.compiler.frontends.rdna.Instruction attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Instruction attribute), [1] (ml_switcheroo.compiler.frontends.sass.Instruction attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Instruction attribute), [1] (ml_switcheroo.core.mlir.nodes.OperationNode attribute) operation (ml_switcheroo.core.dsl.OperationDef attribute) OperationDef (class in ml_switcheroo.core.dsl) OperationNode (class in ml_switcheroo.core.mlir.nodes) operations (ml_switcheroo.core.mlir.nodes.BlockNode attribute) operator (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) OpPageRenderer (class in ml_switcheroo.utils.doc_renderer) ops (ml_switcheroo.core.hooks.AutoWireSpec attribute) (ml_switcheroo.core.html.parser.GridParser attribute) OpSchema (class in ml_switcheroo.core.mlir.dialect) optax (in module ml_switcheroo.frameworks.optax_shim) OptaxScanner (class in ml_switcheroo.frameworks.optax_shim) optimize() (ml_switcheroo.core.graph_optimizer.GraphOptimizer method) OPTIMIZER (ml_switcheroo.frameworks.base.StandardCategory attribute) options (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.tikz.nodes.TikzEdge attribute) (ml_switcheroo.core.tikz.nodes.TikzGraph attribute) (ml_switcheroo.core.tikz.nodes.TikzNode attribute) (ml_switcheroo.core.tikz.TikzEdge attribute) (ml_switcheroo.core.tikz.TikzGraph attribute) (ml_switcheroo.core.tikz.TikzNode attribute) OpType (class in ml_switcheroo.core.dsl) output_cast (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) output_select_index (ml_switcheroo.core.dsl.FrameworkVariant attribute) output_shape_calc (ml_switcheroo.core.dsl.OperationDef attribute) output_var (ml_switcheroo.compiler.differ.ReplaceAction attribute) (ml_switcheroo.core.rewriter.patcher.ReplaceAction attribute), [1] P pack_as (ml_switcheroo.core.dsl.FrameworkVariant attribute) pack_to_tuple (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) ParameterDef (class in ml_switcheroo.core.dsl) params (ml_switcheroo.compiler.frontends.rdna.Directive attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Directive attribute), [1] (ml_switcheroo.compiler.frontends.sass.Directive attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Directive attribute), [1] (ml_switcheroo.core.ghost.GhostRef attribute) parent (ml_switcheroo.analysis.symbol_table.Scope attribute) parent_id (ml_switcheroo.core.tracer.TraceEvent attribute) parent_style (ml_switcheroo.core.html.nodes.SvgArrow attribute) (ml_switcheroo.core.html.SvgArrow attribute) parse() (ml_switcheroo.compiler.frontends.rdna.parser.RdnaParser method) (ml_switcheroo.compiler.frontends.rdna.RdnaParser method) (ml_switcheroo.compiler.frontends.sass.parser.SassParser method) (ml_switcheroo.compiler.frontends.sass.SassParser method) (ml_switcheroo.core.engine.ASTEngine method) (ml_switcheroo.core.html.parser.HtmlParser method) (ml_switcheroo.core.latex.parser.LatexParser method) (ml_switcheroo.core.mlir.parser.MlirParser method) (ml_switcheroo.core.tikz.parser.TikzParser method) parse_arg_def() (in module ml_switcheroo.generated_tests.inputs) parse_block() (ml_switcheroo.core.mlir.parser.MlirParser method) parse_cli_key_values() (in module ml_switcheroo.config) parse_file() (ml_switcheroo.importers.onnx_reader.OnnxSpecImporter method) (ml_switcheroo.importers.sass_reader.SassSpecImporter method) (ml_switcheroo.importers.stablehlo_reader.StableHloSpecImporter method) parse_folder() (ml_switcheroo.importers.array_api_reader.ArrayApiSpecImporter method) parse_operation() (ml_switcheroo.core.mlir.parser.MlirParser method) parse_plugin_settings() (ml_switcheroo.config.RuntimeConfig method) parse_region() (ml_switcheroo.core.mlir.parser.MlirParser method) parse_to_graph() (ml_switcheroo.compiler.frontends.python.PythonFrontend method) (ml_switcheroo.compiler.registry.GraphFrontend method) passes (ml_switcheroo.core.rewriter.pipeline.RewriterPipeline attribute) (ml_switcheroo.core.rewriter.RewriterPipeline attribute) PatchAction (class in ml_switcheroo.compiler.differ) (class in ml_switcheroo.core.rewriter.patcher) path (ml_switcheroo.analysis.symbol_table.ModuleType attribute) path_to_alias (ml_switcheroo.core.import_fixer.resolution.ResolutionPlan attribute) (ml_switcheroo.core.import_fixer.ResolutionPlan attribute) PATTERN_DEFS (ml_switcheroo.core.mlir.parser.Tokenizer attribute) PatternDef (class in ml_switcheroo.core.dsl) PATTERNS (ml_switcheroo.compiler.frontends.rdna.RdnaLexer attribute) (ml_switcheroo.compiler.frontends.rdna.tokens.RdnaLexer attribute) (ml_switcheroo.compiler.frontends.sass.SassLexer attribute) (ml_switcheroo.compiler.frontends.sass.tokens.SassLexer attribute) patterns (ml_switcheroo.core.graph_optimizer.GraphOptimizer attribute) PATTERNS (ml_switcheroo.core.tikz.parser.TikzLexer attribute) patterns (ml_switcheroo.semantics.manager.SemanticsManager attribute) (ml_switcheroo.semantics.schema.SemanticsFile attribute) PaxmlAdapter (class in ml_switcheroo.frameworks.paxml) peek() (ml_switcheroo.core.mlir.parser.MlirParser method) persist() (ml_switcheroo.semantics.autogen.SemanticPersister method) PHASE_END (ml_switcheroo.core.tracer.TraceEventType attribute) PHASE_START (ml_switcheroo.core.tracer.TraceEventType attribute) plan (ml_switcheroo.core.import_fixer.base.BaseImportFixer attribute) (ml_switcheroo.core.rewriter.patcher.GraphPatcher attribute) plugin_paths (ml_switcheroo.config.RuntimeConfig attribute) plugin_settings (ml_switcheroo.config.RuntimeConfig attribute) plugin_traits (ml_switcheroo.core.hooks.HookContext property) (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) (ml_switcheroo.semantics.schema.FrameworkTraits attribute) PluginGenerator (class in injector_plugin) plugins_dir (injector_plugin.PluginGenerator attribute) PluginScaffoldDef (class in ml_switcheroo.core.dsl) PluginTraits (class in ml_switcheroo.semantics.schema) PluginType (class in ml_switcheroo.core.dsl) pos (ml_switcheroo.compiler.frontends.rdna.parser.RdnaParser attribute) (ml_switcheroo.compiler.frontends.rdna.RdnaParser attribute) (ml_switcheroo.compiler.frontends.sass.parser.SassParser attribute) (ml_switcheroo.compiler.frontends.sass.SassParser attribute) (ml_switcheroo.core.mlir.parser.MlirParser attribute) (ml_switcheroo.core.tikz.parser.TikzLexer attribute) (ml_switcheroo.core.tikz.parser.TikzParser attribute) post module praxis (in module ml_switcheroo.frameworks.paxml) pre module preamble_stmts (ml_switcheroo.core.rewriter.types.SignatureContext attribute) PreambleInjectorType (in module ml_switcheroo.core.hooks) Predicate (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) predicate (ml_switcheroo.compiler.frontends.sass.Instruction attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Instruction attribute), [1] PREDICATE (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) preserve_source (ml_switcheroo.core.import_fixer.base.BaseImportFixer attribute) propose_fix() (ml_switcheroo.testing.bisector.SemanticsBisector method) provenance (ml_switcheroo.core.graph.GraphExtractor attribute), [1] (ml_switcheroo.core.rewriter.patcher.GraphPatcher attribute) (ml_switcheroo.core.tikz.analyser.GraphExtractor attribute) PurityScanner (class in ml_switcheroo.analysis.purity) PythonBackend (class in ml_switcheroo.compiler.backends) (class in ml_switcheroo.compiler.backends.python) PythonFrontend (class in ml_switcheroo.compiler.frontends.python) PythonSnippetEmitter (class in ml_switcheroo.compiler.backends.python_snippet) PythonToMlirEmitter (class in ml_switcheroo.core.mlir.emitter) R rank (ml_switcheroo.core.dsl.ParameterDef attribute) raw_config() (ml_switcheroo.core.hooks.HookContext method) RBRACE (ml_switcheroo.core.mlir.tokens.Symbol attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) RBRACKET (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) (ml_switcheroo.core.mlir.tokens.Symbol attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) RdnaAdapter (class in ml_switcheroo.frameworks.rdna) RdnaAnalyzer (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.analysis) RdnaBackend (class in ml_switcheroo.compiler.backends.rdna) (class in ml_switcheroo.compiler.backends.rdna.synthesizer) RdnaEmitter (class in ml_switcheroo.compiler.backends.rdna) (class in ml_switcheroo.compiler.backends.rdna.emitter) RdnaLexer (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.tokens) RdnaLifter (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.lifter) RdnaNode (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) RdnaParser (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.parser) RdnaSynthesizer (class in ml_switcheroo.compiler.backends.rdna) (class in ml_switcheroo.compiler.backends.rdna.synthesizer) readme_path (ml_switcheroo.utils.readme_editor.ReadmeEditor attribute) ReadmeEditor (class in ml_switcheroo.utils.readme_editor) recommended_alias (ml_switcheroo.frameworks.base.ImportConfig attribute) record_type() (ml_switcheroo.analysis.symbol_table.SymbolTable method) REDUNDANT_SEGMENTS (in module ml_switcheroo.core.import_fixer.utils) regex_pairs (ml_switcheroo.compiler.frontends.rdna.RdnaLexer attribute) (ml_switcheroo.compiler.frontends.rdna.tokens.RdnaLexer attribute) (ml_switcheroo.compiler.frontends.sass.SassLexer attribute) (ml_switcheroo.compiler.frontends.sass.tokens.SassLexer attribute) REGION_TYPE (ml_switcheroo.core.mlir.tokens.TokenKind attribute) RegionNode (class in ml_switcheroo.core.mlir.nodes) regions (ml_switcheroo.core.mlir.nodes.OperationNode attribute) Register (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) REGISTER (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) register() (ml_switcheroo.core.mlir.naming.NamingContext method) register_framework() (in module ml_switcheroo.frameworks) (in module ml_switcheroo.frameworks.base) register_hook() (in module ml_switcheroo.core.hooks) RegisterAllocator (class in ml_switcheroo.compiler.backends.rdna.synthesizer) (class in ml_switcheroo.compiler.backends.sass.synthesizer) RegisterAllocatorProtocol (class in ml_switcheroo.compiler.backends.rdna.macros) (class in ml_switcheroo.compiler.backends.sass.macros) RegistryLoader (class in ml_switcheroo.semantics.registry_loader) render() (ml_switcheroo.cli.matrix.CompatibilityMatrix method) render_rst() (ml_switcheroo.utils.doc_renderer.OpPageRenderer method) repack_attn_flax() (in module ml_switcheroo.plugins.attention_packing) repack_attn_keras() (in module ml_switcheroo.plugins.attention_packing) replace_with (ml_switcheroo.core.dsl.PatternDef attribute) ReplaceAction (class in ml_switcheroo.compiler.differ) (class in ml_switcheroo.core.rewriter.patcher) replaced_by (ml_switcheroo.core.dsl.OperationDef attribute) required_attributes (ml_switcheroo.core.mlir.dialect.OpSchema attribute) required_imports (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.core.import_fixer.resolution.ResolutionPlan attribute) (ml_switcheroo.core.import_fixer.ResolutionPlan attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) requires_explicit_rng (ml_switcheroo.semantics.schema.PluginTraits attribute) requires_functional_control_flow (ml_switcheroo.semantics.schema.PluginTraits attribute) requires_functional_state (ml_switcheroo.semantics.schema.PluginTraits attribute) requires_plugin (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) requires_super_init (ml_switcheroo.semantics.schema.StructuralTraits attribute) reset() (ml_switcheroo.compiler.backends.rdna.synthesizer.RegisterAllocator method) (ml_switcheroo.compiler.backends.sass.synthesizer.RegisterAllocator method) reset_console() (in module ml_switcheroo.utils.console) reset_tracer() (in module ml_switcheroo.core.tracer) ResolutionPlan (class in ml_switcheroo.core.import_fixer) (class in ml_switcheroo.core.import_fixer.resolution) resolve() (ml_switcheroo.core.import_fixer.ImportResolver method) (ml_switcheroo.core.import_fixer.resolution.ImportResolver method) resolve_implicit_method() (in module pre) resolve_roots() (in module ml_switcheroo.cli.handlers.audit) resolve_semantics_dir() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.semantics.paths) resolve_snapshots_dir() (in module ml_switcheroo.cli.commands) (in module ml_switcheroo.semantics.paths) resolve_symbolic_shape() (in module ml_switcheroo.testing.fuzzer.utils) resolve_type() (ml_switcheroo.core.hooks.HookContext method) resolve_variant() (ml_switcheroo.semantics.manager.SemanticsManager method) result_types (ml_switcheroo.core.mlir.nodes.OperationNode attribute) results (ml_switcheroo.analysis.audit.CoverageScanner attribute) (ml_switcheroo.core.mlir.nodes.OperationNode attribute) return_type (ml_switcheroo.core.dsl.OperationDef attribute) ReturnNode (class in ml_switcheroo.core.latex) (class in ml_switcheroo.core.latex.nodes) rewrite_as_infix() (in module transformers) rewrite_as_inline_lambda() (in module transformers) rewrite_as_macro() (in module transformers) rewrite_stateful_call() (in module utils) RewriterContext (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.context) RewriterPass (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.interface) RewriterPipeline (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.pipeline) rng_seed_methods (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) root_fw (ml_switcheroo.discovery.harvester.ImportScanner attribute) root_scope (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer attribute) row (ml_switcheroo.core.html.GridBox attribute) (ml_switcheroo.core.html.nodes.GridBox attribute) ROW_HEIGHT (ml_switcheroo.compiler.backends.extras.HtmlBackend attribute) rows (ml_switcheroo.core.tikz.nodes.TikzTable attribute) (ml_switcheroo.core.tikz.TikzTable attribute) RPAREN (ml_switcheroo.core.mlir.tokens.Symbol attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) Rule (class in ml_switcheroo.core.dsl) rules (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) run() (ml_switcheroo.core.engine.ASTEngine method) (ml_switcheroo.core.rewriter.pipeline.RewriterPipeline method) (ml_switcheroo.core.rewriter.RewriterPipeline method) run_all() (ml_switcheroo.testing.batch_runner.BatchValidator method) runner (ml_switcheroo.testing.batch_runner.BatchValidator attribute) (ml_switcheroo.testing.bisector.SemanticsBisector attribute) RuntimeConfig (class in ml_switcheroo.config) S SassAdapter (class in ml_switcheroo.frameworks.sass) SassAnalyzer (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.analysis) SassBackend (class in ml_switcheroo.compiler.backends.sass) (class in ml_switcheroo.compiler.backends.sass.synthesizer) SassEmitter (class in ml_switcheroo.compiler.backends.sass) (class in ml_switcheroo.compiler.backends.sass.emitter) SassHtmlParser (class in ml_switcheroo.importers.sass_reader) SassLexer (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.tokens) SassLifter (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.lifter) SassNode (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.nodes) SassParser (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.parser) SassSpecImporter (class in ml_switcheroo.importers.sass_reader) SassSynthesizer (class in ml_switcheroo.compiler.backends.sass) (class in ml_switcheroo.compiler.backends.sass.synthesizer) scaffold() (ml_switcheroo.discovery.Scaffolder method) (ml_switcheroo.discovery.scaffolder.Scaffolder method) scaffold_plugins (ml_switcheroo.core.dsl.OperationDef attribute) Scaffolder (class in ml_switcheroo.discovery) (class in ml_switcheroo.discovery.scaffolder) scan_losses() (ml_switcheroo.frameworks.optax_shim.OptaxScanner static method) scan_optimizers() (ml_switcheroo.frameworks.optax_shim.OptaxScanner static method) Scope (class in ml_switcheroo.analysis.symbol_table) scope_stack (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) score (ml_switcheroo.discovery.consensus.CandidateStandard attribute) search_modules (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) SemanticHarvester (class in ml_switcheroo.discovery) (class in ml_switcheroo.discovery.harvester) SemanticPersister (class in ml_switcheroo.semantics.autogen) semantics (ml_switcheroo.analysis.audit.CoverageScanner attribute) (ml_switcheroo.analysis.dependencies.DependencyScanner attribute) (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer attribute) (ml_switcheroo.cli.handlers.convert_weights.WeightScriptGenerator attribute) (ml_switcheroo.cli.matrix.CompatibilityMatrix attribute) (ml_switcheroo.cli.wizard.MappingWizard attribute) (ml_switcheroo.compiler.backends.extras.MlirBackend attribute) (ml_switcheroo.compiler.backends.extras.StableHloBackend attribute) (ml_switcheroo.compiler.backends.rdna.RdnaSynthesizer attribute) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaSynthesizer attribute) (ml_switcheroo.compiler.backends.sass.SassSynthesizer attribute) (ml_switcheroo.compiler.backends.sass.synthesizer.SassSynthesizer attribute) (ml_switcheroo.core.engine.ASTEngine attribute) (ml_switcheroo.core.hooks.HookContext attribute) (ml_switcheroo.core.import_fixer.ImportResolver attribute) (ml_switcheroo.core.import_fixer.resolution.ImportResolver attribute) (ml_switcheroo.core.mlir.stablehlo_emitter.StableHloEmitter attribute) (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer property) (ml_switcheroo.core.rewriter.passes.ApiTransformer property) (ml_switcheroo.core.rewriter.RewriterContext attribute) (ml_switcheroo.discovery.harvester.SemanticHarvester attribute) (ml_switcheroo.discovery.Scaffolder attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) (ml_switcheroo.discovery.SemanticHarvester attribute) (ml_switcheroo.testing.batch_runner.BatchValidator attribute) (ml_switcheroo.utils.doc_context.DocContextBuilder attribute) (ml_switcheroo.utils.doc_gen.MigrationGuideGenerator attribute) (ml_switcheroo.utils.readme_editor.ReadmeEditor attribute) semantics_mgr (ml_switcheroo.generated_tests.generator.TestCaseGenerator attribute) SemanticsBisector (class in ml_switcheroo.testing.bisector) SemanticsFile (class in ml_switcheroo.semantics.schema) SemanticsManager (class in ml_switcheroo.semantics.manager) SemanticsManagerType (in module ml_switcheroo.core.hooks) SemanticTier (class in ml_switcheroo.enums) SEMICOLON (ml_switcheroo.compiler.frontends.sass.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.sass.TokenType attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) sequence (ml_switcheroo.core.dsl.PatternDef attribute) set() (ml_switcheroo.analysis.symbol_table.Scope method) set_console() (in module ml_switcheroo.utils.console) SGPR (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) shape (ml_switcheroo.core.latex.ComputeNode attribute) (ml_switcheroo.core.latex.InputNode attribute) (ml_switcheroo.core.latex.nodes.ComputeNode attribute) (ml_switcheroo.core.latex.nodes.InputNode attribute) (ml_switcheroo.core.latex.nodes.StateOpNode attribute) (ml_switcheroo.core.latex.StateOpNode attribute) shape_spec (ml_switcheroo.core.dsl.ParameterDef attribute) should_traverse() (ml_switcheroo.core.scanners.SimpleNameScanner method) signature (ml_switcheroo.core.import_fixer.ImportReq property) (ml_switcheroo.core.import_fixer.resolution.ImportReq property) signature_stack (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) SignatureContext (class in ml_switcheroo.core.rewriter.types) similarity_threshold (ml_switcheroo.discovery.Scaffolder attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) SimpleNameScanner (class in ml_switcheroo.core.scanners) SimulatedReflection (class in ml_switcheroo.core.discovery) skip_fuzzing (ml_switcheroo.core.dsl.OperationDef attribute) snapshot() (ml_switcheroo.analysis.symbol_table.Scope method) SNAPSHOT_DIR (in module ml_switcheroo.frameworks.base) source (ml_switcheroo.compiler.ir.LogicalEdge attribute) (ml_switcheroo.compiler.LogicalEdge attribute) (ml_switcheroo.core.engine.ASTEngine attribute) (ml_switcheroo.core.graph.LogicalEdge attribute) (ml_switcheroo.core.html.parser.HtmlParser attribute) (ml_switcheroo.core.latex.parser.LatexParser attribute) source_adapter (ml_switcheroo.cli.handlers.convert_weights.WeightScriptGenerator attribute) source_flavour (ml_switcheroo.config.RuntimeConfig attribute) source_framework (ml_switcheroo.config.RuntimeConfig attribute) source_fw (ml_switcheroo.analysis.dependencies.DependencyScanner attribute) (ml_switcheroo.analysis.purity.PurityScanner attribute) (ml_switcheroo.cli.handlers.convert_weights.WeightScriptGenerator attribute) (ml_switcheroo.core.hooks.HookContext attribute) (ml_switcheroo.core.rewriter.context.RewriterContext property) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer property) (ml_switcheroo.core.rewriter.passes.ApiTransformer property) (ml_switcheroo.core.rewriter.RewriterContext property) (ml_switcheroo.core.scanners.UsageScanner attribute) source_id (ml_switcheroo.core.tikz.nodes.TikzEdge attribute) (ml_switcheroo.core.tikz.TikzEdge attribute) source_traits (ml_switcheroo.core.rewriter.passes.api.ApiTransformer property) (ml_switcheroo.core.rewriter.passes.ApiTransformer property) SPECIAL_REG (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) specifications (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) split_outside_brackets() (in module ml_switcheroo.testing.fuzzer.utils) SSAContext (class in ml_switcheroo.core.mlir.emitter) StableHloAdapter (class in ml_switcheroo.frameworks.stablehlo) StableHloBackend (class in ml_switcheroo.compiler.backends.extras) StableHloEmitter (class in ml_switcheroo.core.mlir.stablehlo_emitter) StableHloSpecImporter (class in ml_switcheroo.importers.stablehlo_reader) stack (ml_switcheroo.utils.visualizer.MermaidGenerator attribute) staged_mappings (ml_switcheroo.discovery.Scaffolder attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) staged_specs (ml_switcheroo.discovery.Scaffolder attribute) (ml_switcheroo.discovery.scaffolder.Scaffolder attribute) StandardCategory (class in ml_switcheroo.frameworks.base) StandardMap (class in ml_switcheroo.frameworks.base) StandardsInjector (class in injector_spec) start() (ml_switcheroo.cli.wizard.MappingWizard method) START_MARKER (ml_switcheroo.core.escape_hatch.EscapeHatch attribute) start_phase() (ml_switcheroo.core.tracer.TraceLogger method) stateful_call (ml_switcheroo.semantics.schema.FrameworkTraits attribute) StatementGeneratorMixin (class in ml_switcheroo.core.mlir.gen_statements) StateOpNode (class in ml_switcheroo.core.latex) (class in ml_switcheroo.core.latex.nodes) std_args (ml_switcheroo.core.dsl.OperationDef attribute) (ml_switcheroo.discovery.consensus.CandidateStandard attribute) std_args_info (ml_switcheroo.discovery.harvester.TargetCallVisitor attribute) strategies_from_spec() (in module ml_switcheroo.testing.fuzzer.strategies) strategy module strict_materialization_method (ml_switcheroo.semantics.schema.PluginTraits attribute) strict_mode (ml_switcheroo.config.RuntimeConfig attribute) (ml_switcheroo.core.engine.ASTEngine property) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer property) (ml_switcheroo.core.rewriter.passes.ApiTransformer property) STRICT_RANK_HELPER (in module guards) STRING (ml_switcheroo.core.mlir.tokens.TokenKind attribute) strip_kwarg() (in module utils) strip_magic_args (ml_switcheroo.semantics.schema.StructuralTraits attribute) strip_zero_grad() (in module ml_switcheroo.plugins.optimizer_step) structural_traits (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) StructuralLinter (class in ml_switcheroo.testing.linter) StructuralPass (class in ml_switcheroo.core.rewriter) (class in ml_switcheroo.core.rewriter.passes) (class in ml_switcheroo.core.rewriter.passes.structure) StructuralTraits (class in ml_switcheroo.semantics.schema) StructuralTransformer (class in ml_switcheroo.core.rewriter.passes) (class in ml_switcheroo.core.rewriter.passes.structure) style_class (ml_switcheroo.core.html.nodes.SvgArrow attribute) (ml_switcheroo.core.html.SvgArrow attribute) STYLES (ml_switcheroo.utils.visualizer.MermaidGenerator attribute) subcomponent (ml_switcheroo.core.import_fixer.ImportReq attribute) (ml_switcheroo.core.import_fixer.resolution.ImportReq attribute) success (ml_switcheroo.core.conversion_result.ConversionResult attribute) SUCCESS_LEVEL_NUM (in module ml_switcheroo.utils.console) supported_tiers (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) SvgArrow (class in ml_switcheroo.core.html) (class in ml_switcheroo.core.html.nodes) SW_CALL (in module ml_switcheroo.core.mlir.dialect) SW_CONSTANT (in module ml_switcheroo.core.mlir.dialect) SW_FUNC (in module ml_switcheroo.core.mlir.dialect) SW_GETATTR (in module ml_switcheroo.core.mlir.dialect) SW_IMPORT (in module ml_switcheroo.core.mlir.dialect) SW_MODULE (in module ml_switcheroo.core.mlir.dialect) SW_OP (in module ml_switcheroo.core.mlir.dialect) SW_RETURN (in module ml_switcheroo.core.mlir.dialect) SW_SETATTR (in module ml_switcheroo.core.mlir.dialect) SYM_ID (ml_switcheroo.core.mlir.tokens.TokenKind attribute) Symbol (class in ml_switcheroo.core.mlir.tokens) SYMBOL (ml_switcheroo.core.mlir.tokens.TokenKind attribute) symbol_table (ml_switcheroo.core.rewriter.context.RewriterContext attribute) (ml_switcheroo.core.rewriter.RewriterContext attribute) symbols (ml_switcheroo.analysis.symbol_table.Scope attribute) SymbolTable (class in ml_switcheroo.analysis.symbol_table) SymbolTableAnalyzer (class in ml_switcheroo.analysis.symbol_table) SymbolTableType (in module ml_switcheroo.core.hooks) SymbolType (class in ml_switcheroo.analysis.symbol_table) sync() (ml_switcheroo.discovery.FrameworkSyncer method) (ml_switcheroo.discovery.syncer.FrameworkSyncer method) synthesizer (ml_switcheroo.compiler.backends.rdna.RdnaBackend attribute) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaBackend attribute) (ml_switcheroo.compiler.backends.sass.SassBackend attribute) (ml_switcheroo.compiler.backends.sass.synthesizer.SassBackend attribute) T T (in module ml_switcheroo.config) (in module ml_switcheroo.core.hooks) table (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer attribute) target (ml_switcheroo.compiler.ir.LogicalEdge attribute) (ml_switcheroo.compiler.LogicalEdge attribute) (ml_switcheroo.core.engine.ASTEngine attribute) (ml_switcheroo.core.graph.LogicalEdge attribute) target_adapter (ml_switcheroo.cli.handlers.convert_weights.WeightScriptGenerator attribute) target_api (ml_switcheroo.discovery.harvester.TargetCallVisitor attribute) target_arch (ml_switcheroo.compiler.backends.rdna.RdnaBackend attribute) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaBackend attribute) (ml_switcheroo.frameworks.rdna.RdnaAdapter attribute) target_class (ml_switcheroo.compiler.backends.python.ClassBodyReplacer attribute) target_flavour (ml_switcheroo.config.RuntimeConfig attribute) target_framework (ml_switcheroo.config.RuntimeConfig attribute) target_fw (injector_fw.core.FrameworkInjector attribute) (ml_switcheroo.cli.handlers.convert_weights.WeightScriptGenerator attribute) (ml_switcheroo.core.hooks.HookContext attribute) (ml_switcheroo.core.rewriter.context.RewriterContext property) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer property) (ml_switcheroo.core.rewriter.passes.ApiTransformer property) (ml_switcheroo.core.rewriter.RewriterContext property) (ml_switcheroo.discovery.harvester.SemanticHarvester attribute) (ml_switcheroo.discovery.SemanticHarvester attribute) target_id (ml_switcheroo.core.latex.nodes.ReturnNode attribute) (ml_switcheroo.core.latex.ReturnNode attribute) (ml_switcheroo.core.tikz.nodes.TikzEdge attribute) (ml_switcheroo.core.tikz.TikzEdge attribute) target_name (ml_switcheroo.core.scanners.SimpleNameScanner attribute) target_traits (ml_switcheroo.core.rewriter.passes.StructuralTransformer property) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer property) TargetCallVisitor (class in ml_switcheroo.discovery.harvester) TEMPLATE_FUNC_DEF (in module injector_plugin) TEMPLATE_FUNC_DEF_AUTO_WIRE (in module injector_plugin) TEMPLATE_HEADER (in module injector_plugin) TensorFlowAdapter (class in ml_switcheroo.frameworks.tensorflow) TensorType (class in ml_switcheroo.analysis.symbol_table) test_atol (ml_switcheroo.core.dsl.OperationDef attribute) test_config (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) test_rtol (ml_switcheroo.core.dsl.OperationDef attribute) test_templates (ml_switcheroo.semantics.manager.SemanticsManager attribute) TestCaseGenerator (class in ml_switcheroo.generated_tests.generator) text (ml_switcheroo.compiler.frontends.rdna.Comment attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Comment attribute), [1] (ml_switcheroo.compiler.frontends.sass.Comment attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Comment attribute), [1] (ml_switcheroo.core.mlir.parser.Token attribute) (ml_switcheroo.core.mlir.parser.Tokenizer attribute) (ml_switcheroo.core.tikz.nodes.TikzComment attribute) (ml_switcheroo.core.tikz.parser.TikzLexer attribute) (ml_switcheroo.core.tikz.parser.Token attribute) (ml_switcheroo.core.tikz.TikzComment attribute) tf (in module ml_switcheroo.frameworks.tensorflow) tier (injector_spec.StandardsInjector attribute) (ml_switcheroo.frameworks.base.ImportConfig attribute) TIER_PRECEDENCE (in module ml_switcheroo.semantics.merging) tiers (ml_switcheroo.semantics.schema.FrameworkTraits attribute) TikzAdapter (class in ml_switcheroo.frameworks.tikz) TikzBackend (class in ml_switcheroo.compiler.backends.extras) TikzBaseNode (class in ml_switcheroo.core.tikz) (class in ml_switcheroo.core.tikz.nodes) TikzComment (class in ml_switcheroo.core.tikz) (class in ml_switcheroo.core.tikz.nodes) TikzEdge (class in ml_switcheroo.core.tikz) (class in ml_switcheroo.core.tikz.nodes) TikzGraph (class in ml_switcheroo.core.tikz) (class in ml_switcheroo.core.tikz.nodes) TikzLexer (class in ml_switcheroo.core.tikz.parser) TikzNode (class in ml_switcheroo.core.tikz) (class in ml_switcheroo.core.tikz.nodes) TikzOption (class in ml_switcheroo.core.tikz) (class in ml_switcheroo.core.tikz.nodes) TikzParser (class in ml_switcheroo.core.tikz.parser) TikzTable (class in ml_switcheroo.core.tikz) (class in ml_switcheroo.core.tikz.nodes) timestamp (ml_switcheroo.core.tracer.TraceEvent attribute) to_html() (ml_switcheroo.core.html.GridBox method) (ml_switcheroo.core.html.HtmlDocument method) (ml_switcheroo.core.html.HtmlNode method) (ml_switcheroo.core.html.nodes.GridBox method) (ml_switcheroo.core.html.nodes.HtmlDocument method) (ml_switcheroo.core.html.nodes.HtmlNode method) (ml_switcheroo.core.html.nodes.SvgArrow method) (ml_switcheroo.core.html.SvgArrow method) to_latex() (ml_switcheroo.core.latex.ComputeNode method) (ml_switcheroo.core.latex.InputNode method) (ml_switcheroo.core.latex.LatexNode method) (ml_switcheroo.core.latex.MemoryNode method) (ml_switcheroo.core.latex.ModelContainer method) (ml_switcheroo.core.latex.nodes.ComputeNode method) (ml_switcheroo.core.latex.nodes.InputNode method) (ml_switcheroo.core.latex.nodes.LatexNode method) (ml_switcheroo.core.latex.nodes.MemoryNode method) (ml_switcheroo.core.latex.nodes.ModelContainer method) (ml_switcheroo.core.latex.nodes.ReturnNode method) (ml_switcheroo.core.latex.nodes.StateOpNode method) (ml_switcheroo.core.latex.ReturnNode method) (ml_switcheroo.core.latex.StateOpNode method) to_python() (ml_switcheroo.compiler.backends.rdna.RdnaSynthesizer method) (ml_switcheroo.compiler.backends.rdna.synthesizer.RdnaSynthesizer method) (ml_switcheroo.compiler.backends.sass.SassSynthesizer method) (ml_switcheroo.compiler.backends.sass.synthesizer.SassSynthesizer method) to_source() (ml_switcheroo.core.engine.ASTEngine method) to_text() (ml_switcheroo.core.mlir.nodes.AttributeNode method) (ml_switcheroo.core.mlir.nodes.BlockNode method) (ml_switcheroo.core.mlir.nodes.MlirNode method) (ml_switcheroo.core.mlir.nodes.ModuleNode method) (ml_switcheroo.core.mlir.nodes.OperationNode method) (ml_switcheroo.core.mlir.nodes.RegionNode method) (ml_switcheroo.core.mlir.nodes.TriviaNode method) (ml_switcheroo.core.mlir.nodes.TypeNode method) (ml_switcheroo.core.mlir.nodes.ValueNode method) (ml_switcheroo.core.tikz.nodes.TikzBaseNode method) (ml_switcheroo.core.tikz.nodes.TikzComment method) (ml_switcheroo.core.tikz.nodes.TikzEdge method) (ml_switcheroo.core.tikz.nodes.TikzGraph method) (ml_switcheroo.core.tikz.nodes.TikzNode method) (ml_switcheroo.core.tikz.nodes.TikzOption method) (ml_switcheroo.core.tikz.nodes.TikzTable method) (ml_switcheroo.core.tikz.nodes.TriviaNode method) (ml_switcheroo.core.tikz.TikzBaseNode method) (ml_switcheroo.core.tikz.TikzComment method) (ml_switcheroo.core.tikz.TikzEdge method) (ml_switcheroo.core.tikz.TikzGraph method) (ml_switcheroo.core.tikz.TikzNode method) (ml_switcheroo.core.tikz.TikzOption method) (ml_switcheroo.core.tikz.TikzTable method) (ml_switcheroo.core.tikz.TriviaNode method) Token (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.tokens) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.tokens) (class in ml_switcheroo.core.mlir.parser) (class in ml_switcheroo.core.tikz.parser) tokenize() (ml_switcheroo.compiler.frontends.rdna.RdnaLexer method) (ml_switcheroo.compiler.frontends.rdna.tokens.RdnaLexer method) (ml_switcheroo.compiler.frontends.sass.SassLexer method) (ml_switcheroo.compiler.frontends.sass.tokens.SassLexer method) (ml_switcheroo.core.mlir.parser.Tokenizer method) (ml_switcheroo.core.tikz.parser.TikzLexer method) Tokenizer (class in ml_switcheroo.core.mlir.parser) tokenizer (ml_switcheroo.core.mlir.parser.MlirParser attribute) TokenKind (class in ml_switcheroo.core.mlir.tokens) (class in ml_switcheroo.core.tikz.parser) tokens (ml_switcheroo.compiler.frontends.rdna.parser.RdnaParser attribute) (ml_switcheroo.compiler.frontends.rdna.RdnaParser attribute) (ml_switcheroo.compiler.frontends.sass.parser.SassParser attribute) (ml_switcheroo.compiler.frontends.sass.SassParser attribute) (ml_switcheroo.core.mlir.parser.MlirParser attribute) (ml_switcheroo.core.tikz.parser.TikzParser attribute) TokenType (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.tokens) (class in ml_switcheroo.compiler.frontends.sass) (class in ml_switcheroo.compiler.frontends.sass.tokens) tomllib (in module ml_switcheroo.config) topological_sort() (in module ml_switcheroo.compiler.ir) (in module ml_switcheroo.core.graph) torch (in module ml_switcheroo.frameworks.torch) TorchAdapter (class in ml_switcheroo.frameworks.torch) trace_events (ml_switcheroo.core.conversion_result.ConversionResult attribute) TraceEvent (class in ml_switcheroo.core.tracer) TraceEventType (class in ml_switcheroo.core.tracer) TraceLogger (class in ml_switcheroo.core.tracer) trailing_newline (ml_switcheroo.core.tikz.nodes.TikzComment attribute) (ml_switcheroo.core.tikz.TikzComment attribute) trailing_trivia (ml_switcheroo.core.mlir.nodes.OperationNode attribute) traits (ml_switcheroo.semantics.schema.FrameworkTraits attribute) transform() (ml_switcheroo.core.rewriter.ApiPass method) (ml_switcheroo.core.rewriter.AuxiliaryPass method) (ml_switcheroo.core.rewriter.interface.RewriterPass method) (ml_switcheroo.core.rewriter.passes.api.ApiPass method) (ml_switcheroo.core.rewriter.passes.ApiPass method) (ml_switcheroo.core.rewriter.passes.auxiliary.AuxiliaryPass method) (ml_switcheroo.core.rewriter.passes.AuxiliaryPass method) (ml_switcheroo.core.rewriter.passes.StructuralPass method) (ml_switcheroo.core.rewriter.passes.structure.StructuralPass method) (ml_switcheroo.core.rewriter.RewriterPass method) (ml_switcheroo.core.rewriter.StructuralPass method) transform_batch_norm() (in module ml_switcheroo.plugins.batch_norm) transform_casting() (in module ml_switcheroo.plugins.casting) transform_checkpoint_keys() (in module ml_switcheroo.plugins.checkpoint_keys) transform_compiler() (in module ml_switcheroo.plugins.mlx_extras) transform_context_manager() (in module ml_switcheroo.plugins.context_to_function_wrap) transform_cuda_check() (in module ml_switcheroo.plugins.device_checks) transform_dataloader() (in module ml_switcheroo.plugins.data_loader) transform_device_allocator() (in module ml_switcheroo.plugins.device_allocator) transform_flatten() (in module ml_switcheroo.plugins.flatten) transform_gather() (in module ml_switcheroo.plugins.gather) transform_grad_clipping() (in module ml_switcheroo.plugins.clipping) transform_io_calls() (in module ml_switcheroo.plugins.io_handler) transform_keras_sequential() (in module ml_switcheroo.plugins.keras_sequential) transform_loops() (in module ml_switcheroo.plugins.loop_unroll) transform_loss_reduction() (in module ml_switcheroo.plugins.loss_wrapper) transform_method_to_property() (in module ml_switcheroo.plugins.method_property) transform_mlx_optimizer_init() (in module ml_switcheroo.plugins.mlx_optimizers) transform_mlx_optimizer_step() (in module ml_switcheroo.plugins.mlx_optimizers) transform_mlx_zero_grad() (in module ml_switcheroo.plugins.mlx_optimizers) transform_nnx_param() (in module ml_switcheroo.plugins.nnx_to_torch_params) transform_optimizer_init() (in module ml_switcheroo.plugins.optimizer_step) transform_optimizer_step() (in module ml_switcheroo.plugins.optimizer_step) transform_padding() (in module ml_switcheroo.plugins.padding) transform_scatter() (in module ml_switcheroo.plugins.scatter) transform_scheduler_init() (in module ml_switcheroo.plugins.schedulers) transform_scheduler_step() (in module ml_switcheroo.plugins.schedulers) transform_shape_packing() (in module ml_switcheroo.plugins.shape_packing) transform_synchronize() (in module ml_switcheroo.plugins.mlx_extras) transform_tf_dataloader() (in module ml_switcheroo.plugins.tf_data_loader) transform_topk() (in module ml_switcheroo.plugins.topk) transform_view_semantics() (in module ml_switcheroo.plugins.reshape) transformation_type (ml_switcheroo.core.dsl.FrameworkVariant attribute) (ml_switcheroo.frameworks.base.StandardMap attribute) transformers module trivia_buffer (ml_switcheroo.core.mlir.parser.MlirParser attribute) TriviaNode (class in ml_switcheroo.core.mlir.nodes) (class in ml_switcheroo.core.tikz) (class in ml_switcheroo.core.tikz.nodes) TUPLE (ml_switcheroo.core.dsl.ContainerType attribute) type (ml_switcheroo.core.dsl.ParameterDef attribute) (ml_switcheroo.core.dsl.PluginScaffoldDef attribute) TYPE (ml_switcheroo.core.mlir.tokens.TokenKind attribute) type (ml_switcheroo.core.tracer.TraceEvent attribute) type_annotation (ml_switcheroo.core.mlir.nodes.AttributeNode attribute) TypeNode (class in ml_switcheroo.core.mlir.nodes) types (ml_switcheroo.analysis.symbol_table.UnionType attribute) U ui_priority (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.flax_nnx.FlaxNNXAdapter attribute) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter attribute) (ml_switcheroo.frameworks.jax.JaxCoreAdapter attribute) (ml_switcheroo.frameworks.keras.KerasAdapter attribute) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter attribute) (ml_switcheroo.frameworks.mlir.MlirAdapter attribute) (ml_switcheroo.frameworks.mlx.MLXAdapter attribute) (ml_switcheroo.frameworks.numpy.NumpyAdapter attribute) (ml_switcheroo.frameworks.paxml.PaxmlAdapter attribute) (ml_switcheroo.frameworks.rdna.RdnaAdapter attribute) (ml_switcheroo.frameworks.sass.SassAdapter attribute) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter attribute) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter attribute) (ml_switcheroo.frameworks.tikz.TikzAdapter attribute) (ml_switcheroo.frameworks.torch.TorchAdapter attribute) UnionType (class in ml_switcheroo.analysis.symbol_table) unknown_imports (ml_switcheroo.analysis.dependencies.DependencyScanner attribute) unroll_inplace_ops() (in module ml_switcheroo.plugins.inplace_unroll) unroll_static_loops() (in module ml_switcheroo.plugins.static_unroll) unsafe_submodules (ml_switcheroo.frameworks.base.FrameworkAdapter property) (ml_switcheroo.frameworks.FrameworkAdapter property) (ml_switcheroo.frameworks.html_dsl.HtmlDSLAdapter property) (ml_switcheroo.frameworks.jax.JaxCoreAdapter property) (ml_switcheroo.frameworks.keras.KerasAdapter property) (ml_switcheroo.frameworks.latex_dsl.LatexDSLAdapter property) (ml_switcheroo.frameworks.mlir.MlirAdapter property) (ml_switcheroo.frameworks.mlx.MLXAdapter property) (ml_switcheroo.frameworks.numpy.NumpyAdapter property) (ml_switcheroo.frameworks.paxml.PaxmlAdapter property) (ml_switcheroo.frameworks.rdna.RdnaAdapter property) (ml_switcheroo.frameworks.sass.SassAdapter property) (ml_switcheroo.frameworks.stablehlo.StableHloAdapter property) (ml_switcheroo.frameworks.tensorflow.TensorFlowAdapter property) (ml_switcheroo.frameworks.tikz.TikzAdapter property) (ml_switcheroo.frameworks.torch.TorchAdapter property) update_definition() (ml_switcheroo.semantics.manager.SemanticsManager method) update_matrix() (ml_switcheroo.utils.readme_editor.ReadmeEditor method) usage_consumers (ml_switcheroo.core.mlir.gen_statements.StatementGeneratorMixin attribute) (ml_switcheroo.core.mlir.generator.MlirToPythonGenerator attribute) usage_counts (ml_switcheroo.core.mlir.gen_statements.StatementGeneratorMixin attribute) (ml_switcheroo.core.mlir.generator.MlirToPythonGenerator attribute) UsageScanner (class in ml_switcheroo.core.scanners) use_api (ml_switcheroo.core.dsl.Rule attribute) utils module V VAL_ID (ml_switcheroo.core.mlir.tokens.TokenKind attribute) validate() (ml_switcheroo.core.mlir.dialect.OpSchema method) validate_framework() (ml_switcheroo.config.RuntimeConfig class method) validate_op() (ml_switcheroo.core.mlir.dialect.DialectRegistry class method) validate_settings() (ml_switcheroo.core.hooks.HookContext method) validate_transpilation() (in module ml_switcheroo.testing.linter) validation_report (ml_switcheroo.config.RuntimeConfig attribute) value (ml_switcheroo.compiler.frontends.rdna.Immediate attribute), [1] (ml_switcheroo.compiler.frontends.rdna.nodes.Immediate attribute), [1] (ml_switcheroo.compiler.frontends.rdna.Token attribute), [1] (ml_switcheroo.compiler.frontends.rdna.tokens.Token attribute), [1] (ml_switcheroo.compiler.frontends.sass.Immediate attribute), [1] (ml_switcheroo.compiler.frontends.sass.nodes.Immediate attribute), [1] (ml_switcheroo.compiler.frontends.sass.Token attribute), [1] (ml_switcheroo.compiler.frontends.sass.tokens.Token attribute), [1] (ml_switcheroo.core.mlir.nodes.AttributeNode attribute) (ml_switcheroo.core.tikz.nodes.TikzOption attribute) (ml_switcheroo.core.tikz.TikzOption attribute) (ml_switcheroo.plugins.static_unroll.LoopVarReplacer attribute) ValueNode (class in ml_switcheroo.core.mlir.nodes) var_name (ml_switcheroo.plugins.static_unroll.LoopVarReplacer attribute) variant (injector_fw.core.FrameworkInjector attribute) variants (ml_switcheroo.core.dsl.OperationDef attribute) (ml_switcheroo.discovery.consensus.CandidateStandard attribute) verification_mode (ml_switcheroo.core.dsl.OperationDef attribute) verify() (ml_switcheroo.testing.runner.EquivalenceRunner method) verify_results() (in module ml_switcheroo.generated_tests.runtime) VGPR (class in ml_switcheroo.compiler.frontends.rdna) (class in ml_switcheroo.compiler.frontends.rdna.nodes) (ml_switcheroo.compiler.frontends.rdna.tokens.TokenType attribute) (ml_switcheroo.compiler.frontends.rdna.TokenType attribute) violations (ml_switcheroo.testing.linter.StructuralLinter attribute) visit_AnnAssign() (ml_switcheroo.analysis.lifecycle.InitializationTracker method) visit_Annotation() (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) visit_Arg() (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_Assign() (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.core.graph.GraphExtractor method) (ml_switcheroo.core.tikz.analyser.GraphExtractor method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_Attribute() (ml_switcheroo.analysis.audit.CoverageScanner method) (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.testing.linter.StructuralLinter method) visit_Call() (ml_switcheroo.analysis.audit.CoverageScanner method) (ml_switcheroo.analysis.purity.PurityScanner method) (ml_switcheroo.discovery.harvester.TargetCallVisitor method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_ClassDef() (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.graph.GraphExtractor method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) (ml_switcheroo.core.tikz.analyser.GraphExtractor method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_Expr() (ml_switcheroo.core.graph.GraphExtractor method) visit_For() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) visit_FunctionDef() (injector_plugin.BodyExtractor method) (ml_switcheroo.analysis.lifecycle.InitializationTracker method) (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) (ml_switcheroo.core.graph.GraphExtractor method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.StructuralTransformer method) (ml_switcheroo.core.rewriter.passes.structure.StructuralTransformer method) (ml_switcheroo.core.tikz.analyser.GraphExtractor method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_Global() (ml_switcheroo.analysis.purity.PurityScanner method) visit_If() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) visit_Import() (ml_switcheroo.analysis.audit.CoverageScanner method) (ml_switcheroo.analysis.dependencies.DependencyScanner method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.discovery.harvester.ImportScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_ImportFrom() (ml_switcheroo.analysis.audit.CoverageScanner method) (ml_switcheroo.analysis.dependencies.DependencyScanner method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.discovery.harvester.ImportScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_Module() (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_Name() (ml_switcheroo.core.scanners.SimpleNameScanner method) (ml_switcheroo.core.scanners.UsageScanner method) (ml_switcheroo.testing.linter.StructuralLinter method) visit_Nonlocal() (ml_switcheroo.analysis.purity.PurityScanner method) visit_Return() (ml_switcheroo.core.graph.GraphExtractor method) (ml_switcheroo.core.tikz.analyser.GraphExtractor method) visit_SimpleStatementLine() (ml_switcheroo.analysis.purity.PurityScanner method) (ml_switcheroo.core.rewriter.passes.api.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.ApiTransformer method) (ml_switcheroo.core.rewriter.passes.auxiliary.AuxiliaryTransformer method) (ml_switcheroo.core.rewriter.passes.AuxiliaryTransformer method) visit_SimpleString() (ml_switcheroo.utils.visualizer.MermaidGenerator method) visit_While() (ml_switcheroo.analysis.symbol_table.SymbolTableAnalyzer method) W warnings (ml_switcheroo.analysis.lifecycle.InitializationTracker attribute) WeightScriptGenerator (class in ml_switcheroo.cli.handlers.convert_weights) WHITESPACE (ml_switcheroo.core.mlir.tokens.TokenKind attribute) (ml_switcheroo.core.tikz.parser.TokenKind attribute) WORD (ml_switcheroo.core.tikz.parser.TokenKind attribute) X x (ml_switcheroo.core.tikz.nodes.TikzNode attribute) (ml_switcheroo.core.tikz.TikzNode attribute) x1 (ml_switcheroo.core.html.nodes.SvgArrow attribute) (ml_switcheroo.core.html.SvgArrow attribute) x2 (ml_switcheroo.core.html.nodes.SvgArrow attribute) (ml_switcheroo.core.html.SvgArrow attribute) x_spacing (ml_switcheroo.compiler.backends.extras.TikzBackend attribute) Y y (ml_switcheroo.core.tikz.nodes.TikzNode attribute) (ml_switcheroo.core.tikz.TikzNode attribute) y1 (ml_switcheroo.core.html.nodes.SvgArrow attribute) (ml_switcheroo.core.html.SvgArrow attribute) y2 (ml_switcheroo.core.html.nodes.SvgArrow attribute) (ml_switcheroo.core.html.SvgArrow attribute) y_spacing (ml_switcheroo.compiler.backends.extras.TikzBackend attribute) yaml (in module ml_switcheroo.cli.handlers.define) Z z_index (ml_switcheroo.core.html.GridBox attribute) (ml_switcheroo.core.html.nodes.GridBox attribute)