| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reshape_test.cc | 54 XLA_TEST_P(ReshapeTest, CollapseTrivial1x1) { in XLA_TEST_P() function
70 XLA_TEST_P(ReshapeTest, CollapseTrivialR1EmptyDims) { in XLA_TEST_P() function
84 XLA_TEST_P(ReshapeTest, CollapseTrivialR1OnlyDim) { in XLA_TEST_P() function
99 XLA_TEST_P(ReshapeTest, SingleElementArrayToScalar) { in XLA_TEST_P() function
117 XLA_TEST_P(ReshapeTest, ScalarToSingleElementArray) { in XLA_TEST_P() function
133 XLA_TEST_P(ReshapeTest, Trivial0x3) { in XLA_TEST_P() function
147 XLA_TEST_P(ReshapeTest, Trivial0x3WithParameter) { in XLA_TEST_P() function
162 XLA_TEST_P(ReshapeTest, Trivial3x0) { in XLA_TEST_P() function
177 XLA_TEST_P(ReshapeTest, Trivial1x3) { in XLA_TEST_P() function
191 XLA_TEST_P(ReshapeTest, Trivial3x1) { in XLA_TEST_P() function
[all …]
|
| D | slice_test.cc | 250 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 …]
|
| D | reduce_window_test.cc | 108 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 …]
|
| D | pad_test.cc | 87 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 …]
|
| D | batch_normalization_test.cc | 104 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 …]
|
| D | matrix_ops_simple_test.cc | 165 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
|
| D | exhaustive_unary_test_f32_or_smaller.cc | 229 XLA_TEST_P(ExhaustiveF32UnaryTest, test_name) \
234 XLA_TEST_P(ExhaustiveF16UnaryTest, test_name) \
242 XLA_TEST_P(ExhaustiveBF16UnaryTest, test_name) \
370 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
385 XLA_TEST_P(ExhaustiveBF16UnaryTest, Acosh) { Run(Acosh, std::acosh); } in XLA_TEST_P() function
389 XLA_TEST_P(ExhaustiveF32UnaryTest, Asinh) { in XLA_TEST_P() function
399 XLA_TEST_P(ExhaustiveF16UnaryTest, Asinh) { Run(Asinh, std::asinh); } in XLA_TEST_P() function
403 XLA_TEST_P(ExhaustiveBF16UnaryTest, Asinh) { Run(Asinh, std::asinh); } in XLA_TEST_P() function
|
| D | reduce_precision_test.cc | 465 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
|
| D | iota_test.cc | 62 XLA_TEST_P(IotaR1Test, DoIt) { in XLA_TEST_P() function
91 XLA_TEST_P(IotaR2Test, DoIt) { in XLA_TEST_P() function
123 XLA_TEST_P(IotaR3Test, DoIt) { in XLA_TEST_P() function
|
| D | half_test.cc | 50 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
|
| D | exhaustive_binary_16_bit_test.cc | 76 XLA_TEST_P(ExhaustiveF16BinaryTest, test_name) \
85 XLA_TEST_P(ExhaustiveBF16BinaryTest, test_name) \
|
| D | exhaustive_unary_test_complex.cc | 95 XLA_TEST_P(ExhaustiveC64UnaryTest, test_name) \
178 XLA_TEST_P(ExhaustiveC128UnaryTest, test_name) \
|
| D | dot_operation_test.cc | 420 XLA_TEST_P(ParametricDotTest, TestF16) { TestImpl<Eigen::half>(); } in XLA_TEST_P() function
422 XLA_TEST_P(ParametricDotTest, TestF32) { TestImpl<float>(); } in XLA_TEST_P() function
423 XLA_TEST_P(ParametricDotTest, TestF64) { TestImpl<double>(); } in XLA_TEST_P() function
424 XLA_TEST_P(ParametricDotTest, TestC64) { TestImpl<std::complex<float>>(); } in XLA_TEST_P() function
426 XLA_TEST_P(ParametricDotTest, TestC128) { TestImpl<std::complex<double>>(); } in XLA_TEST_P() function
428 XLA_TEST_P(ParametricDotTest, TestS32) { TestImpl<int32>(); } in XLA_TEST_P() function
492 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF16) { in XLA_TEST_P() function
496 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF32) { in XLA_TEST_P() function
500 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, DISABLED_TestF64) { in XLA_TEST_P() function
1205 XLA_TEST_P(EinsumTest, SimpleEinsumTest) { in XLA_TEST_P() function
[all …]
|
| D | test_macros.h | 90 #define XLA_TEST_P(test_case_name, test_name) TEST_P(test_case_name, test_name) macro
|
| D | exhaustive_binary_test_f32_f64.cc | 76 XLA_TEST_P(ExhaustiveF32BinaryTest, test_name) \
167 XLA_TEST_P(ExhaustiveF64BinaryTest, test_name) \
|
| D | cholesky_test.cc | 222 XLA_TEST_P(RandomCholeskyTest, Real) { in XLA_TEST_P() function
261 XLA_TEST_P(RandomCholeskyTest, Complex) { in XLA_TEST_P() function
|
| D | reduce_hlo_test.cc | 76 XLA_TEST_P(ReduceWithLayoutTest, Reduce) { in XLA_TEST_P() function
|
| D | conv_depthwise_test.cc | 86 XLA_TEST_P(DepthwiseConvolution2DTest, DoIt) { in XLA_TEST_P() function
|
| D | convolution_test_1d.cc | 136 XLA_TEST_P(Convolve1D1WindowTestFloat, Convolve1D1Window) { TestImpl<float>(); } in XLA_TEST_P() function
172 XLA_TEST_P(Convolve1D1WindowTestHalf, Convolve1D1Window) { in XLA_TEST_P() function
|
| D | exhaustive_unary_test_f64.cc | 58 XLA_TEST_P(ExhaustiveF64UnaryTest, test_name) \
|
| D | scalar_computations_test.cc | 288 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
|
| D | conv_depthwise_backprop_filter_test.cc | 169 XLA_TEST_P(BatchGroupedConvolution2DTest, DoIt) { in XLA_TEST_P() function
|
| D | conditional_test.cc | 192 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
|
| D | grouped_convolution_test.cc | 229 XLA_TEST_P(GroupedConvolution2DTest, DoIt) { in XLA_TEST_P() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal_test.cc | 38 XLA_TEST_P(TridiagonalTest, Solves) { in XLA_TEST_P() function
|