Home
last modified time | relevance | path

Searched refs:ScopedAnnotation (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/internal/
Dscoped_annotation_test.cc28 TEST(ScopedAnnotation, Simple) { in TEST() argument
30 ScopedAnnotation trace("blah"); in TEST()
36 ScopedAnnotation trace("blah"); in TEST()
42 ScopedAnnotation outer("foo"); in TEST()
43 ScopedAnnotation inner("bar"); in TEST()
58 ScopedAnnotation trace(annotation); in BM_ScopedAnnotationDisabled()
71 ScopedAnnotation trace(annotation); in BM_ScopedAnnotationEnabled()
85 ScopedAnnotation trace(annotation); in BM_ScopedAnnotationEnabled_Nested()
86 { ScopedAnnotation trace(annotation); } in BM_ScopedAnnotationEnabled_Nested()
100 ScopedAnnotation trace(absl::StrCat(i, "-", i * i)); in BM_ScopedAnnotationEnabled_Adhoc()
[all …]
/external/tensorflow/tensorflow/core/profiler/lib/
Dscoped_annotation.h42 class ScopedAnnotation {
44 explicit ScopedAnnotation(absl::string_view name) { in ScopedAnnotation() function
52 explicit ScopedAnnotation(const char* name) in ScopedAnnotation() function
53 : ScopedAnnotation(absl::string_view(name)) {} in ScopedAnnotation()
55 explicit ScopedAnnotation(const string& name) { in ScopedAnnotation() function
63 explicit ScopedAnnotation(string&& name) { in ScopedAnnotation() function
72 explicit ScopedAnnotation(NameGeneratorT name_generator) { in ScopedAnnotation() function
81 ~ScopedAnnotation() { in ~ScopedAnnotation()
104 TF_DISALLOW_COPY_AND_ASSIGN(ScopedAnnotation);
Dannotated_traceme.h34 bool annotation_enabled = ScopedAnnotation::IsEnabled();
49 absl::optional<ScopedAnnotation> scoped_annotation_;
/external/tensorflow/tensorflow/python/profiler/
Dscoped_annotation.py31 class ScopedAnnotation(object): class
35 if _pywrap_scoped_annotation.ScopedAnnotation.IsEnabled():
39 self._scoped_annotation = _pywrap_scoped_annotation.ScopedAnnotation(name)
/external/tensorflow/tensorflow/python/profiler/internal/
Dscoped_annotation_wrapper.cc38 return tensorflow::profiler::ScopedAnnotation::IsEnabled(); in IsEnabled()
43 absl::optional<tensorflow::profiler::ScopedAnnotation> annotation_;
/external/compiler-rt/lib/tsan/rtl/
Dtsan_interface_ann.cc32 class ScopedAnnotation { class
34 ScopedAnnotation(ThreadState *thr, const char *aname, const char *f, int l, in ScopedAnnotation() function in __tsan::ScopedAnnotation
41 ~ScopedAnnotation() { in ~ScopedAnnotation()
56 ScopedAnnotation sa(thr, __func__, f, l, caller_pc); \
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc47 using tensorflow::profiler::ScopedAnnotation;
174 bool scoped_annotation_enabled = ScopedAnnotation::IsEnabled(); in ExecuteThunks()
179 absl::optional<ScopedAnnotation> op_annotation; in ExecuteThunks()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc150 profiler::ScopedAnnotation annotation("SetProtoFromGPU"); in SetProtoFromGPU()
Dgpu_device.cc621 profiler::ScopedAnnotation annotation("MakeTensorFromProto"); in MaybeCopyTensorToGPU()
/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc207 profiler::ScopedAnnotation annotation(edge_name); in ViaDMA()
Dexecutor.cc1677 if (profiler::ScopedAnnotation::IsEnabled()) return true; in MightTrace()