Home
last modified time | relevance | path

Searched refs:SharedDeviceMemory (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dkernel.h232 struct IsDeviceMemoryValueLike<SharedDeviceMemory<U> &> {
247 struct IsDeviceMemoryValueLike<SharedDeviceMemory<U>> {
262 struct IsSharedDeviceMemory<SharedDeviceMemory<U> &> {
267 struct IsSharedDeviceMemory<SharedDeviceMemory<U>> {
631 struct IsCompatibleDeviceMemoryRef<const SharedDeviceMemory<U> &,
632 SharedDeviceMemory<U>> {
Ddevice_memory.h176 class SharedDeviceMemory final : public DeviceMemoryBase {
178 explicit SharedDeviceMemory(uint64 elem_count) in SharedDeviceMemory() function