| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reshape_test.cc | 53 XLA_TEST_P(ReshapeTest, CollapseTrivial1x1) { in XLA_TEST_P() function
69 XLA_TEST_P(ReshapeTest, CollapseTrivialR1EmptyDims) { in XLA_TEST_P() function
83 XLA_TEST_P(ReshapeTest, CollapseTrivialR1OnlyDim) { in XLA_TEST_P() function
98 XLA_TEST_P(ReshapeTest, SingleElementArrayToScalar) { in XLA_TEST_P() function
116 XLA_TEST_P(ReshapeTest, ScalarToSingleElementArray) { in XLA_TEST_P() function
132 XLA_TEST_P(ReshapeTest, Trivial0x3) { in XLA_TEST_P() function
146 XLA_TEST_P(ReshapeTest, Trivial0x3WithParameter) { in XLA_TEST_P() function
161 XLA_TEST_P(ReshapeTest, Trivial3x0) { in XLA_TEST_P() function
176 XLA_TEST_P(ReshapeTest, Trivial1x3) { in XLA_TEST_P() function
190 XLA_TEST_P(ReshapeTest, Trivial3x1) { in XLA_TEST_P() function
[all …]
|
| D | slice_test.cc | 252 XLA_TEST_P(SliceR1Test, DoIt_F32) { Run<float>(GetParam()); } in XLA_TEST_P() function
254 XLA_TEST_P(SliceR1Test, DoIt_F64) { Run<double>(GetParam()); } in XLA_TEST_P() function
256 XLA_TEST_P(SliceR1Test, DoIt_U32) { Run<uint32_t>(GetParam()); } in XLA_TEST_P() function
258 XLA_TEST_P(SliceR1Test, DoIt_S32) { Run<int32_t>(GetParam()); } in XLA_TEST_P() function
260 XLA_TEST_P(SliceR1Test, DoIt_U64) { Run<uint64_t>(GetParam()); } in XLA_TEST_P() function
262 XLA_TEST_P(SliceR1Test, DoIt_S64) { Run<int64_t>(GetParam()); } in XLA_TEST_P() function
266 XLA_TEST_P(SliceR1LargeTest, DISABLED_ON_GPU(DoIt_F32)) { in XLA_TEST_P() function
270 XLA_TEST_P(SliceR1LargeTest, DISABLED_ON_GPU(DoIt_F64)) { in XLA_TEST_P() function
274 XLA_TEST_P(SliceR1LargeTest, DISABLED_ON_GPU(DoIt_U32)) { in XLA_TEST_P() function
278 XLA_TEST_P(SliceR1LargeTest, DISABLED_ON_GPU(DoIt_S32)) { in XLA_TEST_P() function
[all …]
|
| D | reduce_window_test.cc | 106 XLA_TEST_P(ReduceWindowTest, MismatchedRanksGivesErrorStatus) { in XLA_TEST_P() function
123 XLA_TEST_P(ReduceWindowTest, R0ReduceWindow) { in XLA_TEST_P() function
135 XLA_TEST_P(ReduceWindowTest, Min3In5Stride2) { in XLA_TEST_P() function
143 XLA_TEST_P(ReduceWindowTest, Min3In5Stride2Same) { in XLA_TEST_P() function
152 XLA_TEST_P(ReduceWindowTest, Min3In5Stride1WithSamePadding) { in XLA_TEST_P() function
162 XLA_TEST_P(ReduceWindowTest, ZeroElementSmall) { in XLA_TEST_P() function
175 XLA_TEST_P(ReduceWindowTest, NonSquareSmall) { in XLA_TEST_P() function
190 XLA_TEST_P(ReduceWindowTest, MiddleDimsSmall) { in XLA_TEST_P() function
204 XLA_TEST_P(ReduceWindowTest, Along2ndMinorDim) { in XLA_TEST_P() function
221 XLA_TEST_P(ReduceWindowTest, AmongMajor2DimsAdd) { in XLA_TEST_P() function
[all …]
|
| D | pad_test.cc | 85 XLA_TEST_P(PadTestFloat, Pad1DS0ToS0Array) { in XLA_TEST_P() function
100 XLA_TEST_P(PadTestFloat, Pad1DS0ToS5Array) { in XLA_TEST_P() function
115 XLA_TEST_P(PadTestFloat, Pad1DS3Array) { in XLA_TEST_P() function
130 XLA_TEST_P(PadTestFloat, Pad4D_2x0x3x2_FloatArray) { in XLA_TEST_P() function
225 XLA_TEST_P(PadTestFloat, Pad4DFloatArrayMinorFirstNonTrivialMinorDimensions) { in XLA_TEST_P() function
319 XLA_TEST_P(PadTestFloat, Large2DPad) { in XLA_TEST_P() function
338 XLA_TEST_P(PadTestFloat, AllTypes2DPad) { in XLA_TEST_P() function
360 XLA_TEST_P(PadTestFloat, High2DPad) { in XLA_TEST_P() function
387 XLA_TEST_P(PadTestFloat, NegativePadding2D) { in XLA_TEST_P() function
415 XLA_TEST_P(PadTestFloat, NegativeAndInteriorPadding2D) { in XLA_TEST_P() function
[all …]
|
| 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 | matrix_ops_simple_test.cc | 163 XLA_TEST_P(TestLinspaceMaxParametric, TestF16) { TestImpl<Eigen::half>(); } in XLA_TEST_P() function
165 XLA_TEST_P(TestLinspaceMaxParametric, TestF32) { TestImpl<float>(); } in XLA_TEST_P() function
239 XLA_TEST_P(MatOpsDotAddTest, Dot_Add_2x2_2x2BF16) { TestImpl<bfloat16>(); } in XLA_TEST_P() function
241 XLA_TEST_P(MatOpsDotAddTest, Dot_Add_2x2_2x2F16) { TestImpl<Eigen::half>(); } in XLA_TEST_P() function
243 XLA_TEST_P(MatOpsDotAddTest, Dot_Add_2x2_2x2F32) { TestImpl<float>(); } 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 | exhaustive_unary_test_f32_or_smaller.cc | 383 XLA_TEST_P(ExhaustiveF32UnaryTest, Acosh) { in XLA_TEST_P() function
394 XLA_TEST_P(ExhaustiveF16UnaryTest, Acosh) { Run(Acosh, std::acosh); } in XLA_TEST_P() function
398 XLA_TEST_P(ExhaustiveBF16UnaryTest, Acosh) { Run(Acosh, std::acosh); } in XLA_TEST_P() function
402 XLA_TEST_P(ExhaustiveF32UnaryTest, Asinh) { in XLA_TEST_P() function
412 XLA_TEST_P(ExhaustiveF16UnaryTest, Asinh) { Run(Asinh, std::asinh); } in XLA_TEST_P() function
416 XLA_TEST_P(ExhaustiveBF16UnaryTest, Asinh) { Run(Asinh, std::asinh); } in XLA_TEST_P() function
|
| D | half_test.cc | 50 XLA_TEST_P(UnaryOpTest, Ops) { in XLA_TEST_P() function
107 XLA_TEST_P(UnaryPredTest, Ops) { in XLA_TEST_P() function
142 XLA_TEST_P(BinaryOpTest, Ops) { in XLA_TEST_P() function
198 XLA_TEST_P(BinaryPredTest, Ops) { in XLA_TEST_P() function
|
| D | conv_depthwise_test.cc | 86 XLA_TEST_P(DepthwiseConvolution2DTest, DoIt) { in XLA_TEST_P() function
|
| D | dot_operation_test.cc | 419 XLA_TEST_P(ParametricDotTest, TestF16) { TestImpl<Eigen::half>(); } in XLA_TEST_P() function
421 XLA_TEST_P(ParametricDotTest, TestF32) { TestImpl<float>(); } in XLA_TEST_P() function
422 XLA_TEST_P(ParametricDotTest, TestF64) { TestImpl<double>(); } in XLA_TEST_P() function
423 XLA_TEST_P(ParametricDotTest, TestC64) { TestImpl<std::complex<float>>(); } in XLA_TEST_P() function
425 XLA_TEST_P(ParametricDotTest, TestC128) { TestImpl<std::complex<double>>(); } in XLA_TEST_P() function
427 XLA_TEST_P(ParametricDotTest, TestS32) { TestImpl<int32_t>(); } in XLA_TEST_P() function
491 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF16) { in XLA_TEST_P() function
495 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, TestF32) { in XLA_TEST_P() function
499 XLA_TEST_P(ParametricDotTestWithoutLayoutAssignment, DISABLED_TestF64) { in XLA_TEST_P() function
1206 XLA_TEST_P(EinsumTest, SimpleEinsumTest) { in XLA_TEST_P() function
[all …]
|
| 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 | 75 XLA_TEST_P(ReduceWithLayoutTest, Reduce) { in XLA_TEST_P() function
|
| D | test_macros.h | 96 #define XLA_TEST_P(test_case_name, test_name) TEST_P(test_case_name, test_name) macro
|
| D | convolution_test_1d.cc | 132 XLA_TEST_P(Convolve1D1WindowTestFloat, Convolve1D1Window) { TestImpl<float>(); } in XLA_TEST_P() function
168 XLA_TEST_P(Convolve1D1WindowTestHalf, Convolve1D1Window) { in XLA_TEST_P() function
|
| D | conv_depthwise_backprop_filter_test.cc | 170 XLA_TEST_P(BatchGroupedConvolution2DTest, DoIt) { in XLA_TEST_P() function
|
| D | grouped_convolution_test.cc | 230 XLA_TEST_P(GroupedConvolution2DTest, DoIt) { in XLA_TEST_P() function
|
| D | scalar_computations_test.cc | 287 XLA_TEST_P(DivS32Test, DivideTwoScalarsS32) { in XLA_TEST_P() function
296 XLA_TEST_P(DivS32Test, RemainderTwoScalarsS32) { in XLA_TEST_P() function
305 XLA_TEST_P(DivS32Test, DivideTwoScalarsNonConstS32) { in XLA_TEST_P() function
320 XLA_TEST_P(DivS32Test, RemainderTwoScalarsNonConstDivisorS32) { in XLA_TEST_P() function
|
| D | batch_normalization_test.cc | 512 XLA_TEST_P(BatchNormTestManySizes, RandomizedTrainingTests) { in XLA_TEST_P() function
613 XLA_TEST_P(BatchNormTestManySizes, RandomizedInferencingTests) { in XLA_TEST_P() function
721 XLA_TEST_P(BatchNormTestManySizes, RandomizedGradTests) { in XLA_TEST_P() function
|
| D | conditional_test.cc | 194 XLA_TEST_P(CaseOpTest, Parameters0) { in XLA_TEST_P() function
237 XLA_TEST_P(CaseOpTest, Parameters1) { in XLA_TEST_P() function
411 XLA_TEST_P(CaseOpTest, Parameters2Array) { in XLA_TEST_P() function
|
| D | broadcast_simple_test.cc | 324 XLA_TEST_P(BroadcastR3ImplicitTest, Doit) { in XLA_TEST_P() function
568 XLA_TEST_P(BroadcastR2ImplicitTest, Doit) { in XLA_TEST_P() function
|
| D | prng_test.cc | 89 XLA_TEST_P(ScalarBF16Test, in XLA_TEST_P() function
|
| D | select_and_scatter_test.cc | 74 XLA_TEST_P(SelectAndScatterTest, ParamTest) { in XLA_TEST_P() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal_test.cc | 38 XLA_TEST_P(TridiagonalTest, SimpleTridiagonalMatMulOk) { in XLA_TEST_P() function
80 XLA_TEST_P(TridiagonalTest, TridiagonalMatMulWrongShape) { in XLA_TEST_P() function
108 XLA_TEST_P(TridiagonalTest, Solves) { in XLA_TEST_P() function
|
| D | self_adjoint_eig_test.cc | 275 XLA_TEST_P(RandomEighTest, Random) { in XLA_TEST_P() function
|