Searched defs:device_memory_ (Results 1 – 4 of 4) sorted by relevance
97 DeviceMemoryBase device_memory_; variable
107 void* device_memory_; variable
221 absl::InlinedVector<se::DeviceMemoryBase, 1> device_memory_; variable
60 se::DeviceMemoryBase device_memory_; member in xla::__anon31ee37eb0111::CpuInfeedBuffer