Home
last modified time | relevance | path

Searched refs:call_state_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_state.h413 : 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_);
/external/rust/crates/grpcio-sys/grpc/src/core/ext/filters/client_channel/health/
Dhealth_check_client.cc113 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()
Dhealth_check_client.h164 OrphanablePtr<CallState> call_state_; variable
/external/tensorflow/tensorflow/c/eager/
Dtape.h236 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_; variable
903 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 …]