| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | GdsFile.cpp | 40 const at::Storage& storage, in gds_load_storage()
56 const at::Storage& storage, in gds_save_storage()
70 void gds_register_buffer(const at::Storage& storage) { in gds_register_buffer()
82 void gds_deregister_buffer(const at::Storage& storage) { in gds_deregister_buffer()
119 namespace torch::cuda::shared { namespace
|
| H A D | Module.cpp | 1078 for (auto& storage : storages_to_add_deleters_to) { in addStorageDeleterFns() local
1980 namespace torch::cuda { namespace
|
| /aosp_15_r20/external/pytorch/torch/csrc/tensor/ |
| H A D | python_tensor.cpp | 28 namespace torch::tensors { namespace
231 THPObjectPtr storage( in get_storage_obj() local
261 auto torch = THPObjectPtr(PyImport_ImportModule("torch")); in get_tensor_dict() local
305 THPObjectPtr storage = get_storage_obj(backend, dtype); in set_default_storage_type() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | DynamicTypes.cpp | 20 namespace torch { namespace
54 PyObject* createPyObject(const at::Storage& storage) { in createPyObject()
124 auto storage = THPStorage_Unpack(untyped_storage_obj); in createStorageGetType() local
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_util.cpp | 7 namespace torch { namespace
40 std::string storage("String storage"); in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ |
| H A D | tensorpipe_cuda.cpp | 15 namespace torch::distributed::rpc { namespace
76 const c10::Storage& storage, in prepareTensorForSending()
|
| H A D | tensorpipe_utils.cpp | 12 namespace torch::distributed::rpc { namespace
40 const c10::Storage& storage, in prepareTensorForSending()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/serialization/ |
| H A D | serializer.cpp | 11 namespace torch { namespace
36 auto storage = _builder.CreateVector(data_ptr, num_bytes); in serializeData() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
| H A D | LlgaTensorImpl.cpp | 7 namespace torch { namespace
47 at::Storage&& storage, in LlgaTensorImpl()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/ |
| H A D | TracerRunner.cpp | 17 namespace torch::jit::mobile { namespace
89 std::vector<float> storage(20, 1.0); in call_setup_methods() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | memory_planner.cpp | 10 namespace torch::jit { namespace
459 const auto& storage = tensor->storage(); in deallocateManagedTensors() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | pickler.cpp | 14 namespace torch::jit { namespace
291 const at::Storage& storage = tensor.storage(); in pushStorageOfTensor() local
|
| H A D | flatbuffer_serializer.cpp | 31 namespace torch::jit { namespace
608 const at::Storage& storage = tensor.storage(); in tensorToFB() local
|
| H A D | unpickler.cpp | 15 namespace torch::jit { namespace
582 at::Storage storage; in readInstruction() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/ |
| H A D | flatbuffer_loader.cpp | 58 namespace torch::jit { namespace
504 c10::Storage storage; in parseTensorFromMetadata() local
698 auto* storage = module_->storage_data()->GetMutableObject(index); in getStorage() local
|
| /aosp_15_r20/external/pytorch/tools/autograd/templates/ |
| H A D | python_variable_methods.cpp | 64 namespace torch::autograd { namespace
1184 at::Storage storage = _r.storage(0, storage_scalar_type, is_typed_storage); in THPVariable_set_() local
1200 at::Storage storage = _r.storage(0, storage_scalar_type, is_typed_storage); in THPVariable_set_() local
1228 at::Tensor storage = _r.tensor(0); in THPVariable_set_() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_new.cpp | 47 namespace torch::utils { namespace
82 Storage storage) { in new_with_storage()
745 at::Storage storage = r.storage(0, storage_scalar_type, is_typed_storage); in legacy_tensor_generic_ctor_new() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | python_variable.cpp | 840 Storage storage{ in THPVariable_make_wrapper_subclass() local
2337 namespace torch::autograd { namespace
|
| H A D | FunctionsManual.cpp | 39 namespace torch::autograd::generated::details { namespace
3218 auto storage = grad.new_zeros_symint(c10::SymIntArrayRef(base_size)); in as_strided_backward() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | init.cpp | 122 namespace torch::jit { namespace
1591 c10::Storage storage( in initJITBindings() local
1626 c10::Storage storage = self.getStorage(name); in initJITBindings() local
|
| /aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-1B-Instruct/ |
| H A D | tokenizer.json | 18980 "storage": 16568, number
30926 "torch": 28514, number
|
| /aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/ |
| H A D | tokenizer.json | 18980 "storage": 16568, number
30926 "torch": 28514, number
|
| /aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/ |
| H A D | tokenizer.json | 18921 "storage": 16568, number
30867 "torch": 28514, number
|