/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | scalar_computations_test.cc | 72 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 …]
|
D | dynamic_ops_test.cc | 205 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 …]
|
D | reduce_test.cc | 354 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 …]
|
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, 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 …]
|
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, 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 …]
|
D | dot_operation_test.cc | 56 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 …]
|
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 | prng_test.cc | 71 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 …]
|
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 | 59 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 …]
|
D | broadcast_simple_test.cc | 107 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 …]
|
D | fusion_test.cc | 181 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 …]
|
D | tuple_test.cc | 42 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 …]
|
D | transfer_manager_test.cc | 60 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 …]
|
D | unary_op_test.cc | 129 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 …]
|
D | conditional_test.cc | 172 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 …]
|
D | local_client_execute_test.cc | 58 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 …]
|
D | params_test.cc | 43 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 …]
|
D | select_and_scatter_test.cc | 200 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 …]
|
D | convert_test.cc | 89 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
|
D | bfloat16_test.cc | 54 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
|
D | call_test.cc | 76 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
|
D | broadcast_test.cc | 36 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
|
D | compilation_cache_test.cc | 77 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/ |
D | triangular_solve_test.cc | 82 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 …]
|