Searched refs:AsyncCopy (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers_test.cc | 286 EXPECT_THAT(copy_done.get(), op::AsyncCopy(2, 1, op::Parameter(0))); in TEST_F()
288 EXPECT_THAT(Explain(copy_start.get(), op::AsyncCopy(2, 1, op::Parameter(0))), in TEST_F()
291 EXPECT_THAT(Explain(copy_done.get(), op::AsyncCopy(3, 1, op::Parameter(0))), in TEST_F()
296 EXPECT_THAT(Explain(copy_done.get(), op::AsyncCopy(2, 3, op::Parameter(0))), in TEST_F()
|
| D | memory_space_assignment_test.cc | 503 EXPECT_THAT(add, op::Add(op::Negate(), op::AsyncCopy(kAlternateMemorySpace, in TEST_P()
538 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
539 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
727 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
728 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
733 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
734 op::AsyncCopy(kDefaultMemorySpace, in TEST_P()
881 op::Multiply(op::Add(op::Negate(), op::AsyncCopy(kAlternateMemorySpace, in TEST_P()
884 op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
1006 op::Bitcast(op::AsyncCopy(kAlternateMemorySpace, kDefaultMemorySpace, in TEST_P()
[all …]
|
| D | hlo_matchers.h | 452 inline ::testing::Matcher<const ::xla::HloInstruction*> AsyncCopy( in AsyncCopy() function
|
| /external/llvm-project/parallel-libs/acxxel/tests/ |
| D | multi_device_test.cpp | 10 TEST_P(MultiDeviceTest, AsyncCopy) { in TEST_P() argument
|