Home
last modified time | relevance | path

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

/external/executorch/backends/arm/_passes/
Dmatch_arg_ranks_pass.py72 elif arg.name in self.exported_program.graph_signature.inputs_to_parameters:
73 buffer_name = self.exported_program.graph_signature.inputs_to_parameters[
Dunsqueeze_scalar_placeholders_pass.py32 in self.exported_program.graph_signature.inputs_to_parameters
Dconv1d_unsqueeze_pass.py57 parameter_name = self.exported_program.graph_signature.inputs_to_parameters[
/external/pytorch/torch/export/
Dgraph_signature.py294 def inputs_to_parameters(self) -> Mapping[str, str]: member in ExportGraphSignature
501 inputs_to_parameters = graph_signature.inputs_to_parameters
531 elif name in inputs_to_parameters:
535 target=inputs_to_parameters[name], # type: ignore[index]
D_trace.py1449 inputs_to_parameters=dict(zip(input_names[0:param_len], named_parameters)),
1617 sig.inputs_to_parameters = pytree.tree_map(
1618 _strip_root, sig.inputs_to_parameters
Dexported_program.py548 new_node.target in new_graph_signature.inputs_to_parameters
/external/pytorch/torch/distributed/tensor/experimental/
D_tp_transform.py139 num_params_and_buffers = len(graph_signature.inputs_to_parameters) + len(
171 if input_name in graph_signature.inputs_to_parameters:
172 return graph_signature.inputs_to_parameters[input_name]
533 if node.name in graph_signature.inputs_to_parameters:
534 fqn = graph_signature.inputs_to_parameters[node.name]
/external/executorch/exir/capture/
D_unlift.py143 if node.name in ep.graph_signature.inputs_to_parameters:
144 param_name = ep.graph_signature.inputs_to_parameters[node.name]
/external/executorch/exir/
Dlowered_backend_module.py217 not in lowered_exported_program.graph_signature.inputs_to_parameters
240 in lowered_exported_program.graph_signature.inputs_to_parameters
294 inputs_to_parameters = (
295 lowered_exported_program.graph_signature.inputs_to_parameters
305 if user_input not in inputs_to_parameters
399 if node.name in graph_sign.inputs_to_parameters:
/external/pytorch/torch/_functorch/_aot_autograd/
Dschemas.py674 inputs_to_parameters: Dict[GraphInputName, FQN]
727 inputs_to_parameters = dict(zip(graph_inputs[start:stop], parameters))
788 inputs_to_parameters=inputs_to_parameters, # type: ignore[arg-type]
/external/executorch/backends/arm/
Dprocess_node.py137 parameter_name = edge_program.graph_signature.inputs_to_parameters[node.name]
209 elif node.name in edge_program.graph_signature.inputs_to_parameters:
/external/executorch/exir/passes/
Dweights_to_outputs_pass.py46 inputs_to_params = gs.inputs_to_parameters
Dspec_prop_pass.py76 node.target in exported_program.graph_signature.inputs_to_parameters
Dconstant_prop_pass.py215 if name := signature.inputs_to_parameters.get(node.name, None):
/external/executorch/backends/qualcomm/builders/
Dutils.py53 edge_program.graph_signature.inputs_to_parameters[node.name]
/external/pytorch/torch/_export/
Dutils.py164 if node.target in new_sig.inputs_to_parameters:
165 param_name = new_sig.inputs_to_parameters[node.target]
410 return node.name in program.graph_signature.inputs_to_parameters
423 parameter_name = program.graph_signature.inputs_to_parameters[node.name]
/external/executorch/backends/qualcomm/partition/
Dqnn_partitioner.py147 *edge_program.graph_signature.inputs_to_parameters,
/external/executorch/backends/xnnpack/_passes/
Dconv1d_unsqueeze_pass.py86 parameter_name = self.exported_program.graph_signature.inputs_to_parameters[
/external/executorch/docs/source/
Dmodel-inspector.rst104 …rs=[], user_inputs=['arg0_1'], user_outputs=['aten_tan_default'], inputs_to_parameters={}, inputs_…
/external/pytorch/docs/source/
Dexport.rst58 inputs_to_parameters={},
209 inputs_to_parameters={
368 inputs_to_parameters={
/external/pytorch/torch/_inductor/
Dcompile_fx.py187 if node_name in graph_signature.inputs_to_parameters:
188 parameter_name = graph_signature.inputs_to_parameters[node_name]
/external/executorch/exir/emit/
D_emitter.py1502 if target in self.exported_program.graph_signature.inputs_to_parameters:
1503 fqn = self.exported_program.graph_signature.inputs_to_parameters[target]
/external/pytorch/torch/onnx/_internal/exporter/
D_core.py880 exported_program.graph_signature.inputs_to_parameters.items(),
/external/pytorch/test/functorch/
Dtest_aotdispatch.py4692 str(signature.inputs_to_parameters),
/external/pytorch/test/export/
Dtest_export.py2895 or node.name in ep.graph_signature.inputs_to_parameters