Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dlogdet_test.cc49 xla::Literal expected = xla::LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
72 xla::Literal expected = xla::LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
106 xla::Literal expected = xla::LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
152 xla::Literal expected = xla::LiteralUtil::MakeTupleOwned( in XLA_TEST_F()
Dmath_test.cc82 auto expected = LiteralUtil::MakeTupleOwned( in TestIsInfOrNan()
/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_t>(42)), in XLA_TEST_F()
216 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<float>(3.0), in XLA_TEST_F()
220 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32_t>(42)), result)); in XLA_TEST_F()
Dtuple_test.cc391 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR1<float>({1, 2, 3})); in XLA_TEST_F()
394 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR2<float>({{1, 2, 3}})), in XLA_TEST_F()
Dlocal_client_execute_test.cc438 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F()
493 LiteralUtil::MakeTupleOwned(std::move(inner_tuple_elements))); in XLA_TEST_F()
496 LiteralUtil::MakeTupleOwned(std::move(outer_tuple_elements)); in XLA_TEST_F()
539 arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_vector)); in XLA_TEST_F()
Dtransfer_manager_test.cc407 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
436 Literal literal = LiteralUtil::MakeTupleOwned(std::move(tuple_elements)); in Run()
Dtest_utils.cc280 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternal()
384 return LiteralUtil::MakeTupleOwned(std::move(elements)); in MakeFakeLiteralInternalWithBounds()
Dvalue_inference_test.cc746 xla::LiteralUtil::MakeTupleOwned(std::move(bound), std::move(dynamism)); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Darg_op.cc96 xla::Literal tuple = xla::LiteralUtil::MakeTupleOwned( in Compile()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h192 static Literal MakeTupleOwned(std::vector<Literal> elements);
204 static Literal MakeTupleOwned(Ts... elements) { in MakeTupleOwned() function
209 return MakeTupleOwned(std::move(v)); in MakeTupleOwned()
Dliteral_util.cc567 /* static */ Literal LiteralUtil::MakeTupleOwned( in MakeTupleOwned() function in xla::LiteralUtil
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer_test.cc98 Literal literal = LiteralUtil::MakeTupleOwned( in TEST()
/external/tensorflow/tensorflow/compiler/xla/client/
Dvalue_inference.cc715 return LiteralUtil::MakeTupleOwned(std::move(results)); in AnalyzeUpperBound()
1080 return LiteralUtil::MakeTupleOwned(std::move(results)); in AnalyzeIsDynamic()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc3512 *literal = LiteralUtil::MakeTupleOwned(std::move(elements)); in ParseLiteral()
3556 *literal = LiteralUtil::MakeTupleOwned(std::move(elements)); in ParseTupleLiteral()
Dhlo_evaluator_test.cc4150 Literal expected = LiteralUtil::MakeTupleOwned( in TEST_F()