| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_executor.cc | 34 EagerExecutor::EagerExecutor(bool async) in EagerExecutor() function in tensorflow::EagerExecutor
39 std::bind(&EagerExecutor::Run, this)) in EagerExecutor()
45 EagerExecutor::~EagerExecutor() { in ~EagerExecutor()
56 Status EagerExecutor::ShutDown() { in ShutDown()
89 const char* EagerExecutor::StateStringLocked() { in StateStringLocked()
100 Status EagerExecutor::SyncExecute(EagerNode* node) { in SyncExecute()
127 Status EagerExecutor::AddOrExecute(std::unique_ptr<EagerNode> node) { in AddOrExecute()
176 tensorflow::Status EagerExecutor::WaitForAllPendingNodes() { in WaitForAllPendingNodes()
181 tensorflow::Status EagerExecutor::WaitForAllPendingNodesLocked( in WaitForAllPendingNodesLocked()
199 void EagerExecutor::ClearError() { in ClearError()
[all …]
|
| D | eager_executor.h | 112 class EagerExecutor {
114 explicit EagerExecutor(bool async);
116 ~EagerExecutor();
265 inline bool EagerExecutor::Async() const { return thread_ != nullptr; } in Async()
|
| D | eager_operation.h | 142 EagerExecutor* executor,
184 EagerExecutor& Executor() { return *executor_; } in Executor()
254 EagerExecutor* executor_; // Not owned.
|
| D | context.h | 158 void SetExecutorForThread(EagerExecutor* executor) override;
212 EagerExecutor& Executor() override;
643 EagerExecutor default_executor_;
646 std::unordered_map<std::thread::id, EagerExecutor*> thread_local_executor_
648 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 | 27 EagerExecutor executor_;
|
| D | context.cc | 330 EagerExecutor& EagerContext::Executor() { in Executor()
336 void EagerContext::SetExecutorForThread(EagerExecutor* executor) { in SetExecutorForThread()
367 std::unordered_map<std::thread::id, EagerExecutor*> executors_copy; in ClearCachesAndThreadExecutors()
497 std::unordered_map<std::thread::id, EagerExecutor*> executors_copy; in WaitForAndCloseRemoteContexts()
528 std::unordered_map<std::thread::id, EagerExecutor*> executors_copy; in ~EagerContext()
799 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 | execute.cc | 611 EagerExecutor& executor = op->Executor(); in AddOrExecuteNode()
1148 EagerExecutor* executor, Device* dstd, in LocalEagerCopyToDevice()
1216 EagerExecutor* executor, Device* device, bool mirror, in EagerCopyToDevice()
|
| D | eager_operation.cc | 314 EagerExecutor* executor, in Reset()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tfe_executor_internal.h | 24 : owned_executor(new tensorflow::EagerExecutor(async)) {} in TFE_Executor()
26 explicit TFE_Executor(tensorflow::EagerExecutor* executor) in TFE_Executor()
29 tensorflow::EagerExecutor* executor() { in executor()
33 std::unique_ptr<tensorflow::EagerExecutor> owned_executor;
34 tensorflow::EagerExecutor* unowned_executor;
|
| D | immediate_execution_context.h | 40 class EagerExecutor; variable
182 virtual EagerExecutor& Executor() = 0;
184 virtual void SetExecutorForThread(EagerExecutor* executor) = 0;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr.h | 79 EagerExecutor& GetOrCreateExecutorForStream(uint64 stream_id);
130 std::unordered_map<uint64, EagerExecutor> executor_map_
|
| D | remote_copy_node.h | 63 RemoteCopyNode(EagerContext* ctx, EagerExecutor* executor, TensorHandle* src,
165 EagerExecutor* const executor_;
|
| D | eager_service_impl.cc | 97 EagerExecutor* eager_executor, in GetEagerOperationAndNumRetvals()
449 EagerExecutor* eager_executor = &eager_context->Executor(); in RunComponentFunction()
513 EagerExecutor* eager_executor, in ExecuteOp()
562 EagerExecutor& executor = in Enqueue()
|
| D | eager_service_impl.h | 212 EagerContext* eager_context, EagerExecutor* eager_executor,
|
| D | remote_mgr.cc | 192 EagerExecutor& RemoteMgr::GetOrCreateExecutorForStream(uint64 stream_id) { in GetOrCreateExecutorForStream()
|
| D | remote_copy_node.cc | 76 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/tools/def_file_filter/ |
| D | symbols_pybind.txt | 201 tensorflow::EagerExecutor::~EagerExecutor
|