| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_executor.cc | 34 EagerExecutor::EagerExecutor(bool async, bool enable_streaming_enqueue) in EagerExecutor() function in tensorflow::EagerExecutor
39 std::bind(&EagerExecutor::Run, this)) in EagerExecutor()
46 EagerExecutor::~EagerExecutor() { in ~EagerExecutor()
57 Status EagerExecutor::ShutDown() { in ShutDown()
90 const char* EagerExecutor::StateStringLocked() { in StateStringLocked()
101 Status EagerExecutor::SyncExecute(EagerNode* node) { in SyncExecute()
124 Status EagerExecutor::AddOrExecute(std::unique_ptr<EagerNode> node) { in AddOrExecute()
173 tensorflow::Status EagerExecutor::WaitForAllPendingNodes() { in WaitForAllPendingNodes()
178 tensorflow::Status EagerExecutor::WaitForAllPendingNodesLocked( in WaitForAllPendingNodesLocked()
195 void EagerExecutor::ClearError() { in ClearError()
[all …]
|
| D | eager_executor.h | 112 class EagerExecutor {
114 explicit EagerExecutor(bool async, bool enable_streaming_enqueue = true);
116 ~EagerExecutor();
270 inline bool EagerExecutor::Async() const { return thread_ != nullptr; } in Async()
272 inline bool EagerExecutor::StreamingEnqueue() const { in StreamingEnqueue()
|
| D | eager_operation.h | 142 EagerExecutor* executor,
208 EagerExecutor& Executor() { return *executor_; } in Executor()
278 EagerExecutor* executor_; // Not owned.
|
| D | context.h | 174 void SetExecutorForThread(EagerExecutor* executor) override;
229 EagerExecutor& Executor() override;
738 EagerExecutor default_executor_;
741 std::unordered_map<std::thread::id, EagerExecutor*> thread_local_executor_
743 std::unordered_map<std::thread::id, std::unordered_set<EagerExecutor*>>
|
| D | execute.h | 64 EagerExecutor* executor, Device* device, bool mirror,
|
| D | eager_op_rewrite_registry_test.cc | 28 EagerExecutor executor_;
|
| D | context.cc | 424 EagerExecutor& EagerContext::Executor() { in Executor()
430 void EagerContext::SetExecutorForThread(EagerExecutor* executor) { in SetExecutorForThread()
471 std::unordered_map<std::thread::id, EagerExecutor*> executors_copy; in ClearCachesAndThreadExecutors()
608 std::unordered_map<std::thread::id, EagerExecutor*> executors_copy; in WaitForAndCloseRemoteContexts()
639 std::unordered_map<std::thread::id, EagerExecutor*> executors_copy; in ~EagerContext()
1014 std::unordered_map<std::thread::id, EagerExecutor*> executors_copy; in SyncExecutors()
|
| D | mkl_eager_op_rewrite_test.cc | 48 EagerExecutor executor_(false); in CreateOp()
|
| D | eager_operation.cc | 314 EagerExecutor* executor, in Reset()
|
| D | execute.cc | 1310 EagerExecutor& executor = op->Executor(); in AddOrExecuteNode()
1873 EagerExecutor* executor, Device* dstd, in LocalEagerCopyToDevice()
1942 EagerExecutor* executor, Device* device, bool mirror, in EagerCopyToDevice()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tfe_executor_internal.h | 25 new tensorflow::EagerExecutor(async, enable_streaming_enqueue)) {} in TFE_Executor()
27 explicit TFE_Executor(tensorflow::EagerExecutor* executor) in TFE_Executor()
30 tensorflow::EagerExecutor* executor() { in executor()
34 std::unique_ptr<tensorflow::EagerExecutor> owned_executor;
35 tensorflow::EagerExecutor* unowned_executor;
|
| D | immediate_execution_context.h | 40 class EagerExecutor; variable
207 virtual EagerExecutor& Executor() = 0;
209 virtual void SetExecutorForThread(EagerExecutor* executor) = 0;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.h | 64 RemoteCopyNode(EagerContext* ctx, EagerExecutor* executor, TensorHandle* src,
166 EagerExecutor* const executor_;
|
| D | remote_mgr.h | 78 EagerExecutor& GetOrCreateExecutorForStream(uint64 stream_id);
129 std::unordered_map<uint64, EagerExecutor> executor_map_
|
| D | eager_service_impl.cc | 95 EagerExecutor* eager_executor, in GetEagerOperationAndNumRetvals()
483 EagerExecutor* eager_executor = &eager_context->Executor(); in RunComponentFunction()
553 EagerExecutor* eager_executor, in ExecuteOp()
602 EagerExecutor& executor = in Enqueue()
|
| D | eager_service_impl.h | 212 EagerContext* eager_context, EagerExecutor* eager_executor,
|
| D | remote_mgr.cc | 220 EagerExecutor& RemoteMgr::GetOrCreateExecutorForStream(uint64 stream_id) { in GetOrCreateExecutorForStream()
|
| D | remote_copy_node.cc | 77 RemoteCopyNode::RemoteCopyNode(EagerContext* ctx, EagerExecutor* executor, in RemoteCopyNode()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_func.cc | 196 std::unique_ptr<EagerExecutor> new_executor = nullptr; in DoCallPyFunc()
197 EagerExecutor* old_executor = nullptr; in DoCallPyFunc()
204 new_executor.reset(new EagerExecutor(call->eager_async)); in DoCallPyFunc()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.h | 205 tensorflow::EagerExecutor& Executor() override { in Executor()
208 void SetExecutorForThread(tensorflow::EagerExecutor* executor) override;
|
| D | c_api_tfrt.cc | 1236 tensorflow::EagerExecutor* executor) { in SetExecutorForThread()
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 212 tensorflow::EagerExecutor::~EagerExecutor
|