| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | pickle.cpp | 44 std::vector<at::Tensor>* tensor_table) { in pickle() argument
45 Pickler pickler(std::move(writer), tensor_table, nullptr, nullptr); in pickle()
53 std::vector<at::Tensor>* tensor_table) { in pickle() argument
61 tensor_table); in pickle()
168 c10::ArrayRef<at::Tensor> tensor_table, in unpickle() argument
174 tensor_table, in unpickle()
184 c10::ArrayRef<at::Tensor> tensor_table, in unpickle() argument
187 data, size, nullptr, std::move(type_resolver), tensor_table, type_parser); in unpickle()
195 c10::ArrayRef<at::Tensor> tensor_table, in unpickle() argument
211 tensor_table, in unpickle()
|
| H A D | pickle.h | 21 std::vector<at::Tensor>* tensor_table = nullptr);
53 std::vector<at::Tensor>* tensor_table = nullptr);
75 c10::ArrayRef<at::Tensor> tensor_table,
90 c10::ArrayRef<at::Tensor> tensor_table = {},
105 c10::ArrayRef<at::Tensor> tensor_table = {},
|
| H A D | unpickler.h | 39 c10::ArrayRef<at::Tensor> tensor_table,
42 tensor_table_(tensor_table), in reader_()
52 c10::ArrayRef<at::Tensor> tensor_table,
56 tensor_table_(tensor_table), in reader_()
|
| H A D | pickler.h | 131 std::vector<at::Tensor>* tensor_table,
137 tensor_table_(tensor_table), in writer_()
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/_model_report/ |
| H A D | model_report_visualizer.py | 202 tensor_table: List[List[Any]] = []
228 tensor_table.append(tensor_table_row)
231 if len(tensor_table) != 0:
234 return (tensor_headers, tensor_table)
379 tensor_headers, tensor_table = self._generate_tensor_table(
390 self.TABLE_TENSOR_KEY: (tensor_headers, tensor_table),
443 tensor_headers, tensor_table = table_dict[self.TABLE_TENSOR_KEY]
456 table_str += tabulate(tensor_table, headers=tensor_headers)
486 tensor_headers, tensor_table = table_dict[self.TABLE_TENSOR_KEY]
504 table = tensor_table
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ |
| H A D | script_resp.cpp | 16 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
17 auto payload = jit::pickle(value_, &tensor_table); in toMessageImpl()
19 std::move(payload), std::move(tensor_table), MessageType::SCRIPT_RET); in toMessageImpl()
|
| H A D | python_remote_call.cpp | 23 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
25 jit::pickle(c10::ivalue::Tuple::create(ivalues), &tensor_table); in toMessageImpl()
29 std::move(tensor_table), in toMessageImpl()
|
| H A D | script_remote_call.cpp | 56 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
58 c10::ivalue::Tuple::create(std::move(ivalues)), &tensor_table); in toMessageImpl()
62 std::move(tensor_table), in toMessageImpl()
|
| H A D | script_call.cpp | 117 std::vector<torch::Tensor> tensor_table; in toMessageImpl() local
119 c10::ivalue::Tuple::create(std::move(ivalues)), &tensor_table); in toMessageImpl()
122 std::move(payload), std::move(tensor_table), MessageType::SCRIPT_CALL); in toMessageImpl()
|
| H A D | rref_proto.cpp | 32 std::vector<torch::Tensor> tensor_table; in fromIValues() local
34 c10::ivalue::Tuple::create(std::move(ivalues)), &tensor_table); in fromIValues()
36 std::move(payload), std::move(tensor_table), type); in fromIValues()
|
| /aosp_15_r20/external/pytorch/torch/distributed/rpc/ |
| H A D | internal.py | 148 def deserialize(self, binary_data, tensor_table): argument
158 _thread_local_tensor_tables.recv_tables = tensor_table
194 def deserialize(binary_data, tensor_table): argument
195 return _internal_rpc_pickler.deserialize(binary_data, tensor_table)
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_interface.cpp | 38 const std::vector<at::IValue>& tensor_table) { in import_libs() argument
41 &tensor_table, in import_libs()
|
| H A D | test_class_import.cpp | 39 const std::vector<at::IValue>& tensor_table) { in import_libs() argument
42 &tensor_table, in import_libs()
|
| H A D | test_module_api.cpp | 39 const std::vector<at::IValue>& tensor_table) { in import_libs() argument
42 &tensor_table, in import_libs()
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | deep_wide_pt.cc | 63 const std::vector<at::IValue>& tensor_table) { in import_libs() argument
66 &tensor_table, in import_libs()
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | jit.cpp | 118 std::vector<at::Tensor> tensor_table; in TEST() local
119 auto data = torch::jit::pickle(float_value, &tensor_table); in TEST()
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_model_report_fx.py | 1873 tensor_headers, tensor_table = table_dict[ModelReportVisualizer.TABLE_TENSOR_KEY]
1877 tensor_info_modules = {row[1] for row in tensor_table}
1901 tensor_headers, tensor_table = empty_tables_dict[ModelReportVisualizer.TABLE_TENSOR_KEY]
1904 tensor_info_modules = {row[1] for row in tensor_table}
1927 tensor_headers, tensor_table = single_feat_dict[ModelReportVisualizer.TABLE_TENSOR_KEY]
|