Searched refs:kPrimLess (Results 1 – 7 of 7) sorted by relevance
/third_party/mindspore/mindspore/lite/tools/converter/parser/ |
D | tf_bidirection_gru_cf_fusion.cc | 93 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/ |
D | arithmetic_logic_cpu_kernel.cc | 176 {prim::kPrimLess->name(), &ArithmeticLogicCPUKernel<T>::Less}, in InitComputeFunc()
|
/third_party/mindspore/mindspore/core/abstract/ |
D | primitive_infer_map.cc | 239 {prim::kPrimLess, {InferImplLess, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
D | tflite_lstm_cell_fusion.cc | 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()
|
D | tf_bidirection_gru_fusion.cc | 280 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/ |
D | graph_kernel_cluster.cc | 83 {kGPUDevice, OpLevel_0, prim::kPrimLess}, in GetClusterableOpList()
|
/third_party/mindspore/mindspore/core/base/ |
D | core_ops.h | 140 inline const PrimitivePtr kPrimLess = std::make_shared<Primitive>("Less"); variable
|