Home
last modified time | relevance | path

Searched refs:hook_ (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/utils/
Dscoped_long_running.h41 if (hook_ != nullptr) { in ScopedLongRunning()
42 hook_->Enter(); in ScopedLongRunning()
46 if (hook_ != nullptr) { in ~ScopedLongRunning()
47 hook_->Leave(); in ~ScopedLongRunning()
51 if (hook_ == nullptr) { in SetHook()
52 hook_ = std::move(hook); in SetHook()
57 static ScopedLongRunningHookPtr hook_;
Dscoped_long_running.cc19 ScopedLongRunningHookPtr ScopedLongRunning::hook_; member in mindspore::ScopedLongRunning
/third_party/skia/third_party/externals/abseil-cpp/absl/base/internal/
Datomic_hook.h84 : hook_(default_fn), default_fn_(default_fn) {} in AtomicHook()
87 : hook_(kUninitialized), default_fn_(default_fn) {} in AtomicHook()
148 FnPtr DoLoad() const { return hook_.load(std::memory_order_acquire); } in DoLoad()
155 const bool store_succeeded = hook_.compare_exchange_strong( in DoStore()
161 std::atomic<FnPtr> hook_; variable
170 const intptr_t value = hook_.load(std::memory_order_acquire); in DoLoad()
181 const bool store_succeeded = hook_.compare_exchange_strong( in DoStore()
187 std::atomic<intptr_t> hook_; variable
/third_party/abseil-cpp/absl/base/internal/
Datomic_hook.h84 : hook_(default_fn), default_fn_(default_fn) {} in AtomicHook()
87 : hook_(kUninitialized), default_fn_(default_fn) {} in AtomicHook()
148 FnPtr DoLoad() const { return hook_.load(std::memory_order_acquire); } in DoLoad()
155 const bool store_succeeded = hook_.compare_exchange_strong( in DoStore()
161 std::atomic<FnPtr> hook_; variable
170 const intptr_t value = hook_.load(std::memory_order_acquire); in DoLoad()
181 const bool store_succeeded = hook_.compare_exchange_strong( in DoStore()
187 std::atomic<intptr_t> hook_; variable
/third_party/mindspore/mindspore/ccsrc/pybind_api/ir/
Dprimitive_py.h60 void set_hook(const py::function &hook) { hook_ = hook; } in set_hook()
61 py::function hook() const { return hook_; } in hook()
84 py::function hook_; variable
117 py::function hook_; variable
Dprimitive_py.cc71 set_hook(adapter->hook_); in PrimitivePy()
200 inst->NewGraph(hook_, input_args.cast<py::args>()); in RunCellBpropFunction()
201 py::object grads_obj = hook_(*convert_args); in RunCellBpropFunction()
203 inst->EndGraph(hook_, grads_obj, input_args.cast<py::args>()); in RunCellBpropFunction()
222 py::object code_obj = py::getattr(hook_, "__code__"); in RunCellHookFunction()
237 obj = hook_(*hook_args); in RunCellHookFunction()
252 py::object code_obj = py::getattr(hook_, "__code__"); in RunVariableHookFunction()
260 py::object obj = hook_(py::make_tuple(py_args[grad_output_index])); in RunVariableHookFunction()
481 hook_ = hook; in set_hook()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/util/
Dservices.cc116 auto n = hook_.size(); in ~Services()
118 hook_.pop_back(); in ~Services()
119 n = hook_.size(); in ~Services()
Dservices.h93 hook_.push_back(std::move(svc)); in AddHook()
109 std::vector<std::unique_ptr<Service>> hook_; variable
/third_party/boost/libs/intrusive/doc/
Dintrusive.qbk424 list_member_hook<> hook_;
434 typedef member_hook<Foo, list_member_hook<>, &Foo::hook_> MemberHookOption;