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.cc57 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
85 auto input_literal = LiteralUtil::CreateR1<float>({1.0f}); in XLA_TEST_P() local
102 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
135 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
164 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array); in XLA_TEST_P() local
178 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f, 2.0f, 3.0f}}); in XLA_TEST_P() local
192 auto input_literal = LiteralUtil::CreateR2<float>({{1.0f}, {2.0f}, {3.0f}}); in XLA_TEST_P() local
206 auto input_literal = LiteralUtil::CreateR1<float>({}); in XLA_TEST_P() local
221 auto input_literal = in XLA_TEST_P() local
[all …]
Dconvolution_test_1d.cc122 auto input_literal = client_->TransferToServer(input_r3).value(); in TestImpl() local
218 auto input_literal = in XLA_TEST_F() local
253 auto input_literal = in RunTest() local
288 auto input_literal = in XLA_TEST_F() local
319 auto input_literal = in XLA_TEST_F() local
355 auto input_literal = in RunTest() local
Dconvolution_test.cc304 auto input_literal = client_->TransferToServer(input_r5).value(); in XLA_TEST_F() local
365 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest() local
427 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest() local
493 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest() local
556 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest() local
620 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest() local
687 auto input_literal = client_->TransferToServer(input_r4_relaid).value(); in RunTest() local
755 auto input_literal = client_->TransferToServer(input_r4_relaid).value(); in RunTest() local
828 auto input_literal = client_->TransferToServer(input_r4_relaid).value(); in RunTest() local
892 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest() local
[all …]
Dreduce_test.cc111 Literal input_literal = in RunR1ToR0Test() local
140 Literal input_literal = LiteralUtil::CreateR1(input_data); in RunR1ToR0PredTest() local
181 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1PredTest() local
215 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR0Test() local
243 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunR2ToR1Test() local
301 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in RunVectorizedReduceTestForType() local
455 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F() local
487 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data); in XLA_TEST_F() local
535 Literal input_literal = LiteralUtil::CreateR3FromArray3D(input_data); in XLA_TEST_F() local
598 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.cc363 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
422 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
444 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
466 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in XLA_TEST_P() local
633 Literal input_literal = LiteralUtil::CreateR4FromArray4DWithLayout( in DoIt() local
1062 Literal input_literal = LiteralUtil::CreateR3FromArray3DWithLayout( in XLA_TEST_P() local
1296 Literal input_literal = LiteralUtil::CreateR2FromArray2DWithLayout( in DoIt() local
1523 Literal input_literal = in XLA_TEST_P() local
Dreverse_test.cc86 auto input_literal = r1_literal.Reshape(spec.input_dims).value(); in TEST_P() local
Dexhaustive_unary_test_complex.cc60 void FillInput(std::array<Literal, 1>* input_literal) override { in FillInput()
Dcollective_ops_test.cc344 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F() local
366 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F() local
459 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F() local
Dconstants_test.cc146 Literal input_literal = LiteralUtil::CreateR4FromArray4D(input_array); in TEST_F() local
Dbatch_normalization_test.cc580 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
680 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
865 auto input_literal = LiteralUtil::CreateR4FromArray4D<float>(input_array); in XLA_TEST_P() local
Dexhaustive_unary_test_f32_or_smaller.cc200 void FillInput(std::array<Literal, 1>* input_literal) override { in FillInput()
Ddynamic_ops_test.cc769 auto input_literal = LiteralUtil::CreateR4( in BM_DynamicSlice() local
Darray_elementwise_ops_test.cc2453 auto input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2495 Literal input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
2529 Literal input_literal = LiteralUtil::CreateR1<float>( in XLA_TEST_F() local
Ddot_operation_test.cc1844 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.cc92 auto input_literal = LiteralUtil::CreateR1<float>({-1.5, -2.5, -3.0}); in TEST_F() local
142 auto input_literal = LiteralUtil::CreateR1<float>({-1.5, -2.5, -3.0}); in TEST_F() local
239 auto input_literal = LiteralUtil::CreateR1<float>({1.0, 2.0, 3.0}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1102 const Literal* input_literal = arg_literals_[parameter->parameter_number()]; in HandleParameter() local
1611 Status ComputeFft(HloInstruction* fft, const Literal& input_literal, in ComputeFft()
1983 bool CopyDataFromInput(const Literal& input_literal, int64_t input_start, in CopyDataFromInput()
2073 bool CopyDataFromInput(const Literal& input_literal, int64_t input_start, in CopyDataFromInput()
2184 const Literal& input_literal = GetEvaluatedLiteralFor(fft->operand(0)); in HandleFft() local
Dhlo_evaluator_test.cc558 auto input_literal = LiteralUtil::CreateR2<int32_t>({{1, 2}, {3, 4}, {5, 6}}); in TEST_F() local
574 auto input_literal = LiteralUtil::CreateR0<int32_t>(111); in TEST_F() local
637 auto input_literal = LiteralUtil::CreateR2<int32_t>({{1, 2}, {3, 4}, {5, 6}}); in TEST_P() local
656 auto input_literal = LiteralUtil::CreateR2WithLayout<int32_t>( in TEST_P() local