Home
last modified time | relevance | path

Searched refs:MakeTupleOwned (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultioutput_fusion_test.cc214 auto param = LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
215 LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
216 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)), in XLA_TEST_F()
218 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<float>(3.0), in XLA_TEST_F()
222 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)), result)); in XLA_TEST_F()
Dtuple_test.cc530 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR1<float>({1, 2, 3})); in XLA_TEST_F()
533 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR2<float>({{1, 2, 3}})), in XLA_TEST_F()
573 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR1<float>({2, 3})); in XLA_TEST_F()
Dtransfer_manager_test.cc369 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
400 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
Dlocal_client_execute_test.cc436 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F()
487 LiteralUtil::MakeTupleOwned(std::move(inner_tuple_elements))); in XLA_TEST_F()
490 LiteralUtil::MakeTupleOwned(std::move(outer_tuple_elements)); in XLA_TEST_F()
533 arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_vector)); in XLA_TEST_F()
Dtest_utils.cc276 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternal()
378 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternalWithBounds()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_test.cc166 AddParam(LiteralUtil::MakeTupleOwned( in TEST_F()
171 AddParam(LiteralUtil::MakeTupleOwned( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h186 static Literal MakeTupleOwned(std::vector<Literal> elements);
198 static Literal MakeTupleOwned(Ts... elements) { in MakeTupleOwned() function
203 return MakeTupleOwned(std::move(v)); in MakeTupleOwned()
Dliteral_util.cc494 /* static */ Literal LiteralUtil::MakeTupleOwned( in MakeTupleOwned() function in xla::LiteralUtil
/external/tensorflow/tensorflow/compiler/xla/python/
Dshared_device_buffer_test.cc135 Literal literal = LiteralUtil::MakeTupleOwned( in TEST()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath_test.cc82 auto expected = LiteralUtil::MakeTupleOwned( in TestIsInfOrNan()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc2279 *literal = LiteralUtil::MakeTupleOwned(std::move(elements)); in ParseTupleLiteral()