| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| round_trip_transfer_test.cc | 48 RoundTripTest(LiteralUtil::CreateR0<int32>(42));
52 RoundTripTest(LiteralUtil::CreateR0<float>(42.0));
56 RoundTripTest(LiteralUtil::CreateR1<float>({}));
60 RoundTripTest(LiteralUtil::CreateR1<float>({42.0, 64.0}));
66 RoundTripTest(LiteralUtil::CreateR1<float>(values));
72 RoundTripTest(LiteralUtil::CreateR1<float>(values));
78 RoundTripTest(LiteralUtil::CreateR1<float>(values));
84 RoundTripTest(LiteralUtil::CreateR1<float>(values));
88 RoundTripTest(LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(10, 0)));
92 RoundTripTest(LiteralUtil::CreateR2<float>({{42.0, 64.0}, {77.0, 88.0}}))
[all...] |
| all_reduce_test.cc | 48 auto literal = LiteralUtil::CreateR1<float>({1, 2, 3});
69 auto literal0 = LiteralUtil::CreateR1<float>({1, 2, 3});
70 auto literal1 = LiteralUtil::CreateR1<float>({10, 20});
71 EXPECT_EQ(LiteralUtil::MakeTuple({&literal0, &literal1}),
95 auto literal0 = LiteralUtil::CreateR1<float>({1, 2, 3});
96 auto literal1 = LiteralUtil::CreateR1<float>({10, 20});
97 EXPECT_EQ(LiteralUtil::MakeTuple({&literal0, &literal1}),
|
| scatter_test.cc | 66 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
67 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2});
68 Literal updates = LiteralUtil::CreateR2<int32>({{10, 20, 30}, {70, 80, 90}});
97 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
98 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 1});
99 Literal updates = LiteralUtil::CreateR2<int32>({{10, 20, 30}, {70, 80, 90}});
125 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
126 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2});
128 LiteralUtil::CreateR2<int32>({{10, 30}, {40, 60}, {70, 90}});
157 LiteralUtil::CreateR2<int32>({{1, 3, 2, 0}, {3, 0, 2, 1}, {2, 3, 1, 0}})
[all...] |
| gather_operation_test.cc | 62 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
63 Literal start_indices = LiteralUtil::CreateR1<int32>({0, 2});
83 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
84 Literal start_indices = LiteralUtil::CreateR1<int32>({0, 2});
104 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
105 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}});
125 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
127 LiteralUtil::CreateR3<int32>({{{0, 2}, {2, 1}}, {{1, 2}, {2, 0}}});
147 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
149 LiteralUtil::CreateR3<int32>({{{0, 2}, {2, 1}}, {{1, 2}, {2, 0}}})
[all...] |
| multioutput_fusion_test.cc | 72 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(8.0f)));
117 Literal literal_r0 = LiteralUtil::CreateR0<float>(-9.0f);
181 Literal expect = LiteralUtil::CreateR1<float>({size * 1.5f * 3.5f});
212 auto param = LiteralUtil::MakeTupleOwned(
213 LiteralUtil::MakeTupleOwned(
214 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)),
215 LiteralUtil::CreateR0<float>(1.0)),
216 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<float>(3.0)
[all...] |
| literal_test_util_test.cc | 34 Literal literal = LiteralUtil::MakeTupleFromSlices({
35 LiteralUtil::CreateR0<int32>(42),
36 LiteralUtil::CreateR0<int32>(64),
46 Literal lhs = LiteralUtil::MakeTupleFromSlices({
47 LiteralUtil::CreateR0<int32>(42),
48 LiteralUtil::CreateR0<int32>(64),
50 Literal rhs = LiteralUtil::MakeTupleFromSlices({
51 LiteralUtil::CreateR0<int32>(64),
52 LiteralUtil::CreateR0<int32>(42),
61 auto two = LiteralUtil::CreateR0<float>(2)
[all...] |
| multi_device_all_reduce_test.cc | 46 auto literal = LiteralUtil::CreateR1<float>({1, 2, 3});
47 auto expected = LiteralUtil::CreateR1<float>({2, 4, 6});
|
| tuple_test.cc | 54 auto value = LiteralUtil::MakeTupleFromSlices(
55 {LiteralUtil::CreateR0<float>(constant_scalar),
56 LiteralUtil::CreateR1<float>(constant_vector),
57 LiteralUtil::CreateR2<float>(constant_matrix)});
69 auto value = LiteralUtil::MakeTupleFromSlices(
70 {LiteralUtil::CreateR0<float>(constant_scalar1),
71 LiteralUtil::CreateR0<float>(constant_scalar2)});
91 auto expected = LiteralUtil::MakeTupleFromSlices(
92 {LiteralUtil::CreateR0<float>(constant_scalar),
93 LiteralUtil::CreateR1<float>(constant_vector)
[all...] |
| reshape_test.cc | 58 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array);
64 auto expected_literal = LiteralUtil::CreateR1<float>({1.0f});
71 auto input_literal = LiteralUtil::CreateR1<float>({1.0f});
77 auto expected_literal = LiteralUtil::CreateR1<float>({1.0f});
84 auto input_literal = LiteralUtil::CreateR1<float>({1.0f});
90 auto expected_literal = LiteralUtil::CreateR1<float>({1.0f});
100 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array);
108 auto expected_literal = LiteralUtil::CreateR0<float>(1.0f);
116 Literal param0_literal = LiteralUtil::CreateR0<float>(1.0f);
123 auto expected_literal = LiteralUtil::CreateR1<float>({-1.0f})
[all...] |
| transfer_manager_test.cc | 72 Literal literal = LiteralUtil::CreateR0<uint32>(42);
88 LiteralUtil::CreateR1<float>({1.25f, 2.5f, -17.0f, -20.125f});
106 Literal literal = LiteralUtil::CreateR1<float>(test_vector);
122 Literal literal = LiteralUtil::CreateR1U8(test_string);
138 LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}});
155 Literal literal = LiteralUtil::CreateR2WithLayout<float>(
176 Literal literal = LiteralUtil::MakeTupleFromSlices(
177 {LiteralUtil::CreateR0<float>(123.0f),
178 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}),
179 LiteralUtil::CreateR1<float>({44.0f, -10.0f, 3333333.3f})})
[all...] |
| broadcast_test.cc | 40 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
49 EXPECT_TRUE(LiteralTestUtil::Near(LiteralUtil::CreateR0<float>(42.0), result,
56 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
66 LiteralUtil::CreateR2<float>({{42.0, 42.0}, {42.0, 42.0}}), result,
73 LiteralUtil::CreateR1<float>({1.0, 2.0, 3.0})));
89 LiteralUtil::CreateR2<float>({{1.0, 1.0}, {2.0, 2.0}, {3.0, 3.0}}),
93 LiteralUtil::CreateR2<float>({{1.0, 2.0, 3.0}, {1.0, 2.0, 3.0}}),
100 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})));
110 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}), result,
119 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})))
[all...] |
| token_hlo_test.cc | 39 EXPECT_TRUE(LiteralTestUtil::Equal(result, LiteralUtil::CreateToken()));
51 Literal token_literal = LiteralUtil::CreateToken();
53 LiteralTestUtil::Equal(result, LiteralUtil::MakeTuple({&token_literal})));
68 EXPECT_TRUE(LiteralTestUtil::Equal(result, LiteralUtil::CreateToken()));
79 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(42)));
188 auto arg = LiteralUtil::CreateR0<bool>(true);
198 auto arg = LiteralUtil::CreateR0<bool>(false);
226 auto p0 = LiteralUtil::CreateR0<float>(10.0);
227 auto p1 = LiteralUtil::CreateR0<float>(3.0);
228 auto expected = LiteralUtil::CreateR0<float>(-156.0)
[all...] |
| literal_test_util.h | 158 EXPECT_TRUE(Equal(LiteralUtil::CreateR0<NativeT>(expected), actual));
164 EXPECT_TRUE(Equal(LiteralUtil::CreateR1<NativeT>(expected), actual));
171 EXPECT_TRUE(Equal(LiteralUtil::CreateR2<NativeT>(expected), actual));
179 EXPECT_TRUE(Equal(LiteralUtil::CreateR3<NativeT>(expected), actual));
185 EXPECT_TRUE(Equal(LiteralUtil::CreateR2FromArray2D(expected), actual));
191 EXPECT_TRUE(Equal(LiteralUtil::CreateR3FromArray3D(expected), actual));
197 EXPECT_TRUE(Equal(LiteralUtil::CreateR4FromArray4D(expected), actual));
204 EXPECT_TRUE(Near(LiteralUtil::CreateR0<NativeT>(expected), actual, error));
211 EXPECT_TRUE(Near(LiteralUtil::CreateR1<NativeT>(expected), actual, error));
218 EXPECT_TRUE(Near(LiteralUtil::CreateR2<NativeT>(expected), actual, error))
[all...] |
| fusion_test.cc | 99 LiteralUtil::CreateR2FromArray2D(operand_data[i])));
130 auto expected = LiteralUtil::CreateR2FromArray2D(answer_data);
204 LiteralUtil::CreateR2<float>({{1.0}, {2.0}, {3.0}})));
206 LiteralUtil::CreateR2<float>({{-1.0}, {-1.0}, {-1.0}})));
212 LiteralUtil::CreateR2<float>({{1.62, 2.72, 3.14}})));
216 LiteralUtil::CreateR2<float>({{1.0, 1.0, 1.0}, {0.0, 0.0, 0.0}})));
222 LiteralUtil::CreateR2<float>({{0.5, 0.5, 0.5}, {0.5, 0.5, 0.5}})));
224 HloInstruction::CreateConstant(LiteralUtil::CreateR2<bool>(
241 LiteralUtil::CreateR2<float>({{0.5}, {2.72}}),
252 LiteralUtil::CreateR2<float>({{1.0, 2.0, 3.0}})))
[all...] |
| bfloat16_test.cc | 100 auto expected = LiteralUtil::MakeTupleFromSlices(
101 {LiteralUtil::CreateR4<bfloat16>(
107 LiteralUtil::CreateR1<bfloat16>(
109 LiteralUtil::CreateR1<bfloat16>(
143 auto expected = LiteralUtil::MakeTupleFromSlices(
144 {LiteralUtil::CreateR4<bfloat16>(
149 LiteralUtil::CreateR1<bfloat16>(
151 LiteralUtil::CreateR1<bfloat16>(
|
| call_test.cc | 80 auto constant = ConstantLiteral(&builder, LiteralUtil::CreateR0<float>(42.0));
89 auto x = ConstantLiteral(&builder, LiteralUtil::CreateR1<float>({}));
90 auto y = ConstantLiteral(&builder, LiteralUtil::CreateR1<float>({}));
100 ConstantLiteral(&builder, LiteralUtil::CreateR1<float>({1.0f, 2.0f}));
102 ConstantLiteral(&builder, LiteralUtil::CreateR1<float>({2.0f, 3.0f}));
135 client_->TransferToServer(LiteralUtil::CreateR0<float>(1.0f)));
142 auto elem = LiteralUtil::CreateR0<float>(42.0);
143 auto tuple = LiteralUtil::MakeTuple({&elem});
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| infeed_test.cc | 55 TestInfeedRoundTrip(LiteralUtil::CreateR0<bool>(true));
59 TestInfeedRoundTrip(LiteralUtil::CreateR1<uint32>({1, 2, 3}));
63 TestInfeedRoundTrip(LiteralUtil::CreateR2F32Linspace(0.0, 1.0, 128, 64));
68 LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}},
76 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout(
81 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout(
88 TestInfeedRoundTrip(LiteralUtil::CreateR4(
97 TestInfeedRoundTrip(LiteralUtil::CreateR4FromArray4D<float>(array));
101 TestInfeedRoundTrip(LiteralUtil::MakeTupleFromSlices(
102 {LiteralUtil::CreateR1<uint32>({1, 2, 3})
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_infeed_test.cc | 61 TestInfeedRoundTrip(LiteralUtil::CreateR0<bool>(true));
65 TestInfeedRoundTrip(LiteralUtil::CreateR1<uint32>({1, 2, 3}));
69 TestInfeedRoundTrip(LiteralUtil::CreateR2F32Linspace(0.0, 1.0, 128, 64));
74 LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}},
82 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout(
87 TestInfeedRoundTrip(LiteralUtil::CreateR3WithLayout(
94 TestInfeedRoundTrip(LiteralUtil::CreateR4(
100 TestInfeedRoundTrip(LiteralUtil::MakeTupleFromSlices(
101 {LiteralUtil::CreateR1<uint32>({1, 2, 3}),
102 LiteralUtil::CreateR0<bool>(false)}))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal_util.cc | 85 /* static */ Literal LiteralUtil::CreateFromDimensions(
91 /* static */ Literal LiteralUtil::ConvertBF16ToF32(
96 /* static */ Literal LiteralUtil::ConvertF32ToBF16(
101 /* static */ Literal LiteralUtil::CreateToken() {
105 /* static */ Literal LiteralUtil::Zero(PrimitiveType primitive_type) {
108 return LiteralUtil::CreateR0<uint8>(0);
110 return LiteralUtil::CreateR0<uint16>(0);
112 return LiteralUtil::CreateR0<uint32>(0);
114 return LiteralUtil::CreateR0<uint64>(0);
116 return LiteralUtil::CreateR0<int8>(0)
[all...] |
| literal_test.cc | 82 LiteralUtil::CreateR4FromArray4DWithLayout<float>(arr4d,
85 LiteralUtil::CreateR4FromArray4DWithLayout<float>(arr4d,
100 auto true_lit = LiteralUtil::CreateR0<bool>(true);
103 auto false_lit = LiteralUtil::CreateR0<bool>(false);
106 auto u32_lit = LiteralUtil::CreateR0<uint32>(42);
109 auto s32_lit = LiteralUtil::CreateR0<int32>(-999);
112 auto f32_lit = LiteralUtil::CreateR0<float>(3.14f);
115 auto f16_lit = LiteralUtil::CreateR0<half>(static_cast<half>(0.5f));
118 auto c64_lit = LiteralUtil::CreateR0<complex64>({3.14f, 2.78f});
121 auto c128_lit = LiteralUtil::CreateR0<complex128>({3.14f, 2.78f})
[all...] |
| reference_util_test.cc | 56 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
68 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
76 auto actual_literal = LiteralUtil::CreateR1<float>(*result);
84 auto actual_literal = LiteralUtil::CreateR1<float>(*result);
90 auto result = LiteralUtil::CreateR1<float>(ReferenceUtil::Reduce4DTo1D(
99 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
109 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
120 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result);
137 auto actual_literal = LiteralUtil::CreateR4FromArray4D(*result);
147 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result)
[all...] |
| text_literal_writer_test.cc | 34 auto literal = LiteralUtil::CreateR2<float>({
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| util.cc | 35 xla::LiteralUtil::Zero(shape.element_type())),
70 literal = xla::LiteralUtil::CreateR0<uint8>(value);
73 literal = xla::LiteralUtil::CreateR0<uint32>(value);
76 literal = xla::LiteralUtil::CreateR0<uint64>(value);
79 literal = xla::LiteralUtil::CreateR0<int8>(value);
82 literal = xla::LiteralUtil::CreateR0<int32>(value);
85 literal = xla::LiteralUtil::CreateR0<int64>(value);
88 literal = xla::LiteralUtil::CreateR0<float>(value);
91 literal = xla::LiteralUtil::CreateR0<double>(value);
94 literal = xla::LiteralUtil::CreateR0<complex64>(value)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_creation_utils_test.cc | 61 evaluator.Evaluate(*module, {LiteralUtil::CreateR1<int32>({3, 4})}));
62 CHECK_EQ(result_literal, LiteralUtil::CreateR1<int32>({3, 4}));
80 evaluator.Evaluate(*module, {LiteralUtil::CreateR3<int32>(
84 LiteralUtil::CreateR2<int32>(
103 evaluator.Evaluate(*module, {LiteralUtil::CreateR1<int32>({9, 10})}));
104 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9, 10}}));
122 evaluator.Evaluate(*module, {LiteralUtil::CreateR1<int32>({9, 10})}));
123 CHECK_EQ(result_literal, LiteralUtil::CreateR3<int32>({{{9, 10}}}));
141 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<int32>(9)}));
142 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9}}))
[all...] |
| hlo_evaluator_test.cc | 153 auto low = LiteralUtil::CreateR2<float>({{0.f, 2.f}, {2.f, 4.f}});
154 auto value = LiteralUtil::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}});
155 auto high = LiteralUtil::CreateR2<float>({{2.f, 4.f}, {4.f, 4.f}});
168 auto expected = LiteralUtil::CreateR2<float>({{0, 4}, {2, 4}});
178 LiteralUtil::CreateR2<int64>({{0, ones(54)}, {ones(54), ones(58)}});
179 auto value = LiteralUtil::CreateR2<int64>({{0, ones(56)}, {0, ones(58)}});
180 auto high = LiteralUtil::CreateR2<int64>(
195 LiteralUtil::CreateR2<int64>({{0, ones(55)}, {ones(54), ones(58)}});
201 auto low = LiteralUtil::CreateR0<float>(0.f);
202 auto value = LiteralUtil::CreateR2<float>({{-1.f, 0.f}, {1.f, 2.f}})
[all...] |
