Searched defs:enable_tuple_grad (Results 1 – 2 of 2) sorted by relevance
115 void set_enable_tuple_grad(bool enable_tuple_grad) { enable_tuple_grad_ = enable_tuple_grad; } in set_enable_tuple_grad()
408 bool CheckTailGradFristSequence(const abstract::AbstractSequeuePtr &sequeue, bool enable_tuple_grad… in CheckTailGradFristSequence()573 … const std::vector<AnfNodePtr> &forward_graph_params, bool enable_tuple_grad, in GetGrad()602 const AnfNodePtr &weights, bool enable_tuple_grad) { in GradByParameter()