Searched defs:device_memory_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ | ||
| D | temporary_device_memory.h | 97 DeviceMemoryBase device_memory_; variable |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ | ||
| D | cpu_transfer_manager.cc | 60 se::DeviceMemoryBase device_memory_; member in xla::__anonf0769de40111::CpuInfeedBuffer |