| /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 | 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 | 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 | 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 | 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 | 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 …]
|
| 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 | test_utils_test.cc | 33 XLA_TEST_F(TestUtilsTest, UnusedParam) { in XLA_TEST_F() function
59 XLA_TEST_F(TestUtilsTest, MultipleIndexSpacesForDynamicSlices) { in XLA_TEST_F() function
87 XLA_TEST_F(TestUtilsTest, MultipleIndexSpacesForDynamicUpdateSlices) { in XLA_TEST_F() function
118 XLA_TEST_F(TestUtilsTest, NoDuplicatesFloats) { in XLA_TEST_F() function
149 XLA_TEST_F(TestUtilsTest, NoDuplicatesInt32) { in XLA_TEST_F() function
180 XLA_TEST_F(TestUtilsTest, NoDuplicatesBfloat16) { in XLA_TEST_F() function
211 XLA_TEST_F(TestUtilsTest, MakeFakeArgumentsR0InputToDynamicSlice) { in XLA_TEST_F() function
239 XLA_TEST_F(TestUtilsTest, MakeFakeArgumentsForGather) { in XLA_TEST_F() function
273 XLA_TEST_F(TestUtilsTest, MakeFakeArgumentsForScatter) { in XLA_TEST_F() function
|
| D | cpu_gpu_fusion_test.cc | 192 XLA_TEST_F(CpuGpuFusionTest, Test) { in XLA_TEST_F() function
245 XLA_TEST_F(CpuGpuFusionTest, Parameter) { in XLA_TEST_F() function
270 XLA_TEST_F(CpuGpuFusionTest, RandomizedParallelPartition) { in XLA_TEST_F() function
306 XLA_TEST_F(CpuGpuFusionTest, BroadcastIntoBinaryOp) { in XLA_TEST_F() function
330 XLA_TEST_F(CpuGpuFusionTest, ReshapeToScalar) { in XLA_TEST_F() function
345 XLA_TEST_F(CpuGpuFusionTest, Reshape_3by2_1by2by3) { in XLA_TEST_F() function
360 XLA_TEST_F(CpuGpuFusionTest, Reshape_1by2by3_3by2) { in XLA_TEST_F() function
375 XLA_TEST_F(CpuGpuFusionTest, Reshape_1by1by1_) { in XLA_TEST_F() function
390 XLA_TEST_F(CpuGpuFusionTest, Reshape__1by1by1) { in XLA_TEST_F() function
405 XLA_TEST_F(CpuGpuFusionTest, Reshape__) { 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 | 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 | 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 | 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 | 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 | multioutput_fusion_test.cc | 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
340 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() function
363 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() function
386 XLA_TEST_F(MultiOutputFusionTest, in XLA_TEST_F() function
[all …]
|
| D | transfer_manager_test.cc | 71 XLA_TEST_F(TransferManagerTest, TransferR0U32) { in XLA_TEST_F() function
86 XLA_TEST_F(TransferManagerTest, TransferR1F32) { in XLA_TEST_F() function
103 XLA_TEST_F(TransferManagerTest, TransferR1F32AwkwardSizes) { in XLA_TEST_F() function
125 XLA_TEST_F(TransferManagerTest, TransferR1LargeF32) { in XLA_TEST_F() function
142 XLA_TEST_F(TransferManagerTest, TransferR1LargeUnalignedF32) { in XLA_TEST_F() function
162 XLA_TEST_F(TransferManagerTest, TransferR1U8) { in XLA_TEST_F() function
178 XLA_TEST_F(TransferManagerTest, TransferR2F32) { in XLA_TEST_F() function
195 XLA_TEST_F(TransferManagerTest, in XLA_TEST_F() function
217 XLA_TEST_F(TransferManagerTest, TransferTuple) { in XLA_TEST_F() function
234 XLA_TEST_F(TransferManagerTest, TransferEmptyTuple) { 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 | 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 | 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 …]
|
| /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 …]
|