Searched refs:CreateR4 (Results 1 – 15 of 15) 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 | reduce_hlo_test.cc | 93 Literal reduce_input = LiteralUtil::CreateR4<float>( in XLA_TEST_P()
|
| D | reshape_test.cc | 681 auto expected_literal = LiteralUtil::CreateR4<float>({ in XLA_TEST_P()
709 auto expected_literal = LiteralUtil::CreateR4<float>({ in XLA_TEST_P()
830 auto literal_1x2x3x4 = LiteralUtil::CreateR4<float>( in XLA_TEST_P()
845 auto literal_1x2x3x4 = LiteralUtil::CreateR4<float>( in XLA_TEST_P()
858 auto expected_2x4x3x1 = 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 | 191 LiteralUtil::CreateR4<float>({{{{0.f}, {0.f}}, {{0.f}, {0.f}}}}); in XLA_TEST_F()
193 LiteralUtil::CreateR4<float>({{{{0.12}, {0.28}}, {{0.018}, {0.42}}}}); in XLA_TEST_F()
|
| D | batch_normalization_test.cc | 247 {LiteralUtil::CreateR4<float>({{{{-1.6f, -2.0f}}, {{0.1f, 0.6f}}}, in XLA_TEST_P()
301 {LiteralUtil::CreateR4<float>({{{{-1.6f}, {-2.0f}}, {{0.1f}, {0.6f}}}, in XLA_TEST_P()
424 {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 | 766 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/service/ |
| D | dynamic_padder_test.cc | 747 Literal operand = LiteralUtil::CreateR4<int32>( in XLA_TEST_F()
973 Literal operand = LiteralUtil::CreateR4<int32>( in XLA_TEST_F()
1009 Literal operand = LiteralUtil::CreateR4<int32>( in XLA_TEST_F()
|
| D | hlo_evaluator_test.cc | 2171 auto input = LiteralUtil::CreateR4<complex64>({{{{{42.24, 24.42}}}}});
2172 auto expected = LiteralUtil::CreateR4<complex64>({{{{{0.0, 0.0}}}}});
2191 LiteralUtil::CreateR4<complex64>({{{{}}}}).Reshape({1, 1, 1, 0}));
2208 auto input = LiteralUtil::CreateR4<complex64>({{{{{42.24, 24.42}}}}});
2225 auto input = LiteralUtil::CreateR4<complex64>({{{{{42.24, 24.42}}}}});
2226 auto expected = LiteralUtil::CreateR4<complex64>({{{{{0.0, 0.0}}}}});
2245 LiteralUtil::CreateR4<complex64>({{{{}}}}).Reshape({0, 1, 0, 1}));
2262 auto input = LiteralUtil::CreateR4<complex64>({{{{{42.24, 24.42}}}}});
2279 auto input = LiteralUtil::CreateR4<complex64>(
2282 LiteralUtil::CreateR4<complex64>({{{{{42.24, 24.42}}},
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 94 TestInfeedRoundTrip(LiteralUtil::CreateR4( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 93 static Literal CreateR4(
395 /* static */ Literal LiteralUtil::CreateR4( in CreateR4() function
|
| D | literal_test.cc | 713 auto original = LiteralUtil::CreateR4<float>({{ in TEST_F()
|