Searched refs:executor_ (Results 1 – 7 of 7) sorted by relevance
101 EagerExecutor* Executor() { return &executor_; } in Executor()128 Status AsyncWait() { return executor_.WaitForAllPendingNodes(); } in AsyncWait()130 Status GetStatus() { return executor_.status(); } in GetStatus()132 void ClearAsyncError() { executor_.ClearError(); } in ClearAsyncError()147 uint64 NextId() { return executor_.NextId(); } in NextId()149 void ExecutorAdd(EagerNode* node) { executor_.Add(node); } in ExecutorAdd()296 EagerExecutor executor_; variable
95 executor_.EnableAsync(); in InitDeviceMapAndAsync()130 executor_.EnableAsync(); in SetAsyncForThread()136 return executor_.WaitForAllPendingNodes(); in SetAsyncForThread()146 TF_RETURN_IF_ERROR(executor_.WaitForAllPendingNodes()); in ClearCaches()217 executor_.WaitForAllPendingNodes().IgnoreError(); in ~EagerContext()
730 : executor_(executor), module_handle_(module_handle) {} in ScopedModuleHandle()733 executor_ = other.executor_; in ScopedModuleHandle()735 other.executor_ = nullptr; in ScopedModuleHandle()740 executor_ = other.executor_;742 other.executor_ = nullptr;749 CHECK(executor_->UnloadModule(module_handle_)); in ~ScopedModuleHandle()754 StreamExecutor *executor_;
110 se::StreamExecutor* executor() const { return executor_; } in executor()134 se::StreamExecutor* executor_; // not owned variable
359 bool ok = executor_->SynchronousMemZero( in InitScratchBuffers()379 executor_ = executor_status.ValueOrDie(); in Init()380 em_.reset(new EventMgr(executor_, options.config.gpu_options())); in Init()389 tf_gpu_id_, i, executor_, options.config.gpu_options())); in Init()
104 std::unique_ptr<TfLiteModelExecutor> executor_; variable
95 : executor_(std::move(executor)), in TFLiteEmbeddingExecutor()