| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_hlo_test.cc | 95 Literal reduce_input = LiteralUtil::CreateR4<float>(
|
| bfloat16_test.cc | 101 {LiteralUtil::CreateR4<bfloat16>(
144 {LiteralUtil::CreateR4<bfloat16>(
|
| round_trip_transfer_test.cc | 102 RoundTripTest(LiteralUtil::CreateR4<float>({{
|
| reshape_test.cc | 647 auto expected_literal = LiteralUtil::CreateR4<float>({
674 auto expected_literal = LiteralUtil::CreateR4<float>({
[all...] |
| scatter_test.cc | 192 LiteralUtil::CreateR4<float>({{{{0.f}, {0.f}}, {{0.f}, {0.f}}}});
194 LiteralUtil::CreateR4<float>({{{{0.12}, {0.28}}, {{0.018}, {0.42}}}});
|
| literal_test_util.h | 235 EXPECT_TRUE(Near(LiteralUtil::CreateR4<NativeT>(expected), actual, error));
|
| copy_test.cc | 78 TestCopyOp(LiteralUtil::CreateR4(
|
| batch_normalization_test.cc | 246 {LiteralUtil::CreateR4<float>({{{{-1.6f, -2.0f}}, {{0.1f, 0.6f}}},
270 {LiteralUtil::CreateR4<float>({{{{-1.6f}, {-2.0f}}, {{0.1f}, {0.6f}}},
363 {LiteralUtil::CreateR4<float>({{{{-3.f}, {-3.f}}, {{-1.f}, {-1.f}}},
[all...] |
| dynamic_ops_test.cc | 749 auto input_literal = LiteralUtil::CreateR4(
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| infeed_test.cc | 88 TestInfeedRoundTrip(LiteralUtil::CreateR4(
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_infeed_test.cc | 94 TestInfeedRoundTrip(LiteralUtil::CreateR4(
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal_util.h | 94 static Literal CreateR4(
428 /* static */ Literal LiteralUtil::CreateR4(
|
| literal_test.cc | 717 auto original = LiteralUtil::CreateR4<float>({{
[all...] |
