Searched refs:kPrimSelect (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | lamb_update_with_lr_v2.cc | 31 VectorRef select0({prim::kPrimSelect, greater1, real_div0, input_varptr_[kIndex6]}); in DefinePattern()
32 VectorRef select1({prim::kPrimSelect, greater0, select0, input_varptr_[kIndex6]}); in DefinePattern()
|
| D | clip_by_norm_no_div_square_sum_fusion.cc | 37 VectorRef({prim::kPrimSelect, greater_pattern, in DefinePattern()
38 …VectorRef({sqrt, VectorRef({prim::kPrimSelect, greater_pattern, input_, constant_select_})}), inpu… in DefinePattern()
|
| D | lamb_update_with_lr_rule_fusion.cc | 36 …VectorRef pattern_select0({prim::kPrimSelect, pattern_greater0, pattern_real_div0, constant_select… in DefinePattern()
37 …VectorRef pattern_select1({prim::kPrimSelect, pattern_greater1, pattern_select0, constant_select_}… in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | value_based_eliminate.cc | 107 …MATCH_REPLACE_IF(node, PPrimitive(prim::kPrimSelect, PPrimitive(prim::kPrimGreaterEqual, x, zero_)… in operator ()()
110 …MATCH_REPLACE_IF(node, PPrimitive(prim::kPrimSelect, PPrimitive(prim::kPrimGreaterEqual, x, zero_s… in operator ()()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | reorder_ops.cc | 37 prim::kPrimMaximum, prim::kPrimMinimum, prim::kPrimSelect}; in IsTypeInsensitive()
69 } else if (IsPrimitiveCNode(node, prim::kPrimSelect)) { in GetOpDataInputIndexes()
|
| D | graph_kernel_cluster.cc | 92 {kGPUDevice, OpLevel_0, prim::kPrimSelect}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 135 …auto select1 = VectorRef({GetPrim(prim::kPrimSelect), greater_equal, std::make_shared<SeqVar>()});… in DefineBidirectionRnnPattern()
155 …auto select = VectorRef({GetPrim(prim::kPrimSelect), greater_equal, select_enter, rnn_cell_out}); … in DefineBidirectionRnnPattern()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | select.cc | 182 REGISTER_PRIMITIVE_EVAL_IMPL(Select, prim::kPrimSelect, SelectInfer, SelectInferValue, true);
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 251 … {prim::kPrimSelect, prim::kPrimMinimum, prim::kPrimMaximum}); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | selection_ops_declare.cc | 132 REG_ADPT_DESC(Select, prim::kPrimSelect->name(), ADPT_DESC(Select))
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 516 auto prim = prim::kPrimSelect; in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 542 inline const PrimitivePtr kPrimSelect = std::make_shared<Primitive>(kSelect); variable
|