Home
last modified time | relevance | path

Searched refs:torch_module (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_layouts.cpp16 if (PyModule_AddObject(torch_module, "" #layout, layout##_layout) != 0) { \
22 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in initializeLayouts() local
23 if (!torch_module) in initializeLayouts()
29 if (PyModule_AddObject(torch_module, "strided", strided_layout) != 0) { in initializeLayouts()
37 if (PyModule_AddObject(torch_module, "sparse_coo", sparse_coo_layout) != 0) { in initializeLayouts()
49 if (PyModule_AddObject(torch_module, "_mkldnn", mkldnn_layout) != 0) { in initializeLayouts()
H A Dtensor_dtypes.cpp11 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in initializeDtypes() local
12 if (!torch_module) in initializeDtypes()
27 if (PyModule_AddObject(torch_module.get(), primary_name.c_str(), dtype) != in initializeDtypes()
33 if (PyModule_AddObject(torch_module.get(), legacy_name.c_str(), dtype) != in initializeDtypes()
H A Dtensor_qschemes.cpp18 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in initializeQSchemes() local
19 if (!torch_module) { in initializeQSchemes()
29 torch_module, toString(qscheme).c_str(), qscheme_obj) != 0) { in initializeQSchemes()
H A Dtensor_memoryformats.cpp29 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in initializeMemoryFormats() local
30 if (!torch_module) { in initializeMemoryFormats()
38 if (PyModule_AddObject(torch_module, name, memory_format) != 0) { in initializeMemoryFormats()
/aosp_15_r20/external/pytorch/torch/csrc/tensor/
H A Dpython_tensor.cpp307 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in set_default_storage_type() local
308 if (!torch_module) in set_default_storage_type()
311 if (PyObject_SetAttrString(torch_module.get(), "Storage", storage) != 0) { in set_default_storage_type()
393 auto torch_module = THPObjectPtr(PyImport_ImportModule("torch")); in py_bind_tensor_types() local
394 if (!torch_module) in py_bind_tensor_types()
398 PyObject_GetAttrString(torch_module.get(), "_tensor_classes")); in py_bind_tensor_types()
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_testing.py55 torch_module = exported_program.module()
56 torch_outputs, _ = _pytree.tree_flatten(torch_module(*args, **kwargs))
H A D_verification.py81 torch_module = exported_program.module()
82 torch_outputs, _ = _pytree.tree_flatten(torch_module(*args, **kwargs))
/aosp_15_r20/external/pytorch/torch/csrc/
H A DDevice.cpp158 py::object torch_module = py::module::import("torch"); in THPDevice_reduce() local
159 py::object torch_device = torch_module.attr("device"); in THPDevice_reduce()
H A DGenerator.cpp262 py::object torch_module = py::module::import("torch"); in THPGenerator_reduce() local
263 py::object torch_generator = torch_module.attr("Generator"); in THPGenerator_reduce()
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Daccelerator_partitioner.py306 torch_module: torch.nn.Module,
314 self.torch_module = torch_module
571 self.torch_module,
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_export.py2974 self.torch_module = torch.nn.LayerNorm(
2980 return input.cos() * self.int_val * self.torch_module.eps