Home
last modified time | relevance | path

Searched refs:arg_literal (Results 1 – 8 of 8) 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.cc350 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local
355 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F()
382 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local
385 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F()
436 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F() local
437 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F()
489 auto arg_literal = in XLA_TEST_F() local
491 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F()
529 Literal arg_literal = LiteralUtil::CreateR0<float>(123.0); in XLA_TEST_F() local
532 arg_vector.push_back(std::move(arg_literal)); in XLA_TEST_F()
[all …]
Dreduce_window_test.cc386 Literal arg_literal(shape); in XLA_TEST_P() local
387 arg_literal.PopulateWithValue(1.0f); in XLA_TEST_P()
388 const auto input = CreateConstantFromLiteral(arg_literal, &builder_); in XLA_TEST_P()
406 Literal arg_literal = in XLA_TEST_P() local
409 const auto input = CreateConstantFromLiteral(arg_literal, &builder_); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc112 TF_ASSIGN_OR_RETURN(Literal arg_literal, in ExecuteAsyncOnStream()
118 arg_literal = arg_literal.ToBoundedDynamic(expected_shape); in ExecuteAsyncOnStream()
120 arg_literals.push_back(std::move(arg_literal)); in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc159 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in ReduceWindowMaxIotaTest() local
162 HloInstruction::CreateConstant(std::move(arg_literal))); in ReduceWindowMaxIotaTest()
2596 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
2599 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
2638 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
2641 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
2755 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
2758 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
2807 Literal arg_literal = in TEST_P() local
2811 b.AddInstruction(HloInstruction::CreateConstant(std::move(arg_literal))); in TEST_P()
Dspace_to_batch_converter.cc1585 auto arg_literal = LiteralUtil::CreateR1(b); in Propagate() local
1586 VLOG(4) << "Slice mask created: arg literal " << arg_literal.ToString(); in Propagate()
1588 HloInstruction::CreateConstant(std::move(arg_literal))); in Propagate()
1679 auto arg_literal = LiteralUtil::CreateR1(b); in SelectValidPortion() local
1680 VLOG(4) << "Slice mask created: arg literal " << arg_literal.ToString(); in SelectValidPortion()
1682 HloInstruction::CreateConstant(std::move(arg_literal))); in SelectValidPortion()
Dhlo_evaluator.cc1924 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleCall() local
1925 arg_literals.push_back(&arg_literal); in HandleCall()
1959 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleFusion() local
1960 arg_literals.push_back(&arg_literal); in HandleFusion()
Dhlo_evaluator_typed_visitor.h1742 const Literal& arg_literal =
1745 auto curr_val = arg_literal.Get<NativeT>(multi_index);