Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpickler.h320 BackendMetaSerialization; in GetBackendMetaSerialization() local
321 return BackendMetaSerialization; in GetBackendMetaSerialization()
341 auto& BackendMetaSerialization = GetBackendMetaSerialization(); in TensorBackendMetaRegistry() local
343 !BackendMetaSerialization[device_type].has_value(), in TensorBackendMetaRegistry()
347 BackendMetaSerialization[device_type] = in TensorBackendMetaRegistry()
374 const auto& BackendMetaSerialization = GetBackendMetaSerialization(); in getTensorMetadata() local
375 if (BackendMetaSerialization[device_type].has_value()) { in getTensorMetadata()
378 BackendMetaPtr fptr = BackendMetaSerialization[device_type].value().first; in getTensorMetadata()
403 const auto& BackendMetaSerialization = GetBackendMetaSerialization(); in setTensorMetadata() local
404 if (BackendMetaSerialization[device_type].has_value()) { in setTensorMetadata()
[all …]