Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc58 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
60 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", in XLA_TEST_P()
71 auto input_literal = LiteralUtil::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 = LiteralUtil::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 = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
102 auto input = CreateParameterAndTransferLiteral(0, input_literal, "parameter", in XLA_TEST_P()
131 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
133 auto input = CreateParameterAndTransferLiteral(0, input_literal, "input", in XLA_TEST_P()
[all …]
Dreduce_test.cc107 Literal input_literal = LiteralUtil::CreateR1(AsSlice(input_data)); in RunR1ToR0Test() local
109 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR1ToR0Test()
138 Literal input_literal = LiteralUtil::CreateR1(input_data); in RunR1ToR0PredTest() local
140 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR1ToR0PredTest()
179 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1PredTest() local
180 input_literal = in RunR2ToR1PredTest()
181 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunR2ToR1PredTest()
183 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR2ToR1PredTest()
212 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR0Test() local
213 input_literal = in RunR2ToR0Test()
[all …]
Dconvolution_test.cc258 auto input_literal = in XLA_TEST_F() local
266 {input_literal.get(), filter_literal.get()}, in XLA_TEST_F()
293 auto input_literal = in RunTest() local
301 {input_literal.get(), filter_literal.get()}, in RunTest()
328 auto input_literal = in XLA_TEST_F() local
336 {input_literal.get(), filter_literal.get()}, in XLA_TEST_F()
359 auto input_literal = in XLA_TEST_F() local
367 {input_literal.get(), filter_literal.get()}, in XLA_TEST_F()
395 auto input_literal = in RunTest() local
403 {input_literal.get(), filter_literal.get()}, in RunTest()
[all …]
Dexhaustive_op_test.cc259 Literal input_literal = LiteralUtil::CreateFromDimensions(ty, {input_size}); in RunImpl() local
260 absl::Span<T> input_arr = input_literal.data<T>(); in RunImpl()
274 BuildAndRunComputation(enqueue_op, input_literal)); in RunImpl()
275 ExpectNear<T>(input_literal, result_literal, evaluate_op); in RunImpl()
280 const Literal& input_literal) { in BuildAndRunComputation() argument
282 auto input = Parameter(&builder, 0, input_literal.shape(), "input"); in BuildAndRunComputation()
297 client_->Compile(comp, {&input_literal.shape()}, build_opts)); in BuildAndRunComputation()
301 client_->LiteralToShapedBuffer(input_literal, /*device_ordinal=*/0)); in BuildAndRunComputation()
332 void ExpectNear(const Literal& input_literal, const Literal& result_literal, in ExpectNear() argument
342 absl::Span<const T> input_arr = input_literal.data<T>(); in ExpectNear()
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 Literal expected = input_literal.Clone(); in TEST_P()
98 float value = input_literal.Get<float>(indices); in TEST_P()
Dreduce_window_test.cc342 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in TEST_P() local
346 0, input_literal, "parameter", &builder_, &input); in TEST_P()
400 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
404 0, input_literal, "parameter", &builder_, &input); in XLA_TEST_P()
421 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
425 0, input_literal, "parameter", &builder_, &input); in XLA_TEST_P()
442 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
446 0, input_literal, "parameter", &builder_, &input); in XLA_TEST_P()
620 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in DoIt() local
623 auto input_arg = CreateParameterAndTransferLiteral(0, input_literal, "p0", in DoIt()
[all …]
Dbatch_normalization_test.cc517 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
520 Parameter(&builder, 0, input_literal.shape(), "input"); in XLA_TEST_P()
531 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
617 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
620 Parameter(&builder, 0, input_literal.shape(), "input"); in XLA_TEST_P()
632 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
802 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
809 auto input_parameter = Parameter(&builder, 0, input_literal.shape(), "input"); in XLA_TEST_P()
817 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
Dconstants_test.cc146 Literal input_literal = LiteralUtil::CreateR4FromArray4D(input_array); in TEST_F() local
150 ConstantLiteral(&builder, input_literal); in TEST_F()
Darray_elementwise_ops_test.cc2268 auto input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2277 client_->TransferToServer(input_literal)); in XLA_TEST_F()
2279 auto input = Parameter(&builder, 0, input_literal.shape(), "input"); in XLA_TEST_F()
2310 Literal input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2323 client_->TransferToServer(input_literal)); in XLA_TEST_F()
2325 auto input = Parameter(&builder, 0, input_literal.shape(), "input"); in XLA_TEST_F()
2329 int64 input_size = input_literal.shape().dimensions(0); in XLA_TEST_F()
2332 expected_result.push_back(std::exp(input_literal.Get<float>({i}))); in XLA_TEST_F()
2344 Literal input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2361 client_->TransferToServer(input_literal)); in XLA_TEST_F()
[all …]
Ddynamic_ops_test.cc749 auto input_literal = LiteralUtil::CreateR4( in BM_DynamicSlice() local
752 auto input = ConstantLiteral(&builder, 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/cpu/tests/
Dcpu_fusion_test.cc93 auto input_literal = LiteralUtil::CreateR1<float>({-1.5, -2.5, -3.0}); in TEST_F() local
94 Shape vshape = input_literal.shape(); in TEST_F()
97 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_F()
143 auto input_literal = LiteralUtil::CreateR1<float>({-1.5, -2.5, -3.0}); in TEST_F() local
144 Shape vshape = input_literal.shape(); in TEST_F()
147 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_F()
240 auto input_literal = LiteralUtil::CreateR1<float>({1.0, 2.0, 3.0}); in TEST_F() local
241 Shape vshape = input_literal.shape(); in TEST_F()
244 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc443 auto input_literal = LiteralUtil::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); in TEST_F() local
447 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_F()
459 auto input_literal = LiteralUtil::CreateR0<int32>(111); in TEST_F() local
464 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_F()
522 auto input_literal = LiteralUtil::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); in TEST_P() local
525 ASSERT_TRUE(LayoutUtil::LayoutsInShapesEqual(input_literal.shape(), in TEST_P()
529 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_P()
541 auto input_literal = LiteralUtil::CreateR2WithLayout<int32>( in TEST_P() local
545 ASSERT_FALSE(LayoutUtil::LayoutsInShapesEqual(input_literal.shape(), in TEST_P()
549 HloInstruction::CreateConstant(std::move(input_literal))); in TEST_P()
Dhlo_evaluator.cc423 const Literal* input_literal = arg_literals_[parameter->parameter_number()]; in HandleParameter() local
424 VLOG(2) << "Parameter evaluated to: " << input_literal->ToString(); in HandleParameter()
425 DCHECK(ShapeUtil::Equal(parameter->shape(), input_literal->shape())) in HandleParameter()
428 << ShapeUtil::HumanString(input_literal->shape()); in HandleParameter()