Searched refs:ScopedMemoryDebugAnnotation (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.h | 81 class ScopedMemoryDebugAnnotation {
87 explicit ScopedMemoryDebugAnnotation(const char* op_name) { in ScopedMemoryDebugAnnotation() function
93 explicit ScopedMemoryDebugAnnotation(const char* op_name, int64 step_id) { in ScopedMemoryDebugAnnotation() function
102 explicit ScopedMemoryDebugAnnotation(const char* op_name, in ScopedMemoryDebugAnnotation() function
114 explicit ScopedMemoryDebugAnnotation(const char* op_name, int64 step_id, in ScopedMemoryDebugAnnotation() function
125 ~ScopedMemoryDebugAnnotation() { annotation_ = last_annotation_; } in ~ScopedMemoryDebugAnnotation()
142 TF_DISALLOW_COPY_AND_ASSIGN(ScopedMemoryDebugAnnotation);
|
| D | allocator.cc | 30 thread_local MemoryDebugAnnotation ScopedMemoryDebugAnnotation::annotation_;
|
| D | op_kernel.cc | 764 ScopedMemoryDebugAnnotation op_annotation(op_kernel().name_view().data(), in allocate_output()
795 ScopedMemoryDebugAnnotation op_annotation(op_kernel().name_view().data(), in allocate_temp()
825 ScopedMemoryDebugAnnotation op_annotation(op_kernel().name_view().data(), in allocate_persistent()
933 ScopedMemoryDebugAnnotation op_annotation(op_kernel().name_view().data(), in maybe_set_output_by_allocate_and_copy()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 77 ScopedMemoryDebugAnnotation op_annotation("SameWorkerRecvDone", 0, "dynamic", in SameWorkerRecvDone()
122 ScopedMemoryDebugAnnotation op_annotation("RecvAsync"); in IntraProcessRecvAsyncImpl()
|
| D | bfc_allocator_test.cc | 57 ScopedMemoryDebugAnnotation annotation(test_op_name.data()); in BM_Allocator()
|
| D | bfc_allocator.cc | 461 << ScopedMemoryDebugAnnotation::CurrentAnnotation().pending_op_name in AllocateRawInternal()
499 ScopedMemoryDebugAnnotation::CurrentAnnotation(); in AddTraceMe()
570 ScopedMemoryDebugAnnotation::CurrentAnnotation(); in FindChunkPtr()
|
| D | hierarchical_tree_broadcaster.cc | 410 ScopedMemoryDebugAnnotation op_annotation( in DispatchSend()
|
| D | base_collective_executor.cc | 154 ScopedMemoryDebugAnnotation op_annotation( in TempChunk()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | copy_to_device_node.h | 53 ScopedMemoryDebugAnnotation op_annotation( in Run()
|
| D | execute.cc | 696 ScopedMemoryDebugAnnotation op_annotation( in EagerLocalExecute()
1367 ScopedMemoryDebugAnnotation op_annotation( in EagerLocalExecuteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 262 ScopedMemoryDebugAnnotation op_annotation("SameWorkerRecvDone", step_id_, in SameWorkerRecvDone()
312 ScopedMemoryDebugAnnotation op_annotation("RecvAsync", step_id_); in RecvAsync()
|
| D | collective_rma_distributed.cc | 135 ScopedMemoryDebugAnnotation op_annotation( in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op.cc | 76 ScopedMemoryDebugAnnotation op_annotation(name_view().data()); in ConstantOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 408 ScopedMemoryDebugAnnotation op_annotation("ScratchBuffer"); in InitScratchBuffers()
567 ScopedMemoryDebugAnnotation op_annotation(op_kernel->name_view().data(), in Compute()
699 ScopedMemoryDebugAnnotation op_annotation("MakeTensorFromProto", "dynamic", in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 672 ScopedMemoryDebugAnnotation op_annotation( in RecvBufAsync()
|