Searched refs:arg_literal (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | convert_test.cc | 213 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 …]
|
D | local_client_execute_test.cc | 350 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 …]
|
D | reduce_window_test.cc | 386 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/ |
D | executable_base.cc | 112 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/ |
D | hlo_evaluator_test.cc | 159 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()
|
D | space_to_batch_converter.cc | 1585 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()
|
D | hlo_evaluator.cc | 1924 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()
|
D | hlo_evaluator_typed_visitor.h | 1742 const Literal& arg_literal = 1745 auto curr_val = arg_literal.Get<NativeT>(multi_index);
|