Home
last modified time | relevance | path

Searched refs:ExecutesBefore (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_schedule_test.cc97 EXPECT_TRUE(order->ExecutesBefore(x, dot1)); in TEST_F()
98 EXPECT_TRUE(order->ExecutesBefore(x, dot2)); in TEST_F()
99 EXPECT_TRUE(order->ExecutesBefore(y, dot1)); in TEST_F()
100 EXPECT_TRUE(order->ExecutesBefore(y, dot2)); in TEST_F()
101 EXPECT_TRUE(order->ExecutesBefore(z, dot2)); in TEST_F()
102 EXPECT_TRUE(order->ExecutesBefore(dot1, dot2)); in TEST_F()
104 EXPECT_FALSE(order->ExecutesBefore(x, x)); in TEST_F()
105 EXPECT_FALSE(order->ExecutesBefore(x, y)); in TEST_F()
106 EXPECT_FALSE(order->ExecutesBefore(x, z)); in TEST_F()
107 EXPECT_FALSE(order->ExecutesBefore(y, x)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering_test.cc86 EXPECT_TRUE(ordering.ExecutesBefore(x, y)); in TEST_F()
87 EXPECT_FALSE(ordering.ExecutesBefore(y, x)); in TEST_F()
89 EXPECT_TRUE(ordering.ExecutesBefore(a, b)); in TEST_F()
90 EXPECT_FALSE(ordering.ExecutesBefore(b, a)); in TEST_F()
92 EXPECT_FALSE(ordering.ExecutesBefore(a, x)); in TEST_F()
93 EXPECT_TRUE(ordering.ExecutesBefore(a, y)); in TEST_F()
94 EXPECT_FALSE(ordering.ExecutesBefore(x, a)); in TEST_F()
95 EXPECT_FALSE(ordering.ExecutesBefore(y, a)); in TEST_F()
97 EXPECT_FALSE(ordering.ExecutesBefore(b, x)); in TEST_F()
98 EXPECT_FALSE(ordering.ExecutesBefore(b, y)); in TEST_F()
[all …]
Dhlo_memory_scheduler_test.cc109 EXPECT_TRUE(ordering.ExecutesBefore(add, negate)); in TEST_F()
169 EXPECT_TRUE(ordering.ExecutesBefore(instructions_by_name.at("d"), in TEST_F()
213 EXPECT_TRUE(ordering.ExecutesBefore(instructions_by_name.at("send-done"), in TEST_F()
259 EXPECT_TRUE(ordering.ExecutesBefore(abs_abs2, tuple)); in TEST_F()
309 EXPECT_TRUE(ordering.ExecutesBefore(exp, fusion)); in TEST_F()
Dhlo_ordering.h66 bool ExecutesBefore(const HloInstruction* a, const HloInstruction* b) const;
Dhlo_ordering.cc35 bool HloOrdering::ExecutesBefore(const HloInstruction* a, in ExecutesBefore() function in xla::HloOrdering
208 return ExecutesBefore(a.defining_instruction(), b.defining_instruction()); in IsDefinedBefore()