Home
last modified time | relevance | path

Searched refs:XLA_TEST_P (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc54 XLA_TEST_P(ReshapeTest, CollapseTrivial1x1) { in XLA_TEST_P() function
69 XLA_TEST_P(ReshapeTest, CollapseTrivialR1EmptyDims) { in XLA_TEST_P() function
82 XLA_TEST_P(ReshapeTest, CollapseTrivialR1OnlyDim) { in XLA_TEST_P() function
96 XLA_TEST_P(ReshapeTest, SingleElementArrayToScalar) { in XLA_TEST_P() function
113 XLA_TEST_P(ReshapeTest, ScalarToSingleElementArray) { in XLA_TEST_P() function
128 XLA_TEST_P(ReshapeTest, Trivial0x3) { in XLA_TEST_P() function
141 XLA_TEST_P(ReshapeTest, Trivial0x3WithParameter) { in XLA_TEST_P() function
155 XLA_TEST_P(ReshapeTest, Trivial3x0) { in XLA_TEST_P() function
169 XLA_TEST_P(ReshapeTest, Trivial1x3) { in XLA_TEST_P() function
182 XLA_TEST_P(ReshapeTest, Trivial3x1) { in XLA_TEST_P() function
[all …]
Dreduce_window_test.cc108 XLA_TEST_P(ReduceWindowTest, MismatchedRanksGivesErrorStatus) { in XLA_TEST_P() function
125 XLA_TEST_P(ReduceWindowTest, R0ReduceWindow) { in XLA_TEST_P() function
137 XLA_TEST_P(ReduceWindowTest, Min3In5Stride2) { in XLA_TEST_P() function
145 XLA_TEST_P(ReduceWindowTest, Min3In5Stride2Same) { in XLA_TEST_P() function
154 XLA_TEST_P(ReduceWindowTest, Min3In5Stride1WithSamePadding) { in XLA_TEST_P() function
164 XLA_TEST_P(ReduceWindowTest, ZeroElementSmall) { in XLA_TEST_P() function
177 XLA_TEST_P(ReduceWindowTest, NonSquareSmall) { in XLA_TEST_P() function
192 XLA_TEST_P(ReduceWindowTest, MiddleDimsSmall) { in XLA_TEST_P() function
206 XLA_TEST_P(ReduceWindowTest, Along2ndMinorDim) { in XLA_TEST_P() function
223 XLA_TEST_P(ReduceWindowTest, AmongMajor2DimsAdd) { in XLA_TEST_P() function
[all …]
Dslice_test.cc250 XLA_TEST_P(SliceR1Test, DoIt_F32) { Run<float>(GetParam()); } in XLA_TEST_P() function
252 XLA_TEST_P(SliceR1Test, DoIt_F64) { Run<double>(GetParam()); } in XLA_TEST_P() function
254 XLA_TEST_P(SliceR1Test, DoIt_U32) { Run<uint32>(GetParam()); } in XLA_TEST_P() function
256 XLA_TEST_P(SliceR1Test, DoIt_S32) { Run<int32>(GetParam()); } in XLA_TEST_P() function
258 XLA_TEST_P(SliceR1Test, DoIt_U64) { Run<uint64>(GetParam()); } in XLA_TEST_P() function
260 XLA_TEST_P(SliceR1Test, DoIt_S64) { Run<int64>(GetParam()); } in XLA_TEST_P() function
264 XLA_TEST_P(SliceR1LargeTest, DISABLED_ON_GPU(DoIt_F32)) { in XLA_TEST_P() function
268 XLA_TEST_P(SliceR1LargeTest, DISABLED_ON_GPU(DoIt_F64)) { in XLA_TEST_P() function
272 XLA_TEST_P(SliceR1LargeTest, DISABLED_ON_GPU(DoIt_U32)) { in XLA_TEST_P() function
276 XLA_TEST_P(SliceR1LargeTest, DISABLED_ON_GPU(DoIt_S32)) { in XLA_TEST_P() function
[all …]
Dpad_test.cc87 XLA_TEST_P(PadTestFloat, Pad1DS0ToS0Array) { in XLA_TEST_P() function
102 XLA_TEST_P(PadTestFloat, Pad1DS0ToS5Array) { in XLA_TEST_P() function
117 XLA_TEST_P(PadTestFloat, Pad1DS3Array) { in XLA_TEST_P() function
132 XLA_TEST_P(PadTestFloat, Pad4D_2x0x3x2_FloatArray) { in XLA_TEST_P() function
227 XLA_TEST_P(PadTestFloat, Pad4DFloatArrayMinorFirstNonTrivialMinorDimensions) { in XLA_TEST_P() function
321 XLA_TEST_P(PadTestFloat, Large2DPad) { in XLA_TEST_P() function
340 XLA_TEST_P(PadTestFloat, AllTypes2DPad) { in XLA_TEST_P() function
362 XLA_TEST_P(PadTestFloat, High2DPad) { in XLA_TEST_P() function
389 XLA_TEST_P(PadTestFloat, NegativePadding2D) { in XLA_TEST_P() function
417 XLA_TEST_P(PadTestFloat, NegativeAndInteriorPadding2D) { in XLA_TEST_P() function
[all …]
Dbatch_normalization_test.cc104 XLA_TEST_P(BatchNormalizationTest, SubtractInZ) { in XLA_TEST_P() function
120 XLA_TEST_P(BatchNormalizationTest, SquareTesseractElementwise) { in XLA_TEST_P() function
137 XLA_TEST_P(BatchNormalizationTest, SumToZ) { in XLA_TEST_P() function
148 XLA_TEST_P(BatchNormalizationTest, SquareAndReduce) { in XLA_TEST_P() function
162 XLA_TEST_P(BatchNormalizationTest, VarianceToStddev) { in XLA_TEST_P() function
173 XLA_TEST_P(BatchNormalizationTest, SpecComparisonForward) { in XLA_TEST_P() function
232 XLA_TEST_P(BatchNormalizationTest, BasicTraining) { in XLA_TEST_P() function
255 XLA_TEST_P(BatchNormalizationTest, BasicTraining_fp16) { in XLA_TEST_P() function
285 XLA_TEST_P(BatchNormalizationTest, BasicTrainingOnDimension2) { in XLA_TEST_P() function
309 XLA_TEST_P(BatchNormalizationTest, BasicTrainingOnDimension2_fp16) { in XLA_TEST_P() function
[all …]
Dmatrix_ops_simple_test.cc165 XLA_TEST_P(TestLinspaceMaxParametric, TestF16) { TestImpl<Eigen::half>(); } in XLA_TEST_P() function
167 XLA_TEST_P(TestLinspaceMaxParametric, TestF32) { TestImpl<float>(); } in XLA_TEST_P() function
241 XLA_TEST_P(MatOpsDotAddTest, Dot_Add_2x2_2x2BF16) { TestImpl<bfloat16>(); } in XLA_TEST_P() function
243 XLA_TEST_P(MatOpsDotAddTest, Dot_Add_2x2_2x2F16) { TestImpl<Eigen::half>(); } in XLA_TEST_P() function
245 XLA_TEST_P(MatOpsDotAddTest, Dot_Add_2x2_2x2F32) { TestImpl<float>(); } in XLA_TEST_P() function
Dexhaustive_unary_test.cc239 XLA_TEST_P(ExhaustiveF32UnaryTest, test_name) \
247 XLA_TEST_P(ExhaustiveF16UnaryTest, test_name) \
255 XLA_TEST_P(ExhaustiveBF16UnaryTest, test_name) \
372 XLA_TEST_P(ExhaustiveF32UnaryTest, Acosh) { in XLA_TEST_P() function
381 XLA_TEST_P(ExhaustiveF16UnaryTest, Acosh) { Run(Acosh, std::acosh); } in XLA_TEST_P() function
382 XLA_TEST_P(ExhaustiveBF16UnaryTest, Acosh) { Run(Acosh, std::acosh); } in XLA_TEST_P() function
385 XLA_TEST_P(ExhaustiveF32UnaryTest, Asinh) { in XLA_TEST_P() function
393 XLA_TEST_P(ExhaustiveF16UnaryTest, Asinh) { Run(Asinh, std::asinh); } in XLA_TEST_P() function
394 XLA_TEST_P(ExhaustiveBF16UnaryTest, Asinh) { Run(Asinh, std::asinh); } in XLA_TEST_P() function
656 XLA_TEST_P(ExhaustiveF64UnaryTest, test_name) \
[all …]
Diota_test.cc37 XLA_TEST_P(IotaR1Test, DoIt) { in XLA_TEST_P() function
66 XLA_TEST_P(IotaR2Test, DoIt) { in XLA_TEST_P() function
98 XLA_TEST_P(IotaR3Test, DoIt) { in XLA_TEST_P() function
Dreduce_precision_test.cc465 XLA_TEST_P(ReducedPrecisionAccuracyTest, ReducePrecisionHalf) { in XLA_TEST_P() function
472 XLA_TEST_P(ReducedPrecisionAccuracyTest, ReducePrecisionBfloat16) { in XLA_TEST_P() function
479 XLA_TEST_P(ReducedPrecisionAccuracyTest, ReducePrecisionFloat) { in XLA_TEST_P() function
486 XLA_TEST_P(ReducedPrecisionAccuracyTest, ReducePrecisionDouble) { in XLA_TEST_P() function
Dhalf_test.cc50 XLA_TEST_P(UnaryOpTest, Ops) { in XLA_TEST_P() function
105 XLA_TEST_P(UnaryPredTest, Ops) { in XLA_TEST_P() function
140 XLA_TEST_P(BinaryOpTest, Ops) { in XLA_TEST_P() function
194 XLA_TEST_P(BinaryPredTest, Ops) { in XLA_TEST_P() function
Dexhaustive_binary_test.cc98 XLA_TEST_P(ExhaustiveF16BinaryTest, test_name) \
102 XLA_TEST_P(ExhaustiveBF16BinaryTest, test_name) \
247 XLA_TEST_P(ExhaustiveF32BinaryTest, test_name) \
337 XLA_TEST_P(ExhaustiveF64BinaryTest, test_name) \
Ddot_operation_test.cc415 XLA_TEST_P(ParametricDotTest, TestF16) { TestImpl<Eigen::half>(); } in XLA_TEST_P() function
417 XLA_TEST_P(ParametricDotTest, TestF32) { TestImpl<float>(); } in XLA_TEST_P() function
418 XLA_TEST_P(ParametricDotTest, TestF64) { TestImpl<double>(); } in XLA_TEST_P() function
419 XLA_TEST_P(ParametricDotTest, TestS32) { TestImpl<int32>(); } in XLA_TEST_P() function
483 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF16) { in XLA_TEST_P() function
487 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF32) { in XLA_TEST_P() function
491 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, DISABLED_TestF64) { in XLA_TEST_P() function
1196 XLA_TEST_P(EinsumTest, SimpleEinsumTest) { in XLA_TEST_P() function
1285 XLA_TEST_P(BatchDotTest, BroadcastingBatchDotTest) { in XLA_TEST_P() function
Dreduce_hlo_test.cc76 XLA_TEST_P(ReduceWithLayoutTest, Reduce) { in XLA_TEST_P() function
Dconv_depthwise_test.cc86 XLA_TEST_P(DepthwiseConvolution2DTest, DoIt) { in XLA_TEST_P() function
Dscalar_computations_test.cc288 XLA_TEST_P(DivS32Test, DivideTwoScalarsS32) { in XLA_TEST_P() function
297 XLA_TEST_P(DivS32Test, RemainderTwoScalarsS32) { in XLA_TEST_P() function
306 XLA_TEST_P(DivS32Test, DivideTwoScalarsNonConstS32) { in XLA_TEST_P() function
321 XLA_TEST_P(DivS32Test, RemainderTwoScalarsNonConstDivisorS32) { in XLA_TEST_P() function
Dtest_macros.h166 #define XLA_TEST_P(test_case_name, test_name) \ macro
Dcholesky_test.cc183 XLA_TEST_P(RandomCholeskyTest, Random) { in XLA_TEST_P() function
Dconv_depthwise_backprop_filter_test.cc154 XLA_TEST_P(BatchGroupedConvolution2DTest, DoIt) { in XLA_TEST_P() function
Dconditional_test.cc192 XLA_TEST_P(CaseOpTest, Parameters0) { in XLA_TEST_P() function
235 XLA_TEST_P(CaseOpTest, Parameters1) { in XLA_TEST_P() function
410 XLA_TEST_P(CaseOpTest, Parameters2Array) { in XLA_TEST_P() function
Dgrouped_convolution_test.cc224 XLA_TEST_P(GroupedConvolution2DTest, DoIt) { in XLA_TEST_P() function
Dbroadcast_simple_test.cc326 XLA_TEST_P(BroadcastR3ImplicitTest, Doit) { in XLA_TEST_P() function
570 XLA_TEST_P(BroadcastR2ImplicitTest, Doit) { in XLA_TEST_P() function
Dprng_test.cc90 XLA_TEST_P(ScalarBF16Test, in XLA_TEST_P() function
Dtriangular_solve_test.cc453 XLA_TEST_P(TriangularSolveParametricTest, Random) { in XLA_TEST_P() function
Dselect_and_scatter_test.cc72 XLA_TEST_P(SelectAndScatterTest, ParamTest) { in XLA_TEST_P() function
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtridiagonal_test.cc38 XLA_TEST_P(TridiagonalTest, Solves) { in XLA_TEST_P() function

12