Home
last modified time | relevance | path

Searched defs:TF_GUARDED_BY (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/debug/
Ddebug_io_utils_test.cc360 std::vector<string> dump_roots TF_GUARDED_BY(mu); in TEST_F() local
361 std::vector<string> dump_file_paths TF_GUARDED_BY(mu); in TEST_F() local
363 int dump_count TF_GUARDED_BY(mu) = 0; in TEST_F() local
364 int done_count TF_GUARDED_BY(mu) = 0; in TEST_F() local
Ddebug_grpc_io_utils_test.cc195 int tensor_count TF_GUARDED_BY(mu) = 0; in TEST_F() local
196 std::vector<Status> statuses TF_GUARDED_BY(mu); in TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Dsession_state.h82 std::atomic<bool> dirty_ TF_GUARDED_BY(lock_){false}; in TF_GUARDED_BY() function
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_scheduler.h131 std::atomic<bool> empty_ TF_GUARDED_BY(mu_){true}; in TF_GUARDED_BY() function
Dshared_batch_scheduler.h367 std::atomic<bool> closed_ TF_GUARDED_BY(mu_){false}; in TF_GUARDED_BY() function
/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.cc43 static std::unordered_set<std::string>* seen_ptxas_paths TF_GUARDED_BY(mu) = in WarnIfBadPtxasVersion() local
114 static auto& ptx_cache TF_GUARDED_BY(ptx_cache_mutex) = in CompileGpuAsmOrGetCached() local
/external/tensorflow/tensorflow/core/platform/
Dthread_annotations.h52 #define TF_GUARDED_BY(x) TF_INTERNAL_THREAD_ANNOTATION_ATTRIBUTE(guarded_by(x)) macro
/external/tensorflow/tensorflow/compiler/xla/service/
Ddump.cc495 static auto& module_id_to_execution_count TF_GUARDED_BY(mu) = in DumpHloSnapshotIfEnabled() local
532 static auto& module_name_to_execution_count TF_GUARDED_BY(mu) = in DumpHloSnapshotIfEnabled() local