Home
last modified time | relevance | path

Searched refs:XLA_TEST_F (Results 1 – 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/tests/
Dscalar_computations_test.cc72 XLA_TEST_F(ScalarComputationsTest, ReturnScalarF32) { in XLA_TEST_F() function
79 XLA_TEST_F(ScalarComputationsTest, NegateScalarF32) { in XLA_TEST_F() function
86 XLA_TEST_F(ScalarComputationsTest, NegateScalarS32) { in XLA_TEST_F() function
93 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsF32) { in XLA_TEST_F() function
100 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsS32) { in XLA_TEST_F() function
107 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsU32) { in XLA_TEST_F() function
114 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsU8) { in XLA_TEST_F() function
121 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsU64) { in XLA_TEST_F() function
130 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsS64) { in XLA_TEST_F() function
139 XLA_TEST_F(ScalarComputationsTest, AddTwoScalarsF64) { in XLA_TEST_F() function
[all …]
Ddynamic_ops_test.cc205 XLA_TEST_F(DynamicSliceTest, Int32R1BF16) { TestR1<int32, bfloat16>(); } in XLA_TEST_F() function
206 XLA_TEST_F(DynamicSliceTest, Int32R1) { TestR1<int32, int32>(); } in XLA_TEST_F() function
207 XLA_TEST_F(DynamicSliceTest, Int32R1Wrap) { TestR1Wrap<int32, int32>(); } in XLA_TEST_F() function
208 XLA_TEST_F(DynamicSliceTest, Int64R1) { TestR1<int64, float>(); } in XLA_TEST_F() function
209 XLA_TEST_F(DynamicSliceTest, UInt64R1) { TestR1<uint64, double>(); } in XLA_TEST_F() function
211 XLA_TEST_F(DynamicSliceTest, Int32R2BF16) { TestR2<int32, bfloat16>(); } in XLA_TEST_F() function
212 XLA_TEST_F(DynamicSliceTest, Int32R2) { TestR2<int32, int32>(); } in XLA_TEST_F() function
213 XLA_TEST_F(DynamicSliceTest, Int32R2Wrap) { TestR2Wrap<int32, int32>(); } in XLA_TEST_F() function
214 XLA_TEST_F(DynamicSliceTest, Int64R2) { TestR2<int64, double>(); } in XLA_TEST_F() function
215 XLA_TEST_F(DynamicSliceTest, UInt64R2) { TestR2<uint64, int32>(); } in XLA_TEST_F() function
[all …]
Dreduce_test.cc354 XLA_TEST_F(ReduceTest, ReduceR1_0_F32_To_R0) { RunR1ToR0Test(0); } in XLA_TEST_F() function
355 XLA_TEST_F(ReduceTest, ReduceR1_1_F32_To_R0) { RunR1ToR0Test(1); } in XLA_TEST_F() function
356 XLA_TEST_F(ReduceTest, ReduceR1_2_F32_To_R0) { RunR1ToR0Test(2); } in XLA_TEST_F() function
357 XLA_TEST_F(ReduceTest, ReduceR1_16_F32_To_R0) { RunR1ToR0Test(16); } in XLA_TEST_F() function
358 XLA_TEST_F(ReduceTest, ReduceR1_128_F32_To_R0) { RunR1ToR0Test(128); } in XLA_TEST_F() function
359 XLA_TEST_F(ReduceTest, ReduceR1_129_F32_To_R0) { RunR1ToR0Test(129); } in XLA_TEST_F() function
360 XLA_TEST_F(ReduceTest, ReduceR1_240_F32_To_R0) { RunR1ToR0Test(240); } in XLA_TEST_F() function
361 XLA_TEST_F(ReduceTest, ReduceR1_256_F32_To_R0) { RunR1ToR0Test(256); } in XLA_TEST_F() function
362 XLA_TEST_F(ReduceTest, ReduceR1_1024_F32_To_R0) { RunR1ToR0Test(1024); } in XLA_TEST_F() function
363 XLA_TEST_F(ReduceTest, ReduceR1_2048_F32_To_R0) { RunR1ToR0Test(2048); } in XLA_TEST_F() function
[all …]
Darray_elementwise_ops_test.cc52 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, IsFiniteZeroElementF32s) { in XLA_TEST_F() function
115 XLA_TEST_F(ArrayElementwiseOpTest, IsFiniteScalarF32) { in XLA_TEST_F() function
136 XLA_TEST_F(ArrayElementwiseOpTest, IsFiniteR1F32s) { in XLA_TEST_F() function
148 XLA_TEST_F(ArrayElementwiseOpTest, AddTwoConstantF32s) { in XLA_TEST_F() function
158 XLA_TEST_F(ArrayElementwiseOpTest, AddTwoConstantZeroElementF32s) { in XLA_TEST_F() function
[all …]
Dvector_ops_simple_test.cc51 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, SquareTenValues) { in XLA_TEST_F() function
149 XLA_TEST_F(VecOpsSimpleTest, ReciprocalTenValues) { in XLA_TEST_F() function
161 XLA_TEST_F(VecOpsSimpleTest, SqrtZeroes) { in XLA_TEST_F() function
169 XLA_TEST_F(VecOpsSimpleTest, SqrtSixValues) { in XLA_TEST_F() function
[all …]
Ddot_operation_test.cc56 XLA_TEST_F(DotOperationTest, ZeroElementVectorDotF32) { in XLA_TEST_F() function
65 XLA_TEST_F(DotOperationTest, TrivialMatrixVectorDotF32) { in XLA_TEST_F() function
84 XLA_TEST_F(DotOperationTest, OneElementVectorDotF32) { in XLA_TEST_F() function
88 XLA_TEST_F(DotOperationTest, OneElementVectorDotF64) { in XLA_TEST_F() function
102 XLA_TEST_F(DotOperationTest, VectorDotF32) { TestVectorDot<float>(); } in XLA_TEST_F() function
104 XLA_TEST_F(DotOperationTest, VectorDotF64) { TestVectorDot<double>(); } in XLA_TEST_F() function
114 XLA_TEST_F(DotOperationTest, Dot_0x2_2x0) { in XLA_TEST_F() function
123 XLA_TEST_F(DotOperationTest, Dot_0x2_2x3) { in XLA_TEST_F() function
132 XLA_TEST_F(DotOperationTest, Dot_3x2_2x0) { in XLA_TEST_F() function
142 XLA_TEST_F(DotOperationTest, Dot_2x0_0x2) { in XLA_TEST_F() function
[all …]
Dconcat_test.cc40 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 …]
Dprng_test.cc71 XLA_TEST_F(PrngTest, ScalarU01) { UniformTest<float>(0, 1, {}); } in XLA_TEST_F() function
72 XLA_TEST_F(PrngTest, ScalarU01limits) { in XLA_TEST_F() function
76 XLA_TEST_F(PrngTest, ZeroValuesU01) { UniformTest<float>(0, 1, {0}); } in XLA_TEST_F() function
77 XLA_TEST_F(PrngTest, TenValuesU01) { UniformTest<float>(0, 1, {10}); } in XLA_TEST_F() function
78 XLA_TEST_F(PrngTest, TenValuesU37) { UniformTest<float>(3, 7, {10}); } in XLA_TEST_F() function
79 XLA_TEST_F(PrngTest, ZeroValuesR2) { UniformTest<float>(0, 1, {0, 20}); } in XLA_TEST_F() function
80 XLA_TEST_F(PrngTest, LargeU01) { UniformTest<float>(0, 1, {0x100, 0x100}); } in XLA_TEST_F() function
81 XLA_TEST_F(PrngTest, TwelveValuesU524) { UniformTest<int32>(5, 24, {12}); } in XLA_TEST_F() function
84 XLA_TEST_F(PrngTest, DISABLED_ON_GPU(DISABLED_ON_CPU_PARALLEL( in XLA_TEST_F() function
108 XLA_TEST_F(PrngTest, DISABLED_ON_GPU(DISABLED_ON_CPU( in XLA_TEST_F() function
[all …]
Dconvolution_variants_test.cc54 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 …]
Dcopy_test.cc59 XLA_TEST_F(CopyOpTest, CopyR0Bool) { in XLA_TEST_F() function
63 XLA_TEST_F(CopyOpTest, CopyR1S0U32) { in XLA_TEST_F() function
67 XLA_TEST_F(CopyOpTest, CopyR1S3U32) { in XLA_TEST_F() function
71 XLA_TEST_F(CopyOpTest, CopyR3F32_2x2x3) { in XLA_TEST_F() function
76 XLA_TEST_F(CopyOpTest, CopyR4S32_2x2x3x2) { in XLA_TEST_F() function
82 XLA_TEST_F(CopyOpTest, CopyR4S32_0x2x3x2) { in XLA_TEST_F() function
86 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 …]
Dbroadcast_simple_test.cc107 XLA_TEST_F(BroadcastSimpleTest, ScalarNoOpBroadcast) { in XLA_TEST_F() function
113 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_2x3) { in XLA_TEST_F() function
120 XLA_TEST_F(BroadcastSimpleTest, ScalarParamTo2D_2x3) { in XLA_TEST_F() function
133 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_2x0) { in XLA_TEST_F() function
140 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_0x2) { in XLA_TEST_F() function
147 XLA_TEST_F(BroadcastSimpleTest, 1DTo2D) {
162 XLA_TEST_F(BroadcastSimpleTest, BooleanAnd2DTo3D_Pred) { in XLA_TEST_F() function
188 XLA_TEST_F(BroadcastSimpleTest, ZeroElement_1DTo2D) { in XLA_TEST_F() function
196 XLA_TEST_F(BroadcastSimpleTest, 1DToZeroElement2D) {
204 XLA_TEST_F(BroadcastSimpleTest, InDimensionAndDegenerateBroadcasting) { in XLA_TEST_F() function
[all …]
Dfusion_test.cc181 XLA_TEST_F(FusionTest, Test) { in XLA_TEST_F() function
233 XLA_TEST_F(FusionTest, Parameter) { in XLA_TEST_F() function
258 XLA_TEST_F(FusionTest, RandomizedParallelPartition) { in XLA_TEST_F() function
294 XLA_TEST_F(FusionTest, BroadcastIntoBinaryOp) { in XLA_TEST_F() function
318 XLA_TEST_F(FusionTest, ReshapeToScalar) { in XLA_TEST_F() function
332 XLA_TEST_F(FusionTest, Reshape_3by2_1by2by3) { in XLA_TEST_F() function
347 XLA_TEST_F(FusionTest, Reshape_1by2by3_3by2) { in XLA_TEST_F() function
362 XLA_TEST_F(FusionTest, Reshape_1by1by1_) { in XLA_TEST_F() function
376 XLA_TEST_F(FusionTest, Reshape__1by1by1) { in XLA_TEST_F() function
390 XLA_TEST_F(FusionTest, Reshape__) { in XLA_TEST_F() function
[all …]
Dtuple_test.cc42 XLA_TEST_F(TupleTest, TupleConstant) { in XLA_TEST_F() function
61 XLA_TEST_F(TupleTest, TupleScalarConstant) { in XLA_TEST_F() function
75 XLA_TEST_F(TupleTest, TupleCreate) { in XLA_TEST_F() function
96 XLA_TEST_F(TupleTest, TupleCreateWithZeroElementEntry) { in XLA_TEST_F() function
108 XLA_TEST_F(TupleTest, EmptyTupleCreate) { in XLA_TEST_F() function
116 XLA_TEST_F(TupleTest, GetTupleElement) { in XLA_TEST_F() function
131 XLA_TEST_F(TupleTest, GetTupleElementWithZeroElements) { in XLA_TEST_F() function
140 XLA_TEST_F(TupleTest, GetTupleElementOfNonTupleFailsGracefully) { in XLA_TEST_F() function
153 XLA_TEST_F(TupleTest, AddTupleElements) { in XLA_TEST_F() function
180 XLA_TEST_F(TupleTest, TupleGTEToTuple) { in XLA_TEST_F() function
[all …]
Dtransfer_manager_test.cc60 XLA_TEST_F(TransferManagerTest, TransferR0U32) { in XLA_TEST_F() function
75 XLA_TEST_F(TransferManagerTest, TransferR1F32) { in XLA_TEST_F() function
92 XLA_TEST_F(TransferManagerTest, TransferR1LargeF32) { in XLA_TEST_F() function
109 XLA_TEST_F(TransferManagerTest, TransferR1U8) { in XLA_TEST_F() function
125 XLA_TEST_F(TransferManagerTest, TransferR2F32) { in XLA_TEST_F() function
142 XLA_TEST_F(TransferManagerTest, in XLA_TEST_F() function
164 XLA_TEST_F(TransferManagerTest, TransferTuple) { in XLA_TEST_F() function
181 XLA_TEST_F(TransferManagerTest, TransferEmptyTuple) { in XLA_TEST_F() function
195 XLA_TEST_F(TransferManagerTest, TransferNestedTuple) { in XLA_TEST_F() function
215 XLA_TEST_F(TransferManagerTest, TransferComplexValue) { in XLA_TEST_F() function
[all …]
Dunary_op_test.cc129 XLA_TEST_F(UnaryOpTest, AbsTestR1Size0) { in XLA_TEST_F() function
135 XLA_TEST_F(UnaryOpTest, AbsTestR1) { in XLA_TEST_F() function
141 XLA_TEST_F(UnaryOpTest, AbsTestR0) { in XLA_TEST_F() function
157 XLA_TEST_F(UnaryOpTest, SignTestR0) { in XLA_TEST_F() function
177 XLA_TEST_F(UnaryOpTest, SignTestR1) { in XLA_TEST_F() function
183 XLA_TEST_F(UnaryOpTest, SignAbsTestR1) { in XLA_TEST_F() function
189 XLA_TEST_F(UnaryOpTest, UnsignedAbsTestR1) { in XLA_TEST_F() function
199 XLA_TEST_F(UnaryOpTest, UnsignedSignTestR1) { in XLA_TEST_F() function
208 XLA_TEST_F(UnaryOpTest, SignAbsTestR2) { in XLA_TEST_F() function
218 XLA_TEST_F(UnaryOpTest, ConvertElementTypePredToS32) { in XLA_TEST_F() function
[all …]
Dconditional_test.cc172 XLA_TEST_F(ConditionalOpTest, Parameters0) { in XLA_TEST_F() function
185 XLA_TEST_F(ConditionalOpTest, Parameters1) { in XLA_TEST_F() function
199 XLA_TEST_F(ConditionalOpTest, DiffComputationsDiffArgs) { in XLA_TEST_F() function
212 XLA_TEST_F(ConditionalOpTest, DiffComputationsSameArg) { in XLA_TEST_F() function
224 XLA_TEST_F(ConditionalOpTest, SameComputationDiffArgs) { in XLA_TEST_F() function
237 XLA_TEST_F(ConditionalOpTest, SameComputationSameArg) { in XLA_TEST_F() function
249 XLA_TEST_F(ConditionalOpTest, SameComputationDiffInstances) { in XLA_TEST_F() function
261 XLA_TEST_F(ConditionalOpTest, ConditionalWithCall) { in XLA_TEST_F() function
283 XLA_TEST_F(ConditionalOpTest, Parameters2TrueBranch) { in XLA_TEST_F() function
298 XLA_TEST_F(ConditionalOpTest, Parameters2FalseBranch) { in XLA_TEST_F() function
[all …]
Dlocal_client_execute_test.cc58 XLA_TEST_F(LocalClientExecuteTest, Constant) { in XLA_TEST_F() function
69 XLA_TEST_F(LocalClientExecuteTest, AddScalars) { in XLA_TEST_F() function
83 XLA_TEST_F(LocalClientExecuteTest, AddZeroElementVectors) { in XLA_TEST_F() function
97 XLA_TEST_F(LocalClientExecuteTest, AddVectors) { in XLA_TEST_F() function
112 XLA_TEST_F(LocalClientExecuteTest, AddVectorsWithProfile) { in XLA_TEST_F() function
131 XLA_TEST_F(LocalClientExecuteTest, AddArraysWithDifferentInputLayouts) { in XLA_TEST_F() function
164 XLA_TEST_F(LocalClientExecuteTest, AddArraysWithDifferentOutputLayouts) { in XLA_TEST_F() function
201 XLA_TEST_F(LocalClientExecuteTest, TupleResult) { in XLA_TEST_F() function
229 XLA_TEST_F(LocalClientExecuteTest, NestedTupleResult) { in XLA_TEST_F() function
262 XLA_TEST_F(LocalClientExecuteTest, TupleResultWithLayout) { in XLA_TEST_F() function
[all …]
Dparams_test.cc43 XLA_TEST_F(ParamsTest, ConstantR0F32Param) { in XLA_TEST_F() function
55 XLA_TEST_F(ParamsTest, ConstantR1S0F32Param) { in XLA_TEST_F() function
67 XLA_TEST_F(ParamsTest, ConstantR1S2F32Param) { in XLA_TEST_F() function
80 XLA_TEST_F(ParamsTest, ConstantR1U8Param) { in XLA_TEST_F() function
93 XLA_TEST_F(ParamsTest, ConstantR2_3x0_F32Param) { in XLA_TEST_F() function
106 XLA_TEST_F(ParamsTest, ConstantR2F32Param) { in XLA_TEST_F() function
121 XLA_TEST_F(ParamsTest, TwoParameters) { in XLA_TEST_F() function
152 XLA_TEST_F(ParamsTest, MissingParameter) { in XLA_TEST_F() function
170 XLA_TEST_F(ParamsTest, UnusedParameter) { in XLA_TEST_F() function
188 XLA_TEST_F(ParamsTest, UnusedParametersInUnusedExpression) { in XLA_TEST_F() function
[all …]
Dselect_and_scatter_test.cc200 XLA_TEST_F(SelectAndScatterTest, R1S0F32) { in XLA_TEST_F() function
210 XLA_TEST_F(SelectAndScatterTest, R1F32) { in XLA_TEST_F() function
222 XLA_TEST_F(SelectAndScatterTest, R1S32) { in XLA_TEST_F() function
233 XLA_TEST_F(SelectAndScatterTest, R1S32OverlappingWindow) { in XLA_TEST_F() function
244 XLA_TEST_F(SelectAndScatterTest, R2S32) { in XLA_TEST_F() function
256 XLA_TEST_F(SelectAndScatterTest, ReshapeR2S32) { in XLA_TEST_F() function
270 XLA_TEST_F(SelectAndScatterTest, R2S32OverlappingWindow) { in XLA_TEST_F() function
282 XLA_TEST_F(SelectAndScatterTest, R2S32SamePadding) { in XLA_TEST_F() function
295 XLA_TEST_F(SelectAndScatterTest, R2S32SamePaddingOverlappingWindow) { in XLA_TEST_F() function
307 XLA_TEST_F(SelectAndScatterTest, R2F32OverlappingR2Source) { in XLA_TEST_F() function
[all …]
Dconvert_test.cc89 XLA_TEST_F(ConvertTest, ConvertR1S0S32ToR1S0F32) { in XLA_TEST_F() function
107 XLA_TEST_F(ConvertTest, ConvertR1S64ToR1F32) { in XLA_TEST_F() function
116 XLA_TEST_F(ConvertTest, ConvertR1U8ToR1F32) { in XLA_TEST_F() function
125 XLA_TEST_F(ConvertTest, ConvertR1U8ToR1S32) { in XLA_TEST_F() function
134 XLA_TEST_F(ConvertTest, ConvertR1U8ToR1U32) { in XLA_TEST_F() function
143 XLA_TEST_F(ConvertTest, ConvertR1F32ToR1F64) { in XLA_TEST_F() function
152 XLA_TEST_F(ConvertTest, ConvertR1F64ToR1F32) { in XLA_TEST_F() function
Dbfloat16_test.cc54 XLA_TEST_F(Bfloat16Test, ScalarOperation) { in XLA_TEST_F() function
64 XLA_TEST_F(Bfloat16Test, LogOperation) { in XLA_TEST_F() function
73 XLA_TEST_F(Bfloat16Test, NegateScalarF16) { in XLA_TEST_F() function
81 XLA_TEST_F(Bfloat16Test, BatchNormTraining) { in XLA_TEST_F() function
118 XLA_TEST_F(Bfloat16Test, BatchNormGrad) { in XLA_TEST_F() function
Dcall_test.cc76 XLA_TEST_F(CallOpTest, CallR0F32IdentityScalar) { in XLA_TEST_F() function
85 XLA_TEST_F(CallOpTest, CallR1S0F32AddArray) { in XLA_TEST_F() function
95 XLA_TEST_F(CallOpTest, CallR1S2F32AddArray) { in XLA_TEST_F() function
105 XLA_TEST_F(CallOpTest, CallTreeTwoDeepBranchFactorThree) { in XLA_TEST_F() function
136 XLA_TEST_F(CallOpTest, CallR0F32Tuple) { in XLA_TEST_F() function
Dbroadcast_test.cc36 XLA_TEST_F(BroadcastTest, BroadcastScalarToScalar) { in XLA_TEST_F() function
53 XLA_TEST_F(BroadcastTest, BroadcastScalarTo2D) { in XLA_TEST_F() function
70 XLA_TEST_F(BroadcastTest, BroadcastVectorTo2D) { in XLA_TEST_F() function
97 XLA_TEST_F(BroadcastTest, Broadcast2DTo2D) { in XLA_TEST_F() function
114 XLA_TEST_F(BroadcastTest, Broadcast2DTo2DTranspose) { in XLA_TEST_F() function
133 XLA_TEST_F(BroadcastTest, Broadcast2DTo3D) { in XLA_TEST_F() function
203 XLA_TEST_F(BroadcastTest, Broadcast_R1_64_To_R4_32x64x7x7) { in XLA_TEST_F() function
Dcompilation_cache_test.cc77 XLA_TEST_F(CompilationCacheTest, ComputationCalledMultipleTimes) { in XLA_TEST_F() function
87 XLA_TEST_F(CompilationCacheTest, ComputationCalledWithDifferentParameters) { in XLA_TEST_F() function
112 XLA_TEST_F(CompilationCacheTest, MultipleComputations) { in XLA_TEST_F() function
136 XLA_TEST_F(CompilationCacheTest, DifferentParameterLayouts) { in XLA_TEST_F() function
172 XLA_TEST_F(CompilationCacheTest, MutatedComputation) { in XLA_TEST_F() function
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dtriangular_solve_test.cc82 XLA_TEST_F(TriangularSolveTest, SimpleRightLowerTranspose) { in XLA_TEST_F() function
104 XLA_TEST_F(TriangularSolveTest, SimpleRightLowerNotranspose) { in XLA_TEST_F() function
126 XLA_TEST_F(TriangularSolveTest, SimpleRightUpperTranspose) { in XLA_TEST_F() function
148 XLA_TEST_F(TriangularSolveTest, SimpleRightUpperNotranspose) { in XLA_TEST_F() function
170 XLA_TEST_F(TriangularSolveTest, SimpleLeftLowerTranspose) { in XLA_TEST_F() function
193 XLA_TEST_F(TriangularSolveTest, SimpleLeftLowerNotranspose) { in XLA_TEST_F() function
216 XLA_TEST_F(TriangularSolveTest, SimpleLeftUpperTranspose) { in XLA_TEST_F() function
239 XLA_TEST_F(TriangularSolveTest, SimpleLeftUpperNotranspose) { in XLA_TEST_F() function
262 XLA_TEST_F(TriangularSolveTest, SimpleRightLowerTransposeConjugate) { in XLA_TEST_F() function
290 XLA_TEST_F(TriangularSolveTest, SimpleLeftUpperTransposeNoconjugate) { in XLA_TEST_F() function
[all …]

123