| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| gather_operation_test.cc | 62 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
83 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
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}});
147 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
172 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}});
195 Literal start_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}});
215 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
236 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}})
[all...] |
| scatter_test.cc | 66 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}});
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}});
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}});
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}});
164 LiteralUtil::CreateR2<int32>({{3, 0, 2, 1}, {1, 3, 2, 0}, {3, 2, 0, 1}});
195 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0, 0}});
222 LiteralUtil::CreateR2<int32>({{1, 2, 3}, {4, 5, 6}, {7, 8, 9}})
[all...] |
| multioutput_fusion_test.cc | 318 LiteralUtil::CreateR2<float>({{3, 7}, {11, 15}}),
319 LiteralUtil::CreateR2<float>({{5, 16}, {36, 64}})),
347 LiteralUtil::CreateR2<float>({{6, 8}, {10, 12}}),
348 LiteralUtil::CreateR2<float>({{25, 36}, {49, 64}})),
408 LiteralUtil::CreateR2<float>({{3, 7}, {11, 15}}),
409 LiteralUtil::CreateR2<float>({{5, 16}, {36, 64}})),
438 LiteralUtil::CreateR2<float>({{6, 8}, {10, 12}}),
441 LiteralUtil::CreateR2<float>({{25, 36}, {49, 64}})),
507 LiteralUtil::CreateR2<float>({{167, 172}, {176, 180}}),
508 LiteralUtil::CreateR2<float>({{6, 6}, {6, 8}}))
[all...] |
| broadcast_test.cc | 66 LiteralUtil::CreateR2<float>({{42.0, 42.0}, {42.0, 42.0}}), result,
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}})));
129 LiteralUtil::CreateR2<float>({{1.0, 3.0}, {2.0, 4.0}}), result,
136 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})));
|
| fusion_test.cc | 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}})));
256 LiteralUtil::CreateR2<float>({{-2.0, -2.0, -2.0}})));
267 LiteralUtil::CreateR2<float>({{-1.0, 0.0, 1.0}})
[all...] |
| local_client_execute_test.cc | 164 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}}));
166 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}}));
201 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}}));
203 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}}));
229 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}}));
231 LiteralUtil::CreateR2<float>({{10.0f, 20.0f}, {30.0f, 40.0f}}));
258 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {3.0f, 4.0f}}));
302 {LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}}),
306 LiteralUtil::CreateR2<float>({{55.0, 44.0}, {33.0, 22.0}})});
349 {LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})
[all...] |
| client_test.cc | 121 LiteralUtil::CreateR2<int32>({{5, 6}, {7, 8}})));
143 auto expected_result = LiteralUtil::CreateR2<int32>({{6, 8}, {10, 12}});
|
| reshape_test.cc | 171 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}});
184 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f}, {2.0f}, {3.0f}});
203 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}});
219 LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}});
233 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}});
280 auto expected_literal = LiteralUtil::CreateR2<float>({{}, {}, {}, {}});
413 auto expected_literal = LiteralUtil::CreateR2<float>({{10, 11, 12},
448 auto expected_literal = LiteralUtil::CreateR2<float>({{10, 20, 30},
500 auto expected_literal = LiteralUtil::CreateR2<float>(
528 LiteralUtil::CreateR2<float>({{0, 1, 2, 3}, {4, 5, 6, 7}})
[all...] |
| transfer_manager_test.cc | 138 LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}});
178 LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}),
210 {LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}),
275 {LiteralUtil::CreateR2<float>({{1.0f, 2.0f}, {4.0f, 5.0f}}),
281 {LiteralUtil::CreateR2<float>({{5.0f, 7.0f}, {9.0f, 4.0f}}),
|
| tuple_test.cc | 57 LiteralUtil::CreateR2<float>(constant_matrix)});
94 LiteralUtil::CreateR2<float>(constant_matrix)});
200 {LiteralUtil::CreateR2<float>(constant_matrix),
485 LiteralUtil::CreateR2<complex64>(
496 LiteralUtil::CreateR2<complex64>({{{111, 222}, {331, 442}},
533 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR2<float>({{1, 2, 3}})),
|
| round_trip_transfer_test.cc | 92 RoundTripTest(LiteralUtil::CreateR2<float>({{42.0, 64.0}, {77.0, 88.0}}));
|
| custom_call_test.cc | 154 Literal argument = LiteralUtil::CreateR2<float>({{1.f, 2.f}, {3.f, 4.f}});
184 Literal argument = LiteralUtil::CreateR2<float>({{1.f, 2.f}, {3.f, 4.f}});
|
| literal_test_util.h | 171 EXPECT_TRUE(Equal(LiteralUtil::CreateR2<NativeT>(expected), actual));
218 EXPECT_TRUE(Near(LiteralUtil::CreateR2<NativeT>(expected), actual, error));
|
| compilation_cache_test.cc | 68 LiteralUtil::CreateR2<float>(expected_result), result, error_spec_));
|
| copy_test.cc | 111 auto literal = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}});
132 Literal literal = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}});
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| text_literal_writer_test.cc | 34 auto literal = LiteralUtil::CreateR2<float>({
|
| literal_test.cc | 143 const auto literal = LiteralUtil::CreateR2({{1, 2}, {3, 4}, {5, 6}});
214 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}});
354 auto literal = LiteralUtil::CreateR2<float>({
388 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}});
389 auto matrix_clone = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}});
391 LiteralUtil::CreateR2<float>({{4.0, 3.0}, {1.0, 2.0}});
441 auto matrix = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}});
492 auto element2 = LiteralUtil::CreateR2<float>({{0.0, 0.0}, {0.0, 0.0}});
503 auto matrix = LiteralUtil::CreateR2<int32>({{0, 0}, {0, 0}});
530 EXPECT_TRUE(LiteralUtil::CreateR2<uint64>({{8, 8}, {8, 8}}).IsAll(8))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| gpu_copy_test.cc | 40 Literal literal = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}});
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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)}});
202 auto value = LiteralUtil::CreateR2<float>({{-1.f, 0.f}, {1.f, 2.f}});
216 auto expected = LiteralUtil::CreateR2<float>({{0, 0}, {1, 1}})
[all...] |
| hlo_creation_utils_test.cc | 84 LiteralUtil::CreateR2<int32>(
104 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9, 10}}));
142 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{9}}));
202 CHECK_EQ(result_literal, LiteralUtil::CreateR2<int32>({{0, 0}, {0, 0}}));
222 LiteralUtil::CreateR2<float>({{0.0f, 0.0f}, {0.0f, 0.0f}}));
|
| map_inliner_test.cc | 91 LiteralUtil::CreateR2<float>({{1, 2, 3, 4}, {5, 6, 7, 8}})));
107 auto expected = LiteralUtil::CreateR2<float>({{2, 2, 2, 2}, {2, 2, 2, 2}});
|
| hlo_cse_test.cc | 107 auto expected = LiteralUtil::CreateR2<float>({{2.0, 4.0}, {6.0, 8.0}});
137 auto expected = LiteralUtil::CreateR2<float>({{2.0, 4.0}, {6.0, 8.0}});
191 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})));
193 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})));
197 LiteralUtil::CreateR2<float>({{2.0, 4.0}, {6.0, 8.0}})));
429 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})));
460 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})));
|
| defuser_test.cc | 131 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})));
170 LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}})));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| sample_harness.cc | 45 xla::Literal param1_literal = xla::LiteralUtil::CreateR2<float>(
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_noalias_test.cc | 44 Literal literal = LiteralUtil::CreateR2<float>({{1.0, 2.0}, {3.0, 4.0}});
|
