Searched refs:kAlternateMemorySpace (Results 1 – 1 of 1) sorted by relevance
44 const int64 kAlternateMemorySpace = 1; member in xla::__anona324b8620111::MemorySpaceAssignmentTest116 subshape.layout().memory_space() == kAlternateMemorySpace) { in AssignMemorySpace()131 options.alternate_memory_space = kAlternateMemorySpace; in AssignMemorySpace()170 EXPECT_EQ(subshape.layout().memory_space(), kAlternateMemorySpace) in CheckPresetAssignments()186 EXPECT_NE(subshape.layout().memory_space(), kAlternateMemorySpace) in CheckParametersInDefaultMemory()213 .memory_space() == kAlternateMemorySpace) { in CountMaximumOutstandingAsyncCopies()221 kAlternateMemorySpace) { in CountMaximumOutstandingAsyncCopies()452 kAlternateMemorySpace); in TEST_P()503 EXPECT_THAT(add, op::Add(op::Negate(), op::AsyncCopy(kAlternateMemorySpace, in TEST_P()513 kAlternateMemorySpace); in TEST_P()[all …]