/aosp_15_r20/external/executorch/backends/xnnpack/_passes/ |
H A D | channels_last_tagged_reshape_pass.py | 156 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 D | conv1d_unsqueeze_pass.py | 161 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 D | freezing.py | 212 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 D | convert_to_linear.py | 108 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 D | fold_qdq.py | 29 user_n.replace_input_with(n, n.args[0]) 55 user_n.replace_input_with(n, n.args[0])
|
H A D | insert_io_qdq.py | 100 user.replace_input_with(node, inserted_node) 115 user.replace_input_with(node, inserted_node)
|
H A D | remove_redundancy.py | 36 user_n.replace_input_with(n, n.args[0])
|
H A D | convert_prelu.py | 38 user.replace_input_with(output_node, prelu_node)
|
H A D | decompose_einsum.py | 47 user.replace_input_with(
|
H A D | decompose_silu.py | 44 user.replace_input_with(silu_node, mul_node)
|
H A D | convert_interpolate_with_upsample2d.py | 50 user.replace_input_with(output_node, upsample2d_node)
|
H A D | replace_index_put_input.py | 48 node.replace_input_with(bad_frozen_node, m_buffer_node)
|
H A D | expand_broadcast_tensor_shape.py | 52 user.replace_input_with(arg, reshape_node)
|
H A D | recompose_rms_norm.py | 70 user.replace_input_with(output_node, rms_node)
|
H A D | insert_requantize.py | 63 user.replace_input_with(n, inserted_n)
|
H A D | convert_bmm_to_matmul.py | 69 user.replace_input_with(output, matmul_node)
|
H A D | recompose_pixel_unshuffle.py | 83 user.replace_input_with(node, pixel_unshuffle_node)
|
/aosp_15_r20/external/executorch/backends/arm/_passes/ |
H A D | conv1d_unsqueeze_pass.py | 132 node.replace_input_with(input_node, unsqueeze_before) 152 user.replace_input_with(node, squeeze_after)
|
H A D | annotate_channels_last_dim_order_pass.py | 97 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 D | convert.py | 568 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.py | 701 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 D | decompose_scaled_dot_product_attention.py | 61 user.replace_input_with(node, new_node)
|
/aosp_15_r20/external/executorch/backends/transforms/ |
H A D | decompose_sdpa.py | 73 user.replace_input_with(
|
/aosp_15_r20/external/pytorch/test/expect/ |
H A D | TestFXAPIBackwardCompatibility.test_class_member_back_compat-fx_backcompat_class_members.expect | 10 …ormalized_arguments', 'prepend', 'prev', 'replace_all_uses_with', 'replace_input_with', 'stack_tra…
|
/aosp_15_r20/external/executorch/exir/passes/ |
H A D | quant_fusion_pass.py | 86 maybe_cat.replace_input_with(tensor_arg, tensor_arg.args[0])
|