| /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 | 218 XLA_TEST_F(DynamicSliceTest, Int32R1BF16) { TestR1<int32, bfloat16>(); } in XLA_TEST_F() function
219 XLA_TEST_F(DynamicSliceTest, Int32R1) { TestR1<int32, int32>(); } in XLA_TEST_F() function
220 XLA_TEST_F(DynamicSliceTest, Int32R1OOB) { TestR1OOB<int32, int32>(); } in XLA_TEST_F() function
221 XLA_TEST_F(DynamicSliceTest, Int64R1) { TestR1<int64, float>(); } in XLA_TEST_F() function
222 XLA_TEST_F(DynamicSliceTest, UInt64R1) { TestR1<uint64, float>(); } in XLA_TEST_F() function
223 XLA_TEST_F(DynamicSliceTest, UInt32R1OOB) { in XLA_TEST_F() function
227 XLA_TEST_F(DynamicSliceTest, Int32R2BF16) { TestR2<int32, bfloat16>(); } in XLA_TEST_F() function
228 XLA_TEST_F(DynamicSliceTest, Int32R2) { TestR2<int32, int32>(); } in XLA_TEST_F() function
229 XLA_TEST_F(DynamicSliceTest, Int32R2OOB) { TestR2OOB<int32, int32>(); } in XLA_TEST_F() function
230 XLA_TEST_F(DynamicSliceTest, Int64R2) { TestR2<int64, float>(); } in XLA_TEST_F() function
[all …]
|
| D | reduce_test.cc | 358 XLA_TEST_F(ReduceTest, ReduceR1_0_F32_To_R0) { RunR1ToR0Test(0); } in XLA_TEST_F() function
359 XLA_TEST_F(ReduceTest, ReduceR1_1_F32_To_R0) { RunR1ToR0Test(1); } in XLA_TEST_F() function
360 XLA_TEST_F(ReduceTest, ReduceR1_2_F32_To_R0) { RunR1ToR0Test(2); } in XLA_TEST_F() function
361 XLA_TEST_F(ReduceTest, ReduceR1_16_F32_To_R0) { RunR1ToR0Test(16); } in XLA_TEST_F() function
362 XLA_TEST_F(ReduceTest, ReduceR1_128_F32_To_R0) { RunR1ToR0Test(128); } in XLA_TEST_F() function
363 XLA_TEST_F(ReduceTest, ReduceR1_129_F32_To_R0) { RunR1ToR0Test(129); } in XLA_TEST_F() function
364 XLA_TEST_F(ReduceTest, ReduceR1_240_F32_To_R0) { RunR1ToR0Test(240); } in XLA_TEST_F() function
365 XLA_TEST_F(ReduceTest, ReduceR1_256_F32_To_R0) { RunR1ToR0Test(256); } in XLA_TEST_F() function
366 XLA_TEST_F(ReduceTest, ReduceR1_1024_F32_To_R0) { RunR1ToR0Test(1024); } in XLA_TEST_F() function
367 XLA_TEST_F(ReduceTest, ReduceR1_2048_F32_To_R0) { RunR1ToR0Test(2048); } in XLA_TEST_F() function
[all …]
|
| D | array_elementwise_ops_test.cc | 53 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantZeroElementF32) { in XLA_TEST_F() function
61 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantF32) { in XLA_TEST_F() function
70 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantF64) { in XLA_TEST_F() function
78 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantS32) { in XLA_TEST_F() function
94 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantZeroElementC64) { in XLA_TEST_F() function
102 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantC64) { in XLA_TEST_F() function
113 XLA_TEST_F(ArrayElementwiseOpTest, NegConstantS64) { in XLA_TEST_F() function
141 XLA_TEST_F(ArrayElementwiseOpTest, IsFiniteZeroElementF32s) { in XLA_TEST_F() function
149 XLA_TEST_F(ArrayElementwiseOpTest, IntPow) { in XLA_TEST_F() function
162 XLA_TEST_F(ArrayElementwiseOpTest, IntPowLarge) { 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, BatchDimInMiddle) { in XLA_TEST_F() function
91 XLA_TEST_F(GatherOperationTest, TensorFlowGatherV2) { in XLA_TEST_F() function
112 XLA_TEST_F(GatherOperationTest, TensorFlowGatherMultipleBatchDims) { in XLA_TEST_F() function
133 XLA_TEST_F(GatherOperationTest, TensorFlowGatherNdMultipleBatchDims_0) { in XLA_TEST_F() function
155 XLA_TEST_F(GatherOperationTest, TensorFlowGatherNdMultipleBatchDims_1) { in XLA_TEST_F() function
177 XLA_TEST_F(GatherOperationTest, TensorFlowGatherNd) { in XLA_TEST_F() function
200 XLA_TEST_F(GatherOperationTest, TensorFlowGatherNdNonDefaultIndexVectorDim) { in XLA_TEST_F() function
223 XLA_TEST_F(GatherOperationTest, DynamicSlice) { in XLA_TEST_F() function
244 XLA_TEST_F(GatherOperationTest, BatchDynamicSlice) { in XLA_TEST_F() function
[all …]
|
| D | scatter_test.cc | 43 XLA_TEST_F(ScatterTest, TensorFlowScatterV1_Update) { in XLA_TEST_F() function
71 XLA_TEST_F(ScatterTest, TensorFlowScatterV1_WithFusedAdds) { in XLA_TEST_F() function
102 XLA_TEST_F(ScatterTest, TensorFlowScatterV2_Update) { in XLA_TEST_F() function
131 XLA_TEST_F(ScatterTest, TensorFlowScatterV2_InversePermutation) { in XLA_TEST_F() function
167 XLA_TEST_F(ScatterTest, SimpleR4) { in XLA_TEST_F() function
198 XLA_TEST_F(ScatterTest, TensorFlowScatter_Add) { in XLA_TEST_F() function
227 XLA_TEST_F(ScatterTest, TensorFlowScatter_Add_UniqueIndices) { in XLA_TEST_F() function
257 XLA_TEST_F(ScatterTest, TensorFlowScatter_Mul) { in XLA_TEST_F() function
286 XLA_TEST_F(ScatterTest, TensorFlowScatter_F32) { in XLA_TEST_F() function
316 XLA_TEST_F(ScatterTest, TensorFlowScatter_RepeatedIndices) { 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
116 XLA_TEST_F(PrngTest, DISABLED_ON_INTERPRETER(DISABLED_ON_GPU( in XLA_TEST_F() function
175 XLA_TEST_F(PrngTest, Uniformity7) { in XLA_TEST_F() function
[all …]
|
| D | concat_test.cc | 43 XLA_TEST_F(ConcatTest, Concat_Nothing) { in XLA_TEST_F() function
53 XLA_TEST_F(ConcatTest, Concat_R1_With_Nothing) { in XLA_TEST_F() function
62 XLA_TEST_F(ConcatTest, Concat_R1_L0_With_Nothing) { in XLA_TEST_F() function
73 XLA_TEST_F(ConcatTest, CannotConcatR0WithR0) { in XLA_TEST_F() function
84 XLA_TEST_F(ConcatTest, Concat_R1_L0_With_R1_L0) { in XLA_TEST_F() function
94 XLA_TEST_F(ConcatTest, Concat_R1_L0_With_R1_L1) { in XLA_TEST_F() function
104 XLA_TEST_F(ConcatTest, Concat_R1_L2_With_R1_L0) { in XLA_TEST_F() function
114 XLA_TEST_F(ConcatTest, Concat_R1_L2_With_R1_L1) { in XLA_TEST_F() function
124 XLA_TEST_F(ConcatTest, Concat_R1_L253_With_R1_L7) { in XLA_TEST_F() function
143 XLA_TEST_F(ConcatTest, Concat_0x0_With_0x0) { 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 | 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 | 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 | 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, DifferentTypesNoFusion) { in XLA_TEST_F() function
193 XLA_TEST_F(MultiOutputFusionTest, DifferentTypesFusion) { RunTest1D(true, 8); } in XLA_TEST_F() function
195 XLA_TEST_F(MultiOutputFusionTest, FusionNodeIsRoot) { in XLA_TEST_F() function
225 XLA_TEST_F(MultiOutputFusionTest, MultiOutputLoopFusion) { in XLA_TEST_F() function
250 XLA_TEST_F(MultiOutputFusionTest, MultiOutputLoopFeedingMap) { in XLA_TEST_F() function
296 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() function
318 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() function
[all …]
|
| D | collective_ops_test.cc | 170 XLA_TEST_F(CollectiveOpsTest, AllReduce_sum_float32_2D) { in XLA_TEST_F() function
177 XLA_TEST_F(CollectiveOpsTest, AllReduceSingleOutput_float32) { in XLA_TEST_F() function
184 XLA_TEST_F(CollectiveOpsTest, AllReduceTwoReplicasOneOperand_int8) { in XLA_TEST_F() function
188 XLA_TEST_F(CollectiveOpsTest, AllReduceTwoReplicasOneOperand_uint8) { in XLA_TEST_F() function
192 XLA_TEST_F(CollectiveOpsTest, AllReduceTwoReplicasOneOperand_uint32) { in XLA_TEST_F() function
196 XLA_TEST_F(CollectiveOpsTest, AllReduceTwoReplicasOneOperand_int32) { in XLA_TEST_F() function
200 XLA_TEST_F(CollectiveOpsTest, AllReduceTwoReplicasOneOperand_int64) { in XLA_TEST_F() function
204 XLA_TEST_F(CollectiveOpsTest, AllReduceTwoReplicasOneOperand_uint64) { in XLA_TEST_F() function
208 XLA_TEST_F(CollectiveOpsTest, AllReduceTwoReplicasOneOperand_float32) { in XLA_TEST_F() function
212 XLA_TEST_F(CollectiveOpsTest, AllReduceTwoReplicasOneOperand_double) { 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 | token_hlo_test.cc | 30 XLA_TEST_F(TokenHloTest, SingleTokenInstruction) { in XLA_TEST_F() function
41 XLA_TEST_F(TokenHloTest, TokenInTuple) { in XLA_TEST_F() function
55 XLA_TEST_F(TokenHloTest, TokenTree) { in XLA_TEST_F() function
70 XLA_TEST_F(TokenHloTest, InvalidTokenShapedEntryParameter) { in XLA_TEST_F() function
91 XLA_TEST_F(TokenHloTest, InvalidTupleTokenShapedEntryParameter) { in XLA_TEST_F() function
111 XLA_TEST_F(TokenHloTest, TokenInWhileLoop) { in XLA_TEST_F() function
153 XLA_TEST_F(TokenHloTest, TokenInConditional) { in XLA_TEST_F() function
203 XLA_TEST_F(TokenHloTest, AddDependency) { in XLA_TEST_F() function
231 XLA_TEST_F(TokenHloTest, AddDependencyOfConstant) { in XLA_TEST_F() function
251 XLA_TEST_F(TokenHloTest, AddDependencyAsRoot) { in XLA_TEST_F() function
[all …]
|
| /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 | sorting_test.cc | 31 XLA_TEST_F(SortingTest, TopK3From8Values) { in XLA_TEST_F() function
39 XLA_TEST_F(SortingTest, TopK3From8Indices) { in XLA_TEST_F() function
47 XLA_TEST_F(SortingTest, TopKFullSortMinInt) { in XLA_TEST_F() function
56 XLA_TEST_F(SortingTest, TopKFullSort) { in XLA_TEST_F() function
71 XLA_TEST_F(SortingTest, TopKFullSortWithDuplicates) { in XLA_TEST_F() function
79 XLA_TEST_F(SortingTest, TopK3From8Values2Partitions) { in XLA_TEST_F() function
87 XLA_TEST_F(SortingTest, TopK3From8Indices2Partitions) { in XLA_TEST_F() function
96 XLA_TEST_F(SortingTest, TopK3From8Values3Partitions) { in XLA_TEST_F() function
104 XLA_TEST_F(SortingTest, TopK3From8Indices3Partitions) { in XLA_TEST_F() function
113 XLA_TEST_F(SortingTest, TopK3From8Values5Partitions) { in XLA_TEST_F() function
[all …]
|
| D | math_test.cc | 186 XLA_TEST_F(MathTest, RealFpOnlyOps) { in XLA_TEST_F() function
226 XLA_TEST_F(MathTest, SqrtF32) { in XLA_TEST_F() function
239 XLA_TEST_F(MathTest, SqrtF64) { in XLA_TEST_F() function
253 XLA_TEST_F(MathTest, ErfInvF64) { in XLA_TEST_F() function
274 XLA_TEST_F(MathTest, SquareTenValues) { in XLA_TEST_F() function
285 XLA_TEST_F(MathTest, ReciprocalTenValues) { in XLA_TEST_F() function
297 XLA_TEST_F(MathTest, SqrtZeroes) { in XLA_TEST_F() function
305 XLA_TEST_F(MathTest, SqrtSixValues) { in XLA_TEST_F() function
314 XLA_TEST_F(MathTest, CbrtSixValues) { in XLA_TEST_F() function
323 XLA_TEST_F(MathTest, SinhSmallValues) { in XLA_TEST_F() function
[all …]
|
| D | matrix_test.cc | 62 XLA_TEST_F(MatrixTest, Triangle) { in XLA_TEST_F() function
110 XLA_TEST_F(MatrixTest, SetMatrixDiagonal_S32) { in XLA_TEST_F() function
113 XLA_TEST_F(MatrixTest, SetMatrixDiagonal_S64) { in XLA_TEST_F() function
116 XLA_TEST_F(MatrixTest, SetMatrixDiagonal_F32) { in XLA_TEST_F() function
120 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_S32) { TestMatrixDiagonal<int32>(); } in XLA_TEST_F() function
122 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_S64) { TestMatrixDiagonal<int64>(); } in XLA_TEST_F() function
124 XLA_TEST_F(MatrixTest, GetMatrixDiagonal_F32) { TestMatrixDiagonal<float>(); } in XLA_TEST_F() function
151 XLA_TEST_F(MatrixTest, GetMatrixDiagonal4D_S32) { in XLA_TEST_F() function
155 XLA_TEST_F(MatrixTest, GetMatrixDiagonal4D_S64) { in XLA_TEST_F() function
159 XLA_TEST_F(MatrixTest, GetMatrixDiagonal4D_F32) { 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 | slicing_test.cc | 59 XLA_TEST_F(SlicingTest, Simple2dLookup) { in XLA_TEST_F() function
73 XLA_TEST_F(SlicingTest, Simple3dLookup) { in XLA_TEST_F() function
88 XLA_TEST_F(SlicingTest, SimpleSliceUpdate) { in XLA_TEST_F() function
107 XLA_TEST_F(SlicingTest, TorchGatherSparse) { in XLA_TEST_F() function
121 XLA_TEST_F(SlicingTest, TorchGatherDense) { in XLA_TEST_F() function
135 XLA_TEST_F(SlicingTest, TorchScatterDense) { in XLA_TEST_F() function
153 XLA_TEST_F(SlicingTest, TorchIndexSelectOn0) { in XLA_TEST_F() function
172 XLA_TEST_F(SlicingTest, TorchIndexSelectOn1) { in XLA_TEST_F() function
191 XLA_TEST_F(SlicingTest, EmptyIndexSelect) { in XLA_TEST_F() function
203 XLA_TEST_F(SlicingTest, DoubleEmptyIndexSelect) { 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 …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter_test.cc | 51 XLA_TEST_F(ElementalIrEmitterExecutionTest, DotFusion) { in XLA_TEST_F() function
75 XLA_TEST_F(ElementalIrEmitterExecutionTest, ScalarDotFusion) { in XLA_TEST_F() function
99 XLA_TEST_F(ElementalIrEmitterExecutionTest, BatchDot) { in XLA_TEST_F() function
132 XLA_TEST_F(ElementalIrEmitterExecutionTest, in XLA_TEST_F() function
156 XLA_TEST_F(ElementalIrEmitterExecutionTest, in XLA_TEST_F() function
179 XLA_TEST_F(ElementalIrEmitterExecutionTest, in XLA_TEST_F() function
203 XLA_TEST_F(ElementalIrEmitterExecutionTest, ConvertFloatsToBF16) { in XLA_TEST_F() function
220 XLA_TEST_F(ElementalIrEmitterExecutionTest, ConvertSignedToBF16) { in XLA_TEST_F() function
239 XLA_TEST_F(ElementalIrEmitterExecutionTest, ConvertUnsignedToBF16) { in XLA_TEST_F() function
258 XLA_TEST_F(ElementalIrEmitterExecutionTest, ConvertBF16ToFloat) { in XLA_TEST_F() function
[all …]
|