Searched full:prim_device (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_device_type.py | 574 prim_device = cls.get_primary_device()
581 return [prim_device] + non_primary_devices
631 prim_device = cls.get_primary_device()
632 return [prim_device]
653 prim_device = cls.get_primary_device()
654 return [prim_device]
690 prim_device = cls.get_primary_device()
697 return [prim_device] + non_primary_devices
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_impls.py | 95 aten.to.prim_Device,
193 @register_op_impl(aten.to.prim_Device)
650 aten.to.prim_Device,
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 596 - aten::to.prim_Device
990 aten::to.prim_Device: 23
|
| H A D | model_ops.yaml | 365 aten::to.prim_Device: 67
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | symbolic_shape_registry_util.cpp | 29 …{"aten::to.prim_Device(Tensor(a) self, Device? device, int? dtype=None, bool non_blocking=False, b… in get_tensorexpr_elementwise_set()
|
| H A D | register_prim_ops.cpp | 1253 …"aten::to.prim_Device(Tensor(a) self, Device? device, int? dtype=None, bool non_blocking=False, bo…
|
| /aosp_15_r20/external/pytorch/ |
| H A D | pt_ops.bzl | 355 "aten::to.prim_Device",
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | native_ops.cpp | 1311 prim_device,
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | conf.py | 1637 "prim_device",
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | lowerings.cpp | 1578 …"aten::to.prim_Device(Tensor(a) self, Device? device, int? dtype=None, bool non_blocking=False, bo… in nnc_lowerings_lazy_registration()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 206 "prim_device",
6371 def prim_device(g: jit_utils.GraphContext, *inputs, **kwargs) -> None: function
|