Searched refs:operation_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 124 operation_ = context->op_kernel().name() + "/EigenAllocator"; in Reinitialize()
149 " bytes for ", operation_))); in allocate()
157 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret, in allocate()
164 LogMemory::RecordRawDeallocation(operation_, step_id_, buffer, allocator_, in deallocate()
168 new AsyncFreeData(allocator_, buffer, operation_, step_id_); in deallocate()
192 : allocator_(a), address_(p), operation_(o), step_id_(s) {} in AsyncFreeData()
195 const string operation_; member
207 LogMemory::RecordRawDeallocation(data->operation_, data->step_id_, in asyncFree()
214 string operation_; in asyncFree() member in tensorflow::EigenGpuStreamDevice
|