Home
last modified time | relevance | path

Searched refs:CreateR2 (Results 1 – 25 of 46) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dscatter_test.cc66 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
68 Literal updates = LiteralUtil::CreateR2<int32>({{10, 20, 30}, {70, 80, 90}}); in XLA_TEST_F()
97 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
99 Literal updates = LiteralUtil::CreateR2<int32>({{10, 20, 30}, {70, 80, 90}}); in XLA_TEST_F()
125 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
128 LiteralUtil::CreateR2<int32>({{10, 30}, {40, 60}, {70, 90}}); in XLA_TEST_F()
157 LiteralUtil::CreateR2<int32>({{1, 3, 2, 0}, {3, 0, 2, 1}, {2, 3, 1, 0}}); in XLA_TEST_F()
164 LiteralUtil::CreateR2<int32>({{3, 0, 2, 1}, {1, 3, 2, 0}, {3, 2, 0, 1}}); in XLA_TEST_F()
195 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0, 0}}); in XLA_TEST_F()
222 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
[all …]
Dgather_operation_test.cc62 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
83 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
104 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
105 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); in XLA_TEST_F()
125 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
147 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
172 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in XLA_TEST_F()
195 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in XLA_TEST_F()
215 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
236 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
[all …]
Dmultioutput_fusion_test.cc318 LiteralUtil::CreateR2<float>({{3, 7}, {11, 15}}), in XLA_TEST_F()
319 LiteralUtil::CreateR2<float>({{5, 16}, {36, 64}})), in XLA_TEST_F()
347 LiteralUtil::CreateR2<float>({{6, 8}, {10, 12}}), in XLA_TEST_F()
348 LiteralUtil::CreateR2<float>({{25, 36}, {49, 64}})), in XLA_TEST_F()
408 LiteralUtil::CreateR2<float>({{3, 7}, {11, 15}}), in XLA_TEST_F()
409 LiteralUtil::CreateR2<float>({{5, 16}, {36, 64}})), in XLA_TEST_F()
438 LiteralUtil::CreateR2<float>({{6, 8}, {10, 12}}), in XLA_TEST_F()
441 LiteralUtil::CreateR2<float>({{25, 36}, {49, 64}})), in XLA_TEST_F()
507 LiteralUtil::CreateR2<float>({{167, 172}, {176, 180}}), in XLA_TEST_F()
508 LiteralUtil::CreateR2<float>({{6, 6}, {6, 8}})), in XLA_TEST_F()
[all …]
Dfusion_test.cc204 LiteralUtil::CreateR2<float>({{1.0}, {2.0}, {3.0}}))); in XLA_TEST_F()
206 LiteralUtil::CreateR2<float>({{-1.0}, {-1.0}, {-1.0}}))); in XLA_TEST_F()
212 LiteralUtil::CreateR2<float>({{1.62, 2.72, 3.14}}))); in XLA_TEST_F()
216 LiteralUtil::CreateR2<float>({{1.0, 1.0, 1.0}, {0.0, 0.0, 0.0}}))); in XLA_TEST_F()
222 LiteralUtil::CreateR2<float>({{0.5, 0.5, 0.5}, {0.5, 0.5, 0.5}}))); in XLA_TEST_F()
224 HloInstruction::CreateConstant(LiteralUtil::CreateR2<bool>( in XLA_TEST_F()
241 LiteralUtil::CreateR2<float>({{0.5}, {2.72}}), in XLA_TEST_F()
252 LiteralUtil::CreateR2<float>({{1.0, 2.0, 3.0}}))); in XLA_TEST_F()
256 LiteralUtil::CreateR2<float>({{-2.0, -2.0, -2.0}}))); in XLA_TEST_F()
267 LiteralUtil::CreateR2<float>({{-1.0, 0.0, 1.0}}), in XLA_TEST_F()
[all …]
Dbroadcast_test.cc66 LiteralUtil::CreateR2<float>({{42.0, 42.0}, {42.0, 42.0}}), result, in XLA_TEST_F()
89 LiteralUtil::CreateR2<float>({{1.0, 1.0}, {2.0, 2.0}, {3.0, 3.0}}), in XLA_TEST_F()
93 LiteralUtil::CreateR2<float>({{1.0, 2.0, 3.0}, {1.0, 2.0, 3.0}}), in XLA_TEST_F()
100 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in XLA_TEST_F()
110 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}), result, in XLA_TEST_F()
119 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in XLA_TEST_F()
129 LiteralUtil::CreateR2<float>({{1.0, 3.0}, {2.0, 4.0}}), result, in XLA_TEST_F()
136 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in XLA_TEST_F()
Dlocal_client_execute_test.cc164 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}})); in XLA_TEST_F()
166 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}})); in XLA_TEST_F()
201 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}})); in XLA_TEST_F()
203 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}})); in XLA_TEST_F()
229 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}})); in XLA_TEST_F()
231 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}})); in XLA_TEST_F()
258 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}})); in XLA_TEST_F()
302 {LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}), in XLA_TEST_F()
306 LiteralUtil::CreateR2<float>({{55.0, 44.0}, {33.0, 22.0}})}); in XLA_TEST_F()
349 {LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}), in XLA_TEST_F()
[all …]
Dreshape_test.cc171 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}}); in XLA_TEST_P()
184 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f}, {2.0f}, {3.0f}}); in XLA_TEST_P()
203 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}}); in XLA_TEST_P()
219 LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}); in XLA_TEST_P()
233 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}}); in XLA_TEST_P()
280 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}, {}, {}}); in XLA_TEST_P()
413 auto expected_literal = LiteralUtil::CreateR2<float>({{10, 11, 12}, in XLA_TEST_P()
448 auto expected_literal = LiteralUtil::CreateR2<float>({{10, 20, 30}, in XLA_TEST_P()
500 auto expected_literal = LiteralUtil::CreateR2<float>( in XLA_TEST_P()
528 LiteralUtil::CreateR2<float>({{0, 1, 2, 3}, {4, 5, 6, 7}}); in XLA_TEST_P()
[all …]
Dtransfer_manager_test.cc138 LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}); in XLA_TEST_F()
178 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}), in XLA_TEST_F()
210 {LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}), in XLA_TEST_F()
275 {LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}), in XLA_TEST_F()
281 {LiteralUtil::CreateR2<float>({{5.0f, 7.0f}, {9.0f, 4.0f}}), in XLA_TEST_F()
Dtuple_test.cc57 LiteralUtil::CreateR2<float>(constant_matrix)}); in XLA_TEST_F()
94 LiteralUtil::CreateR2<float>(constant_matrix)}); in XLA_TEST_F()
200 {LiteralUtil::CreateR2<float>(constant_matrix), in XLA_TEST_F()
485 LiteralUtil::CreateR2<complex64>( in XLA_TEST_F()
496 LiteralUtil::CreateR2<complex64>({{{111, 222}, {331, 442}}, in XLA_TEST_F()
533 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR2<float>({{1, 2, 3}})), in XLA_TEST_F()
Dbroadcast_simple_test.cc629 auto r1 = ConstantLiteral(&b, LiteralUtil::CreateR2<float>({{1, 2}})); in XLA_TEST_F()
630 auto r2 = ConstantLiteral(&b, LiteralUtil::CreateR2<float>({{1, 2}, {3, 4}})); in XLA_TEST_F()
633 auto expected = LiteralUtil::CreateR2<float>({{2, 4}, {4, 6}}); in XLA_TEST_F()
640 auto r1 = ConstantLiteral(&b, LiteralUtil::CreateR2<float>({{1}, {2}})); in XLA_TEST_F()
641 auto r2 = ConstantLiteral(&b, LiteralUtil::CreateR2<float>({{1, 2}, {3, 4}})); in XLA_TEST_F()
644 auto expected = LiteralUtil::CreateR2<float>({{2, 3}, {5, 6}}); in XLA_TEST_F()
Dclient_test.cc121 LiteralUtil::CreateR2<int32>({{5, 6}, {7, 8}}))); in XLA_TEST_F()
143 auto expected_result = LiteralUtil::CreateR2<int32>({{6, 8}, {10, 12}}); in XLA_TEST_F()
Dcustom_call_test.cc154 Literal argument = LiteralUtil::CreateR2<float>({{1.f, 2.f}, {3.f, 4.f}}); in XLA_TEST_F()
184 Literal argument = LiteralUtil::CreateR2<float>({{1.f, 2.f}, {3.f, 4.f}}); in XLA_TEST_F()
Dliteral_test_util.h171 EXPECT_TRUE(Equal(LiteralUtil::CreateR2<NativeT>(expected), actual)); in ExpectR2Equal()
218 EXPECT_TRUE(Near(LiteralUtil::CreateR2<NativeT>(expected), actual, error)); in ExpectR2Near()
Dcopy_test.cc111 auto literal = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in XLA_TEST_F()
132 Literal literal = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc153 auto low = LiteralUtil::CreateR2<float>({{0.f, 2.f}, {2.f, 4.f}}); in TEST_P()
154 auto value = LiteralUtil::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); in TEST_P()
155 auto high = LiteralUtil::CreateR2<float>({{2.f, 4.f}, {4.f, 4.f}}); in TEST_P()
168 auto expected = LiteralUtil::CreateR2<float>({{0, 4}, {2, 4}}); in TEST_P()
178 LiteralUtil::CreateR2<int64>({{0, ones(54)}, {ones(54), ones(58)}}); in TEST_P()
179 auto value = LiteralUtil::CreateR2<int64>({{0, ones(56)}, {0, ones(58)}}); in TEST_P()
180 auto high = LiteralUtil::CreateR2<int64>( in TEST_P()
195 LiteralUtil::CreateR2<int64>({{0, ones(55)}, {ones(54), ones(58)}}); in TEST_P()
202 auto value = LiteralUtil::CreateR2<float>({{-1.f, 0.f}, {1.f, 2.f}}); in TEST_P()
216 auto expected = LiteralUtil::CreateR2<float>({{0, 0}, {1, 1}}); in TEST_P()
[all …]
Dhlo_creation_utils_test.cc84 LiteralUtil::CreateR2<int32>( in TEST_F()
104 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9, 10}})); in TEST_F()
142 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9}})); in TEST_F()
202 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{0, 0}, {0, 0}})); in TEST_F()
222 LiteralUtil::CreateR2<float>({{0.0f, 0.0f}, {0.0f, 0.0f}})); in TEST_F()
Dhlo_cse_test.cc107 auto expected = LiteralUtil::CreateR2<float>({{2.0, 4.0}, {6.0, 8.0}}); in TEST_F()
137 auto expected = LiteralUtil::CreateR2<float>({{2.0, 4.0}, {6.0, 8.0}}); in TEST_F()
191 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in TEST_F()
193 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in TEST_F()
197 LiteralUtil::CreateR2<float>({{2.0, 4.0}, {6.0, 8.0}}))); in TEST_F()
429 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in TEST_F()
460 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in TEST_F()
Dmap_inliner_test.cc91 LiteralUtil::CreateR2<float>({{1, 2, 3, 4}, {5, 6, 7, 8}}))); in TEST_F()
107 auto expected = LiteralUtil::CreateR2<float>({{2, 2, 2, 2}, {2, 2, 2, 2}}); in TEST_F()
Dreshape_mover_test.cc188 HloInstruction::CreateConstant(LiteralUtil::CreateR2<bool>(
270 LiteralUtil::CreateR2<float>({{1, 2, 3}, {4, 5, 6}})));
275 LiteralUtil::CreateR2<float>({{1, 2, 3}, {4, 5, 6}})));
325 LiteralUtil::CreateR2<float>({{1, 2, 3}, {4, 5, 6}})));
Ddefuser_test.cc131 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in TEST_F()
170 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_test.cc143 const auto literal = LiteralUtil::CreateR2({{1, 2}, {3, 4}, {5, 6}}); in TEST_F()
214 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
354 auto literal = LiteralUtil::CreateR2<float>({ in TEST_F()
388 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
389 auto matrix_clone = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
391 LiteralUtil::CreateR2<float>({{4.0, 3.0}, {1.0, 2.0}}); in TEST_F()
441 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
492 auto element2 = LiteralUtil::CreateR2<float>({{0.0, 0.0}, {0.0, 0.0}}); in TEST_F()
503 auto matrix = LiteralUtil::CreateR2<int32>({{0, 0}, {0, 0}}); in TEST_F()
530 EXPECT_TRUE(LiteralUtil::CreateR2<uint64>({{8, 8}, {8, 8}}).IsAll(8)); in TEST_F()
[all …]
Dtext_literal_writer_test.cc34 auto literal = LiteralUtil::CreateR2<float>({ in TEST()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc68 auto matrix = xla::LiteralUtil::CreateR2({{4, 5}, {6, 7}}); in TwoElementTuple()
80 auto matrix = xla::LiteralUtil::CreateR2({{4, 5}, {6, 7}}); in NestedTuple()
90 auto matrix = xla::LiteralUtil::CreateR2({{4, 5}, {6, 7}}); in MakeTuple0()
248 xla::LiteralUtil::CreateR2<float>({{4.0f, 5.0f}, {6.0f, 7.0f}}); in TEST()
274 xla::LiteralUtil::CreateR2<float>({{4.0f, 5.0f}, {6.0f, 7.0f}}); in TEST()
276 xla::LiteralUtil::CreateR2<float>({{14.0f, -5.0f}, {16.0f, 17.0f}}); in TEST()
305 xla::LiteralUtil::CreateR2<float>({{4.0f, 5.0f}, {6.0f, 7.0f}}); in TEST()
331 xla::LiteralUtil::CreateR2<float>({{4.0f, 5.0f}, {6.0f, 7.0f}}); in TEST()
356 xla::LiteralUtil::CreateR2<float>({{4.0f, 6.0f}, {5.0f, 7.0f}}); in TEST()
363 xla::LiteralUtil::CreateR2({{4, 5}, {6, 7}}).ToProto(); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_copy_test.cc40 Literal literal = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc45 xla::Literal param1_literal = xla::LiteralUtil::CreateR2<float>( in main()

12