Searched refs:kDefaultMemorySpace (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_test.cc | 43 const int64 kDefaultMemorySpace = 0; member in xla::__anondbd3670d0111::MemorySpaceAssignmentTest
314 kDefaultMemorySpace, in TEST_P()
348 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
349 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
537 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
538 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
543 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
544 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
692 kDefaultMemorySpace, in TEST_P()
694 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
[all …]
|
| D | layout_assignment.cc | 1975 int64 buffer_memory_space = Layout::kDefaultMemorySpace; in PropagateMemorySpace()
1979 if (memory_space != Layout::kDefaultMemorySpace) { in PropagateMemorySpace()
1980 if (buffer_memory_space != Layout::kDefaultMemorySpace && in PropagateMemorySpace()
1993 if (buffer_memory_space != Layout::kDefaultMemorySpace) { in PropagateMemorySpace()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout.h | 211 static constexpr int64 kDefaultMemorySpace = 0; variable
|