Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultioutput_fusion_test.cc212 auto param = LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
213 LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
214 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)), in XLA_TEST_F()
216 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<float>(3.0), in XLA_TEST_F()
220 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)), result)); in XLA_TEST_F()
317 LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
346 LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
376 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR1<float>({14, 22}), in XLA_TEST_F()
406 LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
437 LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
[all …]
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.cc327 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
358 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
Dlocal_client_execute_test.cc433 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F()
484 LiteralUtil::MakeTupleOwned(std::move(inner_tuple_elements))); in XLA_TEST_F()
487 LiteralUtil::MakeTupleOwned(std::move(outer_tuple_elements)); in XLA_TEST_F()
530 arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_vector)); in XLA_TEST_F()
Dtest_utils.cc205 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternal()
298 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternalWithBounds()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h223 static Literal MakeTupleOwned(std::vector<Literal> elements);
236 static Literal MakeTupleOwned(Ts... elements) { in MakeTupleOwned() function
241 return MakeTupleOwned(std::move(v)); in MakeTupleOwned()
Dliteral_util.cc442 /* static */ Literal LiteralUtil::MakeTupleOwned( in MakeTupleOwned() function in xla::LiteralUtil
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath_test.cc79 auto expected = LiteralUtil::MakeTupleOwned( in TestIsInfOrNan()
/external/tensorflow/tensorflow/compiler/xla/python/
Dnumpy_bridge.cc408 return LiteralUtil::MakeTupleOwned(std::move(elements)); in XlaLiteralFromPyObject()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc2150 *literal = LiteralUtil::MakeTupleOwned(std::move(elements)); in ParseTupleLiteral()