/aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/ |
H A D | expanded_weights_impl.py | 7 from torch._decomp import decomposition_table 19 decomposition_table[aten.rnn_relu.input], 20 decomposition_table[aten.rnn_relu.data], 23 decomposition_table[aten.rnn_tanh.input], 24 decomposition_table[aten.rnn_tanh.data], 27 decomposition_table[aten.lstm.input], 28 decomposition_table[aten.lstm.data], 31 decomposition_table[aten.gru.input], 32 decomposition_table[aten.gru.data],
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/passes/ |
H A D | decomp.py | 25 decomposition_table: Mapping[torch._ops.OpOverload, Callable], 30 self.decomposition_table = decomposition_table 72 decomposition_table=self.decomposition_table,
|
H A D | functionalization.py | 122 decomposition_table={},
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
H A D | decomposition_table.py | 89 decomposition_table: dict[torch._ops.OperatorBase, Callable] = {} 97 …for op_overload, decomp_fn in torch._decomp.decomposition_table.items(): # type: ignore[attr-defi… 107 decomposition_table[op_overload] = decomp_fn 115 decomposition_table[op_overload] = decomp_fn 116 return decomposition_table
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
H A D | _decomp.py | 86 decomposition_table: dict[torch._ops.OperatorBase, Callable] = {} 91 …for op_overload, decomp_fn in torch._decomp.decomposition_table.items(): # type: ignore[attr-defi… 98 decomposition_table[op_overload] = decomp_fn 100 return decomposition_table
|
/aosp_15_r20/external/pytorch/torchgen/decompositions/ |
H A D | gen_jit_decompositions.py | 5 from torch.jit._decompositions import decomposition_table 59 [scripted_func.code for scripted_func in decomposition_table.values()] # type: ignore[misc] 65 for schema, scripted_func in decomposition_table.items():
|
/aosp_15_r20/external/pytorch/torch/fx/experimental/ |
H A D | proxy_tensor.py | 144 decomposition_table: Optional[Mapping[OpOverload, Callable]] 148 CURRENT_DECOMPOSITION_TABLE = decomposition_table or {} 1417 decomposition_table: Optional[Mapping[OpOverload, Callable]] = None, 1424 self.decomposition_table = decomposition_table or {} 1463 with decompose(self.decomposition_table), self.mode: 1783 decomposition_table: Optional[Mapping[OpOverload, Callable]], 1793 self.decomposition_table: Dict[OpOverload, Callable] = dict( 1794 decomposition_table or {} 1796 self.decomposition_table.setdefault( 2023 stack.enter_context(decompose(self.decomposition_table)) [all …]
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/ |
H A D | _exporter_legacy.py | 36 decomposition_table, 301 decomposition_table: dict[torch._ops.OpOverload, Callable] 333 self.decomposition_table = options.decomposition_table 361 self.decomposition_table = ( 362 … decomposition_table.create_onnx_friendly_decomposition_table( # type: ignore[assignment] 1251 options.decomposition_table,
|
H A D | onnxruntime.py | 44 import torch.onnx._internal.fx.decomposition_table 95 decomposition_table, 768 import torch.onnx._internal.fx.decomposition_table 804 …support_dict = torch.onnx._internal.fx.decomposition_table._create_onnx_supports_op_overload_table( 1166 decompositions=self._resolved_onnx_exporter_options.decomposition_table,
|
/aosp_15_r20/external/pytorch/torch/_decomp/ |
H A D | decompositions_for_jvp.py | 12 decomposition_table = torch._decomp.decomposition_table variable 64 elif decomp in decomposition_table: 65 decomposition_table_used = decomposition_table
|
H A D | __init__.py | 34 decomposition_table = global_decomposition_table["post_autograd"] variable
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_decomp.py | 13 from torch._decomp import core_aten_decompositions, decomposition_table 59 for k in decomposition_table 734 cfunc = make_fx(func, decomposition_table=decomposition_table) 781 cfunc = make_fx(func, decomposition_table=decomposition_table) 828 func not in decomposition_table 853 decomposition = decomposition_table[func] 1282 ops_missing_decomp = overloads_wanting_decomp - decomposition_table.keys() 1304 for op in decomposition_table.keys()
|
H A D | test_proxy_tensor.py | 15 from torch._decomp import decomposition_table 354 out = make_fx(f, decomposition_table=factory_func_decomp)(torch.ones(2)) 376 … traced = make_fx(f, decomposition_table=decomp, tracing_mode=self.tracing_mode)(torch.rand(3)) 593 fx_module = make_fx(fn, tracing_mode=self.tracing_mode, decomposition_table=None)(x) 603 …silu_decomp_table = {torch.ops.aten.silu.default: decomposition_table[torch.ops.aten.silu.default]} 607 decomposition_table=silu_decomp_table, 737 decomposed_fx = make_fx(f, decomposition_table={aten.addmm.default: addmm})(*inps) 752 traced = make_fx(f, decomposition_table={torch.ops.aten.t.default: nop})(torch.randn(5)) 1688 …r = str(make_fx(f, tracing_mode="symbolic", decomposition_table=decomposition_table)(torch.empty(4…
|
/aosp_15_r20/external/pytorch/torch/_functorch/ |
H A D | vmap.py | 287 from torch._decomp import decomposition_table 290 if decomp in decomposition_table: 291 VMAP_DECOMPOSITIONS_LIB.impl(decomp, decomposition_table[decomp])
|
/aosp_15_r20/external/pytorch/torch/jit/ |
H A D | _decompositions.py | 15 decomposition_table: Dict[str, torch.jit.ScriptFunction] = {} variable 73 registry = decomposition_table
|
/aosp_15_r20/external/pytorch/torch/_prims/ |
H A D | context.py | 129 func = torch._decomp.decomposition_table.get(orig_func, None) 133 func = torch._decomp.decomposition_table.get(default, None)
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | eval_frame.py | 1231 decomposition_table: Optional[ 1311 if decomposition_table is not None: 1542 decomposition_table=decomposition_table,
|
/aosp_15_r20/external/pytorch/torch/_subclasses/ |
H A D | fake_tensor.py | 225 from torch._decomp import decomposition_table 231 return decomposition_table[func].__module__.startswith( 233 ) and decomposition_table[func].__name__ in dir(decompositions) 1930 from torch._decomp import decomposition_table 1933 if func in decomposition_table and ( 1943 return decomposition_table[func](*args, **kwargs)
|
/aosp_15_r20/external/executorch/backends/transforms/ |
H A D | decompose_sdpa.py | 37 decomposition_table=get_decompositions( # pyre-fixme[6]
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_debug_utils.py | 34 fx = make_fx(fn, decomposition_table=decomps)(x)
|
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
H A D | dispatch_and_compile_graph.py | 56 decomposition_table=aot_config.decompositions,
|
/aosp_15_r20/external/pytorch/torch/_export/ |
H A D | __init__.py | 151 decomposition_table=decomp_table,
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_flex_decoding.py | 590 gm = make_fx(attention, decomposition_table={})(query, key, value) 602 gm = make_fx(attention, decomposition_table=core_aten_decompositions())(
|
H A D | test_flex_attention.py | 600 gm = make_fx(flex_attention, decomposition_table={})(query, key, value) 612 gm = make_fx(flex_attention, decomposition_table=core_aten_decompositions())(
|
/aosp_15_r20/external/executorch/exir/ |
H A D | tracer.py | 671 decomposition_table=(
|