Searched refs:session_state_ (Results 1 – 7 of 7) sorted by relevance
527 args, dump_providers_, session_state_, callback, in CreateProcessDump()533 CHECK(!session_state_ || in CreateProcessDump()534 session_state_->IsDumpModeAllowed(args.level_of_detail)); in CreateProcessDump()884 session_state_ = session_state; in OnTraceLogEnabled()901 if (!session_state_->IsDumpModeAllowed(trigger.level_of_detail)) { in OnTraceLogEnabled()933 session_state_ = nullptr; in OnTraceLogDisabled()954 if (!session_state_) in IsDumpModeAllowed()956 return session_state_->IsDumpModeAllowed(dump_mode); in IsDumpModeAllowed()
151 return session_state_; in session_state()199 scoped_refptr<MemoryDumpSessionState> session_state_; variable
141 return session_state_; in session_state_for_testing()295 scoped_refptr<MemoryDumpSessionState> session_state_;
153 session_state_(std::move(session_state)), in ProcessMemoryDump()
529 args.session_state = &session_state_; in Run()672 run_state.tensor_store.SaveTensors(output_names, &session_state_)); in Run()771 args.session_state = &session_state_; in PRunSetup()857 s = run_state->tensor_store.SaveTensors(output_names, &session_state_); in PRun()905 return session_state_.GetTensor(resource_handle.name(), retrieved_tensor); in ResourceHandleToInputTensor()
311 SessionState session_state_; variable
1176 SessionState* session_state_; member in tensorflow::__anon59a581e80111::ExecutorState1298 session_state_(args.session_state), in ExecutorState()1508 params.session_state = session_state_; in Process()