Home
last modified time | relevance | path

Searched defs:tensor_table (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpickle.cpp44 std::vector<at::Tensor>* tensor_table) { in pickle()
53 std::vector<at::Tensor>* tensor_table) { in pickle()
168 c10::ArrayRef<at::Tensor> tensor_table, in unpickle()
184 c10::ArrayRef<at::Tensor> tensor_table, in unpickle()
195 c10::ArrayRef<at::Tensor> tensor_table, in unpickle()
/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A Dinternal.py148 def deserialize(self, binary_data, tensor_table): argument
194 def deserialize(binary_data, tensor_table): argument
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dscript_resp.cpp16 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
H A Dpython_remote_call.cpp23 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
H A Dscript_remote_call.cpp56 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
H A Dscript_call.cpp117 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
H A Drref_proto.cpp32 std::vector<torch::Tensor> tensor_table; in fromIValues() local
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_interface.cpp38 const std::vector<at::IValue>& tensor_table) { in import_libs()
H A Dtest_class_import.cpp39 const std::vector<at::IValue>& tensor_table) { in import_libs()
H A Dtest_module_api.cpp39 const std::vector<at::IValue>& tensor_table) { in import_libs()
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Ddeep_wide_pt.cc63 const std::vector<at::IValue>& tensor_table) { in import_libs()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Djit.cpp118 std::vector<at::Tensor> tensor_table; in TEST() local