Home
last modified time | relevance | path

Searched refs:MakeTupleFromSlices (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dliteral_test_util_test.cc35 Literal literal = LiteralUtil::MakeTupleFromSlices({ in TEST()
43 Literal literal = LiteralUtil::MakeTupleFromSlices({ in TEST()
51 Literal literal = LiteralUtil::MakeTupleFromSlices({ in TEST()
59 Literal literal0 = LiteralUtil::MakeTupleFromSlices({ in TEST()
63 Literal literal1 = LiteralUtil::MakeTupleFromSlices({ in TEST()
67 Literal literal2 = LiteralUtil::MakeTupleFromSlices({ in TEST()
71 Literal literal3 = LiteralUtil::MakeTupleFromSlices({ in TEST()
82 Literal literal0 = LiteralUtil::MakeTupleFromSlices({ in TEST()
86 Literal literal1 = LiteralUtil::MakeTupleFromSlices({ in TEST()
90 Literal literal2 = LiteralUtil::MakeTupleFromSlices({ in TEST()
[all …]
Dtuple_test.cc54 auto value = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
69 auto value = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
91 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
105 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
199 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
221 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
290 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
334 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
409 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
445 ->TransferToServer(LiteralUtil::MakeTupleFromSlices({ in XLA_TEST_F()
[all …]
Dtransfer_manager_test.cc218 Literal literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
249 Literal literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
251 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
283 Literal literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
314 Literal literal1 = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
316 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
320 Literal literal2 = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
322 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
Dround_trip_transfer_test.cc115 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({1, 2}), in TEST_F()
121 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({}), in TEST_F()
126 RoundTripTest(LiteralUtil::MakeTupleFromSlices( in TEST_F()
Dbuffer_donation_test.cc258 args.push_back(LiteralUtil::MakeTupleFromSlices( in TEST_F()
260 Literal expected = LiteralUtil::MakeTupleFromSlices( in TEST_F()
275 args.push_back(LiteralUtil::MakeTupleFromSlices( in TEST_F()
277 Literal expected = LiteralUtil::MakeTupleFromSlices( in TEST_F()
304 Literal expected = LiteralUtil::MakeTupleFromSlices( in TEST_F()
335 Literal expected = LiteralUtil::MakeTupleFromSlices( in TEST_F()
Dbatch_normalization_test.cc246 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
278 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
300 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
332 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
360 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
392 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
423 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
462 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
624 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
928 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_P()
Dbfloat16_test.cc100 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
143 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
Dconditional_test.cc477 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR0<float>(12.0f), in XLA_TEST_F()
494 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
533 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
579 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
580 {LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
583 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
747 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
784 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR0<int32>(123), in XLA_TEST_F()
797 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR0<int32>(123), in XLA_TEST_F()
Dlocal_client_execute_test.cc304 auto x_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
307 auto y_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
350 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
351 {LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
382 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
858 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
860 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
865 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
866 {LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
891 test_to_device_and_back(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
Dconstants_test.cc164 ConstantLiteral(&builder, LiteralUtil::MakeTupleFromSlices( in TEST_F()
Dparams_test.cc445 ->TransferToServer(LiteralUtil::MakeTupleFromSlices({ in XLA_TEST_F()
Ddot_operation_test.cc79 LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
Darray_elementwise_ops_test.cc2642 auto expected = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc100 TestInfeedRoundTrip(LiteralUtil::MakeTupleFromSlices( in TEST_F()
253 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({1, 2}), in TEST_F()
256 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({3, 4}), in TEST_F()
259 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({5, 6}), in TEST_F()
262 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({7, 8}), in TEST_F()
278 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({1, 2, 3}), in TEST_F()
281 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({7, 8, 9}), in TEST_F()
284 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR1<float>({4, 5, 6}), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc101 TestInfeedRoundTrip(LiteralUtil::MakeTupleFromSlices( in TEST_F()
114 TestInfeedRoundTrip(LiteralUtil::MakeTupleFromSlices( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h177 static Literal MakeTupleFromSlices(absl::Span<const LiteralSlice> elements);
Dliteral_util.cc481 /* static */ Literal LiteralUtil::MakeTupleFromSlices( in MakeTupleFromSlices() function in xla::LiteralUtil