Home
last modified time | relevance | path

Searched refs:decomp_table (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/export/
H A Dexported_program.py177 def _override_composite_implicit_decomp(ops_to_preserve, decomp_table, safe=True): argument
262 if op_overload in decomp_table:
263 removed_decomps[op_overload] = decomp_table[op_overload]
264 del decomp_table[op_overload]
275 decomp_table[op] = decomp
294 decomp_table: Dict[torch._ops.OperatorBase, Callable],
360 decomp_table,
372 decomp_table=decomp_table,
408 decomp_table,
413 decompositions=decomp_table,
[all …]
H A D_trace.py602 decomp_table=None, argument
642 decompositions=decomp_table,
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_fx_passes.py19 decomp_table = _decomp.create_onnx_friendly_decomposition_table(onnx_registered_ops)
24 return exported_program.run_decompositions(decomp_table, _preserve_ops=can_preserve)
/aosp_15_r20/external/pytorch/torch/_export/
H A D__init__.py140 decomp_table = {
151 decomposition_table=decomp_table,
/aosp_15_r20/external/pytorch/test/
H A Dtest_out_dtype_op.py191 decomp_table = torch._inductor.decomposition.select_decomp_table()
192 gm = make_fx(func, decomp_table, tracing_mode="symbolic")(x, w)
H A Dtest_decomp.py612 decomp_table = torch._inductor.decomposition.select_decomp_table()
613 cat_inductor = decomp_table[torch.ops.aten.cat.default]
/aosp_15_r20/external/pytorch/torch/export/experimental/
H A D__init__.py60 decomp_table=core_aten_decompositions(),
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_export.py1486 decomp_table={},
1503 decomp_table=None,
1526 decomp_table=None, _preserve_ops=[torch.ops.aten.conv2d.default]
1571 decomp_table={},
1589 decomp_table=None,
1613 decomp_table=None, _preserve_ops=[torch.ops.aten.conv2d.default]
5230 ep.run_decompositions(decomp_table=torch._decomp.decomposition_table)
5460 ep.run_decompositions(decomp_table=torch._decomp.decomposition_table)
8028 decomp_table=get_decompositions([torch.ops.aten.elu.default]),
8052 decomp_table = get_decompositions([torch.ops.aten.upsample_bilinear2d.vec])
[all …]
/aosp_15_r20/external/executorch/exir/program/test/
H A Dtest_program.py579 reference_decomp_ep = reference_ep.run_decompositions(decomp_table=table)