Searched refs:sequeue (Results 1 – 4 of 4) sorted by relevance
408 bool CheckTailGradFristSequence(const abstract::AbstractSequeuePtr &sequeue, bool enable_tuple_grad… in CheckTailGradFristSequence() argument409 return sequeue->size() > 1 && (*sequeue)[1] != nullptr && in CheckTailGradFristSequence()410 ((*sequeue)[1]->isa<abstract::AbstractUndetermined>() || in CheckTailGradFristSequence()411 …(MsContext::GetInstance()->get_param<bool>(MS_CTX_GRAD_FOR_SCALAR) && (*sequeue)[1]->BuildType() !… in CheckTailGradFristSequence()412 (*sequeue)[1]->BuildType()->isa<Number>()) || in CheckTailGradFristSequence()413 ((*sequeue)[1]->isa<abstract::AbstractTuple>() && enable_tuple_grad && in CheckTailGradFristSequence()414 CheckSequenceAllTensor((*sequeue)[1]->cast<abstract::AbstractTuplePtr>()))); in CheckTailGradFristSequence()417 FuncGraphPtr Tail::GenerateSequeueFuncGraph(const abstract::AbstractSequeuePtr &sequeue) const { in GenerateSequeueFuncGraph()418 MS_EXCEPTION_IF_NULL(sequeue); in GenerateSequeueFuncGraph()427 if (sequeue->isa<AbstractTuple>()) { in GenerateSequeueFuncGraph()[all …]
112 FuncGraphPtr GenerateSequeueFuncGraph(const abstract::AbstractSequeuePtr &sequeue) const;
1838 std::vector<ValuePtr> GetValueSequeue(const ValuePtr &sequeue) { in GetValueSequeue() argument1839 MS_EXCEPTION_IF_NULL(sequeue); in GetValueSequeue()1841 if (!sequeue->isa<ValueTuple>() && !sequeue->isa<ValueList>()) { in GetValueSequeue()1846 if (sequeue->isa<ValueTuple>()) { in GetValueSequeue()1847 auto val_tuple = sequeue->cast<ValueTuplePtr>(); in GetValueSequeue()1850 auto val = sequeue->cast<ValueListPtr>(); in GetValueSequeue()
329 std::vector<ValuePtr> GetValueSequeue(const ValuePtr &sequeue);