Home
last modified time | relevance | path

Searched refs:input_literal (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc58 auto input_literal = Literal::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
60 auto input = CreateParameterAndTransferLiteral(0, *input_literal, "parameter", in XLA_TEST_P()
71 auto input_literal = Literal::CreateR1<float>({1.0f}); in XLA_TEST_P() local
73 auto input = CreateParameterAndTransferLiteral(0, *input_literal, "parameter", in XLA_TEST_P()
84 auto input_literal = Literal::CreateR1<float>({1.0f}); in XLA_TEST_P() local
86 auto input = CreateParameterAndTransferLiteral(0, *input_literal, "parameter", in XLA_TEST_P()
100 auto input_literal = Literal::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
102 auto input = CreateParameterAndTransferLiteral(0, *input_literal, "parameter", in XLA_TEST_P()
135 auto input_literal = Literal::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
137 auto input = CreateParameterAndTransferLiteral(0, *input_literal, "input", in XLA_TEST_P()
[all …]
Dreduce_test.cc98 std::unique_ptr<Literal> input_literal = in RunR1ToR0Test() local
101 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunR1ToR0Test()
131 std::unique_ptr<Literal> input_literal = Literal::CreateR1(input_data); in RunR1ToR0PredTest() local
133 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunR1ToR0PredTest()
172 std::unique_ptr<Literal> input_literal = in RunR2ToR1PredTest() local
174 input_literal = in RunR2ToR1PredTest()
175 input_literal->Relayout(LayoutUtil::MakeLayout({minor, major})); in RunR2ToR1PredTest()
177 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunR2ToR1PredTest()
206 std::unique_ptr<Literal> input_literal = in RunR2ToR0Test() local
208 input_literal = in RunR2ToR0Test()
[all …]
Dexhaustive_f32_elementwise_op_test.cc40 std::unique_ptr<Literal> input_literal = in ExhaustivelyTestF32Op() local
47 input_literal->Set({i - begin}, 0.0f); in ExhaustivelyTestF32Op()
49 input_literal->Set({i - begin}, tensorflow::bit_cast<float, int>(i)); in ExhaustivelyTestF32Op()
54 client_->TransferToServer(*input_literal)); in ExhaustivelyTestF32Op()
56 auto input = builder.Parameter(0, input_literal->shape(), "input"); in ExhaustivelyTestF32Op()
62 expected_result.push_back(evaluate_op(input_literal->Get<float>({i}))); in ExhaustivelyTestF32Op()
Dconvolution_test.cc266 auto input_literal = in XLA_TEST_F() local
274 {input_literal.get(), filter_literal.get()}, in XLA_TEST_F()
301 auto input_literal = in RunTest() local
309 {input_literal.get(), filter_literal.get()}, in RunTest()
336 auto input_literal = in XLA_TEST_F() local
344 {input_literal.get(), filter_literal.get()}, in XLA_TEST_F()
367 auto input_literal = in XLA_TEST_F() local
375 {input_literal.get(), filter_literal.get()}, in XLA_TEST_F()
403 auto input_literal = in RunTest() local
411 {input_literal.get(), filter_literal.get()}, in RunTest()
[all …]
Dreverse_test.cc86 auto input_literal = r1_literal->Reshape(spec.input_dims).ConsumeValueOrDie(); in TEST_P() local
89 auto a = AddParam(*input_literal, &builder); in TEST_P()
92 std::unique_ptr<Literal> expected = input_literal->CloneToUnique(); in TEST_P()
99 float value = input_literal->Get<float>(indices); in TEST_P()
Dreduce_window_test.cc296 std::unique_ptr<Literal> input_literal = in TEST_P() local
301 0, *input_literal, "parameter", &builder_, &input); in TEST_P()
364 std::unique_ptr<Literal> input_literal = in XLA_TEST_P() local
369 0, *input_literal, "parameter", &builder_, &input); in XLA_TEST_P()
386 std::unique_ptr<Literal> input_literal = in XLA_TEST_P() local
391 0, *input_literal, "parameter", &builder_, &input); in XLA_TEST_P()
408 std::unique_ptr<Literal> input_literal = in XLA_TEST_P() local
413 0, *input_literal, "parameter", &builder_, &input); in XLA_TEST_P()
579 std::unique_ptr<Literal> input_literal = in DoIt() local
583 auto input_arg = CreateParameterAndTransferLiteral(0, *input_literal, "p0", in DoIt()
[all …]
Dbatch_normalization_test.cc507 auto input_literal = Literal::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
510 builder.Parameter(0, input_literal->shape(), "input"); in XLA_TEST_P()
521 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
607 auto input_literal = Literal::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
610 builder.Parameter(0, input_literal->shape(), "input"); in XLA_TEST_P()
622 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
792 auto input_literal = Literal::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
799 auto input_parameter = builder.Parameter(0, input_literal->shape(), "input"); in XLA_TEST_P()
807 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
Dconstants_test.cc144 std::unique_ptr<Literal> input_literal = in TEST_F() local
149 builder.ConstantLiteral(*input_literal); in TEST_F()
Darray_elementwise_ops_test.cc2053 auto input_literal = Literal::CreateR1<float>( in XLA_TEST_F() local
2062 client_->TransferToServer(*input_literal)); in XLA_TEST_F()
2064 auto input = builder.Parameter(0, input_literal->shape(), "input"); in XLA_TEST_F()
2095 std::unique_ptr<Literal> input_literal = Literal::CreateR1<float>( in XLA_TEST_F() local
2108 client_->TransferToServer(*input_literal)); in XLA_TEST_F()
2110 auto input = builder.Parameter(0, input_literal->shape(), "input"); in XLA_TEST_F()
2114 int64 input_size = input_literal->shape().dimensions(0); in XLA_TEST_F()
2117 expected_result.push_back(std::exp(input_literal->Get<float>({i}))); in XLA_TEST_F()
2129 std::unique_ptr<Literal> input_literal = Literal::CreateR1<float>( in XLA_TEST_F() local
2146 client_->TransferToServer(*input_literal)); in XLA_TEST_F()
[all …]
Ddynamic_ops_test.cc678 auto input_literal = Literal::CreateR4( in BM_DynamicSlice() local
681 auto input = builder.ConstantLiteral(*input_literal); in BM_DynamicSlice()
/external/pdfium/testing/
Dtest_support.h22 #define STR_IN_TEST_CASE(input_literal, ...) \ argument
24 reinterpret_cast<const unsigned char*>(input_literal), \
25 sizeof(input_literal) - 1, __VA_ARGS__ \
28 #define STR_IN_OUT_CASE(input_literal, expected_literal, ...) \ argument
30 reinterpret_cast<const unsigned char*>(input_literal), \
31 sizeof(input_literal) - 1, \
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc342 auto input_literal = Literal::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); in TEST_P() local
346 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_P()
358 auto input_literal = Literal::CreateR0<int32>(111); in TEST_P() local
363 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_P()
421 auto input_literal = Literal::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); in TEST_P() local
424 ASSERT_TRUE(LayoutUtil::LayoutsInShapesEqual(input_literal->shape(), in TEST_P()
428 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_P()
440 auto input_literal = Literal::CreateR2WithLayout<int32>( in TEST_P() local
444 ASSERT_FALSE(LayoutUtil::LayoutsInShapesEqual(input_literal->shape(), in TEST_P()
448 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_P()
Dhlo_evaluator.cc2121 const Literal* input_literal = arg_literals_[operand->parameter_number()]; in Evaluate() local
2123 << input_literal->ToString(); in Evaluate()
2124 TF_RET_CHECK(ShapeUtil::Equal(operand->shape(), input_literal->shape())); in Evaluate()
2126 evaluated_[operand] = input_literal->CloneToUnique(); in Evaluate()
2204 const Literal* input_literal = arg_literals_[parameter->parameter_number()]; in HandleParameter() local
2205 VLOG(2) << "Parameter evaluated to: " << input_literal->ToString(); in HandleParameter()
2206 DCHECK(ShapeUtil::Equal(parameter->shape(), input_literal->shape())) in HandleParameter()
2209 << ShapeUtil::HumanString(input_literal->shape()); in HandleParameter()
2211 evaluated_[parameter] = input_literal->CloneToUnique(); in HandleParameter()