Home
last modified time | relevance | path

Searched defs:trace_me (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfunctional_ops.cc177 profiler::TraceMe trace_me("IfOp"); in Start() local
307 profiler::TraceMe trace_me("CaseOp"); in Start() local
535 profiler::TraceMe trace_me("WhileOp-EvalCond"); in EvalCond() local
571 profiler::TraceMe trace_me("WhileOp-StartBody"); in StartBody() local
626 profiler::TraceMe trace_me("WhileOp-EvalCond"); in DoComputeSync() local
647 profiler::TraceMe trace_me("WhileOp-StartBody"); in DoComputeSync() local
827 profiler::TraceMe trace_me("ForOp"); in RunNext() local
Dfunction_ops.cc237 profiler::TraceMe trace_me("SymbolicGradientOp"); in ComputeAsync() local
351 profiler::TraceMe trace_me( in ComputeAsync() local
Dpartitioned_function_ops.cc260 profiler::TraceMe trace_me("PartitionedCallOp"); in RunFunction() local
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc64 profiler::TraceMe trace_me("XRTExecuteOp::LookupProto", /*level=*/2); in GetComputationCacheEntry() local
194 profiler::TraceMe trace_me("RunExecutable", /*level=*/2); in RunExecutable() local
282 profiler::TraceMe trace_me( in DoWork() local
403 profiler::TraceMe trace_me( in DoWork() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc73 profiler::TraceMe trace_me("TpuExecuteOp::LookupProto", /*level=*/2); in GetComputationCacheEntry() local
200 profiler::TraceMe trace_me("BuildComputationInputs", /*level=*/2); in BuildComputationInputs() local
402 profiler::TraceMe trace_me("AllocateOutputTensors", /*level=*/2); in AllocateOutputTensors() local
631 profiler::TraceMe trace_me( in DoWork() local
Dinfeed_ops.cc78 profiler::TraceMe trace_me("TransposeTensor", /*level=*/2); in TransposeTensor() local
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_memory_manager.cc103 profiler::TraceMe trace_me("XRTMemoryManager::CompactAllocations", in CompactAllocations() local
152 profiler::TraceMe trace_me("XRTMemoryManager::TryFreeMemory", /*level=*/2); in TryFreeMemory() local
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc326 profiler::TraceMe trace_me([padded_batch_size, padding_amount]() { in ConcatInputTensors() local
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc542 profiler::TraceMe trace_me( in ComputeAsync() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1041 tensorflow::profiler::TraceMe trace_me("PjRtStreamExecutorBuffer::Release"); in Release() local