Searched refs:kDefaultMemorySpace (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_test.cc | 53 const int64_t kDefaultMemorySpace = 0; member in xla::__anon57348ac70111::MemorySpaceAssignmentTest
222 EXPECT_EQ(root->shape().layout().memory_space(), kDefaultMemorySpace); in CheckRootInDefaultMemory()
538 kDefaultMemorySpace, in TEST_P()
573 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
574 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
762 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
763 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
768 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
769 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
917 kDefaultMemorySpace, in TEST_P()
[all …]
|
| D | layout_assignment.cc | 2328 int64_t buffer_memory_space = Layout::kDefaultMemorySpace; in PropagateMemorySpace()
2335 if (memory_space != Layout::kDefaultMemorySpace) { in PropagateMemorySpace()
2336 if (buffer_memory_space != Layout::kDefaultMemorySpace && in PropagateMemorySpace()
2349 if (buffer_memory_space != Layout::kDefaultMemorySpace) { in PropagateMemorySpace()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout.h | 226 static constexpr int64_t kDefaultMemorySpace = 0; variable
|
| D | layout_util.cc | 575 : Layout::kDefaultMemorySpace; in MemorySpace()
|
| D | shape_util.cc | 1732 subshape->mutable_layout()->set_memory_space(Layout::kDefaultMemorySpace); in DeviceShapeToHostShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 187 if (xla_layout.memory_space() != xla::Layout::kDefaultMemorySpace) in ExtractLayoutsFromShapes()
|