Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/serialization/
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()
/external/pytorch/torch/csrc/distributed/rpc/
Dscript_resp.cpp16 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
Dpython_remote_call.cpp23 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
Dscript_remote_call.cpp56 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
Dscript_call.cpp117 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
Drref_proto.cpp32 std::vector<torch::Tensor> tensor_table; in fromIValues() local
/external/pytorch/torch/distributed/rpc/
Dinternal.py148 def deserialize(self, binary_data, tensor_table): argument
194 def deserialize(binary_data, tensor_table): argument
/external/pytorch/test/cpp/jit/
Dtest_interface.cpp38 const std::vector<at::IValue>& tensor_table) { in import_libs()
Dtest_class_import.cpp39 const std::vector<at::IValue>& tensor_table) { in import_libs()
Dtest_module_api.cpp39 const std::vector<at::IValue>& tensor_table) { in import_libs()
/external/pytorch/benchmarks/static_runtime/
Ddeep_wide_pt.cc63 const std::vector<at::IValue>& tensor_table) { in import_libs()
/external/pytorch/test/cpp/api/
Djit.cpp118 std::vector<at::Tensor> tensor_table; in TEST() local