| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | scalar_computations_test.cc | 70 XLA_TEST_F(ScalarComputationsTest, ReturnScalarF32) { in XLA_TEST_F() function
77 XLA_TEST_F(ScalarComputationsTest, NegateScalarF32) { in XLA_TEST_F() function
84 XLA_TEST_F(ScalarComputationsTest, NegateScalarS32) { in XLA_TEST_F() function
91 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsF32) { in XLA_TEST_F() function
98 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsS32) { in XLA_TEST_F() function
105 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsU32) { in XLA_TEST_F() function
112 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsU8) { in XLA_TEST_F() function
119 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsU64) { in XLA_TEST_F() function
128 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsS64) { in XLA_TEST_F() function
137 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsF64) { in XLA_TEST_F() function
[all …]
|
| D | dynamic_ops_test.cc | 217 XLA_TEST_F(DynamicSliceTest, Int32R1BF16) { TestR1<int32, bfloat16>(); } in XLA_TEST_F() function
218 XLA_TEST_F(DynamicSliceTest, Int32R1) { TestR1<int32, int32>(); } in XLA_TEST_F() function
219 XLA_TEST_F(DynamicSliceTest, Int32R1OOB) { TestR1OOB<int32, int32>(); } in XLA_TEST_F() function
220 XLA_TEST_F(DynamicSliceTest, Int64R1) { TestR1<int64, float>(); } in XLA_TEST_F() function
221 XLA_TEST_F(DynamicSliceTest, UInt64R1) { TestR1<uint64, float>(); } in XLA_TEST_F() function
222 XLA_TEST_F(DynamicSliceTest, UInt32R1OOB) { in XLA_TEST_F() function
226 XLA_TEST_F(DynamicSliceTest, Int32R2BF16) { TestR2<int32, bfloat16>(); } in XLA_TEST_F() function
227 XLA_TEST_F(DynamicSliceTest, Int32R2) { TestR2<int32, int32>(); } in XLA_TEST_F() function
228 XLA_TEST_F(DynamicSliceTest, Int32R2OOB) { TestR2OOB<int32, int32>(); } in XLA_TEST_F() function
229 XLA_TEST_F(DynamicSliceTest, Int64R2) { TestR2<int64, float>(); } in XLA_TEST_F() function
[all …]
|
| D | reduce_test.cc | 357 XLA_TEST_F(ReduceTest, ReduceR1_0_F32_To_R0) { RunR1ToR0Test(0); } in XLA_TEST_F() function
358 XLA_TEST_F(ReduceTest, ReduceR1_1_F32_To_R0) { RunR1ToR0Test(1); } in XLA_TEST_F() function
359 XLA_TEST_F(ReduceTest, ReduceR1_2_F32_To_R0) { RunR1ToR0Test(2); } in XLA_TEST_F() function
360 XLA_TEST_F(ReduceTest, ReduceR1_16_F32_To_R0) { RunR1ToR0Test(16); } in XLA_TEST_F() function
361 XLA_TEST_F(ReduceTest, ReduceR1_128_F32_To_R0) { RunR1ToR0Test(128); } in XLA_TEST_F() function
362 XLA_TEST_F(ReduceTest, ReduceR1_129_F32_To_R0) { RunR1ToR0Test(129); } in XLA_TEST_F() function
363 XLA_TEST_F(ReduceTest, ReduceR1_240_F32_To_R0) { RunR1ToR0Test(240); } in XLA_TEST_F() function
364 XLA_TEST_F(ReduceTest, ReduceR1_256_F32_To_R0) { RunR1ToR0Test(256); } in XLA_TEST_F() function
365 XLA_TEST_F(ReduceTest, ReduceR1_1024_F32_To_R0) { RunR1ToR0Test(1024); } in XLA_TEST_F() function
366 XLA_TEST_F(ReduceTest, ReduceR1_2048_F32_To_R0) { RunR1ToR0Test(2048); } in XLA_TEST_F() function
[all …]
|
| D | array_elementwise_ops_test.cc | 52 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantZeroElementF32) { in XLA_TEST_F() function
60 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantF32) { in XLA_TEST_F() function
69 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantS32) { in XLA_TEST_F() function
85 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantZeroElementC64) { in XLA_TEST_F() function
93 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantC64) { in XLA_TEST_F() function
104 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantS64) { in XLA_TEST_F() function
132 XLA_TEST_F(ArrayElementwiseOpTest, IsFiniteZeroElementF32s) { in XLA_TEST_F() function
143 XLA_TEST_F(ArrayElementwiseOpTest, IsFiniteScalarF32) { in XLA_TEST_F() function
163 XLA_TEST_F(ArrayElementwiseOpTest, IsFiniteR1F32s) { in XLA_TEST_F() function
175 XLA_TEST_F(ArrayElementwiseOpTest, AddTwoConstantF32s) { in XLA_TEST_F() function
[all …]
|
| D | gather_operation_test.cc | 46 XLA_TEST_F(GatherOperationTest, TensorFlowGatherV1) { in XLA_TEST_F() function
67 XLA_TEST_F(GatherOperationTest, TensorFlowGatherV2) { in XLA_TEST_F() function
88 XLA_TEST_F(GatherOperationTest, TensorFlowGatherMultipleBatchDims) { in XLA_TEST_F() function
109 XLA_TEST_F(GatherOperationTest, TensorFlowGatherNdMultipleBatchDims_0) { in XLA_TEST_F() function
131 XLA_TEST_F(GatherOperationTest, TensorFlowGatherNdMultipleBatchDims_1) { in XLA_TEST_F() function
153 XLA_TEST_F(GatherOperationTest, TensorFlowGatherNd) { in XLA_TEST_F() function
176 XLA_TEST_F(GatherOperationTest, TensorFlowGatherNdNonDefaultIndexVectorDim) { in XLA_TEST_F() function
199 XLA_TEST_F(GatherOperationTest, DynamicSlice) { in XLA_TEST_F() function
220 XLA_TEST_F(GatherOperationTest, BatchDynamicSlice) { in XLA_TEST_F() function
241 XLA_TEST_F(GatherOperationTest, ZeroDimBounds) { in XLA_TEST_F() function
[all …]
|
| D | scatter_test.cc | 44 XLA_TEST_F(ScatterTest, TensorFlowScatterV1_Update) { in XLA_TEST_F() function
72 XLA_TEST_F(ScatterTest, TensorFlowScatterV1_WithFusedAdds) { in XLA_TEST_F() function
103 XLA_TEST_F(ScatterTest, TensorFlowScatterV2_Update) { in XLA_TEST_F() function
132 XLA_TEST_F(ScatterTest, TensorFlowScatterV2_InversePermutation) { in XLA_TEST_F() function
168 XLA_TEST_F(ScatterTest, SimpleR4) { in XLA_TEST_F() function
199 XLA_TEST_F(ScatterTest, TensorFlowScatter_Add) { in XLA_TEST_F() function
228 XLA_TEST_F(ScatterTest, TensorFlowScatter_Mul) { in XLA_TEST_F() function
257 XLA_TEST_F(ScatterTest, TensorFlowScatter_F32) { in XLA_TEST_F() function
287 XLA_TEST_F(ScatterTest, TensorFlowScatter_RepeatedIndices) { in XLA_TEST_F() function
316 XLA_TEST_F(ScatterTest, TensorFlowScatter_MultipleBatchDims) { in XLA_TEST_F() function
[all …]
|
| D | vector_ops_simple_test.cc | 51 XLA_TEST_F(VecOpsSimpleTest, ExpTenValues) { in XLA_TEST_F() function
64 XLA_TEST_F(VecOpsSimpleTest, ExpManyValues) { in XLA_TEST_F() function
86 XLA_TEST_F(VecOpsSimpleTest, ExpIn4D) { in XLA_TEST_F() function
108 XLA_TEST_F(VecOpsSimpleTest, NegateTenFloatValues) { in XLA_TEST_F() function
119 XLA_TEST_F(VecOpsSimpleTest, NegateTenInt32Values) { in XLA_TEST_F() function
128 XLA_TEST_F(VecOpsSimpleTest, NegateUint32Values) { in XLA_TEST_F() function
138 XLA_TEST_F(VecOpsSimpleTest, InvSqrtSevenValues) { in XLA_TEST_F() function
150 XLA_TEST_F(VecOpsSimpleTest, AddTenValuesViaMap) { in XLA_TEST_F() function
165 XLA_TEST_F(VecOpsSimpleTest, MaxTenValues) { in XLA_TEST_F() function
178 XLA_TEST_F(VecOpsSimpleTest, MaxTenValuesFromParams) { in XLA_TEST_F() function
[all …]
|
| D | prng_test.cc | 70 XLA_TEST_F(PrngTest, ScalarU01) { UniformTest<float>(0, 1, {}); } in XLA_TEST_F() function
71 XLA_TEST_F(PrngTest, ScalarU01limits) { in XLA_TEST_F() function
75 XLA_TEST_F(PrngTest, ZeroValuesU01) { UniformTest<float>(0, 1, {0}); } in XLA_TEST_F() function
76 XLA_TEST_F(PrngTest, TenValuesU01) { UniformTest<float>(0, 1, {10}); } in XLA_TEST_F() function
77 XLA_TEST_F(PrngTest, TenValuesU37) { UniformTest<float>(3, 7, {10}); } in XLA_TEST_F() function
78 XLA_TEST_F(PrngTest, ZeroValuesR2) { UniformTest<float>(0, 1, {0, 20}); } in XLA_TEST_F() function
79 XLA_TEST_F(PrngTest, LargeU01) { UniformTest<float>(0, 1, {0x100, 0x100}); } in XLA_TEST_F() function
80 XLA_TEST_F(PrngTest, TwelveValuesU524) { UniformTest<int32>(5, 24, {12}); } in XLA_TEST_F() function
84 XLA_TEST_F(PrngTest, DISABLED_ON_INTERPRETER( in XLA_TEST_F() function
109 XLA_TEST_F(PrngTest, DISABLED_ON_INTERPRETER(DISABLED_ON_GPU( in XLA_TEST_F() function
[all …]
|
| D | concat_test.cc | 40 XLA_TEST_F(ConcatTest, Concat_Nothing) { in XLA_TEST_F() function
50 XLA_TEST_F(ConcatTest, Concat_R1_With_Nothing) { in XLA_TEST_F() function
59 XLA_TEST_F(ConcatTest, Concat_R1_L0_With_Nothing) { in XLA_TEST_F() function
70 XLA_TEST_F(ConcatTest, CannotConcatR0WithR0) { in XLA_TEST_F() function
81 XLA_TEST_F(ConcatTest, Concat_R1_L0_With_R1_L0) { in XLA_TEST_F() function
91 XLA_TEST_F(ConcatTest, Concat_R1_L0_With_R1_L1) { in XLA_TEST_F() function
101 XLA_TEST_F(ConcatTest, Concat_R1_L2_With_R1_L0) { in XLA_TEST_F() function
111 XLA_TEST_F(ConcatTest, Concat_R1_L2_With_R1_L1) { in XLA_TEST_F() function
121 XLA_TEST_F(ConcatTest, Concat_R1_L253_With_R1_L7) { in XLA_TEST_F() function
140 XLA_TEST_F(ConcatTest, Concat_0x0_With_0x0) { in XLA_TEST_F() function
[all …]
|
| D | convolution_variants_test.cc | 54 XLA_TEST_F(ConvolutionVariantsTest, Minimal) { in XLA_TEST_F() function
69 XLA_TEST_F(ConvolutionVariantsTest, MinimalWithBatch) { in XLA_TEST_F() function
84 XLA_TEST_F(ConvolutionVariantsTest, Flat1x1) { in XLA_TEST_F() function
101 XLA_TEST_F(ConvolutionVariantsTest, Deep1x1) { in XLA_TEST_F() function
116 XLA_TEST_F(ConvolutionVariantsTest, Filter1x2in1x2) { in XLA_TEST_F() function
131 XLA_TEST_F(ConvolutionVariantsTest, Filter1x2in1x3) { in XLA_TEST_F() function
146 XLA_TEST_F(ConvolutionVariantsTest, Filter1x2in2x2) { in XLA_TEST_F() function
161 XLA_TEST_F(ConvolutionVariantsTest, Filter2x1in2x2) { in XLA_TEST_F() function
176 XLA_TEST_F(ConvolutionVariantsTest, Filter2x2in2x2) { in XLA_TEST_F() function
191 XLA_TEST_F(ConvolutionVariantsTest, Filter1x2in2x3WithDepthAndBatch) { in XLA_TEST_F() function
[all …]
|
| D | broadcast_simple_test.cc | 108 XLA_TEST_F(BroadcastSimpleTest, ScalarNoOpBroadcast) { in XLA_TEST_F() function
114 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_2x3) { in XLA_TEST_F() function
121 XLA_TEST_F(BroadcastSimpleTest, ScalarParamTo2D_2x3) { in XLA_TEST_F() function
134 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_2x0) { in XLA_TEST_F() function
141 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_0x2) { in XLA_TEST_F() function
148 XLA_TEST_F(BroadcastSimpleTest, 1DTo2D) {
162 XLA_TEST_F(BroadcastSimpleTest, 1DTo2D_WithDimsUsual) {
175 XLA_TEST_F(BroadcastSimpleTest, 1DTo2D_WithDimsTranspose) {
188 XLA_TEST_F(BroadcastSimpleTest, 2DTo3D_WithDims) {
206 XLA_TEST_F(BroadcastSimpleTest, 2DTo3D_WithDimsNotPossibleWithBroadCast) {
[all …]
|
| D | copy_test.cc | 60 XLA_TEST_F(CopyOpTest, CopyR0Bool) { in XLA_TEST_F() function
64 XLA_TEST_F(CopyOpTest, CopyR1S0U32) { in XLA_TEST_F() function
68 XLA_TEST_F(CopyOpTest, CopyR1S3U32) { in XLA_TEST_F() function
72 XLA_TEST_F(CopyOpTest, CopyR3F32_2x2x3) { in XLA_TEST_F() function
77 XLA_TEST_F(CopyOpTest, CopyR4S32_2x2x3x2) { in XLA_TEST_F() function
83 XLA_TEST_F(CopyOpTest, CopyR4S32_0x2x3x2) { in XLA_TEST_F() function
87 XLA_TEST_F(CopyOpTest, CopyParameterScalar) { in XLA_TEST_F() function
108 XLA_TEST_F(CopyOpTest, CopyConstantR2Twice) { in XLA_TEST_F() function
129 XLA_TEST_F(CopyOpTest, CopyConstantR2DifferentLayouts) { in XLA_TEST_F() function
222 XLA_TEST_F(CopyOpTest, CopyConstantR3Layout021_SingleIncompleteTilePerLayer) { in XLA_TEST_F() function
[all …]
|
| D | tuple_test.cc | 45 XLA_TEST_F(TupleTest, TupleConstant) { in XLA_TEST_F() function
64 XLA_TEST_F(TupleTest, TupleScalarConstant) { in XLA_TEST_F() function
78 XLA_TEST_F(TupleTest, TupleCreate) { in XLA_TEST_F() function
99 XLA_TEST_F(TupleTest, TupleCreateWithZeroElementEntry) { in XLA_TEST_F() function
111 XLA_TEST_F(TupleTest, EmptyTupleCreate) { in XLA_TEST_F() function
119 XLA_TEST_F(TupleTest, GetTupleElement) { in XLA_TEST_F() function
135 XLA_TEST_F(TupleTest, GetTupleElementWithZeroElements) { in XLA_TEST_F() function
145 XLA_TEST_F(TupleTest, GetTupleElementOfNonTupleFailsGracefully) { in XLA_TEST_F() function
158 XLA_TEST_F(TupleTest, AddTupleElements) { in XLA_TEST_F() function
187 XLA_TEST_F(TupleTest, TupleGTEToTuple) { in XLA_TEST_F() function
[all …]
|
| D | fusion_test.cc | 193 XLA_TEST_F(FusionTest, Test) { in XLA_TEST_F() function
246 XLA_TEST_F(FusionTest, Parameter) { in XLA_TEST_F() function
271 XLA_TEST_F(FusionTest, RandomizedParallelPartition) { in XLA_TEST_F() function
307 XLA_TEST_F(FusionTest, BroadcastIntoBinaryOp) { in XLA_TEST_F() function
331 XLA_TEST_F(FusionTest, ReshapeToScalar) { in XLA_TEST_F() function
346 XLA_TEST_F(FusionTest, Reshape_3by2_1by2by3) { in XLA_TEST_F() function
361 XLA_TEST_F(FusionTest, Reshape_1by2by3_3by2) { in XLA_TEST_F() function
376 XLA_TEST_F(FusionTest, Reshape_1by1by1_) { in XLA_TEST_F() function
391 XLA_TEST_F(FusionTest, Reshape__1by1by1) { in XLA_TEST_F() function
406 XLA_TEST_F(FusionTest, Reshape__) { in XLA_TEST_F() function
[all …]
|
| D | token_hlo_test.cc | 31 XLA_TEST_F(TokenHloTest, SingleTokenInstruction) { in XLA_TEST_F() function
42 XLA_TEST_F(TokenHloTest, TokenInTuple) { in XLA_TEST_F() function
56 XLA_TEST_F(TokenHloTest, TokenTree) { in XLA_TEST_F() function
71 XLA_TEST_F(TokenHloTest, InvalidTokenShapedEntryParameter) { in XLA_TEST_F() function
92 XLA_TEST_F(TokenHloTest, InvalidTupleTokenShapedEntryParameter) { in XLA_TEST_F() function
112 XLA_TEST_F(TokenHloTest, TokenInWhileLoop) { in XLA_TEST_F() function
154 XLA_TEST_F(TokenHloTest, TokenInConditional) { in XLA_TEST_F() function
204 XLA_TEST_F(TokenHloTest, AddDependency) { in XLA_TEST_F() function
232 XLA_TEST_F(TokenHloTest, AddDependencyOfConstant) { in XLA_TEST_F() function
252 XLA_TEST_F(TokenHloTest, AddDependencyAsRoot) { in XLA_TEST_F() function
[all …]
|
| D | test_utils_test.cc | 33 XLA_TEST_F(TestUtilsTest, UnusedParam) { in XLA_TEST_F() function
59 XLA_TEST_F(TestUtilsTest, Token) { in XLA_TEST_F() function
77 XLA_TEST_F(TestUtilsTest, MultipleIndexSpacesForDynamicSlices) { in XLA_TEST_F() function
105 XLA_TEST_F(TestUtilsTest, MultipleIndexSpacesForDynamicUpdateSlices) { in XLA_TEST_F() function
136 XLA_TEST_F(TestUtilsTest, NoDuplicatesFloats) { in XLA_TEST_F() function
167 XLA_TEST_F(TestUtilsTest, NoDuplicatesInt32) { in XLA_TEST_F() function
198 XLA_TEST_F(TestUtilsTest, NoDuplicatesBfloat16) { in XLA_TEST_F() function
229 XLA_TEST_F(TestUtilsTest, MakeFakeArgumentsR0InputToDynamicSlice) { in XLA_TEST_F() function
257 XLA_TEST_F(TestUtilsTest, MakeFakeArgumentsForGather) { in XLA_TEST_F() function
291 XLA_TEST_F(TestUtilsTest, MakeFakeArgumentsForScatter) { in XLA_TEST_F() function
|
| D | convert_test.cc | 156 XLA_TEST_F(ConvertTest, ConvertR1S0S32ToR1S0F32) { in XLA_TEST_F() function
174 XLA_TEST_F(ConvertTest, ConvertR1S64ToR1F32) { in XLA_TEST_F() function
227 XLA_TEST_F(ConvertTest, ConvertR1U32ToR1F32) { in XLA_TEST_F() function
246 XLA_TEST_F(ConvertTest, ConvertR1F32ToR1U32) { in XLA_TEST_F() function
264 XLA_TEST_F(ConvertTest, ConvertR1U32ToR1S64) { in XLA_TEST_F() function
281 XLA_TEST_F(ConvertTest, ConvertR1S32ToR1S64) { in XLA_TEST_F() function
298 XLA_TEST_F(ConvertTest, ConvertR1F32ToR1S64) { in XLA_TEST_F() function
335 XLA_TEST_F(ConvertTest, ConvertR1U8ToR1F32) { in XLA_TEST_F() function
344 XLA_TEST_F(ConvertTest, ConvertR1U8ToR1S32) { in XLA_TEST_F() function
353 XLA_TEST_F(ConvertTest, ConvertR1U8ToR1U32) { in XLA_TEST_F() function
[all …]
|
| D | multioutput_fusion_test.cc | 187 XLA_TEST_F(MultiOutputFusionTest, 2DNofusion) { RunTest2D(false, 5); }
188 XLA_TEST_F(MultiOutputFusionTest, 2DFusion) { RunTest2D(true, 5); }
189 XLA_TEST_F(MultiOutputFusionTest, 2DFusionSize129) { RunTest2D(true, 129); }
190 XLA_TEST_F(MultiOutputFusionTest, DiffentTypesNoFusion) { RunTest1D(false, 8); } in XLA_TEST_F() function
191 XLA_TEST_F(MultiOutputFusionTest, DiffentTypesFusion) { RunTest1D(true, 8); } in XLA_TEST_F() function
193 XLA_TEST_F(MultiOutputFusionTest, FusionNodeIsRoot) { in XLA_TEST_F() function
223 XLA_TEST_F(MultiOutputFusionTest, MultiOutputLoopFusion) { in XLA_TEST_F() function
248 XLA_TEST_F(MultiOutputFusionTest, MultiOutputLoopFeedingMap) { in XLA_TEST_F() function
294 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() function
323 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() function
[all …]
|
| D | triangular_solve_test.cc | 101 XLA_TEST_F(TriangularSolveTest, EmptyArrays) { in XLA_TEST_F() function
118 XLA_TEST_F(TriangularSolveTest, SimpleRightLowerTranspose) { in XLA_TEST_F() function
139 XLA_TEST_F(TriangularSolveTest, SimpleRightLowerNotranspose) { in XLA_TEST_F() function
160 XLA_TEST_F(TriangularSolveTest, SimpleRightUpperTranspose) { in XLA_TEST_F() function
181 XLA_TEST_F(TriangularSolveTest, SimpleRightUpperNotranspose) { in XLA_TEST_F() function
202 XLA_TEST_F(TriangularSolveTest, SimpleLeftLowerTranspose) { in XLA_TEST_F() function
224 XLA_TEST_F(TriangularSolveTest, SimpleLeftLowerNotranspose) { in XLA_TEST_F() function
246 XLA_TEST_F(TriangularSolveTest, SimpleLeftLowerNoTransposeUnitDiagonal) { in XLA_TEST_F() function
265 XLA_TEST_F(TriangularSolveTest, SimpleLeftLowerNotransposeIrregularblock) { in XLA_TEST_F() function
287 XLA_TEST_F(TriangularSolveTest, SimpleLeftUpperTranspose) { in XLA_TEST_F() function
[all …]
|
| D | unary_op_test.cc | 136 XLA_TEST_F(UnaryOpTest, AbsTestR1Size0) { in XLA_TEST_F() function
142 XLA_TEST_F(UnaryOpTest, AbsTestR1) { in XLA_TEST_F() function
148 XLA_TEST_F(UnaryOpTest, AbsTestR0) { in XLA_TEST_F() function
163 XLA_TEST_F(UnaryOpTest, SignTestR0) { in XLA_TEST_F() function
180 XLA_TEST_F(UnaryOpTest, SignTestR1) { in XLA_TEST_F() function
187 XLA_TEST_F(UnaryOpTest, SignAbsTestR1) { in XLA_TEST_F() function
193 XLA_TEST_F(UnaryOpTest, SignAbsTestR2) { in XLA_TEST_F() function
203 XLA_TEST_F(UnaryOpTest, ConvertElementTypePredToS32) { in XLA_TEST_F() function
212 XLA_TEST_F(UnaryOpTest, ConvertElementTypePredToF32) { in XLA_TEST_F() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | constants_test.cc | 31 XLA_TEST_F(ConstantsTest, ConstantR0WithTypeS32) { in XLA_TEST_F() function
37 XLA_TEST_F(ConstantsTest, ConstantR0WithTypeS32DoesNotAcceptFloats) { in XLA_TEST_F() function
45 XLA_TEST_F(ConstantsTest, ConstantR0WithTypeF32) { in XLA_TEST_F() function
53 XLA_TEST_F(ConstantsTest, ScalarLikeS32) { in XLA_TEST_F() function
59 XLA_TEST_F(ConstantsTest, ScalarLikeF32) { in XLA_TEST_F() function
65 XLA_TEST_F(ConstantsTest, ZeroS32) { in XLA_TEST_F() function
71 XLA_TEST_F(ConstantsTest, ZeroF32) { in XLA_TEST_F() function
77 XLA_TEST_F(ConstantsTest, ZerosS32) { in XLA_TEST_F() function
83 XLA_TEST_F(ConstantsTest, ZerosLikeF32) { in XLA_TEST_F() function
89 XLA_TEST_F(ConstantsTest, OneS32) { in XLA_TEST_F() function
[all …]
|
| D | pooling_test.cc | 65 XLA_TEST_F(PoolingTest, MaxPool2D) { in XLA_TEST_F() function
78 XLA_TEST_F(PoolingTest, MaxPool2DWithPadding) { in XLA_TEST_F() function
91 XLA_TEST_F(PoolingTest, MaxPool2DWithPaddingAndStride) { in XLA_TEST_F() function
105 XLA_TEST_F(PoolingTest, AvgPool2D) { in XLA_TEST_F() function
121 XLA_TEST_F(PoolingTest, AvgPool2DWithPadding) { in XLA_TEST_F() function
137 XLA_TEST_F(PoolingTest, AvgPool2DWithPaddingAndStride) { in XLA_TEST_F() function
155 XLA_TEST_F(PoolingTest, AvgPool2DWithGeneralPaddingCountNotIncludePadding) { in XLA_TEST_F() function
169 XLA_TEST_F(PoolingTest, in XLA_TEST_F() function
185 XLA_TEST_F(PoolingTest, AvgPool2DGradNoPadding) { in XLA_TEST_F() function
202 XLA_TEST_F(PoolingTest, AvgPool2DGradNoPaddingWithStride) { in XLA_TEST_F() function
[all …]
|
| D | comparators_test.cc | 92 XLA_TEST_F(ComparatorsTest, CompareLtBF16) { in XLA_TEST_F() function
99 XLA_TEST_F(ComparatorsTest, CompareGtBF16) { in XLA_TEST_F() function
106 XLA_TEST_F(ComparatorsTest, CompareLtF16) { in XLA_TEST_F() function
113 XLA_TEST_F(ComparatorsTest, CompareGtF16) { in XLA_TEST_F() function
120 XLA_TEST_F(ComparatorsTest, CompareLtF32) { in XLA_TEST_F() function
127 XLA_TEST_F(ComparatorsTest, CompareGtF32) { in XLA_TEST_F() function
134 XLA_TEST_F(ComparatorsTest, CompareLtF64) { in XLA_TEST_F() function
141 XLA_TEST_F(ComparatorsTest, CompareGtF64) { in XLA_TEST_F() function
|
| D | self_adjoint_eig_test.cc | 145 XLA_TEST_F(SelfAdjointEigTest, Test_VWVt_EQ_A_2x4x4) { in XLA_TEST_F() function
157 XLA_TEST_F(SelfAdjointEigTest, Test_VWVt_EQ_A_Lower_2x4x4) { in XLA_TEST_F() function
170 XLA_TEST_F(SelfAdjointEigTest, Test_VWVt_EQ_A_Upper_2x4x4) { in XLA_TEST_F() function
183 XLA_TEST_F(SelfAdjointEigTest, Test_Orthogonality_2x4x4) { in XLA_TEST_F() function
195 XLA_TEST_F(SelfAdjointEigTest, Test_VtWV_EQ_A_Rank_Deficient_4x4) { in XLA_TEST_F() function
207 XLA_TEST_F(SelfAdjointEigTest, Test_Eigen_8x8) { in XLA_TEST_F() function
223 XLA_TEST_F(SelfAdjointEigTest, Test_Orthogonality_8x8) { in XLA_TEST_F() function
240 XLA_TEST_F(SelfAdjointEigTest, Wrong_Type_Int) { in XLA_TEST_F() function
250 XLA_TEST_F(SelfAdjointEigTest, Various_Size_Random_Matrix_8x8) { in XLA_TEST_F() function
263 XLA_TEST_F(SelfAdjointEigTest, Various_Size_Random_Matrix_16x16) { in XLA_TEST_F() function
[all …]
|
| D | math_test.cc | 174 XLA_TEST_F(MathTest, RealFpOnlyOps) { in XLA_TEST_F() function
214 XLA_TEST_F(MathTest, SqrtF32) { in XLA_TEST_F() function
227 XLA_TEST_F(MathTest, SquareTenValues) { in XLA_TEST_F() function
238 XLA_TEST_F(MathTest, ReciprocalTenValues) { in XLA_TEST_F() function
250 XLA_TEST_F(MathTest, SqrtZeroes) { in XLA_TEST_F() function
258 XLA_TEST_F(MathTest, SqrtSixValues) { in XLA_TEST_F() function
267 XLA_TEST_F(MathTest, Lgamma) { in XLA_TEST_F() function
291 XLA_TEST_F(MathTest, LgammaF16) { in XLA_TEST_F() function
313 XLA_TEST_F(MathTest, Digamma) { in XLA_TEST_F() function
343 XLA_TEST_F(MathTest, RoundToEven) { in XLA_TEST_F() function
[all …]
|