Home
last modified time | relevance | path

Searched defs:input_literal (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc58 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
72 auto input_literal = LiteralUtil::CreateR1<float>({1.0f}); in XLA_TEST_P() local
86 auto input_literal = LiteralUtil::CreateR1<float>({1.0f}); in XLA_TEST_P() local
103 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
136 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
165 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
179 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}}); in XLA_TEST_P() local
193 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f}, {2.0f}, {3.0f}}); in XLA_TEST_P() local
207 auto input_literal = LiteralUtil::CreateR1<float>({}); in XLA_TEST_P() local
222 auto input_literal = in XLA_TEST_P() local
[all …]
Dconvolution_test_1d.cc124 auto input_literal = in TestImpl() local
222 auto input_literal = in XLA_TEST_F() local
257 auto input_literal = in RunTest() local
292 auto input_literal = in XLA_TEST_F() local
323 auto input_literal = in XLA_TEST_F() local
359 auto input_literal = in RunTest() local
Dconvolution_test.cc289 auto input_literal = client_->TransferToServer(input_r5).ConsumeValueOrDie(); in XLA_TEST_F() local
351 auto input_literal = in RunTest() local
415 auto input_literal = in RunTest() local
483 auto input_literal = in RunTest() local
548 auto input_literal = in RunTest() local
614 auto input_literal = in RunTest() local
684 auto input_literal = in RunTest() local
755 auto input_literal = in RunTest() local
830 auto input_literal = in RunTest() local
896 auto input_literal = in RunTest() local
[all …]
Dreduce_test.cc111 Literal input_literal = LiteralUtil::CreateR1(AsSlice(input_data)); in RunR1ToR0Test() local
139 Literal input_literal = LiteralUtil::CreateR1(input_data); in RunR1ToR0PredTest() local
180 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1PredTest() local
213 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR0Test() local
240 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1Test() local
297 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunVectorizedReduceTestForType() local
451 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F() local
482 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F() local
529 Literal input_literal = LiteralUtil::CreateR3FromArray3D(input_data); in XLA_TEST_F() local
591 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F() local
[all …]
Dexhaustive_unary_test_f64.cc41 void FillInput(std::array<Literal, 1>* input_literal) override { in FillInput()
Dreduce_window_test.cc365 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
424 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
446 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
468 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
635 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in DoIt() local
1037 Literal input_literal = LiteralUtil::CreateR3FromArray3DWithLayout( in XLA_TEST_P() local
1247 Literal input_literal = LiteralUtil::CreateR2FromArray2DWithLayout( in DoIt() local
1450 Literal input_literal = in XLA_TEST_P() local
Dcollective_ops_test.cc320 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F() local
340 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F() local
413 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F() local
506 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F() local
Dreverse_test.cc87 auto input_literal = r1_literal.Reshape(spec.input_dims).ConsumeValueOrDie(); in TEST_P() local
Dexhaustive_unary_test_complex.cc60 void FillInput(std::array<Literal, 1>* input_literal) override { in FillInput()
Dconstants_test.cc146 Literal input_literal = LiteralUtil::CreateR4FromArray4D(input_array); in TEST_F() local
Dbatch_normalization_test.cc615 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
715 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
900 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
Dexhaustive_unary_test_f32_or_smaller.cc195 void FillInput(std::array<Literal, 1>* input_literal) override { in FillInput()
Ddynamic_ops_test.cc766 auto input_literal = LiteralUtil::CreateR4( in BM_DynamicSlice() local
Darray_elementwise_ops_test.cc2374 auto input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2416 Literal input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2450 Literal input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
Ddot_operation_test.cc1843 auto input_literal = LiteralUtil::CreateR3FromArray3D<float>(input_arr); in DOT_ReorderContracting() local
/external/pdfium/testing/
Dtest_support.h12 #define STR_IN_TEST_CASE(input_literal, ...) \ argument
18 #define STR_IN_OUT_CASE(input_literal, expected_literal, ...) \ argument
/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
143 auto input_literal = LiteralUtil::CreateR1<float>({-1.5, -2.5, -3.0}); in TEST_F() local
240 auto input_literal = LiteralUtil::CreateR1<float>({1.0, 2.0, 3.0}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc458 const Literal* input_literal = arg_literals_[parameter->parameter_number()]; in HandleParameter() local
1130 bool CopyDataFromInput(const Literal& input_literal, int64 input_start, in CopyDataFromInput()
1220 bool CopyDataFromInput(const Literal& input_literal, int64 input_start, in CopyDataFromInput()
1378 const Literal& input_literal = GetEvaluatedLiteralFor(fft->operand(0)); in HandleFft() local
Dhlo_evaluator_test.cc520 auto input_literal = LiteralUtil::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); in TEST_F() local
536 auto input_literal = LiteralUtil::CreateR0<int32>(111); in TEST_F() local
599 auto input_literal = LiteralUtil::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); in TEST_P() local
618 auto input_literal = LiteralUtil::CreateR2WithLayout<int32>( in TEST_P() local