Home
last modified time | relevance | path

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

/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 …]
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
262 XLA_TEST_P(SliceR1LargeTest, DoIt_F32) { Run<float>(GetParam()); } in XLA_TEST_P() function
264 XLA_TEST_P(SliceR1LargeTest, DoIt_F64) { Run<double>(GetParam()); } in XLA_TEST_P() function
266 XLA_TEST_P(SliceR1LargeTest, DoIt_U32) { Run<uint32>(GetParam()); } in XLA_TEST_P() function
268 XLA_TEST_P(SliceR1LargeTest, DoIt_S32) { Run<int32>(GetParam()); } in XLA_TEST_P() function
[all …]
Dexhaustive_op_test.cc480 XLA_TEST_P(ExhaustiveOpTest, Log) { in XLA_TEST_P() function
489 XLA_TEST_P(ExhaustiveOpTest, Log1p) { in XLA_TEST_P() function
498 XLA_TEST_P(ExhaustiveOpTest, Exp) { in XLA_TEST_P() function
516 XLA_TEST_P(ExhaustiveOpTest, Expm1) { in XLA_TEST_P() function
538 XLA_TEST_P(ExhaustiveOpTest, PowOneHalf) { in XLA_TEST_P() function
543 XLA_TEST_P(ExhaustiveOpTest, Rsqrt) { in XLA_TEST_P() function
548 XLA_TEST_P(ExhaustiveOpTest, Sqrt) { in XLA_TEST_P() function
558 XLA_TEST_P(ExhaustiveOpTest, Tanh) { Run(Tanh, std::tanh); } in XLA_TEST_P() function
560 XLA_TEST_P(ExhaustiveOpTest, Erf) { Run(Erf, std::erf); } in XLA_TEST_P() function
561 XLA_TEST_P(ExhaustiveOpTest, Erfc) { Run(Erfc, std::erfc); } 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 …]
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
Dbatch_normalization_test.cc103 XLA_TEST_P(BatchNormalizationTest, SubtractInZ) { in XLA_TEST_P() function
119 XLA_TEST_P(BatchNormalizationTest, SquareTesseractElementwise) { in XLA_TEST_P() function
136 XLA_TEST_P(BatchNormalizationTest, SumToZ) { in XLA_TEST_P() function
147 XLA_TEST_P(BatchNormalizationTest, SquareAndReduce) { in XLA_TEST_P() function
161 XLA_TEST_P(BatchNormalizationTest, VarianceToStddev) { in XLA_TEST_P() function
172 XLA_TEST_P(BatchNormalizationTest, SpecComparisonForward) { in XLA_TEST_P() function
231 XLA_TEST_P(BatchNormalizationTest, BasicTraining) { in XLA_TEST_P() function
254 XLA_TEST_P(BatchNormalizationTest, BasicTrainingOnDimension2) { in XLA_TEST_P() function
278 XLA_TEST_P(BatchNormalizationTest, TrainingWithFeatureOnLowDimension) { in XLA_TEST_P() function
309 XLA_TEST_P(BatchNormalizationTest, LargeEpsilonTest) { in XLA_TEST_P() function
[all …]
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
Dreduce_window_test.cc155 XLA_TEST_P(ReduceWindowTest, ZeroElementSmall) { in XLA_TEST_P() function
302 XLA_TEST_P(ReduceWindowTest, NonstandardReduceFunction) { in XLA_TEST_P() function
358 XLA_TEST_P(ReduceWindowTest, R6AddMultipleStrides) { in XLA_TEST_P() function
378 XLA_TEST_P(ReduceWindowTest, R6Add) { in XLA_TEST_P() function
397 XLA_TEST_P(ReduceWindowTest, R4SecondMinorStride) { in XLA_TEST_P() function
418 XLA_TEST_P(ReduceWindowTest, R4SecondMinorUnitStride) { in XLA_TEST_P() function
439 XLA_TEST_P(ReduceWindowTest, R4SecondMinorWin) { in XLA_TEST_P() function
481 XLA_TEST_P(ReduceWindowTest, Add24In1152_NoOverlap) { in XLA_TEST_P() function
492 XLA_TEST_P(ReduceWindowTest, Add128In128Stride128) { in XLA_TEST_P() function
509 XLA_TEST_P(ReduceWindowTest, Add128In128) { in XLA_TEST_P() function
[all …]
Ddot_operation_test.cc384 XLA_TEST_P(ParametricDotTest, TestF16) { TestImpl<Eigen::half>(); } in XLA_TEST_P() function
386 XLA_TEST_P(ParametricDotTest, TestF32) { TestImpl<float>(); } in XLA_TEST_P() function
387 XLA_TEST_P(ParametricDotTest, TestF64) { TestImpl<double>(); } in XLA_TEST_P() function
449 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF16) { in XLA_TEST_P() function
453 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF32) { in XLA_TEST_P() function
456 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF64) { in XLA_TEST_P() function
1165 XLA_TEST_P(EinsumTest, SimpleEinsumTest) { in XLA_TEST_P() function
Dreduce_hlo_test.cc78 XLA_TEST_P(ReduceWithLayoutTest, DISABLED_ON_GPU(Reduce)) { in XLA_TEST_P() function
Dscalar_computations_test.cc278 XLA_TEST_P(DivS32Test, DivideTwoScalarsS32) { in XLA_TEST_P() function
287 XLA_TEST_P(DivS32Test, RemainderTwoScalarsS32) { in XLA_TEST_P() function
296 XLA_TEST_P(DivS32Test, DivideTwoScalarsNonConstS32) { in XLA_TEST_P() function
311 XLA_TEST_P(DivS32Test, RemainderTwoScalarsNonConstDivisorS32) { in XLA_TEST_P() function
Dcholesky_test.cc159 XLA_TEST_P(RandomCholeskyTest, Random) { in XLA_TEST_P() function
Dtest_macros.h151 #define XLA_TEST_P(test_case_name, test_name) \ macro
Dconv_depthwise_backprop_filter_test.cc155 XLA_TEST_P(BatchGroupedConvolution2DTest, DoIt) { in XLA_TEST_P() function
Dreduce_precision_test.cc207 XLA_TEST_P(ReducePrecisionAccuracyTest, ReducePrecisionF32) { 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
Dconv_depthwise_test.cc212 XLA_TEST_P(DepthwiseConvolution2DTest, DoIt) { 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
Dselect_and_scatter_test.cc72 XLA_TEST_P(SelectAndScatterTest, ParamTest) { in XLA_TEST_P() function
Dtriangular_solve_test.cc445 XLA_TEST_P(TriangularSolveParametricTest, Random) { in XLA_TEST_P() function
Dconvolution_test.cc1676 XLA_TEST_P(ConvolveWithAndWithoutCanonicalization, in XLA_TEST_P() function
1791 XLA_TEST_P(Convolve1D1WindowTestFloat, Convolve1D1Window) { TestImpl<float>(); } in XLA_TEST_P() function
1826 XLA_TEST_P(Convolve1D1WindowTestHalf, Convolve1D1Window) { in XLA_TEST_P() function
Dreduce_test.cc811 XLA_TEST_P(ReduceR3ToR2Test, ReduceR3ToR2) { in XLA_TEST_P() function