| /external/tensorflow/tensorflow/core/profiler/lib/ |
| D | traceme.h | 82 class TraceMe {
91 explicit TraceMe(absl::string_view name, int level = 1) {
105 explicit TraceMe(std::string&& name, int level = 1) = delete;
112 explicit TraceMe(const std::string& name, int level = 1) = delete;
116 explicit TraceMe(const char* raw, int level = 1)
117 : TraceMe(absl::string_view(raw), level) {} in TraceMe() function
137 explicit TraceMe(NameGeneratorT name_generator, int level = 1) {
148 TraceMe(TraceMe&& other) { *this = std::move(other); } in TraceMe() function
149 TraceMe& operator=(TraceMe&& other) {
160 ~TraceMe() { Stop(); } in ~TraceMe()
[all …]
|
| D | connected_traceme.h | 93 context_id.has_value() ? *context_id : TraceMe::NewActivityId(); in trace_me_()
101 TraceMe trace_me_;
121 TraceMe trace_me_;
|
| D | annotated_traceme.h | 39 bool traceme_enabled = TraceMe::Active(level);
52 absl::optional<TraceMe> trace_me_;
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/platform/external/ |
| D | compat.h | 36 class TraceMe {
38 explicit TraceMe(absl::string_view name, int level = 1) {}
39 explicit TraceMe(std::string&& name, int level = 1) = delete;
40 explicit TraceMe(const std::string& name, int level = 1) = delete;
41 explicit TraceMe(const char* raw, int level = 1)
42 : TraceMe(absl::string_view(raw), level) {} in TraceMe() function
44 explicit TraceMe(NameGeneratorT name_generator, int level = 1) {}
45 ~TraceMe() {} in ~TraceMe()
|
| /external/tensorflow/tensorflow/core/profiler/internal/cpu/ |
| D | host_tracer_test.cc | 89 { TraceMe traceme("hello"); } in TEST()
90 { TraceMe traceme("world"); } in TEST()
91 { TraceMe traceme("contains#inside"); } in TEST()
92 { TraceMe traceme("good#key1=value1#"); } in TEST()
93 { TraceMe traceme("morning#key1=value1,key2=value2#"); } in TEST()
94 { TraceMe traceme("incomplete#key1=value1,key2#"); } in TEST()
134 { TraceMe traceme("hello"); } in TEST()
135 { TraceMe traceme("world"); } in TEST()
136 { TraceMe traceme("contains#inside"); } in TEST()
137 { TraceMe traceme("good#key1=value1#"); } in TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/profiler/ |
| D | traceme.py | 21 from tensorflow.python.profiler.trace import Trace as TraceMe unknown
30 with TraceMe(name):
|
| D | trace.py | 83 self._traceme = _pywrap_traceme.TraceMe(name, **kwargs)
|
| /external/tensorflow/tensorflow/python/profiler/internal/ |
| D | traceme_wrapper.h | 61 static bool IsEnabled() { return tensorflow::profiler::TraceMe::Active(); } in IsEnabled()
76 tensorflow::profiler::TraceMe traceme_;
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_local_lookup.cc | 33 profiler::TraceMe proto_lookup_traceme("Local TPU proto cache lookup", in Lookup()
52 profiler::TraceMe proto_lookup_traceme("Local TPU proto cache lookup by uid", in Lookup()
|
| D | tpu_compilation_cache_rpc_lookup.cc | 67 profiler::TraceMe proto_lookup_traceme("Remote TPU proto cache lookup", in Lookup()
102 profiler::TraceMe proto_lookup_traceme("Remote TPU proto cache lookup by uid", in Lookup()
147 profiler::TraceMe proto_lookup_traceme("Remote TPU proto cache fetch", in RemoteLookupLocked()
|
| D | tpu_compilation_cache_interface.cc | 143 profiler::TraceMe key_release_traceme( in MarkEntryForEviction()
190 profiler::TraceMe key_release_traceme("TPU compilation cache release uid", in Release()
410 profiler::TraceMe subgraph_lookup_traceme( in CompileIfKeyAbsentHelper()
561 profiler::TraceMe proto_lookup_traceme( in Lookup()
586 profiler::TraceMe proto_lookup_traceme("TPU compilation cache proto lookup", in Lookup()
|
| D | tpu_execute_op.cc | 73 profiler::TraceMe trace_me("TpuExecuteOp::LookupProto", /*level=*/2); in GetComputationCacheEntry()
200 profiler::TraceMe trace_me("BuildComputationInputs", /*level=*/2); in BuildComputationInputs()
402 profiler::TraceMe trace_me("AllocateOutputTensors", /*level=*/2); in AllocateOutputTensors()
631 profiler::TraceMe trace_me( in DoWork()
639 profiler::TraceMe trace_me_init("TPUExecuteOp::Init", /*level=*/2); in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_backend_independent_test.py | 130 with xla_client.profiler.TraceMe("test1"):
132 with xla_client.profiler.TraceMe("test2", foo=123):
135 with xla_client.profiler.TraceMe("test3"):
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | memory_checker.py | 22 from tensorflow.python.profiler.traceme import TraceMe
77 self._trace_me = TraceMe('with MemoryChecker():')
|
| /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;
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | profiler_options.proto | 32 // - Level 1 enables tracing of only user instrumented (or default) TraceMe.
33 // - Level 2 enables tracing of all level 1 TraceMe(s) and instrumented high
36 // - Level 3 enables tracing of all level 2 TraceMe(s) and more verbose
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| D | grpc_tpu_driver.cc | 660 TraceMe activity("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("GrpcTpuStream::Allocate(num_bytes)"); in Allocate()
740 TraceMe activity("GrpcTpuStream::Allocate(shape)"); in Allocate()
757 TraceMe activity("GrpcTpuStream::AllocateTuple"); in AllocateTuple()
774 TraceMe activity("GrpcTpuStream::Deallocate"); in Deallocate()
787 TraceMe activity("GrpcTpuStream::TransferToDevice"); in TransferToDevice()
802 TraceMe activity("GrpcTpuStream::TransferFromDevice"); in TransferFromDevice()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer.cc | 102 profiler::TraceMe activity("MemCpyAsync", profiler::TraceMeLevel::kInfo); in Run()
147 profiler::TraceMe activity("WaitForQueuedEvents", in RunAsyncParts()
169 profiler::TraceMe activity("Loop", profiler::TraceMeLevel::kInfo); in RunAsyncParts()
|
| D | ring_reducer.cc | 96 profiler::TraceMe activity("MemCpyAsync", profiler::TraceMeLevel::kInfo); in Run()
197 profiler::TraceMe activity("WaitForQueuedEvents", in RunAsyncParts()
218 profiler::TraceMe activity("Loop", profiler::TraceMeLevel::kInfo); in RunAsyncParts()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 107 profiler::TraceMe activity("Nccl", profiler::TraceMeLevel::kInfo); in Run()
111 profiler::TraceMe activity("GroupSizeCopy", profiler::TraceMeLevel::kInfo); in Run()
|
| D | function_ops.cc | 237 profiler::TraceMe trace_me("SymbolicGradientOp"); in ComputeAsync()
299 profiler::TraceMe activity( in ComputeAsync()
351 profiler::TraceMe trace_me( in ComputeAsync()
362 profiler::TraceMe activity( in ComputeAsync()
|
| D | functional_ops.cc | 177 profiler::TraceMe trace_me("IfOp"); in Start()
307 profiler::TraceMe trace_me("CaseOp"); in Start()
535 profiler::TraceMe trace_me("WhileOp-EvalCond"); in EvalCond()
571 profiler::TraceMe trace_me("WhileOp-StartBody"); in StartBody()
626 profiler::TraceMe trace_me("WhileOp-EvalCond"); in DoComputeSync()
647 profiler::TraceMe trace_me("WhileOp-StartBody"); in DoComputeSync()
827 profiler::TraceMe trace_me("ForOp"); in RunNext()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 177 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()
442 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::AllocateBuffer"); in AllocateBuffer()
474 tensorflow::profiler::TraceMe traceme("PyTpuBuffer::CreateBuffer"); in CreateBuffer()
551 tensorflow::profiler::TraceMe traceme("PyTpuExecutable::Execute"); in ExecuteHelper()
669 tensorflow::profiler::TraceMe traceme( in ExecuteOnLocalDevices()
799 tensorflow::profiler::TraceMe traceme("PyTpuExecutable::Compile"); in Compile()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 186 profiler::TraceMe activity("WaitForDependencies", in Enqueue()
196 profiler::TraceMe activity("Schedule", profiler::TraceMeLevel::kInfo); in Enqueue()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 64 profiler::TraceMe trace_me("XRTExecuteOp::LookupProto", /*level=*/2); in GetComputationCacheEntry()
194 profiler::TraceMe trace_me("RunExecutable", /*level=*/2); in RunExecutable()
282 profiler::TraceMe trace_me( in DoWork()
288 profiler::TraceMe trace_me_init("XRTExecuteOp::Init", /*level=*/2); in DoWork()
403 profiler::TraceMe trace_me( in DoWork()
|