Home
last modified time | relevance | path

Searched refs:ScopedMemoryDebugAnnotation (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/lib/
H A Dscoped_memory_debug_annotation.h41 class ScopedMemoryDebugAnnotation {
47 explicit ScopedMemoryDebugAnnotation(const char* op_name) { in ScopedMemoryDebugAnnotation() function
55 explicit ScopedMemoryDebugAnnotation(const char* op_name, int64_t step_id) { in ScopedMemoryDebugAnnotation() function
66 explicit ScopedMemoryDebugAnnotation( in ScopedMemoryDebugAnnotation() function
80 explicit ScopedMemoryDebugAnnotation( in ScopedMemoryDebugAnnotation() function
93 ~ScopedMemoryDebugAnnotation() { in ~ScopedMemoryDebugAnnotation()
104 ScopedMemoryDebugAnnotation(const ScopedMemoryDebugAnnotation&) = delete;
105 ScopedMemoryDebugAnnotation& operator=(const ScopedMemoryDebugAnnotation&) =
H A Dscoped_memory_debug_annotation.cc21 ScopedMemoryDebugAnnotation::ThreadMemoryDebugAnnotation() { in ThreadMemoryDebugAnnotation()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dtensorflow_profiler_logger.cc117 tensorflow::profiler::ScopedMemoryDebugAnnotation annotation( in OnTfLiteOpPrepare()
143 tensorflow::profiler::ScopedMemoryDebugAnnotation annotation( in OnTfLiteOpInvoke()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drendezvous_mgr.cc91 profiler::ScopedMemoryDebugAnnotation op_annotation( in SameWorkerRecvDone()
139 profiler::ScopedMemoryDebugAnnotation op_annotation("RecvAsync"); in IntraProcessRecvAsyncImpl()
H A Dbfc_allocator.cc484 << profiler::ScopedMemoryDebugAnnotation::CurrentAnnotation() in AllocateRawInternal()
526 profiler::ScopedMemoryDebugAnnotation::CurrentAnnotation(); in AddTraceMe()
610 profiler::ScopedMemoryDebugAnnotation::CurrentAnnotation(); in FindChunkPtr()
H A Dhierarchical_tree_broadcaster.cc411 profiler::ScopedMemoryDebugAnnotation op_annotation( in DispatchSend()
H A Dbase_collective_executor.cc156 profiler::ScopedMemoryDebugAnnotation op_annotation( in TempChunk()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dbase_rendezvous_mgr.cc275 profiler::ScopedMemoryDebugAnnotation op_annotation( in SameWorkerRecvDone()
326 profiler::ScopedMemoryDebugAnnotation op_annotation("RecvAsync", step_id_); in RecvAsync()
H A Dcollective_rma_distributed.cc155 profiler::ScopedMemoryDebugAnnotation op_annotation( in RecvFromPeer()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcopy_to_device_node.h53 profiler::ScopedMemoryDebugAnnotation op_annotation( in Run()
H A Dexecute.cc1395 profiler::ScopedMemoryDebugAnnotation op_annotation( in EagerLocalExecute()
2099 profiler::ScopedMemoryDebugAnnotation op_annotation( in EagerLocalExecuteAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcpu_allocator_impl.cc129 profiler::ScopedMemoryDebugAnnotation::CurrentAnnotation(); in AddTraceMe()
H A Dop_kernel.cc808 profiler::ScopedMemoryDebugAnnotation op_annotation( in allocate_output()
839 profiler::ScopedMemoryDebugAnnotation op_annotation( in allocate_temp()
944 profiler::ScopedMemoryDebugAnnotation op_annotation( in maybe_set_output_by_allocate_and_copy()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconstant_op.cc78 profiler::ScopedMemoryDebugAnnotation op_annotation(name_view().data()); in ConstantOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc419 profiler::ScopedMemoryDebugAnnotation op_annotation("ScratchBuffer"); in InitScratchBuffers()
675 profiler::ScopedMemoryDebugAnnotation op_annotation( in Compute()
837 profiler::ScopedMemoryDebugAnnotation op_annotation( in MakeTensorFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_worker_service.cc677 profiler::ScopedMemoryDebugAnnotation op_annotation( in RecvBufAsync()