| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDAGuard.h | 24 explicit CUDAGuard(DeviceIndex device_index) : guard_(device_index) {} in CUDAGuard()
28 explicit CUDAGuard(Device device) : guard_(device) {} in CUDAGuard()
41 guard_.set_device(device); in set_device()
48 guard_.reset_device(device); in reset_device()
53 guard_.set_index(device_index); in set_index()
58 return guard_.original_device(); in original_device()
64 return guard_.current_device(); in current_device()
69 c10::impl::InlineDeviceGuard<impl::CUDAGuardImpl> guard_;
76 explicit OptionalCUDAGuard() : guard_() {} in OptionalCUDAGuard()
80 : guard_(device_opt) {} in OptionalCUDAGuard()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/ |
| H A D | HIPGuardImplMasqueradingAsCUDA.h | 253 explicit HIPGuardMasqueradingAsCUDA(DeviceIndex device_index) : guard_(device_index) {} in HIPGuardMasqueradingAsCUDA()
254 explicit HIPGuardMasqueradingAsCUDA(Device device) : guard_(device) {} in HIPGuardMasqueradingAsCUDA()
261 void set_device(Device device) { guard_.set_device(device); } in set_device()
262 void reset_device(Device device) { guard_.reset_device(device); } in reset_device()
263 void set_index(DeviceIndex device_index) { guard_.set_index(device_index); } in set_index()
264 Device original_device() const { return guard_.original_device(); } in original_device()
265 Device current_device() const { return guard_.current_device(); } in current_device()
268 c10::impl::InlineDeviceGuard<HIPGuardImplMasqueradingAsCUDA> guard_;
272 explicit OptionalHIPGuardMasqueradingAsCUDA() : guard_() {} in OptionalHIPGuardMasqueradingAsCUDA()
273 …explicit OptionalHIPGuardMasqueradingAsCUDA(std::optional<Device> device_opt) : guard_(device_opt)… in OptionalHIPGuardMasqueradingAsCUDA()
[all …]
|
| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A D | InlineDeviceGuard.h | 222 : guard_() // See Note [Explicit initialization of optional fields] in InlineOptionalDeviceGuard()
227 : guard_() { // See Note [Explicit initialization of optional fields] in InlineOptionalDeviceGuard()
229 guard_.emplace(device_opt.value()); in InlineOptionalDeviceGuard()
240 : guard_() { // See Note [Explicit initialization of optional fields] in InlineOptionalDeviceGuard()
242 guard_.emplace(device_index_opt.value()); in InlineOptionalDeviceGuard()
250 : guard_(std::in_place, std::forward<Args>(args)...) {} in InlineOptionalDeviceGuard()
348 if (!guard_.has_value()) { in set_device()
349 guard_.emplace(device); in set_device()
351 guard_->set_device(device); in set_device()
368 if (!guard_.has_value()) {
[all …]
|
| H A D | InlineStreamGuard.h | 136 : guard_() // See Note [Explicit initialization of optional fields] in InlineOptionalStreamGuard()
143 : guard_() { in InlineOptionalStreamGuard()
145 guard_.emplace(stream_opt.value()); in InlineOptionalStreamGuard()
152 : guard_(std::in_place, std::forward<Args>(args)...) {} in InlineOptionalStreamGuard()
167 if (guard_.has_value()) { in reset_stream()
168 guard_->reset_stream(stream); in reset_stream()
170 guard_.emplace(stream); in reset_stream()
177 return guard_.has_value() ? std::make_optional(guard_->original_stream()) in original_stream()
185 return guard_.has_value() ? std::make_optional(guard_->current_stream()) in current_stream()
192 guard_.reset(); in reset()
[all …]
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | StreamGuard.h | 33 explicit StreamGuard(Stream stream) : guard_(stream) {} in StreamGuard()
56 guard_.reset_stream(stream); in reset_stream()
61 return guard_.original_stream(); in original_stream()
67 return guard_.current_stream(); in current_stream()
73 return guard_.current_device(); in current_device()
79 return guard_.original_device(); in original_device()
83 c10::impl::InlineStreamGuard<impl::VirtualGuardImpl> guard_;
97 explicit OptionalStreamGuard(Stream stream) : guard_(stream) {} in OptionalStreamGuard()
103 : guard_(stream_opt) {} in OptionalStreamGuard()
121 guard_.reset_stream(stream); in reset_stream()
[all …]
|
| H A D | DeviceGuard.h | 29 explicit DeviceGuard(Device device) : guard_(device) {} in DeviceGuard()
35 : guard_(device, impl) {} in DeviceGuard()
54 guard_.reset_device(device); in reset_device()
61 guard_.reset_device(device, impl); in reset_device()
67 guard_.set_index(index); in set_index()
72 return guard_.original_device(); in original_device()
78 return guard_.current_device(); in current_device()
82 impl::InlineDeviceGuard<impl::VirtualGuardImpl> guard_;
134 explicit OptionalDeviceGuard(Device device) : guard_(device) {} in OptionalDeviceGuard()
138 explicit OptionalDeviceGuard(std::optional<Device> device) : guard_(device) {} in OptionalDeviceGuard()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | Dispatcher.cpp | 72 , guard_(std::make_shared<Guard>()) in Dispatcher()
76 std::lock_guard<std::mutex> lock(guard_->mutex); in ~Dispatcher()
77 guard_->alive.store(false); in ~Dispatcher()
100 std::unique_lock<std::mutex> lock(guard_->mutex); in waitForDef()
114 std::unique_lock<std::mutex> lock(guard_->mutex); in waitForImpl()
194 std::lock_guard<std::mutex> lock(guard_->mutex); in registerLibrary()
208 return RegistrationHandleRAII([guard = this->guard_, this, ns] { in registerLibrary()
224 std::lock_guard<std::mutex> lock(guard_->mutex); in registerDef()
241 return RegistrationHandleRAII([guard = this->guard_, this, op, op_name] { in registerDef()
285 std::lock_guard<std::mutex> lock(guard_->mutex); in getPyStub()
[all …]
|
| /aosp_15_r20/art/runtime/base/ |
| H A D | mutex.cc | 1025 : name_(name), guard_(guard) { in ConditionVariable()
1064 DCHECK_EQ(guard_.GetExclusiveOwnerTid(), SafeGetTid(self)); in Broadcast()
1082 guard_.state_and_contenders_.Address(), in RequeueWaiters()
1094 guard_.AssertExclusiveHeld(self); in Signal()
1103 guard_.CheckSafeToWait(self); in Wait()
1109 guard_.AssertExclusiveHeld(self); in WaitHoldingLocks()
1110 unsigned int old_recursion_count = guard_.recursion_count_; in WaitHoldingLocks()
1114 guard_.increment_contenders(); in WaitHoldingLocks()
1115 guard_.recursion_count_ = 1; in WaitHoldingLocks()
1117 guard_.ExclusiveUnlock(self); in WaitHoldingLocks()
[all …]
|
| H A D | mutex.h | 483 guard_.CheckSafeToWait(self); in CheckSafeToWait()
491 Mutex& guard_; variable
|
| /aosp_15_r20/external/private-join-and-compute/third_party/abseil-cpp-20230125.2/absl/log/internal/ |
| H A D | log_sink_set.cc | 187 ABSL_LOCKS_EXCLUDED(guard_) { in LogToSinks()
195 absl::ReaderMutexLock global_sinks_lock(&guard_); in LogToSinks()
206 void AddLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in AddLogSink()
208 absl::WriterMutexLock global_sinks_lock(&guard_); in AddLogSink()
218 void RemoveLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in RemoveLogSink()
220 absl::WriterMutexLock global_sinks_lock(&guard_); in RemoveLogSink()
230 void FlushLogSinks() ABSL_LOCKS_EXCLUDED(guard_) { in FlushLogSinks()
235 guard_.AssertReaderHeld(); in FlushLogSinks()
238 absl::ReaderMutexLock global_sinks_lock(&guard_); in FlushLogSinks()
250 void FlushLogSinksLocked() ABSL_SHARED_LOCKS_REQUIRED(guard_) { in FlushLogSinksLocked()
[all …]
|
| /aosp_15_r20/external/webrtc/third_party/abseil-cpp/absl/log/internal/ |
| H A D | log_sink_set.cc | 187 ABSL_LOCKS_EXCLUDED(guard_) { in LogToSinks()
195 absl::ReaderMutexLock global_sinks_lock(&guard_); in LogToSinks()
206 void AddLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in AddLogSink()
208 absl::WriterMutexLock global_sinks_lock(&guard_); in AddLogSink()
218 void RemoveLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in RemoveLogSink()
220 absl::WriterMutexLock global_sinks_lock(&guard_); in RemoveLogSink()
230 void FlushLogSinks() ABSL_LOCKS_EXCLUDED(guard_) { in FlushLogSinks()
235 guard_.AssertReaderHeld(); in FlushLogSinks()
238 absl::ReaderMutexLock global_sinks_lock(&guard_); in FlushLogSinks()
250 void FlushLogSinksLocked() ABSL_SHARED_LOCKS_REQUIRED(guard_) { in FlushLogSinksLocked()
[all …]
|
| /aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/log/internal/ |
| D | log_sink_set.cc | 187 ABSL_LOCKS_EXCLUDED(guard_) { in LogToSinks()
195 absl::ReaderMutexLock global_sinks_lock(&guard_); in LogToSinks()
206 void AddLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in AddLogSink()
208 absl::WriterMutexLock global_sinks_lock(&guard_); in AddLogSink()
218 void RemoveLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in RemoveLogSink()
220 absl::WriterMutexLock global_sinks_lock(&guard_); in RemoveLogSink()
230 void FlushLogSinks() ABSL_LOCKS_EXCLUDED(guard_) { in FlushLogSinks()
235 guard_.AssertReaderHeld(); in FlushLogSinks()
238 absl::ReaderMutexLock global_sinks_lock(&guard_); in FlushLogSinks()
250 void FlushLogSinksLocked() ABSL_SHARED_LOCKS_REQUIRED(guard_) { in FlushLogSinksLocked()
[all …]
|
| /aosp_15_r20/external/abseil-cpp/absl/log/internal/ |
| H A D | log_sink_set.cc | 187 ABSL_LOCKS_EXCLUDED(guard_) { in LogToSinks()
195 absl::ReaderMutexLock global_sinks_lock(&guard_); in LogToSinks()
206 void AddLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in AddLogSink()
208 absl::WriterMutexLock global_sinks_lock(&guard_); in AddLogSink()
218 void RemoveLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in RemoveLogSink()
220 absl::WriterMutexLock global_sinks_lock(&guard_); in RemoveLogSink()
230 void FlushLogSinks() ABSL_LOCKS_EXCLUDED(guard_) { in FlushLogSinks()
235 guard_.AssertReaderHeld(); in FlushLogSinks()
238 absl::ReaderMutexLock global_sinks_lock(&guard_); in FlushLogSinks()
250 void FlushLogSinksLocked() ABSL_SHARED_LOCKS_REQUIRED(guard_) { in FlushLogSinksLocked()
[all …]
|
| /aosp_15_r20/external/cronet/third_party/abseil-cpp/absl/log/internal/ |
| H A D | log_sink_set.cc | 187 ABSL_LOCKS_EXCLUDED(guard_) { in LogToSinks()
195 absl::ReaderMutexLock global_sinks_lock(&guard_); in LogToSinks()
206 void AddLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in AddLogSink()
208 absl::WriterMutexLock global_sinks_lock(&guard_); in AddLogSink()
218 void RemoveLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in RemoveLogSink()
220 absl::WriterMutexLock global_sinks_lock(&guard_); in RemoveLogSink()
230 void FlushLogSinks() ABSL_LOCKS_EXCLUDED(guard_) { in FlushLogSinks()
235 guard_.AssertReaderHeld(); in FlushLogSinks()
238 absl::ReaderMutexLock global_sinks_lock(&guard_); in FlushLogSinks()
250 void FlushLogSinksLocked() ABSL_SHARED_LOCKS_REQUIRED(guard_) { in FlushLogSinksLocked()
[all …]
|
| /aosp_15_r20/external/angle/third_party/abseil-cpp/absl/log/internal/ |
| H A D | log_sink_set.cc | 187 ABSL_LOCKS_EXCLUDED(guard_) { in LogToSinks()
195 absl::ReaderMutexLock global_sinks_lock(&guard_); in LogToSinks()
206 void AddLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in AddLogSink()
208 absl::WriterMutexLock global_sinks_lock(&guard_); in AddLogSink()
218 void RemoveLogSink(absl::LogSink* sink) ABSL_LOCKS_EXCLUDED(guard_) { in RemoveLogSink()
220 absl::WriterMutexLock global_sinks_lock(&guard_); in RemoveLogSink()
230 void FlushLogSinks() ABSL_LOCKS_EXCLUDED(guard_) { in FlushLogSinks()
235 guard_.AssertReaderHeld(); in FlushLogSinks()
238 absl::ReaderMutexLock global_sinks_lock(&guard_); in FlushLogSinks()
250 void FlushLogSinksLocked() ABSL_SHARED_LOCKS_REQUIRED(guard_) { in FlushLogSinksLocked()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/mps/ |
| H A D | MPSGuardImpl.h | 118 explicit OptionalMPSGuard() : guard_() {} in OptionalMPSGuard()
121 : guard_(device_opt) {} in OptionalMPSGuard()
126 : guard_(device_index_opt) {} in OptionalMPSGuard()
138 guard_.set_device(device); in set_device()
144 guard_.reset_device(device); in reset_device()
150 guard_.set_index(device_index); in set_index()
156 return guard_.original_device(); in original_device()
163 return guard_.current_device(); in current_device()
169 guard_.reset(); in reset()
173 c10::impl::InlineOptionalDeviceGuard<MPSGuardImpl> guard_;
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| H A D | utils_cuda.h | 27 AOTICudaGuard(int32_t device_index) : guard_(nullptr, delete_cuda_guard) { in AOTICudaGuard()
31 guard_.reset(ptr); in AOTICudaGuard()
36 aoti_torch_cuda_guard_set_index(guard_.get(), device_index)); in set_index()
40 std::unique_ptr<CUDAGuardOpaque, DeleterFnPtr> guard_;
46 : guard_(nullptr, delete_cuda_stream_guard) { in AOTICudaStreamGuard()
50 guard_.reset(ptr); in AOTICudaStreamGuard()
54 std::unique_ptr<CUDAStreamGuardOpaque, DeleterFnPtr> guard_;
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_raii.h | 14 guard_.emplace(std::forward<Args>(args)...); in enter()
20 guard_ = std::nullopt; in exit()
24 std::optional<GuardT> guard_;
47 guard_.emplace(std::forward<Args>(args)...); in DeprecatedRAIIContextManager()
53 guard_ = std::nullopt; in exit()
57 std::optional<GuardT> guard_;
|
| H A D | disable_torch_function.h | 23 : guard_(c10::DispatchKeySet( in DisableTorchDispatch()
26 c10::impl::ExcludeDispatchKeyGuard guard_; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/gen/cpp/renderers/ |
| H A D | guard_renderer.cc | 31 guard_ = toUpperSnake(with_underscores) + "_"; in GuardRenderer()
35 CodeLine("#ifndef $0", guard_); in Open()
36 CodeLine("#define $0", guard_); in Open()
42 CodeLine("#endif // $0", guard_); in Close()
|
| H A D | cpp_file_renderer.cc | 49 guard_(context), in CppFileRenderer()
61 guard_.Open(); in Render()
76 guard_.Close(); in Render()
|
| H A D | guard_renderer.h | 33 string guard_;
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| H A D | api.cc | 65 std::lock_guard<std::mutex> lock(guard_); in Execute()
74 std::lock_guard<std::mutex> lock(guard_); in Reset()
85 mutable std::mutex guard_; member in tflite::gpu::gl::__anon655d0df00111::InferenceContextImpl
101 std::lock_guard<std::mutex> lock(guard_); in Execute()
153 std::lock_guard<std::mutex> lock(guard_); in Reset()
169 mutable std::mutex guard_; member in tflite::gpu::gl::__anon655d0df00111::InferenceContextWithBatchImpl
|
| /aosp_15_r20/external/cronet/base/allocator/partition_allocator/src/partition_alloc/starscan/ |
| H A D | pcscan_internal.cc | 80 PA_CHECK(!guard_); in ReentrantScannerGuard()
81 guard_ = true; in ReentrantScannerGuard()
83 ~ReentrantScannerGuard() { guard_ = false; } in ~ReentrantScannerGuard()
88 static thread_local size_t guard_; member
90 thread_local size_t ReentrantScannerGuard::guard_ = 0; member in partition_alloc::internal::__anon9b4c4c840111::ReentrantScannerGuard
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | PythonFallbackKernel.h | 13 c10::impl::ForceDispatchKeyGuard guard_; member
|