Home
last modified time | relevance | path

Searched defs: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
71 auto input_literal = LiteralUtil::CreateR1<float>({1.0f}); in XLA_TEST_P() local
84 auto input_literal = LiteralUtil::CreateR1<float>({1.0f}); in XLA_TEST_P() local
100 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
131 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
158 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
171 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}}); in XLA_TEST_P() local
184 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f}, {2.0f}, {3.0f}}); in XLA_TEST_P() local
197 auto input_literal = LiteralUtil::CreateR1<float>({}); in XLA_TEST_P() local
211 auto input_literal = in XLA_TEST_P() local
[all …]
Dconvolution_test.cc258 auto input_literal = in XLA_TEST_F() local
293 auto input_literal = in RunTest() local
328 auto input_literal = in XLA_TEST_F() local
359 auto input_literal = in XLA_TEST_F() local
395 auto input_literal = in RunTest() local
457 auto input_literal = client_->TransferToServer(input_r5).ConsumeValueOrDie(); in XLA_TEST_F() local
519 auto input_literal = in RunTest() local
583 auto input_literal = in RunTest() local
651 auto input_literal = in RunTest() local
716 auto input_literal = in RunTest() local
[all …]
Dreduce_test.cc107 Literal input_literal = LiteralUtil::CreateR1(AsSlice(input_data)); in RunR1ToR0Test() local
138 Literal input_literal = LiteralUtil::CreateR1(input_data); in RunR1ToR0PredTest() local
179 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1PredTest() local
212 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR0Test() local
239 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1Test() local
296 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunVectorizedReduceTestForType() local
450 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F() local
481 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F() local
528 Literal input_literal = LiteralUtil::CreateR3FromArray3D(input_data); in XLA_TEST_F() local
590 auto input_literal = LiteralUtil::CreateR2FromArray2D(input); in XLA_TEST_F() local
[all …]
Dreduce_window_test.cc342 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in TEST_P() local
400 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
421 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
442 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
620 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in DoIt() local
1021 Literal input_literal = LiteralUtil::CreateR3FromArray3DWithLayout( in TEST_P() local
1163 Literal input_literal = LiteralUtil::CreateR2FromArray2DWithLayout( in DoIt() local
1357 Literal input_literal = in TEST_P() local
Dexhaustive_op_test.cc259 Literal input_literal = LiteralUtil::CreateFromDimensions(ty, {input_size}); in RunImpl() local
280 const Literal& input_literal) { in BuildAndRunComputation()
332 void ExpectNear(const Literal& input_literal, const Literal& result_literal, in ExpectNear()
Dreverse_test.cc86 auto input_literal = r1_literal.Reshape(spec.input_dims).ConsumeValueOrDie(); in TEST_P() local
Dconstants_test.cc146 Literal input_literal = LiteralUtil::CreateR4FromArray4D(input_array); in TEST_F() local
Dbatch_normalization_test.cc517 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
617 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
802 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
Ddynamic_ops_test.cc749 auto input_literal = LiteralUtil::CreateR4( in BM_DynamicSlice() local
Darray_elementwise_ops_test.cc2268 auto input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2310 Literal input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2344 Literal input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
/external/pdfium/testing/
Dtest_support.h22 #define STR_IN_TEST_CASE(input_literal, ...) \ argument
28 #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_test.cc443 auto input_literal = LiteralUtil::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); in TEST_F() local
459 auto input_literal = LiteralUtil::CreateR0<int32>(111); in TEST_F() local
522 auto input_literal = LiteralUtil::CreateR2<int32>({{1, 2}, {3, 4}, {5, 6}}); in TEST_P() local
541 auto input_literal = LiteralUtil::CreateR2WithLayout<int32>( in TEST_P() local
Dhlo_evaluator.cc423 const Literal* input_literal = arg_literals_[parameter->parameter_number()]; in HandleParameter() local