__init__.py __pycache__/ _trace_wrapped_higher_order_op.py aot_compile.py aot_compile_types.py backends/ bytecode_analysis.py bytecode_transformation.py cache_size.py callback.py code_context.py codegen.py compiled_autograd.py comptime.py config.py convert_frame.py create_parameter_op.py current_scope_id.py dce_extra_outputs.py debug_utils.py decorators.py device_interface.py distributed.py eval_frame.py exc.py external_utils.py funcname_cache.py functional_export.py graph_break_hints.py graph_break_registry.json graph_bytecode_inputs.py graph_deduplication.py graph_region_tracker.py graph_utils.py guards.py hooks.py logging.py metrics_context.py mutation_guard.py output_graph.py package.py pgo.py polyfills/ precompile_context.py profiler.py replay_record.py repro/ resume_execution.py side_effects.py source.py symbolic_convert.py tensor_version_op.py test_case.py test_dont_skip_tracing_functions.py test_minifier_common.py testing.py trace_rules.py types.py utils.py variables/