Searched refs:get_param_tensor (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/executorch/backends/transforms/ |
| H A D | fuse_batch_norm_with_conv.py | 11 from executorch.backends.transforms.utils import get_param_tensor, is_param_node
59 conv_weight = get_param_tensor(self.exported_program, conv.args[1])
62 conv_bias = get_param_tensor(self.exported_program, conv.args[2])
73 bn_weight = get_param_tensor(self.exported_program, bn.args[1])
74 bn_bias = get_param_tensor(self.exported_program, bn.args[2])
76 running_mean = get_param_tensor(self.exported_program, bn.args[3])
79 running_var = get_param_tensor(self.exported_program, bn.args[4])
|
| H A D | utils.py | 38 def get_param_tensor( function
|
| /aosp_15_r20/external/executorch/backends/xnnpack/_passes/ |
| H A D | fuse_batch_norm_with_conv.py | 13 from executorch.backends.xnnpack.utils.utils import get_param_tensor, is_param_node
57 conv_weight = get_param_tensor(self.exported_program, conv.args[1])
60 conv_bias = get_param_tensor(self.exported_program, conv.args[2])
71 bn_weight = get_param_tensor(self.exported_program, bn.args[1])
72 bn_bias = get_param_tensor(self.exported_program, bn.args[2])
74 running_mean = get_param_tensor(self.exported_program, bn.args[3])
77 running_var = get_param_tensor(self.exported_program, bn.args[4])
|
| H A D | prelu_reshape_pass.py | 11 get_param_tensor,
40 weight_data = get_param_tensor(self.exported_program, weight_node)
|
| H A D | conv1d_unsqueeze_pass.py | 12 from executorch.backends.xnnpack.utils.utils import get_param_tensor, is_param_node
76 kernel_param_3d = get_param_tensor(self.exported_program, kernel_node)
|
| /aosp_15_r20/external/executorch/backends/arm/_passes/ |
| H A D | conv1d_unsqueeze_pass.py | 14 get_param_tensor,
47 kernel_param_3d = get_param_tensor(self.exported_program, kernel_node)
|
| H A D | arm_pass_utils.py | 46 def get_param_tensor( function
|
| /aosp_15_r20/external/executorch/backends/xnnpack/operators/ |
| H A D | quant_params.py | 26 get_param_tensor,
188 param = get_param_tensor(ep, node)
|
| H A D | node_visitor.py | 36 get_param_tensor,
530 const_val = get_param_tensor(self.exported_program, get_attr_node)
|
| /aosp_15_r20/external/executorch/backends/apple/mps/operators/ |
| H A D | node_visitor.py | 26 get_param_tensor,
245 tensor = get_param_tensor(self.exported_program, node)
|
| /aosp_15_r20/external/executorch/backends/vulkan/serialization/ |
| H A D | vulkan_graph_builder.py | 99 def get_param_tensor(self, node: Node) -> torch.Tensor: member in VkGraphBuilder
125 self.const_tensors.append(self.get_param_tensor(node))
|
| /aosp_15_r20/external/executorch/backends/apple/mps/utils/ |
| H A D | mps_utils.py | 44 def get_param_tensor( function
|
| /aosp_15_r20/external/executorch/backends/xnnpack/utils/ |
| H A D | utils.py | 114 def get_param_tensor( function
|