| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | multi_head_attention_fusion.cc | 30 auto is_matmul = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMatMul)); in DefineEmbedding()
33 auto is_reshape = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape)); in DefineEmbedding()
38 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineEmbedding()
40 auto is_param = std::make_shared<CondVar>(IsParamNode); in DefineEmbedding()
46 auto is_expand_dims = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimExpandDims)); in DefineMask()
48 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefineMask()
51 auto is_sub = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimSubFusion)); in DefineMask()
53 auto is_param2 = std::make_shared<CondVar>(IsParamNode); in DefineMask()
56 auto is_mul = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMulFusion)); in DefineMask()
58 auto is_param3 = std::make_shared<CondVar>(IsParamNode); in DefineMask()
[all …]
|
| D | transpose_fusion.cc | 37 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineBNPattern()
39 auto is_conv = std::make_shared<CondVar>(IsConvNode); in DefineBNPattern()
41 auto transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineBNPattern()
44 auto is_bn = std::make_shared<CondVar>(IsBNCNode); in DefineBNPattern()
46 auto bn_mean_var = std::make_shared<CondVar>(IsParamNode); in DefineBNPattern()
48 auto bn_variable_var = std::make_shared<CondVar>(IsParamNode); in DefineBNPattern()
57 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineActivationscalePattern()
59 auto is_conv = std::make_shared<CondVar>(IsConvNode); in DefineActivationscalePattern()
61 auto transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineActivationscalePattern()
64 auto is_scale = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimScaleFusion>); in DefineActivationscalePattern()
[all …]
|
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 49 query_prim_ = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimFullConnection)); in Init()
51 key_prim_ = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimFullConnection)); in Init()
53 value_prim_ = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimFullConnection)); in Init()
55 output_prim_ = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimFullConnection)); in Init()
57 pos_prim_ = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimFullConnection)); in Init()
83 auto is_add1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefinePatterns()
86 auto is_transpose1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefinePatterns()
88 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefinePatterns()
94 auto is_matmul1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMatMul)); in DefinePatterns()
98 auto is_add2 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefinePatterns()
[all …]
|
| D | tf_bidirection_gru_fusion.cc | 80 auto is_param = std::make_shared<CondVar>(IsParameterNode); in GenerateBodyGraphHiddenPattern()
127 auto is_reduce = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReduceFusion)); in DefineFowardPattern()
129 auto is_param1 = std::make_shared<CondVar>(IsParameterNode); in DefineFowardPattern()
132 auto is_maximum = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMaximum)); in DefineFowardPattern()
134 auto is_param2 = std::make_shared<CondVar>(IsParameterNode); in DefineFowardPattern()
138 auto is_shape = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimShape)); in DefineFowardPattern()
141 …auto is_strided_slice = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimStridedSlice)… in DefineFowardPattern()
146 auto is_minimum = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMinimum)); in DefineFowardPattern()
149 …auto is_tensor_list_reserve = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorL… in DefineFowardPattern()
151 auto is_param3 = std::make_shared<CondVar>(IsParameterNode); in DefineFowardPattern()
[all …]
|
| D | squeeze_fusion.cc | 27 auto is_squeeze = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSqueeze>); in DefinePattern()
29 auto is_bn = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimFusedBatchNorm>); in DefinePattern()
31 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefinePattern()
33 auto is_param2 = std::make_shared<CondVar>(IsParamNode); in DefinePattern()
38 auto is_activation = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimActivation>); in DefinePattern()
41 auto is_unsqueeze = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimUnsqueeze>); in DefinePattern()
|
| D | onnx_gelu_fusion.cc | 49 auto is_div = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimDivFusion>); in DefinePattern()
52 auto is_erf = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimErf>); in DefinePattern()
55 auto is_add = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
58 auto is_mul1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
61 auto is_mul2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
|
| D | tf_gelu_fusion.cc | 76 auto is_mul1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
79 auto is_add1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
82 auto is_mul2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
86 auto is_add2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
89 auto is_mul3 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
92 auto is_mul4 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
|
| D | glu_fusion.cc | 51 split_prim_ = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSplit>); in Init()
62 auto is_tuple_getitem1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTupleGetItem>); in DefinePattern()
67 auto is_tuple_getitem2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTupleGetItem>); in DefinePattern()
72 auto is_activation = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimActivation>); in DefinePattern()
75 auto is_mul = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
|
| D | norm_fusion.cc | 490 …auto is_squared_diffference = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSquaredDiffere… in DefinePattern()
494 auto is_add1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
497 auto is_rsqrt = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimRsqrt>); in DefinePattern()
500 auto is_mul2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
503 auto is_mul1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
506 auto is_mul3 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
509 auto is_sub = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSubFusion>); in DefinePattern()
512 auto is_add2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
524 auto is_sub1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSubFusion>); in DefinePattern()
527 auto is_sub2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSubFusion>); in DefinePattern()
[all …]
|
| D | conv_pad_fusion.cc | 156 auto is_pad = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimPadFusion>); in DefinePadConvPattern()
158 auto is_conv = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimConv2DFusion>); in DefinePadConvPattern()
160 auto is_param = std::make_shared<CondVar>(IsParamNode); in DefinePadConvPattern()
168 auto is_pad = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimPadFusion>); in DefinePadTransposeConvPattern()
170 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefinePadTransposeConvPattern()
172 auto is_param_perm = std::make_shared<CondVar>(IsParamNode); in DefinePadTransposeConvPattern()
176 auto is_conv = std::make_shared<CondVar>(IsConvNode); in DefinePadTransposeConvPattern()
178 auto is_param_weight = std::make_shared<CondVar>(IsParamNode); in DefinePadTransposeConvPattern()
|
| D | reshape_reshape_fusion.cc | 33 auto is_reshape1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape)); in DefinePattern()
35 auto is_param = std::make_shared<CondVar>(IsParamNode); in DefinePattern()
38 auto is_reshape2 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape)); in DefinePattern()
|
| D | tflite_lstm_cell_fusion.cc | 62 auto is_param3 = std::make_shared<CondVar>(IsParameterNode); in GenerateBodyGraphCellPattern()
186 auto is_parameter1 = std::make_shared<CondVar>(IsParameterNode); in GetCondGraphPattern()
188 auto is_parameter2 = std::make_shared<CondVar>(IsParameterNode); in GetCondGraphPattern()
190 auto is_parameter3 = std::make_shared<CondVar>(IsParameterNode); in GetCondGraphPattern()
192 auto is_parameter4 = std::make_shared<CondVar>(IsParameterNode); in GetCondGraphPattern()
194 auto is_less1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimLess)); in GetCondGraphPattern()
196 auto is_less2 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimLess)); in GetCondGraphPattern()
198 auto is_logical_and = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimLogicalAnd)); in GetCondGraphPattern()
200 auto is_return = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReturn)); in GetCondGraphPattern()
215 placeholders.emplace_back(std::make_shared<CondVar>(IsParameterNode)); in GetBodyGraphPattern()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/util/ |
| D | cond_var.cc | 23 CondVar::CondVar() : svc_(nullptr), my_name_(Services::GetUniqueID()) {} in CondVar() function in mindspore::dataset::CondVar
25 Status CondVar::Wait(std::unique_lock<std::mutex> *lck, const std::function<bool()> &pred) { in Wait()
48 CondVar::~CondVar() noexcept { in ~CondVar()
55 void CondVar::NotifyOne() noexcept { cv_.notify_one(); } in NotifyOne()
57 void CondVar::NotifyAll() noexcept { cv_.notify_all(); } in NotifyAll()
59 Status CondVar::Register(std::shared_ptr<IntrpService> svc) { in Register()
67 void CondVar::Interrupt() { in Interrupt()
72 std::string CondVar::my_name() const { return my_name_; } in my_name()
74 Status CondVar::Deregister() { in Deregister()
|
| D | cond_var.h | 30 class CondVar : public IntrpResource {
32 CondVar();
34 ~CondVar() noexcept;
|
| /third_party/grpc/src/core/lib/gprpp/ |
| D | sync.h | 48 using CondVar = absl::CondVar; variable
76 friend class CondVar;
123 class CondVar {
125 CondVar() { gpr_cv_init(&cv_); }
126 ~CondVar() { gpr_cv_destroy(&cv_); }
128 CondVar(const CondVar&) = delete;
129 CondVar& operator=(const CondVar&) = delete;
149 static void WaitUntil(CondVar* cv, Mutex* mu, Predicate pred) { in WaitUntil()
157 static bool WaitUntilWithTimeout(CondVar* cv, Mutex* mu, Predicate pred, in WaitUntilWithTimeout()
167 static bool WaitUntilWithDeadline(CondVar* cv, Mutex* mu, Predicate pred, in WaitUntilWithDeadline()
|
| /third_party/grpc/src/core/lib/gpr/ |
| D | sync_abseil.cc | 72 static_assert(sizeof(gpr_cv) == sizeof(absl::CondVar), in gpr_cv_init()
74 new (cv) absl::CondVar; in gpr_cv_init()
78 reinterpret_cast<absl::CondVar*>(cv)->~CondVar(); in gpr_cv_destroy()
85 reinterpret_cast<absl::CondVar*>(cv)->Wait( in gpr_cv_wait()
92 int ret = reinterpret_cast<absl::CondVar*>(cv)->WaitWithDeadline( in gpr_cv_wait()
100 reinterpret_cast<absl::CondVar*>(cv)->Signal(); in gpr_cv_signal()
105 reinterpret_cast<absl::CondVar*>(cv)->SignalAll(); in gpr_cv_broadcast()
|
| /third_party/grpc/include/grpcpp/impl/codegen/ |
| D | sync.h | 118 class CondVar {
120 CondVar() { g_core_codegen_interface->gpr_cv_init(&cv_); } in CondVar() function
121 ~CondVar() { g_core_codegen_interface->gpr_cv_destroy(&cv_); } in ~CondVar()
123 CondVar(const CondVar&) = delete;
124 CondVar& operator=(const CondVar&) = delete;
|
| /third_party/abseil-cpp/absl/synchronization/internal/ |
| D | mutex_nonprod.cc | 261 CondVar::CondVar() {} in CondVar() function in absl::CondVar
263 CondVar::~CondVar() {} in ~CondVar()
265 void CondVar::Signal() { impl()->Signal(); } in Signal()
267 void CondVar::SignalAll() { impl()->SignalAll(); } in SignalAll()
269 void CondVar::Wait(Mutex* mu) { return impl()->Wait(mu->impl()); } in Wait()
271 bool CondVar::WaitWithDeadline(Mutex* mu, absl::Time deadline) { in WaitWithDeadline()
276 bool CondVar::WaitWithTimeout(Mutex* mu, absl::Duration timeout) { in WaitWithTimeout()
280 void CondVar::EnableDebugLog(const char*) {} in EnableDebugLog()
|
| /third_party/skia/third_party/externals/abseil-cpp/absl/synchronization/ |
| D | lifetime_test.cc | 41 void ThreadOne(absl::Mutex* mutex, absl::CondVar* condvar, in ThreadOne()
59 void ThreadTwo(absl::Mutex* mutex, absl::CondVar* condvar, in ThreadTwo()
76 void RunTests(absl::Mutex* mutex, absl::CondVar* condvar) { in RunTests()
78 absl::CondVar default_condvar; in RunTests()
95 absl::CondVar condvar; in TestLocals()
|
| D | mutex.h | 484 friend class CondVar; // for access to Trans()/Fer(). variable
791 class CondVar {
795 CondVar();
796 ~CondVar();
861 CondVar(const CondVar&) = delete;
862 CondVar& operator=(const CondVar&) = delete;
940 inline CondVar::CondVar() : cv_(0) {} in CondVar() function
|
| /third_party/abseil-cpp/absl/synchronization/ |
| D | lifetime_test.cc | 41 void ThreadOne(absl::Mutex* mutex, absl::CondVar* condvar, in ThreadOne()
59 void ThreadTwo(absl::Mutex* mutex, absl::CondVar* condvar, in ThreadTwo()
76 void RunTests(absl::Mutex* mutex, absl::CondVar* condvar) { in RunTests()
78 absl::CondVar default_condvar; in RunTests()
95 absl::CondVar condvar; in TestLocals()
|
| D | mutex.h | 465 friend class CondVar;
503 friend class CondVar; // for access to Trans()/Fer().
775 class CondVar {
779 CondVar();
780 ~CondVar();
852 CondVar(const CondVar&) = delete;
853 CondVar& operator=(const CondVar&) = delete;
921 inline CondVar::CondVar() : cv_(0) {} in CondVar() function
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 34 auto ptr = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim)); in GetPrim()
75 …auto sub = VectorRef({GetPrim(prim::kPrimSubFusion), std::make_shared<CondVar>(IsParameterNode), z… in DefineGruCellPattern()
89 …auto fw_maximum = VectorRef({GetPrim(prim::kPrimMaximum), std::make_shared<CondVar>(IsParameterNod… in DefineBidirectionRnnPattern()
98 …auto fw_add = VectorRef({GetPrim(prim::kPrimAddFusion), fw_switch_true, std::make_shared<CondVar>(… in DefineBidirectionRnnPattern()
100 …auto fw_merge_enter = VectorRef({GetPrim(lite::kNameEnter), std::make_shared<CondVar>(IsParameterN… in DefineBidirectionRnnPattern()
111 …auto fw_unstack_range = VectorRef({GetPrim(prim::kPrimRange), std::make_shared<CondVar>(IsParamete… in DefineBidirectionRnnPattern()
112 … fw_unstack_strided_slice, std::make_shared<CondVar>(IsParameterNode)}); in DefineBidirectionRnnPattern()
116 …VectorRef({GetPrim(prim::kPrimAddFusion), std::make_shared<SeqVar>(), std::make_shared<CondVar>(Is… in DefineBidirectionRnnPattern()
117 …auto counter_zero = VectorRef({GetPrim(lite::kNameEnter), std::make_shared<CondVar>(IsParameterNod… in DefineBidirectionRnnPattern()
153 …auto zeros1 = VectorRef({GetPrim(prim::kPrimFill), std::make_shared<CondVar>(IsParameterNode), con… in DefineBidirectionRnnPattern()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | pattern_engine.h | 42 class CondVar; variable
44 using CondVarPtr = std::shared_ptr<CondVar>;
108 class CondVar : public Var {
110 explicit CondVar(const ConditionFunc &cond) : cond_fn_(cond) {} in CondVar() function
111 ~CondVar() override = default;
112 MS_DECLARE_PARENT(CondVar, Var);
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | fused_batch_norm_fusion.cc | 246 VarPtr index0 = std::make_shared<CondVar>(IsC); in DefinePattern()
247 VarPtr index1 = std::make_shared<CondVar>(IsC); in DefinePattern()
248 VarPtr index2 = std::make_shared<CondVar>(IsC); in DefinePattern()
265 VarPtr index0 = std::make_shared<CondVar>(IsC); in DefinePattern()
266 VarPtr index1 = std::make_shared<CondVar>(IsC); in DefinePattern()
267 VarPtr index2 = std::make_shared<CondVar>(IsC); in DefinePattern()
288 VarPtr index0 = std::make_shared<CondVar>(IsC); in DefinePattern()
289 VarPtr index1 = std::make_shared<CondVar>(IsC); in DefinePattern()
290 VarPtr index2 = std::make_shared<CondVar>(IsC); in DefinePattern()
|