/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | deconstruct_tuple_test.cc | 69 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 …]
|
D | collective_ops_test.cc | 620 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 …]
|
D | transfer_manager_test.cc | 99 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()
|
D | literal_test_util.h | 65 static void ExpectR1Equal(absl::Span<const NativeT> expected, 162 /* static */ void LiteralTestUtil::ExpectR1Equal( in ExpectR1Equal() function
|
D | local_client_execute_test.cc | 323 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()
|
D | multioutput_fusion_test.cc | 247 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()
|
D | replay_test.cc | 148 LiteralTestUtil::ExpectR1Equal<int32>({3, 4, 5}, literal); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
D | gpu_multistream_test.cc | 99 LiteralTestUtil::ExpectR1Equal<int32>(expected_outputs, *out_literal); in TEST() 101 LiteralTestUtil::ExpectR1Equal<int32>(expected_outputs, *out_literal); in TEST()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | reference_util_test.cc | 93 LiteralTestUtil::ExpectR1Equal<float>({0}, result); in TEST_F()
|