Searched refs:function_state_ (Results 1 – 2 of 2) sorted by relevance
1295 function_state_.reset(); in Reset()1314 assert(op_ != nullptr || function_state_); in Execute()1324 if (function_state_) { in Execute()1326 auto arg_types = function_state_->GetArgTypes(); in Execute()1353 function_state_->GetRunnerTable(), &request_ctx)); in Execute()1363 function_state_->GetFunc()(exec_ctx, th_args, OpAttrsRef(attrs_), in Execute()1417 if (function_state_ && context_->IsAsync()) { in Execute()1418 retvals[i] = new TensorHandleInterface(function_state_->GetRetTypes()[i], in Execute()1554 function_state_ = std::move(result.function_state); in Initialize()
596 RCReference<FunctionState> function_state_; variable