Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport_module.cpp503 const std::string& tensor_dir, in writeArchive() argument
552 writer_.writeRecord(tensor_dir + tensor_name, nullptr, 0); in writeArchive()
561 tensor_dir + tensor_name, in writeArchive()
H A Dexport.h96 const std::string& tensor_dir,
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/compatibility/
H A Dbackport_manager.cpp86 const std::string& tensor_dir, in write_archive_current() argument
125 std::string fname = tensor_dir + tensor_names[i++]; in write_archive_current()