Searched refs:op_annotation (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 77 ScopedMemoryDebugAnnotation op_annotation("SameWorkerRecvDone", 0, "dynamic", in SameWorkerRecvDone() local
122 ScopedMemoryDebugAnnotation op_annotation("RecvAsync"); in IntraProcessRecvAsyncImpl() local
|
| D | hierarchical_tree_broadcaster.cc | 410 ScopedMemoryDebugAnnotation op_annotation( in DispatchSend() local
|
| D | base_collective_executor.cc | 154 ScopedMemoryDebugAnnotation op_annotation( in TempChunk() local
|
| /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() local
1367 ScopedMemoryDebugAnnotation op_annotation( in EagerLocalExecuteAsync() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 262 ScopedMemoryDebugAnnotation op_annotation("SameWorkerRecvDone", step_id_, in SameWorkerRecvDone() local
312 ScopedMemoryDebugAnnotation op_annotation("RecvAsync", step_id_); in RecvAsync() local
|
| D | collective_rma_distributed.cc | 135 ScopedMemoryDebugAnnotation op_annotation( in RecvFromPeer() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op.cc | 76 ScopedMemoryDebugAnnotation op_annotation(name_view().data()); in ConstantOp() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 764 ScopedMemoryDebugAnnotation op_annotation(op_kernel().name_view().data(), in allocate_output() local
795 ScopedMemoryDebugAnnotation op_annotation(op_kernel().name_view().data(), in allocate_temp() local
825 ScopedMemoryDebugAnnotation op_annotation(op_kernel().name_view().data(), in allocate_persistent() local
933 ScopedMemoryDebugAnnotation op_annotation(op_kernel().name_view().data(), in maybe_set_output_by_allocate_and_copy() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 408 ScopedMemoryDebugAnnotation op_annotation("ScratchBuffer"); in InitScratchBuffers() local
567 ScopedMemoryDebugAnnotation op_annotation(op_kernel->name_view().data(), in Compute() local
699 ScopedMemoryDebugAnnotation op_annotation("MakeTensorFromProto", "dynamic", in MakeTensorFromProto() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 672 ScopedMemoryDebugAnnotation op_annotation( in RecvBufAsync() local
|