Searched refs:CreateR4 (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | bfloat16_test.cc | 101 {LiteralUtil::CreateR4<bfloat16>( in XLA_TEST_F()
144 {LiteralUtil::CreateR4<bfloat16>( in XLA_TEST_F()
|
| D | reshape_test.cc | 647 auto expected_literal = LiteralUtil::CreateR4<float>({ in XLA_TEST_P()
674 auto expected_literal = LiteralUtil::CreateR4<float>({ in XLA_TEST_P()
791 auto literal_1x2x3x4 = LiteralUtil::CreateR4<float>( in XLA_TEST_P()
805 auto literal_1x2x3x4 = LiteralUtil::CreateR4<float>( in XLA_TEST_P()
817 auto expected_2x4x3x1 = LiteralUtil::CreateR4<float>( in XLA_TEST_P()
|
| D | reduce_hlo_test.cc | 95 Literal reduce_input = LiteralUtil::CreateR4<float>( in XLA_TEST_P()
|
| D | round_trip_transfer_test.cc | 102 RoundTripTest(LiteralUtil::CreateR4<float>({{ in TEST_F()
|
| D | scatter_test.cc | 192 LiteralUtil::CreateR4<float>({{{{0.f}, {0.f}}, {{0.f}, {0.f}}}}); in XLA_TEST_F()
194 LiteralUtil::CreateR4<float>({{{{0.12}, {0.28}}, {{0.018}, {0.42}}}}); in XLA_TEST_F()
|
| D | batch_normalization_test.cc | 246 {LiteralUtil::CreateR4<float>({{{{-1.6f, -2.0f}}, {{0.1f, 0.6f}}}, in XLA_TEST_P()
270 {LiteralUtil::CreateR4<float>({{{{-1.6f}, {-2.0f}}, {{0.1f}, {0.6f}}}, in XLA_TEST_P()
363 {LiteralUtil::CreateR4<float>({{{{-3.f}, {-3.f}}, {{-1.f}, {-1.f}}}, in XLA_TEST_P()
|
| D | literal_test_util.h | 235 EXPECT_TRUE(Near(LiteralUtil::CreateR4<NativeT>(expected), actual, error)); in ExpectR4Near()
|
| D | copy_test.cc | 78 TestCopyOp(LiteralUtil::CreateR4( in XLA_TEST_F()
|
| D | dynamic_ops_test.cc | 749 auto input_literal = LiteralUtil::CreateR4( in BM_DynamicSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | infeed_test.cc | 88 TestInfeedRoundTrip(LiteralUtil::CreateR4( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 94 static Literal CreateR4(
428 /* static */ Literal LiteralUtil::CreateR4( in CreateR4() function
|
| D | literal_test.cc | 717 auto original = LiteralUtil::CreateR4<float>({{ in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 94 TestInfeedRoundTrip(LiteralUtil::CreateR4( in TEST_F()
|