Home
last modified time | relevance | path

Searched refs:AbstractSequeuePtr (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dzip_operation.cc35 using mindspore::abstract::AbstractSequeuePtr;
57 return (x->cast<AbstractSequeuePtr>()->size() < y->cast<AbstractSequeuePtr>()->size()); in GenerateFuncGraph()
68 for (size_t idx = 0; idx < (*min_abs)->cast<AbstractSequeuePtr>()->size(); idx++) { in GenerateFuncGraph()
Dcomposite.h112 FuncGraphPtr GenerateSequeueFuncGraph(const abstract::AbstractSequeuePtr &sequeue) const;
Dcomposite.cc408 bool CheckTailGradFristSequence(const abstract::AbstractSequeuePtr &sequeue, bool enable_tuple_grad… in CheckTailGradFristSequence()
417 FuncGraphPtr Tail::GenerateSequeueFuncGraph(const abstract::AbstractSequeuePtr &sequeue) const { in GenerateSequeueFuncGraph()
469 return GenerateSequeueFuncGraph(a->cast<abstract::AbstractSequeuePtr>()); in GenerateFuncGraph()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Ditem_tuple_or_list_eliminate.h73 auto sequeue_abstract = sequeue_->abstract()->cast<abstract::AbstractSequeuePtr>(); in Visit()
333 auto sequeue_abstract = tuple_->abstract()->cast<abstract::AbstractSequeuePtr>(); in Visit()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc365 AbstractSequeuePtr axis = nullptr; in InferImplReduceShape()
372 axis = args_spec_list[1]->cast<AbstractSequeuePtr>(); in InferImplReduceShape()
/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc1167 auto input_tuple = args_spec_list[0]->cast<abstract::AbstractSequeuePtr>(); in InferImplDynamicStitch()
1175 auto input_tuple_1 = args_spec_list[1]->cast<abstract::AbstractSequeuePtr>(); in InferImplDynamicStitch()
Dabstract_value.h391 using AbstractSequeuePtr = std::shared_ptr<AbstractSequeue>; variable
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprogram_specialize.cc64 const auto &abs_seq = abs_base->cast<AbstractSequeuePtr>(); in CheckAbstractTensor()
Dstatic_analysis.cc753 auto elements = abstract->cast<AbstractSequeuePtr>()->elements(); in NeedWaitForBranches()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.cc951 const auto &abs_seq = abs->cast<abstract::AbstractSequeuePtr>(); in SetNonCostantValueAbs()