| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reshape_test.cc | 54 XLA_TEST_P(ReshapeTest, CollapseTrivial1x1) { function in namespace:xla::__anon25777
69 XLA_TEST_P(ReshapeTest, CollapseTrivialR1EmptyDims) { function in namespace:xla::__anon25777
82 XLA_TEST_P(ReshapeTest, CollapseTrivialR1OnlyDim) { function in namespace:xla::__anon25777
96 XLA_TEST_P(ReshapeTest, SingleElementArrayToScalar) { function in namespace:xla::__anon25777
113 XLA_TEST_P(ReshapeTest, ScalarToSingleElementArray) { function in namespace:xla::__anon25777
132 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(Trivial0x3)) { function in namespace:xla::__anon25777
148 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(Trivial0x3WithParameter)) { function in namespace:xla::__anon25777
165 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(Trivial3x0)) { function in namespace:xla::__anon25777
179 XLA_TEST_P(ReshapeTest, Trivial1x3) { function in namespace:xla::__anon25777
192 XLA_TEST_P(ReshapeTes function in namespace:xla::__anon25777
209 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(R1ToR2_0_To_2x0)) { function in namespace:xla::__anon25777
223 XLA_TEST_P(ReshapeTest, R1ToR2_6_To_2x3) { function in namespace:xla::__anon25777
243 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(Reshape0x2To2x0)) { function in namespace:xla::__anon25777
257 XLA_TEST_P(ReshapeTest, ReshapeRowToCol) { function in namespace:xla::__anon25777
274 XLA_TEST_P(ReshapeTest, TransposeAsReshape) { function in namespace:xla::__anon25777
295 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(Transpose0x4)) { function in namespace:xla::__anon25777
308 XLA_TEST_P(ReshapeTest, Transpose4x3) { function in namespace:xla::__anon25777
329 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(ReshapeSplitNoShuffleZeroElements)) { function in namespace:xla::__anon25777
345 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(ReshapeR4ToR2ZeroElements)) { function in namespace:xla::__anon25777
360 XLA_TEST_P(ReshapeTest, ReshapeSplitNoShuffle) { function in namespace:xla::__anon25777
380 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(ReshapeSplitAndShuffleZeroElements)) { function in namespace:xla::__anon25777
395 XLA_TEST_P(ReshapeTest, ReshapeSplitAndShuffle) { function in namespace:xla::__anon25777
422 XLA_TEST_P(ReshapeTest, DocR3_R1_Collapse_012) { function in namespace:xla::__anon25777
437 XLA_TEST_P(ReshapeTest, DocR3_R2_Collapse_012_Refine_83) { function in namespace:xla::__anon25777
457 XLA_TEST_P(ReshapeTest, DocR3_R1_Collapse_120) { function in namespace:xla::__anon25777
472 XLA_TEST_P(ReshapeTest, DocR3_R2_Collapse_120_Refine_83) { function in namespace:xla::__anon25777
492 XLA_TEST_P(ReshapeTest, DocR3_R3_Collapse_120_Refine_262) { function in namespace:xla::__anon25777
522 XLA_TEST_P(ReshapeTest, FullyConnectedCollapse) { function in namespace:xla::__anon25777
541 XLA_TEST_P(ReshapeTest, FullyConnectedCollapseDesugared) { function in namespace:xla::__anon25777
566 XLA_TEST_P(ReshapeTest, ToScalar) { function in namespace:xla::__anon25777
587 XLA_TEST_P(ReshapeTest, BadDimensions) { function in namespace:xla::__anon25777
599 XLA_TEST_P(ReshapeTest, BadNewSizes) { function in namespace:xla::__anon25777
610 XLA_TEST_P(ReshapeTest, R4Dim0MinorLayoutToR2Dim0MajorLayout) { function in namespace:xla::__anon25777
665 XLA_TEST_P(ReshapeTest, R2ToR4_3x8_To_3x2x1x4) { function in namespace:xla::__anon25777
692 XLA_TEST_P(ReshapeTest, R2ToR4_3x8_To_3x2x1x4_Dimensions_10) { function in namespace:xla::__anon25777
718 XLA_TEST_P(ReshapeTest, R4ToR2_2x1x1x1_To_2x1) { function in namespace:xla::__anon25777
740 XLA_TEST_P(ReshapeTest, R4ToR2_2x1x4x1_To_4x2) { function in namespace:xla::__anon25777
763 XLA_TEST_P(ReshapeTest, R4ToR2_5x10x2x3_To_5x60_Dimensions_0213) { function in namespace:xla::__anon25777
790 XLA_TEST_P(ReshapeTest, NoopReshape) { function in namespace:xla::__anon25777
828 XLA_TEST_P(ReshapeTest, R4ToR4Reshape_Trivial) { function in namespace:xla::__anon25777
843 XLA_TEST_P(ReshapeTest, R4ToR4Reshape) { function in namespace:xla::__anon25777
870 XLA_TEST_P(ReshapeTest, R4TwoMinorTransposeSimple) { function in namespace:xla::__anon25777
899 XLA_TEST_P(ReshapeTest, R4TwoMinorTransposeMajorFirstEffectiveR2) { function in namespace:xla::__anon25777
928 XLA_TEST_P(ReshapeTest, R4TwoMinorTransposeMajorFirstMinorEffectiveR1) { function in namespace:xla::__anon25777
957 XLA_TEST_P(ReshapeTest, R4TwoMinorTransposeMajorFirstMinorEffectiveR1InR2) { function in namespace:xla::__anon25777
987 XLA_TEST_P(ReshapeTest, R4TwoMinorTransposeTrivialR2) { function in namespace:xla::__anon25777
[all...] |
| H A D | exhaustive_f32_elementwise_op_test.cc | 70 XLA_TEST_P(ExhaustiveF32ElementwiseOpTest, LogF32) { function in namespace:xla::__anon25746
87 XLA_TEST_P(ExhaustiveF32ElementwiseOpTest, ExpF32) { function in namespace:xla::__anon25746
105 XLA_TEST_P(ExhaustiveF32ElementwiseOpTest, TanhF32) { function in namespace:xla::__anon25746
|
| H A D | slice_test.cc | 221 XLA_TEST_P(SliceR1Test, DoIt_F32) { Run<float>(GetParam()); } function in namespace:xla::__anon25787
223 XLA_TEST_P(SliceR1Test, DoIt_F64) { Run<double>(GetParam()); } function in namespace:xla::__anon25787
225 XLA_TEST_P(SliceR1Test, DoIt_U32) { Run<uint32>(GetParam()); } function in namespace:xla::__anon25787
227 XLA_TEST_P(SliceR1Test, DoIt_S32) { Run<int32>(GetParam()); } function in namespace:xla::__anon25787
229 XLA_TEST_P(SliceR1Test, DoIt_U64) { Run<uint64>(GetParam()); } function in namespace:xla::__anon25787
231 XLA_TEST_P(SliceR1Test, DoIt_S64) { Run<int64>(GetParam()); } function in namespace:xla::__anon25787
308 XLA_TEST_P(SliceR2Test, DoIt) { function in namespace:xla::__anon25787
415 XLA_TEST_P(SliceR4Test, DoIt) { Run(GetParam()); } function in namespace:xla::__anon25787
|
| H A D | pad_test.cc | 87 XLA_TEST_P(PadTestFloat, Pad1DS0ToS0Array) { function in namespace:xla::__anon25765
102 XLA_TEST_P(PadTestFloat, Pad1DS0ToS5Array) { function in namespace:xla::__anon25765
117 XLA_TEST_P(PadTestFloat, Pad1DS3Array) { function in namespace:xla::__anon25765
132 XLA_TEST_P(PadTestFloat, Pad4D_2x0x3x2_FloatArray) { function in namespace:xla::__anon25765
225 XLA_TEST_P(PadTestFloat, Pad4DFloatArrayMinorFirstNonTrivialMinorDimensions) { function in namespace:xla::__anon25765
319 XLA_TEST_P(PadTestFloat, Large2DPad) { function in namespace:xla::__anon25765
339 XLA_TEST_P(PadTestFloat, AllTypes2DPad) { function in namespace:xla::__anon25765
362 XLA_TEST_P(PadTestFloat, High2DPad) { function in namespace:xla::__anon25765
389 XLA_TEST_P(PadTestFloat, NegativePadding2D) { function in namespace:xla::__anon25765
417 XLA_TEST_P(PadTestFloa function in namespace:xla::__anon25765
446 XLA_TEST_P(PadTestFloat, ReducePad) { function in namespace:xla::__anon25765
[all...] |
| H A D | batch_normalization_test.cc | 93 XLA_TEST_P(BatchNormalizationTest, SubtractInZ) { function in namespace:xla::__anon25717
109 XLA_TEST_P(BatchNormalizationTest, SquareTesseractElementwise) { function in namespace:xla::__anon25717
126 XLA_TEST_P(BatchNormalizationTest, SumToZ) { function in namespace:xla::__anon25717
138 XLA_TEST_P(BatchNormalizationTest, SquareAndReduce) { function in namespace:xla::__anon25717
153 XLA_TEST_P(BatchNormalizationTest, VarianceToStddev) { function in namespace:xla::__anon25717
164 XLA_TEST_P(BatchNormalizationTest, SpecComparisonForward) { function in namespace:xla::__anon25717
220 XLA_TEST_P(BatchNormalizationTest, BasicTraining) { function in namespace:xla::__anon25717
244 XLA_TEST_P(BatchNormalizationTest, BasicTrainingOnSublane) { function in namespace:xla::__anon25717
268 XLA_TEST_P(BatchNormalizationTest, TrainingWithFeatureOnLowDimension) { function in namespace:xla::__anon25717
300 XLA_TEST_P(BatchNormalizationTes function in namespace:xla::__anon25717
333 XLA_TEST_P(BatchNormalizationTest, BatchNormGradBasic) { function in namespace:xla::__anon25717
440 XLA_TEST_P(BatchNormTestManySizes, RandomizedTrainingTests) { function in namespace:xla::__anon25717
540 XLA_TEST_P(BatchNormTestManySizes, RandomizedInferencingTests) { function in namespace:xla::__anon25717
648 XLA_TEST_P(BatchNormTestManySizes, RandomizedGradTests) { function in namespace:xla::__anon25717
[all...] |
| H A D | half_test.cc | 52 XLA_TEST_P(UnaryOpTest, Ops) { function in namespace:xla::__anon25750
116 XLA_TEST_P(UnaryPredTest, Ops) { function in namespace:xla::__anon25750
153 XLA_TEST_P(BinaryOpTest, Ops) { function in namespace:xla::__anon25750
214 XLA_TEST_P(BinaryPredTest, Ops) { function in namespace:xla::__anon25750
|
| H A D | reduce_hlo_test.cc | 81 XLA_TEST_P(ReduceWithLayoutTest, DISABLED_ON_GPU(Reduce)) { function in namespace:xla::__anon25771
|
| H A D | reduce_window_test.cc | 151 XLA_TEST_P(ReduceWindowTest, ZeroElementSmall) { function in namespace:xla::__anon25774
256 XLA_TEST_P(ReduceWindowTest, NonstandardReduceFunction) { function in namespace:xla::__anon25774
313 XLA_TEST_P(ReduceWindowTest, R6AddMultipleStrides) { function in namespace:xla::__anon25774
342 XLA_TEST_P(ReduceWindowTest, R6Add) { function in namespace:xla::__anon25774
361 XLA_TEST_P(ReduceWindowTest, R4SecondMinorStride) { function in namespace:xla::__anon25774
383 XLA_TEST_P(ReduceWindowTest, R4SecondMinorUnitStride) { function in namespace:xla::__anon25774
405 XLA_TEST_P(ReduceWindowTest, R4SecondMinorWin) { function in namespace:xla::__anon25774
448 XLA_TEST_P(ReduceWindowTest, Add24In1152_NoOverlap) { function in namespace:xla::__anon25774
459 XLA_TEST_P(ReduceWindowTest, Add128In128Stride128) { function in namespace:xla::__anon25774
476 XLA_TEST_P(ReduceWindowTes function in namespace:xla::__anon25774
784 XLA_TEST_P(R4ReduceWindowLargeTest, DISABLED_ON_INTERPRETER(DoIt)) { DoIt(); } function in namespace:xla::__anon25774
822 XLA_TEST_P(R4ReduceWindowAnyDimsTest, function in namespace:xla::__anon25774
1070 XLA_TEST_P(R2ReduceWindowFailingCpuGpuBf16Test, function in namespace:xla::__anon25774
[all...] |
| H A D | test_macros.h | 162 #define XLA_TEST_P(test_case_name, test_name) \ macro
|
| H A D | reduce_precision_test.cc | 207 XLA_TEST_P(ReducePrecisionAccuracyTest, ReducePrecisionF32) { function in namespace:xla::__anon25772
|
| H A D | scalar_computations_test.cc | 283 XLA_TEST_P(DivS32Test, DivideTwoScalarsS32) { function in namespace:xla::__anon25783
292 XLA_TEST_P(DivS32Test, RemainderTwoScalarsS32) { function in namespace:xla::__anon25783
301 XLA_TEST_P(DivS32Test, DivideTwoScalarsNonConstS32) { function in namespace:xla::__anon25783
316 XLA_TEST_P(DivS32Test, RemainderTwoScalarsNonConstDivisorS32) { function in namespace:xla::__anon25783
|
| H A D | convolution_test.cc | 549 XLA_TEST_P(ConvolveWithAndWithoutCanonicalization, function in namespace:xla::__anon25734
667 XLA_TEST_P(Convolve1D1WindowTestFloat, Convolve1D1Window) { TestImpl<float>(); } function in namespace:xla::__anon25734
703 XLA_TEST_P(Convolve1D1WindowTestHalf, function in namespace:xla::__anon25734
|
| H A D | broadcast_simple_test.cc | 248 XLA_TEST_P(BroadcastR3ImplicitTest, Doit) { function in namespace:xla::__anon25721
491 XLA_TEST_P(BroadcastR2ImplicitTest, Doit) { function in namespace:xla::__anon25721
|
| H A D | dot_operation_test.cc | 230 XLA_TEST_P(ParametricDotTest, TestF32) { function in namespace:xla::__anon25742
|
| H A D | select_and_scatter_test.cc | 72 XLA_TEST_P(SelectAndScatterTest, ParamTest) { function in namespace:xla::__anon25784
|
| H A D | reduce_test.cc | 810 XLA_TEST_P(ReduceR3ToR2Test, ReduceR3ToR2) { function in namespace:xla::__anon25773
|