Searched refs:grad_required_backprop_ids_ (Results 1 – 1 of 1) sorted by relevance
60 …std::vector<BackpropId>& grad_required_backprop_ids() const { return grad_required_backprop_ids_; } in grad_required_backprop_ids()75 return grad_required_backprop_ids_.end() != in IsGradRequired()76 … std::find(grad_required_backprop_ids_.begin(), grad_required_backprop_ids_.end(), backprop_id); in IsGradRequired()85 if (body->grad_required_backprop_ids_.end() == in RequireGrad()86 …std::find(body->grad_required_backprop_ids_.begin(), body->grad_required_backprop_ids_.end(), back… in RequireGrad()87 body->grad_required_backprop_ids_.emplace_back(backprop_id); in RequireGrad()175 std::vector<BackpropId> grad_required_backprop_ids_; variable