Home
last modified time | relevance | path

Searched defs:current_stream (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Ddevice_interface.py95 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 DInlineStreamGuard.h103 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 DStreamGuard.h66 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 DCUDAGuard.h184 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 DHIPGuardImplMasqueradingAsCUDA.h305 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 Dscanctx.h44 FILE *current_stream; member
/aosp_15_r20/external/pytorch/torch/cpu/
H A D__init__.py112 def current_stream(device: _device_t = None) -> Stream: function
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/triton/
H A Ddevice_interface.py97 def current_stream(): member in DeviceInterface
/aosp_15_r20/external/pytorch/torch/mtia/
H A D__init__.py132 def current_stream(device: Optional[_device_t] = None) -> Stream: function
/aosp_15_r20/hardware/google/camera/devices/EmulatedCamera/hwl/
DEmulatedLogicalRequestState.cpp338 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__.py360 def current_stream(device: Optional[_device_t] = None) -> Stream: function
/aosp_15_r20/external/libdrm/etnaviv/
H A Detnaviv_priv.h108 struct etna_cmd_stream *current_stream; member
/aosp_15_r20/external/pytorch/torch/cuda/
H A D__init__.py970 def current_stream(device: Optional[_device_t] = None) -> Stream: function