Home
last modified time | relevance | path

Searched refs:is_param_node (Results 1 – 19 of 19) sorted by relevance

/external/executorch/backends/vulkan/_passes/
Dinsert_prepack_nodes.py16 from executorch.backends.vulkan.utils import is_param_node
35 if not is_param_node(program, node):
/external/executorch/backends/xnnpack/_passes/
Dprelu_reshape_pass.py12 is_param_node,
36 is_param_node(self.exported_program, weight_node),
Dchannels_last_tagged_reshape_pass.py11 from executorch.backends.xnnpack.utils.utils import is_param_node
106 is_param_node(self.exported_program, node)
257 if is_param_node(self.exported_program, input_node):
308 if is_param_node(self.exported_program, input_node):
Dfuse_batch_norm_with_conv.py13 from executorch.backends.xnnpack.utils.utils import get_param_tensor, is_param_node
154 if [is_param_node(program, node) for node in {conv_weights, bn_weights}].count(
Dtag_implicit_q_dq_pass.py20 from executorch.backends.xnnpack.utils.utils import is_param_node
172 is_param_node(self.exported_program, n)
Dconv1d_unsqueeze_pass.py12 from executorch.backends.xnnpack.utils.utils import get_param_tensor, is_param_node
129 if not is_param_node(self.exported_program, kernel_node):
/external/executorch/backends/xnnpack/operators/
Dop_dynamic_dequantize_ops.py22 is_param_node,
97 is_param_node(self._exported_program, node.all_input_nodes[0]),
Dnode_visitor.py37 is_param_node,
110 return tensor.op == "placeholder" and not is_param_node(
159 and not is_param_node(self.exported_program, quant_params.q_input)
174 and not is_param_node(self.exported_program, quant_params.q_input)
522 if not is_param_node(self.exported_program, get_attr_node):
Dquant_params.py27 is_param_node,
281 if is_param_node(ep, q_input):
/external/executorch/backends/transforms/
Dfuse_batch_norm_with_conv.py11 from executorch.backends.transforms.utils import get_param_tensor, is_param_node
156 if [is_param_node(program, node) for node in {conv_weights, bn_weights}].count(
Dutils.py29 def is_param_node(exp_prog: ExportedProgram, node: torch.fx.Node) -> bool: function
/external/executorch/backends/xnnpack/
Dxnnpack_preprocess.py28 from executorch.backends.xnnpack.utils.utils import is_param_node
67 if node.op == "placeholder" and not is_param_node(exported_program, node):
/external/executorch/backends/xnnpack/partition/config/
Dnode_configs.py19 from executorch.backends.xnnpack.utils.utils import is_param_node
133 is_param = is_param_node(ep, weight)
Dgemm_configs.py30 is_param_node,
128 if not is_param_node(ep, weight_node):
140 if not is_param_node(ep, weight):
199 if not is_param_node(ep, bias_node):
/external/executorch/backends/arm/_passes/
Dconv1d_unsqueeze_pass.py16 is_param_node,
100 if not is_param_node(self.exported_program, kernel_node):
Darm_pass_utils.py37 def is_param_node(exp_prog: ExportedProgram, node: torch.fx.Node) -> bool: function
/external/executorch/backends/vulkan/serialization/
Dvulkan_graph_builder.py23 is_param_node,
123 if is_param_node(self.program, node):
282 if not is_param_node(self.program, node):
/external/executorch/backends/xnnpack/utils/
Dutils.py105 def is_param_node(exp_prog: ExportedProgram, node: torch.fx.Node) -> bool: function
/external/executorch/backends/vulkan/
Dutils.py38 def is_param_node(program: ExportedProgram, node: torch.fx.Node) -> bool: function