| /external/tensorflow/tensorflow/stream_executor/ |
| D | temporary_memory_manager.cc | 32 DeviceMemoryBase device_memory = it->first; in ForceDeallocateAll() local
38 const DeviceMemoryBase& device_memory, uint64 generation, bool must_exist) { in MarkFinalized()
56 DeviceMemoryBase device_memory = it->first; in DeallocateFinalizedTemporaries() local
67 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory, in IsFinalized()
83 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory, in HasAllocated()
97 DeviceMemoryBase device_memory = in AllocateArrayBase() local
|
| D | temporary_device_memory.cc | 35 const DeviceMemoryBase& TemporaryDeviceMemoryBase::device_memory() const { in device_memory() function in perftools::gputools::TemporaryDeviceMemoryBase
59 Stream* parent, DeviceMemoryBase device_memory, in TemporaryDeviceMemoryBase()
|
| D | temporary_device_memory.h | 120 const DeviceMemory<T>& device_memory() const { in device_memory() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | allocation_tracker.cc | 159 perftools::gputools::DeviceMemoryBase device_memory, int device_ordinal) { in AddAllocationOrIncrementRefCount()
171 perftools::gputools::DeviceMemoryBase device_memory, int device_ordinal) { in DecrementRefCount()
|
| D | allocation_tracker.h | 65 perftools::gputools::DeviceMemoryBase device_memory; member
|
| D | generic_transfer_manager.cc | 122 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in TransferLiteralToDevice() local
|
| D | transfer_manager.cc | 125 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in WriteTupleIndexTables() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_manager.h | 63 perftools::gputools::DeviceMemoryBase* device_memory() { in device_memory() function
|
| D | gpu_executable.cc | 299 const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 261 [&](const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in CreateResultShapedBuffer()
|
| D | cpu_transfer_manager.cc | 55 se::DeviceMemoryBase* device_memory() { return &device_memory_; } in device_memory() function in xla::__anon108200010111::CpuInfeedBuffer
|
| D | parallel_cpu_executable.cc | 478 [&](const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| D | cudnn_rnn_ops.cc | 198 DeviceMemoryBase SliceDeviceMemory(const DeviceMemoryBase& device_memory, in SliceDeviceMemory()
|
| /external/vulkan-validation-layers/tests/ |
| D | layer_validation_tests.cpp | 801 VkDeviceMemory device_memory; in ~VkBufferTest() member
|