/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | device_interface.py | 95 def current_stream(): member in DeviceInterface 197 current_stream = staticmethod(torch.cuda.current_stream) variable in CudaInterface 268 current_stream = staticmethod(torch.xpu.current_stream) variable in XpuInterface
|
/aosp_15_r20/external/pytorch/c10/core/impl/ |
H A D | InlineStreamGuard.h | 103 Stream current_stream() const { in current_stream() function 184 std::optional<Stream> current_stream() const { in current_stream() function
|
/aosp_15_r20/external/pytorch/c10/core/ |
H A D | StreamGuard.h | 66 Stream current_stream() const { in current_stream() function 133 std::optional<Stream> current_stream() const { in current_stream() function
|
/aosp_15_r20/external/pytorch/c10/cuda/ |
H A D | CUDAGuard.h | 184 CUDAStream current_stream() const { in current_stream() function 254 std::optional<CUDAStream> current_stream() const { in current_stream() function
|
/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/ |
H A D | HIPGuardImplMasqueradingAsCUDA.h | 305 HIPStreamMasqueradingAsCUDA current_stream() const { in current_stream() function 337 std::optional<HIPStreamMasqueradingAsCUDA> current_stream() const { in current_stream() function
|
/aosp_15_r20/external/libconfig/lib/ |
H A D | scanctx.h | 44 FILE *current_stream; member
|
/aosp_15_r20/external/pytorch/torch/cpu/ |
H A D | __init__.py | 112 def current_stream(device: _device_t = None) -> Stream: function
|
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/triton/ |
H A D | device_interface.py | 97 def current_stream(): member in DeviceInterface
|
/aosp_15_r20/external/pytorch/torch/mtia/ |
H A D | __init__.py | 132 def current_stream(device: Optional[_device_t] = None) -> Stream: function
|
/aosp_15_r20/hardware/google/camera/devices/EmulatedCamera/hwl/ |
D | EmulatedLogicalRequestState.cpp | 338 auto& current_stream = current_pipeline.streams.at(output_buffer.stream_id); in UpdateRequestForDynamicStreams() local
|
/aosp_15_r20/external/pytorch/torch/xpu/ |
H A D | __init__.py | 360 def current_stream(device: Optional[_device_t] = None) -> Stream: function
|
/aosp_15_r20/external/libdrm/etnaviv/ |
H A D | etnaviv_priv.h | 108 struct etna_cmd_stream *current_stream; member
|
/aosp_15_r20/external/pytorch/torch/cuda/ |
H A D | __init__.py | 970 def current_stream(device: Optional[_device_t] = None) -> Stream: function
|