| /external/tensorflow/tensorflow/core/profiler/lib/ |
| D | traceme.h | 75 class TraceMe {
84 explicit TraceMe(absl::string_view activity_name, int level = 1) {
99 explicit TraceMe(string &&activity_name, int level = 1) {
113 explicit TraceMe(const string &activity_name, int level = 1) = delete;
118 explicit TraceMe(const char *raw, int level = 1)
119 : TraceMe(absl::string_view(raw), level) {} in TraceMe() function
129 explicit TraceMe(NameGeneratorT name_generator, int level = 1) {
163 ~TraceMe() { Stop(); } in ~TraceMe()
209 TF_DISALLOW_COPY_AND_ASSIGN(TraceMe);
225 return TraceMe::Active(TraceMeLevel::kVerbose); in TfOpDetailsEnabled()
|
| D | annotated_traceme.h | 35 bool traceme_enabled = TraceMe::Active(level);
48 absl::optional<TraceMe> trace_me_;
|
| /external/tensorflow/tensorflow/core/profiler/internal/cpu/ |
| D | host_tracer_test.cc | 83 { TraceMe traceme("hello"); } in TEST()
84 { TraceMe traceme("world"); } in TEST()
85 { TraceMe traceme("contains#inside"); } in TEST()
86 { TraceMe traceme("good#key1=value1#"); } in TEST()
87 { TraceMe traceme("morning#key1=value1,key2=value2#"); } in TEST()
88 { TraceMe traceme("incomplete#key1=value1,key2#"); } in TEST()
117 { TraceMe traceme("hello"); } in TEST()
118 { TraceMe traceme("world"); } in TEST()
119 { TraceMe traceme("contains#inside"); } in TEST()
120 { TraceMe traceme("good#key1=value1#"); } in TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/profiler/ |
| D | traceme.py | 31 class TraceMe(object): class
35 if _pywrap_traceme.TraceMe.IsEnabled():
39 self._traceme = _pywrap_traceme.TraceMe(name)
55 with TraceMe(name):
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/platform/external/ |
| D | compat.h | 36 class TraceMe {
38 explicit TraceMe(absl::string_view tag, int level = 1) {}
39 ~TraceMe() {} in ~TraceMe()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 172 profiler::TraceMe activity("WaitForDependencies", in Run()
184 profiler::TraceMe activity("Schedule", profiler::TraceMeLevel::kInfo); in Run()
199 profiler::TraceMe activity("Nccl", profiler::TraceMeLevel::kInfo); in Run()
203 profiler::TraceMe activity("GroupSizeCopy", profiler::TraceMeLevel::kInfo); in Run()
|
| D | collective_nccl_gatherer.cc | 53 profiler::TraceMe activity("WaitForDependencies", in Run()
63 profiler::TraceMe activity("Schedule", profiler::TraceMeLevel::kInfo); in Run()
|
| D | collective_nccl_broadcaster.cc | 63 profiler::TraceMe activity("WaitForDependencies", in Run()
73 profiler::TraceMe activity("Schedule", profiler::TraceMeLevel::kInfo); in Run()
|
| D | function_ops.cc | 265 profiler::TraceMe trace_me( in ComputeAsync()
349 profiler::TraceMe activity( in ComputeAsync()
398 profiler::TraceMe trace_me( in ComputeAsync()
409 profiler::TraceMe activity( in ComputeAsync()
|
| D | functional_ops.cc | 172 profiler::TraceMe trace_me( in Start()
271 profiler::TraceMe trace_me( in Start()
385 profiler::TraceMe trace_me( in EvalCond()
452 profiler::TraceMe trace_me( in StartBody()
609 profiler::TraceMe trace_me( in RunNext()
|
| /external/tensorflow/tensorflow/python/profiler/internal/ |
| D | traceme_wrapper.cc | 36 static bool IsEnabled() { return tensorflow::profiler::TraceMe::Active(); } in IsEnabled()
40 absl::optional<tensorflow::profiler::TraceMe> traceme_;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_remote_master.cc | 116 profiler::TraceMe* NewTraceRpc(StringPiece name, ::grpc::ClientContext* ctx) { in NewTraceRpc()
119 return new profiler::TraceMe( in NewTraceRpc()
138 std::unique_ptr<profiler::TraceMe> trace;
|
| D | grpc_master_service.cc | 287 profiler::TraceMe* TraceRpc( in TraceRpc()
295 return new profiler::TraceMe([&] { return strings::StrCat(name, ":", id); }, in TraceRpc()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 178 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::FromLiterals"); in FromLiterals()
345 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::ToLiteral"); in ToLiteral()
367 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::DestructureTuple"); in DestructureTuple()
392 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::CopyToDevice"); in CopyToDevice()
429 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::BlockHostUntilReady"); in BlockHostUntilReady()
436 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::AllocateBuffer"); in AllocateBuffer()
467 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::CreateBuffer"); in CreateBuffer()
540 tensorflow::profiler::TraceMe traceme("PyTpuExecutable::Execute"); in ExecuteHelper()
642 tensorflow::profiler::TraceMe traceme("PyTpuExecutable::ExecutePerReplica"); in ExecutePerReplica()
655 tensorflow::profiler::TraceMe traceme( in ExecuteOnLocalDevices()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| D | grpc_tpu_driver.cc | 660 TraceMe activity(absl::StrCat("GrpcTpuStream::Send ")); in StreamWriterFn()
676 TraceMe activity("GrpcTpuStream::RequestComplete"); in StreamReaderFn()
678 TraceMe activity("GrpcTpuStream::TransferFromComplete"); in StreamReaderFn()
699 TraceMe activity("GrpcTpuStream::CompileComplete"); in StreamReaderFn()
724 TraceMe activity(absl::StrCat("GrpcTpuStream::Allocate(num_bytes)")); in Allocate()
740 TraceMe activity(absl::StrCat("GrpcTpuStream::Allocate(shape)")); in Allocate()
757 TraceMe activity(absl::StrCat("GrpcTpuStream::AllocateTuple")); in AllocateTuple()
774 TraceMe activity(absl::StrCat("GrpcTpuStream::Deallocate")); in Deallocate()
787 TraceMe activity(absl::StrCat("GrpcTpuStream::TransferToDevice")); in TransferToDevice()
802 TraceMe activity(absl::StrCat("GrpcTpuStream::TransferFromDevice")); in TransferFromDevice()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer.cc | 103 profiler::TraceMe activity("MemCpyAsync", profiler::TraceMeLevel::kInfo); in Run()
148 profiler::TraceMe activity("WaitForQueuedEvents", in RunAsyncParts()
170 profiler::TraceMe activity("Loop", profiler::TraceMeLevel::kInfo); in RunAsyncParts()
|
| D | ring_reducer.cc | 97 profiler::TraceMe activity("MemCpyAsync", profiler::TraceMeLevel::kInfo); in Run()
198 profiler::TraceMe activity("WaitForQueuedEvents", in RunAsyncParts()
219 profiler::TraceMe activity("Loop", profiler::TraceMeLevel::kInfo); in RunAsyncParts()
|
| D | hierarchical_tree_broadcaster.cc | 320 profiler::TraceMe activity( in RunTree()
336 profiler::TraceMe activity( in RunTree()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_client.cc | 282 tensorflow::profiler::TraceMe traceme("PyLocalBuffer::FromLiterals"); in FromHostBuffer()
515 tensorflow::profiler::TraceMe traceme("PyLocalBuffer::ToLiteral"); in ToLiteral()
549 tensorflow::profiler::TraceMe traceme("PyLocalBuffer::DestructureTuple"); in DestructureTuple()
573 tensorflow::profiler::TraceMe traceme("PyLocalBuffer::CopyToDevice"); in CopyToDevice()
647 tensorflow::profiler::TraceMe traceme("PyLocalBuffer::BlockHostUntilReady"); in BlockHostUntilReady()
732 tensorflow::profiler::TraceMe traceme("LocalExecutable::Execute"); in ExecuteHelper()
850 tensorflow::profiler::TraceMe traceme("LocalExecutable::ExecutePerReplica"); in ExecutePerReplica()
863 tensorflow::profiler::TraceMe traceme( in ExecuteOnLocalDevices()
1016 tensorflow::profiler::TraceMe traceme("LocalExecutable::Compile"); in Compile()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 140 profiler::TraceMe activity( in CopyInputToExpectedDevice()
168 profiler::TraceMe activity("ValidateInputTypeAndPlacement", in ValidateInputTypeAndPlacement()
350 profiler::TraceMe activity( in EagerLocalExecute()
379 profiler::TraceMe activity("EagerCopyToDeviceAndAddCacheKey", in EagerLocalExecute()
656 profiler::TraceMe activity("CopyInputToExpectedDevice", in EagerRemoteExecute()
901 profiler::TraceMe activity( in EagerExecute()
950 profiler::TraceMe activity("EagerKernelExecute", in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 169 tensorflow::profiler::TraceMe hlo_module_activity( in ExecuteThunks()
318 tensorflow::profiler::TraceMe hlo_module_activity( in ExecuteAsyncOnStream()
330 tensorflow::profiler::TraceMe hlo_module_activity( in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_memory_manager.cc | 103 profiler::TraceMe trace_me("XRTMemoryManager::CompactAllocations", in CompactAllocations()
152 profiler::TraceMe trace_me("XRTMemoryManager::TryFreeMemory", /*level=*/2); in TryFreeMemory()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | snapshot_dataset_op.cc | 146 profiler::TraceMe activity( in WriteRecord()
157 profiler::TraceMe activity(absl::StrCat(kClassName, kSeparator, kWriteCord), in WriteRecord()
227 profiler::TraceMe activity( in ReadRecord()
238 profiler::TraceMe activity(absl::StrCat(kClassName, kSeparator, kReadCord), in ReadRecord()
820 profiler::TraceMe activity( in GetNextInternal()
985 profiler::TraceMe activity( in ReadFile()
1229 profiler::TraceMe activity( in GetNextInternal()
1476 profiler::TraceMe activity( in ProcessOneElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 157 return tensorflow::profiler::TraceMe::ActivityStart(name); in __xla_cpu_runtime_TracingStart()
164 tensorflow::profiler::TraceMe::ActivityEnd(id); in __xla_cpu_runtime_TracingEnd()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 660 profiler::TraceMe activity( in Run()
699 profiler::TraceMe activity( in RunWithBorrowedArgs()
738 profiler::TraceMe activity( in RunInstantiated()
836 profiler::TraceMe activity( in RunAsync()
|