Home
last modified time | relevance | path

Searched refs:replace_input_with (Results 1 – 25 of 48) sorted by relevance

12

/aosp_15_r20/external/executorch/backends/xnnpack/_passes/
H A Dchannels_last_tagged_reshape_pass.py156 after.replace_input_with(before, dq)
182 copy.replace_input_with(before, dq)
183 after.replace_input_with(before, q)
244 target_node.replace_input_with(original_input, copy_node)
H A Dconv1d_unsqueeze_pass.py161 node.replace_input_with(input_node, unsqueeze_before)
181 user.replace_input_with(node, squeeze_after)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dfreezing.py212 output_node.replace_input_with(n, new_node)
237 n.replace_input_with(n.args[0], new_node)
266 conv.replace_input_with(weight_node, new_node)
/aosp_15_r20/external/executorch/backends/qualcomm/_passes/
H A Dconvert_to_linear.py108 user.replace_input_with(fn_node, linear_node)
133 user.replace_input_with(input_node, squeeze_node)
155 user.replace_input_with(linear_node, unsqueeze_node)
H A Dfold_qdq.py29 user_n.replace_input_with(n, n.args[0])
55 user_n.replace_input_with(n, n.args[0])
H A Dinsert_io_qdq.py100 user.replace_input_with(node, inserted_node)
115 user.replace_input_with(node, inserted_node)
H A Dremove_redundancy.py36 user_n.replace_input_with(n, n.args[0])
H A Dconvert_prelu.py38 user.replace_input_with(output_node, prelu_node)
H A Ddecompose_einsum.py47 user.replace_input_with(
H A Ddecompose_silu.py44 user.replace_input_with(silu_node, mul_node)
H A Dconvert_interpolate_with_upsample2d.py50 user.replace_input_with(output_node, upsample2d_node)
H A Dreplace_index_put_input.py48 node.replace_input_with(bad_frozen_node, m_buffer_node)
H A Dexpand_broadcast_tensor_shape.py52 user.replace_input_with(arg, reshape_node)
H A Drecompose_rms_norm.py70 user.replace_input_with(output_node, rms_node)
H A Dinsert_requantize.py63 user.replace_input_with(n, inserted_n)
H A Dconvert_bmm_to_matmul.py69 user.replace_input_with(output, matmul_node)
H A Drecompose_pixel_unshuffle.py83 user.replace_input_with(node, pixel_unshuffle_node)
/aosp_15_r20/external/executorch/backends/arm/_passes/
H A Dconv1d_unsqueeze_pass.py132 node.replace_input_with(input_node, unsqueeze_before)
152 user.replace_input_with(node, squeeze_after)
H A Dannotate_channels_last_dim_order_pass.py97 node.replace_input_with(input_node, permute_node)
114 user.replace_input_with(node, permute_node)
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dconvert.py568 node.replace_input_with(arg, quantize_node)
644 user_node.replace_input_with(node, dequantize_node)
700 node.replace_input_with(arg, quantize_node)
885 node.replace_input_with(prev_node, prev_node.args[0])
H A D_lower_to_native_backend.py701 ref_node.replace_input_with(dq_node, dq_node.args[0]) # type: ignore[arg-type]
764 ref_node.replace_input_with(dq_node, dq_node.args[0]) # type: ignore[arg-type]
816 ref_node.replace_input_with(dq_node, input_dynamic_q_node.args[0])
1108 bop_node.replace_input_with(dq_node, dn_input) # type: ignore[arg-type]
1227 ref_node.replace_input_with(dq_node, dn_input)
/aosp_15_r20/external/executorch/backends/mediatek/_passes/
H A Ddecompose_scaled_dot_product_attention.py61 user.replace_input_with(node, new_node)
/aosp_15_r20/external/executorch/backends/transforms/
H A Ddecompose_sdpa.py73 user.replace_input_with(
/aosp_15_r20/external/pytorch/test/expect/
H A DTestFXAPIBackwardCompatibility.test_class_member_back_compat-fx_backcompat_class_members.expect10 …ormalized_arguments', 'prepend', 'prev', 'replace_all_uses_with', 'replace_input_with', 'stack_tra…
/aosp_15_r20/external/executorch/exir/passes/
H A Dquant_fusion_pass.py86 maybe_cat.replace_input_with(tensor_arg, tensor_arg.args[0])

12