Home
last modified time | relevance | path

Searched refs:kPrimLess (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dtf_bidirection_gru_cf_fusion.cc93 auto fw_less1 = VectorRef({GetPrim(prim::kPrimLess), std::make_shared<SeqVar>(), fw_less1_enter}); in DefineBidirectionRnnPattern()
103 auto fw_less = VectorRef({GetPrim(prim::kPrimLess), fw_merge, fw_less_enter}); in DefineBidirectionRnnPattern()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Darithmetic_logic_cpu_kernel.cc176 {prim::kPrimLess->name(), &ArithmeticLogicCPUKernel<T>::Less}, in InitComputeFunc()
/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.cc239 {prim::kPrimLess, {InferImplLess, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.cc194 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()
Dtf_bidirection_gru_fusion.cc280 auto is_less1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimLess)); in GetCondGraphPattern()
287 auto is_less2 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimLess)); in GetCondGraphPattern()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_cluster.cc83 {kGPUDevice, OpLevel_0, prim::kPrimLess}, in GetClusterableOpList()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h140 inline const PrimitivePtr kPrimLess = std::make_shared<Primitive>("Less"); variable