Home
last modified time | relevance | path

Searched refs:ExpectR1Equal (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Ddeconstruct_tuple_test.cc69 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal); in TEST_F()
71 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal); in TEST_F()
91 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal); in TEST_F()
93 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal); in TEST_F()
99 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal); in TEST_F()
101 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal); in TEST_F()
121 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal); in XLA_TEST_F()
123 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal); in XLA_TEST_F()
125 LiteralTestUtil::ExpectR1Equal<float>({2.0, 4.0, 6.0, 8.0}, literal); in XLA_TEST_F()
127 LiteralTestUtil::ExpectR1Equal<float>({1.0, 2.0, 3.0, 4.0}, literal); in XLA_TEST_F()
[all …]
Dcollective_ops_test.cc620 LiteralTestUtil::ExpectR1Equal<uint32>({10, 15, 11, 16, 12, 17, 13, 18}, in XLA_TEST_F()
622 LiteralTestUtil::ExpectR1Equal<uint32>({20, 25, 21, 26, 22, 27, 23, 28}, in XLA_TEST_F()
624 LiteralTestUtil::ExpectR1Equal<uint32>({30, 35, 31, 36, 32, 37, 33, 38}, in XLA_TEST_F()
626 LiteralTestUtil::ExpectR1Equal<uint32>({40, 45, 41, 46, 42, 47, 43, 48}, in XLA_TEST_F()
661 LiteralTestUtil::ExpectR1Equal<uint32>({43, 48, 42, 47, 41, 46, 40, 45}, in XLA_TEST_F()
663 LiteralTestUtil::ExpectR1Equal<uint32>({33, 38, 32, 37, 31, 36, 30, 35}, in XLA_TEST_F()
665 LiteralTestUtil::ExpectR1Equal<uint32>({23, 28, 22, 27, 21, 26, 20, 25}, in XLA_TEST_F()
667 LiteralTestUtil::ExpectR1Equal<uint32>({13, 18, 12, 17, 11, 16, 10, 15}, in XLA_TEST_F()
696 LiteralTestUtil::ExpectR1Equal<uint32>({23, 28, 20, 25}, results[0]); in XLA_TEST_F()
697 LiteralTestUtil::ExpectR1Equal<uint32>({22, 27, 21, 26}, results[1]); in XLA_TEST_F()
[all …]
Dtransfer_manager_test.cc99 LiteralTestUtil::ExpectR1Equal<float>({1.25f, 2.5f, -17.0f, -20.125f}, in XLA_TEST_F()
121 LiteralTestUtil::ExpectR1Equal<float>(inputs, result); in XLA_TEST_F()
139 LiteralTestUtil::ExpectR1Equal<float>(test_vector, result); in XLA_TEST_F()
159 LiteralTestUtil::ExpectR1Equal<float>(expected_output, result); in XLA_TEST_F()
Dliteral_test_util.h65 static void ExpectR1Equal(absl::Span<const NativeT> expected,
162 /* static */ void LiteralTestUtil::ExpectR1Equal( in ExpectR1Equal() function
Dlocal_client_execute_test.cc323 LiteralTestUtil::ExpectR1Equal<float>({40.0f, 71.0f, 117.0f}, in XLA_TEST_F()
362 LiteralTestUtil::ExpectR1Equal<float>({264.0, 73.0, 133.0}, in XLA_TEST_F()
749 LiteralTestUtil::ExpectR1Equal<float>({2.0f, 4.0f, 6.0f}, in XLA_TEST_F()
751 LiteralTestUtil::ExpectR1Equal<float>({1.0f, 2.0f, 3.0f}, in XLA_TEST_F()
919 LiteralTestUtil::ExpectR1Equal<float>({-4.0, 125.0, 45.0}, result); in XLA_TEST_F()
944 LiteralTestUtil::ExpectR1Equal<float>({-4.0, 125.0, 45.0}, result); in XLA_TEST_F()
Dmultioutput_fusion_test.cc247 LiteralTestUtil::ExpectR1Equal<float>({0.0, 4.0, 9.0, 1.0}, result); in XLA_TEST_F()
277 LiteralTestUtil::ExpectR1Equal<float>({0.0, 4.0, 9.0}, result); in XLA_TEST_F()
Dreplay_test.cc148 LiteralTestUtil::ExpectR1Equal<int32>({3, 4, 5}, literal); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_multistream_test.cc99 LiteralTestUtil::ExpectR1Equal<int32>(expected_outputs, *out_literal); in TEST()
101 LiteralTestUtil::ExpectR1Equal<int32>(expected_outputs, *out_literal); in TEST()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc93 LiteralTestUtil::ExpectR1Equal<float>({0}, result); in TEST_F()