Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dkernel.h231 struct IsDeviceMemoryValueLike<SharedDeviceMemory<U> &> {
246 struct IsDeviceMemoryValueLike<SharedDeviceMemory<U>> {
261 struct IsSharedDeviceMemory<SharedDeviceMemory<U> &> {
266 struct IsSharedDeviceMemory<SharedDeviceMemory<U>> {
620 struct IsCompatibleDeviceMemoryRef<const SharedDeviceMemory<U> &,
621 SharedDeviceMemory<U>> {
Ddevice_memory.h171 class SharedDeviceMemory final : public DeviceMemoryBase {
173 explicit SharedDeviceMemory(uint64 elem_count) in SharedDeviceMemory() function