Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dscatter_test.cc65 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
67 Literal updates = LiteralUtil::CreateR2<int32>({{10, 20, 30}, {70, 80, 90}}); in XLA_TEST_F()
96 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
98 Literal updates = LiteralUtil::CreateR2<int32>({{10, 20, 30}, {70, 80, 90}}); in XLA_TEST_F()
124 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
127 LiteralUtil::CreateR2<int32>({{10, 30}, {40, 60}, {70, 90}}); in XLA_TEST_F()
156 LiteralUtil::CreateR2<int32>({{1, 3, 2, 0}, {3, 0, 2, 1}, {2, 3, 1, 0}}); in XLA_TEST_F()
163 LiteralUtil::CreateR2<int32>({{3, 0, 2, 1}, {1, 3, 2, 0}, {3, 2, 0, 1}}); in XLA_TEST_F()
194 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0, 0}}); in XLA_TEST_F()
221 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()
107 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
128 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
129 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); in XLA_TEST_F()
149 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
171 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
196 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in XLA_TEST_F()
219 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in XLA_TEST_F()
239 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
260 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
[all …]
Dcpu_gpu_fusion_test.cc203 LiteralUtil::CreateR2<float>({{1.0}, {2.0}, {3.0}}))); in XLA_TEST_F()
205 LiteralUtil::CreateR2<float>({{-1.0}, {-1.0}, {-1.0}}))); in XLA_TEST_F()
211 LiteralUtil::CreateR2<float>({{1.62, 2.72, 3.14}}))); in XLA_TEST_F()
215 LiteralUtil::CreateR2<float>({{1.0, 1.0, 1.0}, {0.0, 0.0, 0.0}}))); in XLA_TEST_F()
221 LiteralUtil::CreateR2<float>({{0.5, 0.5, 0.5}, {0.5, 0.5, 0.5}}))); in XLA_TEST_F()
223 HloInstruction::CreateConstant(LiteralUtil::CreateR2<bool>( in XLA_TEST_F()
240 LiteralUtil::CreateR2<float>({{0.5}, {2.72}}), in XLA_TEST_F()
251 LiteralUtil::CreateR2<float>({{1.0, 2.0, 3.0}}))); in XLA_TEST_F()
255 LiteralUtil::CreateR2<float>({{-2.0, -2.0, -2.0}}))); in XLA_TEST_F()
266 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.cc165 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}})); in XLA_TEST_F()
167 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}})); in XLA_TEST_F()
204 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}})); in XLA_TEST_F()
206 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}})); in XLA_TEST_F()
232 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}})); in XLA_TEST_F()
234 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}})); in XLA_TEST_F()
261 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}})); in XLA_TEST_F()
305 {LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}), in XLA_TEST_F()
309 LiteralUtil::CreateR2<float>({{55.0, 44.0}, {33.0, 22.0}})}); in XLA_TEST_F()
352 {LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}), in XLA_TEST_F()
[all …]
Dreshape_test.cc179 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}}); in XLA_TEST_P()
193 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f}, {2.0f}, {3.0f}}); in XLA_TEST_P()
214 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}}); in XLA_TEST_P()
231 LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}); in XLA_TEST_P()
246 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}}); in XLA_TEST_P()
296 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}, {}, {}}); in XLA_TEST_P()
437 auto expected_literal = LiteralUtil::CreateR2<float>({{10, 11, 12}, in XLA_TEST_P()
474 auto expected_literal = LiteralUtil::CreateR2<float>({{10, 20, 30}, in XLA_TEST_P()
528 auto expected_literal = LiteralUtil::CreateR2<float>( in XLA_TEST_P()
557 LiteralUtil::CreateR2<float>({{0, 1, 2, 3}, {4, 5, 6, 7}}); in XLA_TEST_P()
[all …]
Dtransfer_manager_test.cc180 LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}); in XLA_TEST_F()
220 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}), in XLA_TEST_F()
252 {LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}), in XLA_TEST_F()
317 {LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}), in XLA_TEST_F()
323 {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()
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()
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()
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.cc223 auto low = LiteralUtil::CreateR2<float>({{0.f, 2.f}, {2.f, 4.f}}); in TEST_P()
224 auto value = LiteralUtil::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); in TEST_P()
225 auto high = LiteralUtil::CreateR2<float>({{2.f, 4.f}, {4.f, 4.f}}); in TEST_P()
238 auto expected = LiteralUtil::CreateR2<float>({{0, 4}, {2, 4}}); in TEST_P()
248 LiteralUtil::CreateR2<int64>({{0, ones(54)}, {ones(54), ones(58)}}); in TEST_P()
249 auto value = LiteralUtil::CreateR2<int64>({{0, ones(56)}, {0, ones(58)}}); in TEST_P()
250 auto high = LiteralUtil::CreateR2<int64>( in TEST_P()
265 LiteralUtil::CreateR2<int64>({{0, ones(55)}, {ones(54), ones(58)}}); in TEST_P()
272 auto value = LiteralUtil::CreateR2<float>({{-1.f, 0.f}, {1.f, 2.f}}); in TEST_P()
286 auto expected = LiteralUtil::CreateR2<float>({{0, 0}, {1, 1}}); in TEST_P()
[all …]
Dhlo_creation_utils_test.cc99 LiteralUtil::CreateR2<int32>( in TEST_F()
119 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9, 10}})); in TEST_F()
157 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9}})); in TEST_F()
217 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{0, 0}, {0, 0}})); in TEST_F()
237 LiteralUtil::CreateR2<float>({{0.0f, 0.0f}, {0.0f, 0.0f}})); in TEST_F()
249 LiteralUtil::CreateR2<int32>({{0, 0}, {0, 0}}))); in TEST_F()
258 {LiteralUtil::CreateR2<int32>({{0, 0}, {0, 0}})})); in TEST_F()
260 LiteralUtil::CreateR2<float>({{0.0f, 0.0f}, {0.0f, 0.0f}})); in TEST_F()
279 LiteralUtil::CreateR2<float>({{0.0f, 0.0f}, {1.0f, 1.0f}})); in TEST_F()
298 LiteralUtil::CreateR2<float>({{0.0f, 0.0f}, {0.0f, 0.0f}})); in TEST_F()
[all …]
Ddynamic_padder_test.cc427 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}}); in XLA_TEST_F()
429 Literal updates = LiteralUtil::CreateR2<int32>({{10, 20, 30}, {70, 80, 90}}); in XLA_TEST_F()
449 Literal updates_padded = LiteralUtil::CreateR2<int32>( in XLA_TEST_F()
488 Literal operand = LiteralUtil::CreateR2<float>( in XLA_TEST_F()
492 LiteralUtil::CreateR2<float>({{10.0, 20.0, 30.0}, {70.0, 80.0, 90.0}}); in XLA_TEST_F()
511 EXPECT_EQ(LiteralUtil::CreateR2<float>( in XLA_TEST_F()
594 Literal operand = LiteralUtil::CreateR2<int32>({{1, 2}, {4, 5}}); in XLA_TEST_F()
612 Literal operand_padded = LiteralUtil::CreateR2<int32>( in XLA_TEST_F()
825 Literal expected = LiteralUtil::CreateR2<int32>({{2, 4}, {10, 12}}); in XLA_TEST_F()
855 LiteralUtil::CreateR2<int32>({{0, 1, 2, 3, 4, 5}, {6, 7, 8, 9, 10, 11}}); in XLA_TEST_F()
[all …]
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()
Delemental_ir_emitter_test.cc94 Literal lhs = LiteralUtil::CreateR2<int32>({{1, 2}, {3, 4}}); in XLA_TEST_F()
95 Literal rhs = LiteralUtil::CreateR2<int32>({{10, 20}, {30, 40}}); in XLA_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()
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()
153 auto literal = LiteralUtil::CreateR2({{1, 2}, {3, 4}, {5, 6}}); in TEST_F()
224 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
328 auto literal = LiteralUtil::CreateR2<float>({ in TEST_F()
362 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
363 auto matrix_clone = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
365 LiteralUtil::CreateR2<float>({{4.0, 3.0}, {1.0, 2.0}}); in TEST_F()
415 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
438 auto r2 = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
450 auto r2_clone = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}); in TEST_F()
[all …]
Dtext_literal_writer_test.cc35 auto literal = LiteralUtil::CreateR2<float>({ in TEST()
/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/xrt/tests/
Draw_api_test.cc164 auto matrix = xla::LiteralUtil::CreateR2({{4, 5}, {6, 7}}); in TwoElementTuple()
171 auto matrix = xla::LiteralUtil::CreateR2<float>( in BasedTwoElementTuple()
184 auto matrix = xla::LiteralUtil::CreateR2({{4, 5}, {6, 7}}); in NestedTuple()
194 auto matrix = xla::LiteralUtil::CreateR2({{4, 5}, {6, 7}}); in MakeTuple0()
387 xla::LiteralUtil::CreateR2<float>({{4.0f, 5.0f}, {6.0f, 7.0f}}); in TEST()
413 xla::LiteralUtil::CreateR2<float>({{4.0f, 5.0f}, {6.0f, 7.0f}}); in TEST()
455 xla::LiteralUtil::CreateR2({{9.0f, 2.0f}, {4.0f, 1.0f}}).ToProto(); in TEST()
479 xla::LiteralUtil::CreateR2<float>({{4.0f, 5.0f}, {6.0f, 7.0f}}); in TEST()
481 xla::LiteralUtil::CreateR2<float>({{14.0f, -5.0f}, {16.0f, 17.0f}}); in TEST()
510 xla::LiteralUtil::CreateR2<float>({{4.0f, 5.0f}, {6.0f, 7.0f}}); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc45 xla::Literal param1_literal = xla::LiteralUtil::CreateR2<float>( in main()

12