Searched defs:MemorySpace (Results 1 – 5 of 5) sorted by relevance
34 enum class MemorySpace { kHost, kDevice }; enum
150 using MemorySpace = gpu::MemorySpace; variable
242 MemorySpace: Optional[enum.Enum] = None variable in MemoryConfig
573 /*static*/ int64_t LayoutUtil::MemorySpace(const Shape& shape) { in MemorySpace() function in xla::LayoutUtil
500 enum class MemorySpace { kDefault, kAlternate }; enum