Home
last modified time | relevance | path

Searched refs:arg_literal (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvert_test.cc213 Literal arg_literal = LiteralUtil::CreateR1<int64>({arg}); in XLA_TEST_F() local
214 auto arg_param = Parameter(&builder, 0, arg_literal.shape(), "arg_param"); in XLA_TEST_F()
216 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); in XLA_TEST_F()
232 Literal arg_literal = LiteralUtil::CreateR1<uint32>({arg}); in XLA_TEST_F() local
233 auto arg_param = Parameter(&builder, 0, arg_literal.shape(), "arg_param"); in XLA_TEST_F()
235 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); in XLA_TEST_F()
250 Literal arg_literal = LiteralUtil::CreateR1<float>({arg}); in XLA_TEST_F() local
251 auto arg_param = Parameter(&builder, 0, arg_literal.shape(), "arg_param"); in XLA_TEST_F()
253 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); in XLA_TEST_F()
267 Literal arg_literal = LiteralUtil::CreateR1<uint32>({arg}); in XLA_TEST_F() local
[all …]
Dlocal_client_execute_test.cc347 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local
352 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F()
379 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local
382 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F()
433 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F() local
434 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F()
486 auto arg_literal = in XLA_TEST_F() local
488 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F()
526 Literal arg_literal = LiteralUtil::CreateR0<float>(123.0); in XLA_TEST_F() local
529 arg_vector.push_back(std::move(arg_literal)); in XLA_TEST_F()
[all …]
Dreduce_window_test.cc362 Literal arg_literal(shape); in XLA_TEST_P() local
363 arg_literal.PopulateWithValue(1.0f); in XLA_TEST_P()
364 const auto input = CreateConstantFromLiteral(arg_literal, &builder_); in XLA_TEST_P()
382 Literal arg_literal = in XLA_TEST_P() local
385 const auto input = CreateConstantFromLiteral(arg_literal, &builder_); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.cc90 TF_ASSIGN_OR_RETURN(Literal arg_literal, in ExecuteOnStream()
93 arg_literals.push_back(std::move(arg_literal)); in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc1510 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
1513 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
1552 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
1555 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
1604 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
1607 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
1655 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
1658 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
1707 Literal arg_literal = in TEST_P() local
1711 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
Dhlo_evaluator.cc1212 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleCall() local
1213 arg_literals.push_back(&arg_literal); in HandleCall()
1247 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleFusion() local
1248 arg_literals.push_back(&arg_literal); in HandleFusion()
Dhlo_evaluator_typed_visitor.h1598 const Literal& arg_literal = in MapImpl()
1601 auto curr_val = arg_literal.Get<NativeT>(multi_index); in MapImpl()