| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| literal_test_util.h | 44 class LiteralTestUtil {
152 TF_DISALLOW_COPY_AND_ASSIGN(LiteralTestUtil);
156 /* static */ void LiteralTestUtil::ExpectR0Equal(NativeT expected,
162 /* static */ void LiteralTestUtil::ExpectR1Equal(
168 /* static */ void LiteralTestUtil::ExpectR2Equal(
175 /* static */ void LiteralTestUtil::ExpectR3Equal(
183 /* static */ void LiteralTestUtil::ExpectR2EqualArray2D(
189 /* static */ void LiteralTestUtil::ExpectR3EqualArray3D(
195 /* static */ void LiteralTestUtil::ExpectR4EqualArray4D(
201 /* static */ void LiteralTestUtil::ExpectR0Near(NativeT expected
[all...] |
| literal_test_util_test.cc | 38 EXPECT_TRUE(LiteralTestUtil::Equal(literal, literal));
54 CHECK(LiteralTestUtil::Equal(lhs, rhs)) << "LHS and RHS are unequal";
64 CHECK(LiteralTestUtil::Near(two, four, error)) << "two is not near four";
106 ::testing::AssertionResult result = LiteralTestUtil::Equal(expected, actual);
118 EXPECT_TRUE(LiteralTestUtil::Near(a, b, ErrorSpec{0.0001}));
126 EXPECT_TRUE(LiteralTestUtil::Near(a, b, ErrorSpec{0.0001}));
129 TEST(LiteralTestUtil, NearComparatorDifferentLengths) {
134 EXPECT_FALSE(LiteralTestUtil::Near(a, b, ErrorSpec{0.0001}));
135 EXPECT_FALSE(LiteralTestUtil::Near(b, a, ErrorSpec{0.0001}));
|
| deconstruct_tuple_test.cc | 69 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal);
71 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal);
91 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal);
93 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal);
99 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal);
101 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal);
121 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal);
123 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal);
125 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal);
127 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal)
[all...] |
| literal_test_util.cc | 67 /* static */ ::testing::AssertionResult LiteralTestUtil::EqualShapes(
72 /* static */ ::testing::AssertionResult LiteralTestUtil::EqualShapesAndLayouts(
82 /* static */ ::testing::AssertionResult LiteralTestUtil::Equal(
87 /* static */ ::testing::AssertionResult LiteralTestUtil::Near(
94 /* static */ ::testing::AssertionResult LiteralTestUtil::NearOrEqual(
|
| broadcast_test.cc | 49 EXPECT_TRUE(LiteralTestUtil::Near(LiteralUtil::CreateR0<float>(42.0), result,
65 EXPECT_TRUE(LiteralTestUtil::Near(
88 EXPECT_TRUE(LiteralTestUtil::Near(
92 EXPECT_TRUE(LiteralTestUtil::Near(
109 EXPECT_TRUE(LiteralTestUtil::Near(
128 EXPECT_TRUE(LiteralTestUtil::Near(
145 EXPECT_TRUE(LiteralTestUtil::Near(
169 EXPECT_TRUE(LiteralTestUtil::Near(
199 EXPECT_TRUE(LiteralTestUtil::Near(
221 EXPECT_TRUE(LiteralTestUtil::Near(LiteralUtil::CreateR4FromArray4D(r4_array)
[all...] |
| client_test.cc | 65 ASSERT_TRUE(LiteralTestUtil::EqualShapesAndLayouts(
67 EXPECT_TRUE(LiteralTestUtil::Equal(expected_literal, computed));
94 LiteralTestUtil::ExpectR2Equal<int32>({{1, 2}, {3, 4}},
96 LiteralTestUtil::ExpectR2Equal<int32>({{10, 20}, {30, 40}},
149 EXPECT_TRUE(LiteralTestUtil::Equal(expected_result, result_literal));
|
| multioutput_fusion_test.cc | 120 EXPECT_TRUE(LiteralTestUtil::Near(expect, actual, error_spec_));
183 EXPECT_TRUE(LiteralTestUtil::Near(expect, actual, error_spec_));
219 EXPECT_TRUE(LiteralTestUtil::Equal(
245 LiteralTestUtil::ExpectR1Equal<float>({0.0, 4.0, 9.0, 1.0}, result);
275 LiteralTestUtil::ExpectR1Equal<float>({0.0, 4.0, 9.0}, result);
316 EXPECT_TRUE(LiteralTestUtil::Equal(
345 EXPECT_TRUE(LiteralTestUtil::Equal(
375 EXPECT_TRUE(LiteralTestUtil::Equal(
405 EXPECT_TRUE(LiteralTestUtil::Equal(
436 EXPECT_TRUE(LiteralTestUtil::Equal
[all...] |
| transfer_manager_test.cc | 83 LiteralTestUtil::ExpectR0Equal<uint32>(42, result);
99 LiteralTestUtil::ExpectR1Equal<float>({1.25f, 2.5f, -17.0f, -20.125f},
117 LiteralTestUtil::ExpectR1Equal<float>(test_vector, result);
149 LiteralTestUtil::ExpectR2Equal<float>(
171 LiteralTestUtil::ExpectR2Equal<float>(
189 EXPECT_TRUE(LiteralTestUtil::Equal(literal, result));
203 EXPECT_TRUE(LiteralTestUtil::Equal(literal, result));
222 EXPECT_TRUE(LiteralTestUtil::Equal(literal, result));
237 EXPECT_TRUE(LiteralTestUtil::Equal(literal, result));
255 EXPECT_TRUE(LiteralTestUtil::Equal(literal, result))
[all...] |
| local_client_execute_test.cc | 61 LiteralTestUtil::ExpectR0Near<float>(123.f, ShapedBufferToLiteral(result),
74 LiteralTestUtil::ExpectR0Near<float>(165.f, ShapedBufferToLiteral(result),
87 LiteralTestUtil::ExpectR1Near<float>({}, ShapedBufferToLiteral(result),
101 LiteralTestUtil::ExpectR1Near<float>(
118 LiteralTestUtil::ExpectR1Near<float>(
144 LiteralTestUtil::ExpectR2Near<float>({{11.0f, 22.0f}, {33.0f, 44.0f}},
151 LiteralTestUtil::ExpectR2Near<float>({{11.0f, 22.0f}, {33.0f, 44.0f}},
176 LiteralTestUtil::ExpectR2Near<float>({{11.0f, 22.0f}, {33.0f, 44.0f}},
188 LiteralTestUtil::ExpectR2Near<float>({{11.0f, 22.0f}, {33.0f, 44.0f}},
212 LiteralTestUtil::ExpectR2Equal<float>({{1.0f, 2.0f}, {3.0f, 4.0f}}
[all...] |
| local_client_allocation_test.cc | 60 LiteralTestUtil::ExpectR1Near<float>(
94 LiteralTestUtil::ExpectR1Near<float>(
|
| copy_test.cc | 52 EXPECT_TRUE(LiteralTestUtil::Equal(literal, result));
105 LiteralTestUtil::ExpectR0Near<float>(42.0f, result, error_spec_);
125 LiteralTestUtil::ExpectR2Near<float>({{1.0, 2.0}, {3.0, 4.0}}, result,
154 LiteralTestUtil::ExpectR2Near<float>({{1.0, 3.0}, {2.0, 4.0}}, result,
185 LiteralTestUtil::ExpectR3EqualArray3D(a, result);
219 LiteralTestUtil::ExpectR4EqualArray4D(a, result);
255 EXPECT_TRUE(LiteralTestUtil::Equal(empty, actual));
|
| custom_call_test.cc | 93 LiteralTestUtil::ExpectR0Near<float>(44.0f, result, error_spec_);
114 LiteralTestUtil::ExpectR0Near<float>(10.0f, result, error_spec_);
137 LiteralTestUtil::ExpectR3EqualArray3D<float>(
160 LiteralTestUtil::ExpectR2Equal<float>({{2.f, 4.f}, {3.f, 5.f}}, result);
187 LiteralTestUtil::ExpectR2Equal<float>({{3.f, 4.f}, {5.f, 6.f}}, result);
|
| fusion_test.cc | 133 EXPECT_TRUE(LiteralTestUtil::Near(expected, actual, ErrorSpec(1e-4)));
135 EXPECT_TRUE(LiteralTestUtil::Equal(expected, actual));
240 EXPECT_TRUE(LiteralTestUtil::Near(
266 EXPECT_TRUE(LiteralTestUtil::Near(
326 EXPECT_TRUE(LiteralTestUtil::Near(
342 LiteralTestUtil::Equal(LiteralUtil::CreateR0<int32>(5),
356 EXPECT_TRUE(LiteralTestUtil::Equal(
371 EXPECT_TRUE(LiteralTestUtil::Equal(
387 LiteralTestUtil::Equal(LiteralUtil::CreateR0<int32>(7),
402 LiteralTestUtil::Equal(LiteralUtil::CreateR3<int32>({{{7}}})
[all...] |
| replay_test.cc | 68 LiteralTestUtil::ExpectR0Equal<int32>(4, literal);
109 LiteralTestUtil::ExpectR0Equal<int32>(5, literal);
148 LiteralTestUtil::ExpectR1Equal<int32>({3, 4, 5}, literal);
|
| prng_test.cc | 276 EXPECT_TRUE(LiteralTestUtil::Equal(result1, result2));
277 EXPECT_TRUE(LiteralTestUtil::Equal(result1, result3));
278 EXPECT_FALSE(LiteralTestUtil::Equal(result1, result4));
279 EXPECT_FALSE(LiteralTestUtil::Equal(result4, result5));
280 EXPECT_FALSE(LiteralTestUtil::Equal(result5, result6));
313 EXPECT_FALSE(LiteralTestUtil::Equal(results[0], results[1]));
|
| compilation_cache_test.cc | 51 EXPECT_TRUE(LiteralTestUtil::Near(
67 EXPECT_TRUE(LiteralTestUtil::Near(
|
| round_trip_packed_literal_test.cc | 100 EXPECT_TRUE(LiteralTestUtil::Equal(round_tripped, actual));
133 EXPECT_TRUE(LiteralTestUtil::Equal(round_tripped, actual));
|
| compute_constant_test.cc | 226 EXPECT_TRUE(LiteralTestUtil::Equal(expected_literal, computed));
240 EXPECT_TRUE(LiteralTestUtil::Equal(expected_literal, computed));
261 ASSERT_TRUE(LiteralTestUtil::EqualShapesAndLayouts(
263 EXPECT_TRUE(LiteralTestUtil::Equal(expected_literal, computed));
|
| constants_test.cc | 170 LiteralTestUtil::ExpectR2Near<float>({{1.0}, {2.0}},
172 LiteralTestUtil::ExpectR1Near<float>({2.0, 42.0}, LiteralSlice(result, {1}),
230 EXPECT_TRUE(LiteralTestUtil::Equal(param, result));
|
| token_hlo_test.cc | 39 EXPECT_TRUE(LiteralTestUtil::Equal(result, LiteralUtil::CreateToken()));
53 LiteralTestUtil::Equal(result, LiteralUtil::MakeTuple({&token_literal})));
68 EXPECT_TRUE(LiteralTestUtil::Equal(result, LiteralUtil::CreateToken()));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| reference_util_test.cc | 57 LiteralTestUtil::ExpectR2Near<float>({{1.f, 4.f}, {2.f, 5.f}, {3.f, 6.f}},
69 LiteralTestUtil::ExpectR2Near<float>({{58.f, 64.f}, {139.f, 154.f}},
77 LiteralTestUtil::ExpectR1Near<float>({6.f, 15.f}, actual_literal,
85 LiteralTestUtil::ExpectR1Near<float>({5.f, 7.f, 9.f}, actual_literal,
93 LiteralTestUtil::ExpectR1Equal<float>({0}, result);
100 LiteralTestUtil::ExpectR2NearArray2D(*matrix_, actual_literal,
110 LiteralTestUtil::ExpectR2Near<float>({{1.f, 3.f, 5.f}, {5.f, 7.f, 9.f}},
124 LiteralTestUtil::ExpectR4NearArray4D(expected, actual_literal,
141 LiteralTestUtil::ExpectR4NearArray4D(expected, actual_literal,
149 LiteralTestUtil::ExpectR2Near<float>({{1.f, 2.f}, {4.f, 5.f}}, actual_literal
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator_test.cc | 94 EXPECT_TRUE(LiteralTestUtil::Near(expected, result, error));
96 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result));
111 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result));
129 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result));
170 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result));
197 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result));
218 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result));
243 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result));
411 EXPECT_TRUE(LiteralTestUtil::Equal(expected, result));
454 EXPECT_TRUE(LiteralTestUtil::Equal(result, output_literal))
[all...] |
| map_inliner_test.cc | 74 EXPECT_TRUE(LiteralTestUtil::Equal(result, expected));
108 EXPECT_TRUE(LiteralTestUtil::Equal(result, expected));
146 EXPECT_TRUE(LiteralTestUtil::Equal(result, expected));
177 EXPECT_TRUE(LiteralTestUtil::Equal(result, expected));
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| grpc_client_test.cc | 102 EXPECT_TRUE(LiteralTestUtil::Near(expected_literal, result_literal,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_fusion_test.cc | 88 LiteralTestUtil::ExpectR1Near<float>({1.0, 40.0, -5.0}, result, error_spec_);
135 LiteralTestUtil::ExpectR1Near<float>({14.0, 40.0, 40.0}, result, error_spec_);
221 LiteralTestUtil::ExpectR1Near<float>({14.0, 40.0, 40.0, 14.0, 40.0, 40.0},
|
