Searched refs:trace_me (Results 1 – 12 of 12) sorted by relevance
177 profiler::TraceMe trace_me("IfOp"); in Start() local307 profiler::TraceMe trace_me("CaseOp"); in Start() local535 profiler::TraceMe trace_me("WhileOp-EvalCond"); in EvalCond() local571 profiler::TraceMe trace_me("WhileOp-StartBody"); in StartBody() local626 profiler::TraceMe trace_me("WhileOp-EvalCond"); in DoComputeSync() local647 profiler::TraceMe trace_me("WhileOp-StartBody"); in DoComputeSync() local827 profiler::TraceMe trace_me("ForOp"); in RunNext() local
237 profiler::TraceMe trace_me("SymbolicGradientOp"); in ComputeAsync() local351 profiler::TraceMe trace_me( in ComputeAsync() local
260 profiler::TraceMe trace_me("PartitionedCallOp"); in RunFunction() local
64 profiler::TraceMe trace_me("XRTExecuteOp::LookupProto", /*level=*/2); in GetComputationCacheEntry() local194 profiler::TraceMe trace_me("RunExecutable", /*level=*/2); in RunExecutable() local282 profiler::TraceMe trace_me( in DoWork() local403 profiler::TraceMe trace_me( in DoWork() local
649 profiler::TraceMeProducer trace_me( in ScheduleWithoutSplit()680 profiler::TraceMe trace_me([task] { in ScheduleWithSplit()740 profiler::TraceMeProducer trace_me( in ScheduleWithSplit()815 profiler::TraceMeConsumer trace_me( in ProcessBatch()
535 profiler::TraceMeConsumer trace_me( in CallbackWrapper()683 profiler::TraceMeProducer trace_me( in Schedule()
326 profiler::TraceMe trace_me([padded_batch_size, padding_amount]() { in ConcatInputTensors() local
103 profiler::TraceMe trace_me("XRTMemoryManager::CompactAllocations", in CompactAllocations() local152 profiler::TraceMe trace_me("XRTMemoryManager::TryFreeMemory", /*level=*/2); in TryFreeMemory() local
73 profiler::TraceMe trace_me("TpuExecuteOp::LookupProto", /*level=*/2); in GetComputationCacheEntry() local200 profiler::TraceMe trace_me("BuildComputationInputs", /*level=*/2); in BuildComputationInputs() local402 profiler::TraceMe trace_me("AllocateOutputTensors", /*level=*/2); in AllocateOutputTensors() local631 profiler::TraceMe trace_me( in DoWork() local
78 profiler::TraceMe trace_me("TransposeTensor", /*level=*/2); in TransposeTensor() local
542 profiler::TraceMe trace_me( in ComputeAsync() local
1041 tensorflow::profiler::TraceMe trace_me("PjRtStreamExecutorBuffer::Release"); in Release() local