Home
last modified time | relevance | path

Searched refs:_dataclass_to_dict (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dextern_node_serializer.py5 from torch._export.serde.serialize import _dataclass_to_dict, EnumEncoder
25 return json.dumps(_dataclass_to_dict(serialized_nodes), cls=EnumEncoder)
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Ddynamic_shapes.py16 from .serialize import _dataclass_to_dict
60 return _dataclass_to_dict(spec)
H A Dserialize.py275 _dataclass_to_dict(tensor_meta), cls=EnumEncoder
2351 def _dataclass_to_dict(obj): function
2353 return {obj.type: _dataclass_to_dict(obj.value)}
2356 f.name: _dataclass_to_dict(getattr(obj, f.name))
2361 return [_dataclass_to_dict(x) for x in obj]
2363 return tuple(_dataclass_to_dict(x) for x in obj)
2365 return {k: _dataclass_to_dict(v) for k, v in obj.items()}
2380 _dataclass_to_dict(serialized_program.exported_program), cls=EnumEncoder
/aosp_15_r20/external/executorch/exir/serde/
H A Dexport_serialize.py283 _dataclass_to_dict(tensor_meta), cls=EnumEncoder
2360 def _dataclass_to_dict(obj): function
2362 return {obj.type: _dataclass_to_dict(obj.value)}
2365 f.name: _dataclass_to_dict(getattr(obj, f.name))
2370 return [_dataclass_to_dict(x) for x in obj]
2372 return tuple(_dataclass_to_dict(x) for x in obj)
2374 return {k: _dataclass_to_dict(v) for k, v in obj.items()}
2389 _dataclass_to_dict(serialized_program.exported_program), cls=EnumEncoder
H A Dserialize.py282 export_serialize._dataclass_to_dict(serialized_lowered_module),
660 export_serialize._dataclass_to_dict(serialized_artifact.exported_program),
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_export.py7310 from torch._export.serde.serialize import _dataclass_to_dict
7328 spec_dict = _dataclass_to_dict(spec)