Searched refs:TF_GUARDED_BY (Results 1 – 25 of 316) sorted by relevance
12345678910>>...13
/external/tensorflow/tensorflow/core/common_runtime/ |
D | collective_param_resolver_local.h | 73 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 …]
|
D | pool_allocator.h | 127 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;
|
D | scoped_allocator.h | 78 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_);
|
D | direct_session.h | 201 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/ |
D | subprocess.h | 116 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_);
|
D | unbounded_work_queue.h | 57 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/ |
D | worker_impl.h | 65 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_);
|
D | task_runner.h | 102 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_);
|
D | dispatcher_impl.h | 247 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/ |
D | tracking_allocator.h | 101 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_);
|
D | cancellation.h | 170 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/ |
D | subprocess.h | 111 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/ |
D | master_session.h | 145 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/ |
D | xla_device.h | 225 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/ |
D | gcs_dns_cache.h | 65 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_);
|
D | ram_file_block_cache.h | 181 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/ |
D | c_api_internal.h | 74 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/ |
D | xrt_compilation_cache.h | 217 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/ |
D | eager_executor.h | 221 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_) =
|
D | context.h | 578 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/ |
D | local_device_state.h | 186 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/ |
D | gpu_process_state.h | 151 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/ |
D | remote_profiler_session_manager.h | 72 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/ |
D | device_event_mgr.h | 90 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/ |
D | remote_tensor_handle_data.h | 70 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