.. currentmodule:: torch.fx.experimental
:::{warning} These APIs are experimental and subject to change without notice. :::
.. autoclass:: torch.fx.experimental.sym_node.DynamicInt
.. currentmodule:: torch.fx.experimental.sym_node
.. automodule:: torch.fx.experimental.sym_node
.. autosummary::
:toctree: generated
:nosignatures:
is_channels_last_contiguous_2d
is_channels_last_contiguous_3d
is_channels_last_strides_2d
is_channels_last_strides_3d
is_contiguous
is_non_overlapping_and_dense_indicator
method_to_operator
sympy_is_channels_last_contiguous_2d
sympy_is_channels_last_contiguous_3d
sympy_is_channels_last_strides_2d
sympy_is_channels_last_strides_3d
sympy_is_channels_last_strides_generic
sympy_is_contiguous
sympy_is_contiguous_generic
to_node
.. currentmodule:: torch.fx.experimental.symbolic_shapes
.. automodule:: torch.fx.experimental.symbolic_shapes
.. autosummary::
:toctree: generated
:nosignatures:
ShapeEnv
DimDynamic
StrictMinMaxConstraint
RelaxedUnspecConstraint
EqualityConstraint
SymbolicContext
StatelessSymbolicContext
StatefulSymbolicContext
SubclassSymbolicContext
DimConstraints
ShapeEnvSettings
ConvertIntKey
CallMethodKey
PropagateUnbackedSymInts
DivideByKey
InnerTensorKey
Specialization
hint_int
size_hint
is_concrete_int
is_concrete_bool
is_concrete_float
has_free_symbols
has_free_unbacked_symbols
guard_or_true
guard_or_false
guard_size_oblivious
sym_and
sym_eq
sym_or
constrain_range
constrain_unify
canonicalize_bool_expr
statically_known_true
statically_known_false
has_static_value
lru_cache
check_consistent
compute_unbacked_bindings
rebind_unbacked
resolve_unbacked_bindings
is_accessor_node
cast_symbool_to_symint_guardless
create_contiguous
error
eval_guards
eval_is_non_overlapping_and_dense
find_symbol_binding_fx_nodes
free_symbols
free_unbacked_symbols
fx_placeholder_targets
fx_placeholder_vals
guard_bool
guard_float
guard_int
guard_scalar
has_hint
has_symbolic_sizes_strides
is_nested_int
is_symbol_binding_fx_node
is_symbolic
expect_true
log_lru_cache_stats
.. currentmodule:: torch.fx.experimental.proxy_tensor
.. automodule:: torch.fx.experimental.proxy_tensor
.. autosummary::
:toctree: generated
:nosignatures:
make_fx
handle_sym_dispatch
get_proxy_mode
maybe_enable_thunkify
maybe_disable_thunkify
thunkify
track_tensor
track_tensor_tree
decompose
disable_autocast_cache
disable_proxy_modes_tracing
extract_val
fake_signature
fetch_object_proxy
fetch_sym_proxy
has_proxy_slot
is_sym_node
maybe_handle_decomp
proxy_call
set_meta
set_original_aten_op
set_proxy_slot
snapshot_fake
.. currentmodule:: torch.fx.experimental.optimization
.. automodule:: torch.fx.experimental.optimization
.. autosummary::
:toctree: generated
:nosignatures:
extract_subgraph
modules_to_mkldnn
optimize_for_inference
remove_dropout
replace_node_module
reset_modules
use_mkl_length
.. currentmodule:: torch.fx.experimental.recording
.. automodule:: torch.fx.experimental.recording
.. autosummary::
:toctree: generated
:nosignatures:
record_shapeenv_event
replay_shape_env_events
shape_env_check_state_equal
.. currentmodule:: torch.fx.experimental.unification.core
.. automodule:: torch.fx.experimental.unification.core
.. autosummary::
:toctree: generated
:nosignatures:
reify
.. currentmodule:: torch.fx.experimental.unification.unification_tools
.. automodule:: torch.fx.experimental.unification.unification_tools
.. autosummary::
:toctree: generated
:nosignatures:
assoc
assoc_in
dissoc
first
keyfilter
keymap
merge
merge_with
update_in
valfilter
valmap
itemfilter
itemmap
.. currentmodule:: torch.fx.experimental.migrate_gradual_types.transform_to_z3
.. automodule:: torch.fx.experimental.migrate_gradual_types.transform_to_z3
.. autosummary::
:toctree: generated
:nosignatures:
transform_algebraic_expression
transform_all_constraints
transform_all_constraints_trace_time
transform_dimension
transform_to_z3
transform_var
evaluate_conditional_with_constraints
.. currentmodule:: torch.fx.experimental.migrate_gradual_types.constraint
.. automodule:: torch.fx.experimental.migrate_gradual_types.constraint
.. autosummary::
:toctree: generated
:nosignatures:
is_algebraic_expression
is_bool_expr
is_dim
.. currentmodule:: torch.fx.experimental.migrate_gradual_types.constraint_generator
.. automodule:: torch.fx.experimental.migrate_gradual_types.constraint_generator
.. autosummary::
:toctree: generated
:nosignatures:
adaptive_inference_rule
assert_inference_rule
batchnorm_inference_rule
bmm_inference_rule
embedding_inference_rule
embedding_inference_rule_functional
eq_inference_rule
equality_inference_rule
expand_inference_rule
full_inference_rule
gt_inference_rule
lt_inference_rule
masked_fill_inference_rule
neq_inference_rule
tensor_inference_rule
torch_dim_inference_rule
torch_linear_inference_rule
type_inference_rule
view_inference_rule
register_inference_rule
transpose_inference_rule
.. currentmodule:: torch.fx.experimental.migrate_gradual_types.constraint_transformation
.. automodule:: torch.fx.experimental.migrate_gradual_types.constraint_transformation
.. autosummary::
:toctree: generated
:nosignatures:
apply_padding
calc_last_two_dims
create_equality_constraints_for_broadcasting
is_target_div_by_dim
no_broadcast_dim_with_index
register_transformation_rule
transform_constraint
transform_get_item
transform_get_item_tensor
transform_index_select
transform_transpose
valid_index
valid_index_tensor
is_dim_div_by_target
.. currentmodule:: torch.fx.experimental.graph_gradual_typechecker
.. automodule:: torch.fx.experimental.graph_gradual_typechecker
.. autosummary::
:toctree: generated
:nosignatures:
adaptiveavgpool2d_check
adaptiveavgpool2d_inference_rule
all_eq
bn2d_inference_rule
calculate_out_dimension
conv_refinement_rule
conv_rule
element_wise_eq
expand_to_tensor_dim
first_two_eq
register_algebraic_expressions_inference_rule
register_inference_rule
register_refinement_rule
transpose_inference_rule
.. currentmodule:: torch.fx.experimental.meta_tracer
.. automodule:: torch.fx.experimental.meta_tracer
.. autosummary::
:toctree: generated
:nosignatures:
embedding_override
functional_relu_override
nn_layernorm_override
proxys_to_metas
symbolic_trace
torch_abs_override
torch_nn_relu_override
torch_relu_override
torch_where_override