Searched refs:call_state_ (Results 1 – 4 of 4) sorted by relevance
413 : context_(context), call_(std::move(call)), call_state_(State::kActive) { in StreamingRPCState()445 if (call_state_ != State::kActive) { in SendNextRequest()466 call_state_ = State::kDone; in CallStarted()478 if (call_state_ != State::kActive) { in RequestWriteCompleted()500 if (call_state_ != State::kActive) { in ResponseReadCompleted()532 DCHECK(call_state_ != State::kActive); in CallFinished()533 if (call_state_ != State::kFinishing) { in CallFinished()563 call_state_ = State::kDone; in MarkDoneAndCompleteExchanges()601 call_state_ = State::kFinishing; in IssueCallFinishLocked()622 State call_state_ TF_GUARDED_BY(mu_);
113 call_state_.reset(); in Orphan()128 GPR_ASSERT(call_state_ == nullptr); in StartCallLocked()130 call_state_ = MakeOrphanable<CallState>(Ref(), interested_parties_); in StartCallLocked()133 call_state_.get()); in StartCallLocked()135 call_state_->StartCall(); in StartCallLocked()166 self->call_state_ == nullptr) { in OnRetryTimer()596 if (this == health_check_client_->call_state_.get()) { in CallEndedLocked()597 health_check_client_->call_state_.reset(); in CallEndedLocked()
164 OrphanablePtr<CallState> call_state_; variable
236 call_state_.emplace(nullptr, false); in ForwardAccumulator()294 bool BusyAccumulating() const { return call_state_.top().accumulating; } in BusyAccumulating()316 void PushState() { call_state_.emplace(nullptr, false); } in PushState()317 void PopState() { call_state_.pop(); } in PopState()363 std::stack<AccumulatorCallState> call_state_; variable903 if (call_state_.top().backward_tape != nullptr) { in ShouldRecord()906 return call_state_.top().backward_tape->ShouldRecord(tensor_ids, dtypes); in ShouldRecord()908 if (call_state_.top().accumulating) { in ShouldRecord()940 AccumulatorCallState& call_state = call_state_.top(); in ForwardpropFromTape()1030 if (call_state_.top().backward_tape != nullptr) { in Accumulate()[all …]