Home
last modified time | relevance | path

Searched refs:TF_GUARDED_BY (Results 1 – 25 of 316) sorted by relevance

12345678910>>...13

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.h73 CollGroupParams group TF_GUARDED_BY(mu);
74 Status status TF_GUARDED_BY(mu);
75 std::unordered_map<string, DeviceAttributes> devices TF_GUARDED_BY(mu);
76 std::vector<StatusCallback> waiting TF_GUARDED_BY(mu);
106 Status status TF_GUARDED_BY(mu);
111 int source_rank TF_GUARDED_BY(mu);
112 string communicator_key TF_GUARDED_BY(mu);
113 int known_count TF_GUARDED_BY(mu);
114 std::vector<bool> known TF_GUARDED_BY(mu);
115 std::vector<IRConsumer> known_waiters TF_GUARDED_BY(mu);
[all …]
Dpool_allocator.h127 std::multimap<const size_t, PtrRecord*> pool_ TF_GUARDED_BY(mutex_);
128 PtrRecord* lru_head_ TF_GUARDED_BY(mutex_) = nullptr;
129 PtrRecord* lru_tail_ TF_GUARDED_BY(mutex_) = nullptr;
130 int64 get_from_pool_count_ TF_GUARDED_BY(mutex_) = 0;
131 int64 put_count_ TF_GUARDED_BY(mutex_) = 0;
132 int64 allocated_count_ TF_GUARDED_BY(mutex_) = 0;
133 int64 evicted_count_ TF_GUARDED_BY(mutex_) = 0;
Dscoped_allocator.h78 int32 expected_call_count_ TF_GUARDED_BY(mu_);
79 int32 live_alloc_count_ TF_GUARDED_BY(mu_);
120 bool allocated_ TF_GUARDED_BY(mu_);
121 bool deallocated_ TF_GUARDED_BY(mu_);
122 bool in_table_ TF_GUARDED_BY(mu_);
Ddirect_session.h201 Status status TF_GUARDED_BY(mu);
346 bool graph_created_ TF_GUARDED_BY(graph_state_lock_) = false;
347 bool finalized_ TF_GUARDED_BY(graph_state_lock_) = false;
359 TF_GUARDED_BY(executor_lock_);
368 TF_GUARDED_BY(executor_lock_);
377 int64 next_callable_handle_ TF_GUARDED_BY(callables_lock_) = 0;
378 std::unordered_map<int64, Callable> callables_ TF_GUARDED_BY(callables_lock_);
382 TF_GUARDED_BY(executor_lock_);
396 TF_GUARDED_BY(graph_state_lock_);
400 TF_GUARDED_BY(graph_state_lock_);
[all …]
/external/tensorflow/tensorflow/core/platform/default/
Dsubprocess.h116 bool running_ TF_GUARDED_BY(proc_mu_);
117 pid_t pid_ TF_GUARDED_BY(proc_mu_);
120 char* exec_path_ TF_GUARDED_BY(data_mu_);
121 char** exec_argv_ TF_GUARDED_BY(data_mu_);
122 ChannelAction action_[kNFds] TF_GUARDED_BY(data_mu_);
123 int parent_pipe_[kNFds] TF_GUARDED_BY(data_mu_);
124 int child_pipe_[kNFds] TF_GUARDED_BY(data_mu_);
Dunbounded_work_queue.h57 condition_variable work_queue_cv_ TF_GUARDED_BY(work_queue_mu_);
58 size_t num_idle_threads_ TF_GUARDED_BY(work_queue_mu_) = 0;
59 bool cancelled_ TF_GUARDED_BY(work_queue_mu_) = false;
60 std::deque<WorkFunction> work_queue_ TF_GUARDED_BY(work_queue_mu_);
63 TF_GUARDED_BY(thread_pool_mu_);
/external/tensorflow/tensorflow/core/data/service/
Dworker_impl.h65 bool initialized TF_GUARDED_BY(mu) = false;
90 absl::flat_hash_map<int64, std::unique_ptr<Task>> tasks_ TF_GUARDED_BY(mu_);
92 absl::flat_hash_set<int64> finished_tasks_ TF_GUARDED_BY(mu_);
94 absl::flat_hash_set<int64> pending_completed_tasks_ TF_GUARDED_BY(mu_);
95 bool cancelled_ TF_GUARDED_BY(mu_) = false;
97 bool registered_ TF_GUARDED_BY(mu_) = false;
100 condition_variable task_completion_cv_ TF_GUARDED_BY(mu_);
103 condition_variable heartbeat_cv_ TF_GUARDED_BY(mu_);
Dtask_runner.h102 std::vector<std::vector<Tensor>> buffer_ TF_GUARDED_BY(mu_);
104 Status status_ TF_GUARDED_BY(mu_) = Status::OK();
111 bool cancelled_ TF_GUARDED_BY(mu_) = false;
156 requests_ TF_GUARDED_BY(mu_);
159 int64 first_round_ TF_GUARDED_BY(mu_) = kint64max;
160 int64 current_round_ TF_GUARDED_BY(mu_) = -1;
161 bool round_skipped_ TF_GUARDED_BY(mu_) = false;
163 std::vector<std::vector<Tensor>> buffer_ TF_GUARDED_BY(mu_);
Ddispatcher_impl.h247 bool started_ TF_GUARDED_BY(mu_) = false;
248 bool cancelled_ TF_GUARDED_BY(mu_) = false;
252 worker_stubs_ TF_GUARDED_BY(mu_);
254 std::unique_ptr<DatasetStore> dataset_store_ TF_GUARDED_BY(mu_);
258 TF_GUARDED_BY(mu_);
261 absl::flat_hash_map<int64, int64> round_robin_rounds_ TF_GUARDED_BY(mu_);
264 TF_GUARDED_BY(mu_);
265 DispatcherState state_ TF_GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/framework/
Dtracking_allocator.h101 int ref_ TF_GUARDED_BY(mu_);
105 size_t allocated_ TF_GUARDED_BY(mu_);
109 size_t high_watermark_ TF_GUARDED_BY(mu_);
114 size_t total_bytes_ TF_GUARDED_BY(mu_);
116 gtl::InlinedVector<AllocRecord, 4> allocations_ TF_GUARDED_BY(mu_);
126 std::unordered_map<const void*, Chunk> in_use_ TF_GUARDED_BY(mu_);
127 int64 next_allocation_id_ TF_GUARDED_BY(mu_);
Dcancellation.h170 bool is_removed_from_parent_ TF_GUARDED_BY(parent_->mu_) = false;
176 CancellationManager* prev_sibling_ TF_GUARDED_BY(parent_->mu_) =
178 CancellationManager* next_sibling_ TF_GUARDED_BY(parent_->mu_) =
182 std::unique_ptr<State> state_ TF_GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/platform/windows/
Dsubprocess.h111 bool running_ TF_GUARDED_BY(proc_mu_);
112 void* win_pi_ TF_GUARDED_BY(proc_mu_);
115 char* exec_path_ TF_GUARDED_BY(data_mu_);
116 char** exec_argv_ TF_GUARDED_BY(data_mu_);
117 ChannelAction action_[kNFds] TF_GUARDED_BY(data_mu_);
118 void* parent_pipe_[kNFds] TF_GUARDED_BY(data_mu_);
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.h145 std::unique_ptr<GraphExecutionState> execution_state_ TF_GUARDED_BY(mu_);
155 RCGMap run_graphs_ TF_GUARDED_BY(mu_);
156 RCGMap partial_run_graphs_ TF_GUARDED_BY(mu_);
157 int64 next_callable_handle_ TF_GUARDED_BY(mu_) = 0;
158 RCGMap callables_ TF_GUARDED_BY(mu_);
193 TF_GUARDED_BY(mu_);
197 int32 num_running_ TF_GUARDED_BY(mu_) = 0;
199 bool closed_ TF_GUARDED_BY(mu_) = false;
200 bool garbage_collected_ TF_GUARDED_BY(mu_) = false;
203 TF_GUARDED_BY(mu_);
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h225 Allocator* xla_allocator_ TF_GUARDED_BY(mu_) = nullptr; // Not owned.
231 std::shared_ptr<se::Stream> stream_ TF_GUARDED_BY(mu_);
239 std::shared_ptr<se::Stream> host_to_device_stream_ TF_GUARDED_BY(mu_);
243 TF_GUARDED_BY(mu_);
250 XlaDeviceContext* device_context_ TF_GUARDED_BY(mu_) = nullptr;
254 XlaDeviceContext* fast_mem_device_context_ TF_GUARDED_BY(mu_) = nullptr;
257 bool use_gpu_device_info_ TF_GUARDED_BY(mu_) = false;
258 std::unique_ptr<GpuDeviceInfo> gpu_device_info_ TF_GUARDED_BY(mu_);
265 bool sync_on_completion_ TF_GUARDED_BY(mu_) = true;
268 std::function<Status()> device_error_callback_ TF_GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/platform/cloud/
Dgcs_dns_cache.h65 std::default_random_engine random_ TF_GUARDED_BY(mu_);
66 bool started_ TF_GUARDED_BY(mu_) = false;
67 bool cancelled_ TF_GUARDED_BY(mu_) = false;
68 std::unique_ptr<Thread> worker_ TF_GUARDED_BY(mu_); // After mutable vars.
72 std::vector<std::vector<string>> addresses_ TF_GUARDED_BY(mu_);
Dram_file_block_cache.h181 FetchState state TF_GUARDED_BY(mu) = FetchState::CREATED;
228 BlockMap block_map_ TF_GUARDED_BY(mu_);
232 std::list<Key> lru_list_ TF_GUARDED_BY(mu_);
239 std::list<Key> lra_list_ TF_GUARDED_BY(mu_);
242 size_t cache_size_ TF_GUARDED_BY(mu_) = 0;
245 std::map<string, int64> file_signature_map_ TF_GUARDED_BY(mu_);
/external/tensorflow/tensorflow/c/
Dc_api_internal.h74 tensorflow::Graph graph TF_GUARDED_BY(mu);
77 tensorflow::ShapeRefiner refiner TF_GUARDED_BY(mu);
81 TF_GUARDED_BY(mu);
97 TF_GUARDED_BY(mu);
98 bool delete_requested TF_GUARDED_BY(mu); // set true by TF_DeleteGraph
173 tensorflow::ApiDefMap api_def_map TF_GUARDED_BY(lock);
175 bool update_docs_called TF_GUARDED_BY(lock);
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_compilation_cache.h217 int cache_entries_ TF_GUARDED_BY(mu_) = 0;
219 int marked_for_eviction_entries_ TF_GUARDED_BY(mu_) = 0;
222 int64 use_counter_ TF_GUARDED_BY(mu_) = 0;
226 std::unordered_map<string, CompiledSubgraph*> cache_ TF_GUARDED_BY(mu_);
230 TF_GUARDED_BY(mu_);
234 std::map<int64, CompiledSubgraph*> entries_by_last_use_ TF_GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_executor.h221 condition_variable nodes_pending_ TF_GUARDED_BY(node_queue_mutex_);
225 TF_GUARDED_BY(node_queue_mutex_);
229 unfinished_nodes_ TF_GUARDED_BY(node_queue_mutex_);
233 Status status_ TF_GUARDED_BY(node_queue_mutex_);
234 std::atomic<bool> ok_ TF_GUARDED_BY(node_queue_mutex_);
241 node_done_notifications_ TF_GUARDED_BY(node_queue_mutex_);
249 ExecutorState state_ TF_GUARDED_BY(node_queue_mutex_) =
Dcontext.h578 device_placement_policy_ TF_GUARDED_BY(policy_map_mu_);
591 TF_GUARDED_BY(device_type_list_mu_);
626 kernel_cache_ TF_GUARDED_BY(cache_mu_);
628 TF_GUARDED_BY(cache_mu_);
633 std::unique_ptr<RunMetadata> run_metadata_ TF_GUARDED_BY(metadata_mu_);
641 TF_GUARDED_BY(metadata_mu_);
647 TF_GUARDED_BY(executor_map_mu_);
649 has_cleanup_ TF_GUARDED_BY(executor_map_mu_);
687 uint64 context_id_ TF_GUARDED_BY(remote_state_mu_);
691 uint64 context_view_id_ TF_GUARDED_BY(remote_state_mu_);
[all …]
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dlocal_device_state.h186 int next_device_to_host_stream_ TF_GUARDED_BY(mu_) = 0;
187 int next_device_to_device_stream_ TF_GUARDED_BY(mu_) = 0;
188 std::stack<std::unique_ptr<se::Stream>> usage_stream_pool_ TF_GUARDED_BY(mu_);
190 std::random_device prng_seed_device_ TF_GUARDED_BY(mu_);
191 std::mt19937 prng_seed_generator_ TF_GUARDED_BY(mu_);
192 std::uniform_int_distribution<> prng_seed_distribution_ TF_GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.h151 std::vector<AllocatorParts> gpu_allocators_ TF_GUARDED_BY(mu_);
153 TF_GUARDED_BY(mu_);
155 std::vector<AllocatorParts> gpu_host_allocators_ TF_GUARDED_BY(mu_);
157 TF_GUARDED_BY(mu_);
159 TF_GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/profiler/rpc/client/
Dremote_profiler_session_manager.h72 RemoteProfilerSessionManagerOptions options_ TF_GUARDED_BY(mutex_);
73 ProfileRequest request_ TF_GUARDED_BY(mutex_);
76 TF_GUARDED_BY(mutex_);
78 AddressResolver resolver_ TF_GUARDED_BY(mutex_);
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr.h90 condition_variable events_pending_ TF_GUARDED_BY(mu_);
136 std::vector<se::Event*> free_events_ TF_GUARDED_BY(mu_);
139 std::deque<InUse> used_events_ TF_GUARDED_BY(mu_);
141 bool stop_polling_ TF_GUARDED_BY(mu_);
160 std::map<se::StreamExecutor*, EventMgr*> event_mgr_map_ TF_GUARDED_BY(mu_);
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_tensor_handle_data.h70 bool is_ready_ TF_GUARDED_BY(mu_);
71 Status is_poisoned_ TF_GUARDED_BY(mu_);
72 TensorShape shape_ TF_GUARDED_BY(mu_);
77 string remote_task_ TF_GUARDED_BY(mu_);

12345678910>>...13