Home
last modified time | relevance | path

Searched refs:serialize_tensor_output (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/executorch/exir/serde/
H A Dexport_serialize.py900 def serialize_tensor_output(self, name, meta_val) -> TensorArgument: member in GraphModuleSerializer
1138 tensor_args.append(self.serialize_tensor_output(name, meta))
1188 args.append(self.serialize_tensor_output(sub_user_node.name, m))
1227 as_tensors=[self.serialize_tensor_output(name, meta_val[0])]
1251 tensors.append(self.serialize_tensor_output(name, m))
1274 as_tensor=self.serialize_tensor_output(name, meta_val)
1310 self.serialize_tensor_output(idx_to_name[i], element_meta_val)
H A Dserialize.py180 as_tensor=self.serialize_tensor_output(node.name, meta_val)
207 self.serialize_tensor_output(idx_to_name[i], element_meta_val)
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Dserialize.py930 def serialize_tensor_output(self, name, meta_val) -> TensorArgument: member in GraphModuleSerializer
1161 tensor_args.append(self.serialize_tensor_output(name, meta))
1202 args.append(self.serialize_tensor_output(sub_user_node_name, m))
1231 … return [Argument.create(as_tensors=[self.serialize_tensor_output(name, meta_val[0])])]
1246 tensors.append(self.serialize_tensor_output(name, m))
1269 as_tensor=self.serialize_tensor_output(name, meta_val)
1305 self.serialize_tensor_output(idx_to_name[i], element_meta_val)