Search

Home

Sort by

Searched defs:input_literal (Results 1 - 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A D	constants_test.cc	`144 std::unique_ptr<Literal> input_literal = local 149 builder.ConstantLiteral(*input_literal);`
H A D	exhaustive_f32_elementwise_op_test.cc	`40 std::unique_ptr<Literal> input_literal = local 47 input_literal->Set({i - begin}, 0.0f); 49 input_literal->Set({i - begin}, tensorflow::bit_cast<float, int>(i)); 54 client_->TransferToServer(*input_literal)); 56 auto input = builder.Parameter(0, input_literal->shape(), "input"); 62 expected_result.push_back(evaluate_op(input_literal->Get<float>({i})));`
H A D	reverse_test.cc	`86 auto input_literal = r1_literal->Reshape(spec.input_dims).ConsumeValueOrDie(); local 89 auto a = AddParam(*input_literal, &builder); 92 std::unique_ptr<Literal> expected = input_literal->CloneToUnique(); 99 float value = input_literal->Get<float>(indices);`
H A D	reshape_test.cc	58 auto input_literal = Literal::CreateR2FromArray2D(input_array); local 60 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", 71 auto input_literal = Literal::CreateR1<float>({1.0f}); local 73 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", 84 auto input_literal = Literal::CreateR1<float>({1.0f}); local 86 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", 100 auto input_literal = Literal::CreateR2FromArray2D(input_array); local 102 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", 135 auto input_literal = Literal::CreateR2FromArray2D(input_array); local 137 auto input = CreateParameterAndTransferLiteral(0, input_literal, "inpu 168 auto input_literal = Literal::CreateR2FromArray2D(input_array); local 181 auto input_literal = Literal::CreateR2<float>({{1.0f, 2.0f, 3.0f}}); local 194 auto input_literal = Literal::CreateR2<float>({{1.0f}, {2.0f}, {3.0f}}); local 211 auto input_literal = Literal::CreateR1<float>({}); local 225 auto input_literal = local 245 auto input_literal = Literal::CreateFromArray(Array2D<float>(0, 2)); local 260 auto input_literal = Literal::CreateFromArray(simple); local 277 auto input_literal = Literal::CreateFromArray(a4x3); local 297 auto input_literal = Literal::CreateFromArray(Array2D<float>(0, 4)); local 311 auto input_literal = Literal::CreateFromArray(a4x3); local 331 auto input_literal = Literal::CreateFromArray(Array2D<float>(6, 0)); local 347 auto input_literal = Literal::CreateFromArray(Array4D<float>(2, 3, 4, 0)); local 363 auto input_literal = Literal::CreateFromArray(a4x3); local 382 auto input_literal = Literal::CreateFromArray(Array2D<float>(0, 6)); local 398 auto input_literal = Literal::CreateFromArray(a4x3); local 424 auto input_literal = Literal::CreateFromArray(ArrayForDocR3Tests()); local 439 auto input_literal = Literal::CreateFromArray(ArrayForDocR3Tests()); local 459 auto input_literal = Literal::CreateFromArray(ArrayForDocR3Tests()); local 474 auto input_literal = Literal::CreateFromArray(ArrayForDocR3Tests()); local 494 auto input_literal = Literal::CreateFromArray(ArrayForDocR3Tests()); local 527 auto input_literal = Literal::CreateFromArray(t2x2x2x3); local 552 auto input_literal = Literal::CreateFromArray(t); local 589 auto input_literal = Literal::CreateR1<float>({1.0f}); local 601 auto input_literal = Literal::CreateR1<float>({1.0f, 2.0f}); local 613 auto input_literal = Literal::CreateR4FromArray4DWithLayout(Array4D<float>{ local 667 std::unique_ptr<Literal> input_literal = Literal::CreateR2<float>({ local 694 std::unique_ptr<Literal> input_literal = Literal::CreateR2<float>({ local 726 std::unique_ptr<Literal> input_literal = local 748 std::unique_ptr<Literal> input_literal = local 771 std::unique_ptr<Literal> input_literal = local 798 std::unique_ptr<Literal> input_literal = local 879 std::unique_ptr<Literal> input_literal = local 908 std::unique_ptr<Literal> input_literal = local 937 std::unique_ptr<Literal> input_literal = local 967 std::unique_ptr<Literal> input_literal = local 996 std::unique_ptr<Literal> input_literal = local [all...]
H A D	dynamic_ops_test.cc	`678 auto input_literal = Literal::CreateR4( local 681 auto input = builder.ConstantLiteral(*input_literal);`
H A D	convolution_test.cc	266 auto input_literal = local 274 {input_literal.get(), filter_literal.get()}, 301 auto input_literal = local 309 {input_literal.get(), filter_literal.get()}, 336 auto input_literal = local 344 {input_literal.get(), filter_literal.get()}, 367 auto input_literal = local 375 {input_literal.get(), filter_literal.get()}, 403 auto input_literal = local 411 {input_literal 466 auto input_literal = client_->TransferToServer(*input_r5).ConsumeValueOrDie(); local 529 auto input_literal = local 655 auto input_literal = local [all...]
H A D	reduce_test.cc	98 std::unique_ptr<Literal> input_literal = local 101 client_->TransferToServer(input_literal).ConsumeValueOrDie(); 131 std::unique_ptr<Literal> input_literal = Literal::CreateR1(input_data); local 133 client_->TransferToServer(input_literal).ConsumeValueOrDie(); 172 std::unique_ptr<Literal> input_literal = local 174 input_literal = 175 input_literal->Relayout(LayoutUtil::MakeLayout({minor, major})); 177 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); 206 std::unique_ptr<Literal> input_literal = local 208 input_literal 234 std::unique_ptr<Literal> input_literal = local 292 std::unique_ptr<Literal> input_literal = local 447 std::unique_ptr<Literal> input_literal = local 479 std::unique_ptr<Literal> input_literal = local 531 std::unique_ptr<Literal> input_literal = local 595 auto input_literal = Literal::CreateR2FromArray2D(input); local 610 auto input_literal = Literal::CreateR2FromArray2D(input); local 624 auto input_literal = Literal::CreateR2FromArray2D(input); local 637 auto input_literal = Literal::CreateR2FromArray2D(input); local 817 auto input_literal = Literal::CreateR3FromArray3D(input_array); local [all...]
H A D	batch_normalization_test.cc	507 auto input_literal = Literal::CreateR4FromArray4D<float>(input_array); local 510 builder.Parameter(0, input_literal->shape(), "input"); 521 client_->TransferToServer(input_literal).ConsumeValueOrDie(); 607 auto input_literal = Literal::CreateR4FromArray4D<float>(input_array); local 610 builder.Parameter(0, input_literal->shape(), "input"); 622 client_->TransferToServer(input_literal).ConsumeValueOrDie(); 792 auto input_literal = Literal::CreateR4FromArray4D<float>(input_array); local 799 auto input_parameter = builder.Parameter(0, input_literal->shape(), "input"); 807 client_->TransferToServer(*input_literal).ConsumeValueOrDie();
H A D	reduce_window_test.cc	296 std::unique_ptr<Literal> input_literal = local 301 0, input_literal, "parameter", &builder_, &input); 364 std::unique_ptr<Literal> input_literal = local 369 0, input_literal, "parameter", &builder_, &input); 386 std::unique_ptr<Literal> input_literal = local 391 0, input_literal, "parameter", &builder_, &input); 408 std::unique_ptr<Literal> input_literal = local 413 0, input_literal, "parameter", &builder_, &input); 579 std::unique_ptr<Literal> input_literal = local 583 auto input_arg = CreateParameterAndTransferLiteral(0, *input_literal, "p 930 std::unique_ptr<Literal> input_literal = local 1035 std::unique_ptr<Literal> input_literal = local 1221 std::unique_ptr<Literal> input_literal = local [all...]
H A D	array_elementwise_ops_test.cc	2053 auto input_literal = Literal::CreateR1<float>( local 2062 client_->TransferToServer(input_literal)); 2064 auto input = builder.Parameter(0, input_literal->shape(), "input"); 2095 std::unique_ptr<Literal> input_literal = Literal::CreateR1<float>( local 2108 client_->TransferToServer(input_literal)); 2110 auto input = builder.Parameter(0, input_literal->shape(), "input"); 2114 int64 input_size = input_literal->shape().dimensions(0); 2117 expected_result.push_back(std::exp(input_literal->Get<float>({i}))); 2129 std::unique_ptr<Literal> input_literal = Literal::CreateR1<float>( local 2146 client_->TransferToServer(*input_literal)); [all...]
/external/tensorflow/tensorflow/compiler/xla/service/
H A D	hlo_evaluator_test.cc	342 auto input_literal = Literal::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); local 346 HloInstruction::CreateConstant(std::move(input_literal))); 358 auto input_literal = Literal::CreateR0<int32>(111); local 363 HloInstruction::CreateConstant(std::move(input_literal))); 421 auto input_literal = Literal::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); local 424 ASSERT_TRUE(LayoutUtil::LayoutsInShapesEqual(input_literal->shape(), 428 HloInstruction::CreateConstant(std::move(input_literal))); 440 auto input_literal = Literal::CreateR2WithLayout<int32>( local 444 ASSERT_FALSE(LayoutUtil::LayoutsInShapesEqual(input_literal->shape(), 448 HloInstruction::CreateConstant(std::move(input_literal))); [all...]

Completed in 263 milliseconds