Searched refs:MakeTupleOwned (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multioutput_fusion_test.cc | 214 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()
|
| D | tuple_test.cc | 530 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()
604 auto param = LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR1<float>({2, 3}), in XLA_TEST_F()
606 auto expected = LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
|
| D | transfer_manager_test.cc | 369 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
399 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
|
| D | local_client_execute_test.cc | 436 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()
|
| D | test_utils.cc | 276 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternal()
378 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternalWithBounds()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 186 static Literal MakeTupleOwned(std::vector<Literal> elements);
198 static Literal MakeTupleOwned(Ts... elements) { in MakeTupleOwned() function
203 return MakeTupleOwned(std::move(v)); in MakeTupleOwned()
|
| D | literal_util.cc | 494 /* static */ Literal LiteralUtil::MakeTupleOwned( in MakeTupleOwned() function in xla::LiteralUtil
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer_test.cc | 98 Literal literal = LiteralUtil::MakeTupleOwned( in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math_test.cc | 82 auto expected = LiteralUtil::MakeTupleOwned( in TestIsInfOrNan()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.cc | 3100 *literal = LiteralUtil::MakeTupleOwned(std::move(elements)); in ParseTupleLiteral()
|