Searched refs:kPrimRange (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 111 …auto fw_unstack_range = VectorRef({GetPrim(prim::kPrimRange), std::make_shared<CondVar>(IsParamete… in DefineBidirectionRnnPattern()
166 …auto range = VectorRef({GetPrim(prim::kPrimRange), std::make_shared<Var>(), ta_size, std::make_sha… in DefineBidirectionRnnPattern()
168 auto range1 = VectorRef({GetPrim(prim::kPrimRange), std::make_shared<SeqVar>()}); in DefineBidirectionRnnPattern()
183 auto bw_range = VectorRef({GetPrim(prim::kPrimRange), std::make_shared<SeqVar>()}); in DefinePattern()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 53 const auto kRange = prim::kPrimRange->name(); in GetDependsFormMap()
108 {prim::kPrimRange, {InferImplRange, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 238 inline const PrimitivePtr kPrimRange = std::make_shared<Primitive>("Range"); variable
|