Searched refs:MakeTupleOwned (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | multioutput_fusion_test.cc | 212 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 …]
|
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()
|
D | transfer_manager_test.cc | 327 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run() 358 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
|
D | local_client_execute_test.cc | 433 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()
|
D | test_utils.cc | 205 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternal() 298 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternalWithBounds()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | literal_util.h | 223 static Literal MakeTupleOwned(std::vector<Literal> elements); 236 static Literal MakeTupleOwned(Ts... elements) { in MakeTupleOwned() function 241 return MakeTupleOwned(std::move(v)); in MakeTupleOwned()
|
D | literal_util.cc | 442 /* static */ Literal LiteralUtil::MakeTupleOwned( in MakeTupleOwned() function in xla::LiteralUtil
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | math_test.cc | 79 auto expected = LiteralUtil::MakeTupleOwned( in TestIsInfOrNan()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | numpy_bridge.cc | 408 return LiteralUtil::MakeTupleOwned(std::move(elements)); in XlaLiteralFromPyObject()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_parser.cc | 2150 *literal = LiteralUtil::MakeTupleOwned(std::move(elements)); in ParseTupleLiteral()
|