Home
last modified time | relevance | path

Searched refs:DISABLED_ON_CPU (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultioutput_fusion_test.cc297 DISABLED_ON_CPU(MultiOutputReduceFusionMinor)) { in XLA_TEST_F()
319 DISABLED_ON_CPU(MultiOutputReduceFusionMajor)) { in XLA_TEST_F()
341 DISABLED_ON_CPU(MultiOutputReduceFusionScalar)) { in XLA_TEST_F()
364 DISABLED_ON_CPU(MultiOutputReduceFusionMinorWithExtraOutput)) { in XLA_TEST_F()
387 DISABLED_ON_CPU(MultiOutputReduceFusionMajorWithExtraOutput)) { in XLA_TEST_F()
410 DISABLED_ON_CPU(MultiOutputReduceFusionScalarWithExtraOutput)) { in XLA_TEST_F()
441 DISABLED_ON_CPU(MultiOutputReduceFusionNonConstInit)) { in XLA_TEST_F()
464 DISABLED_ON_CPU(MultiOutputReduceFusionDifferentElementTypes)) { in XLA_TEST_F()
Dexhaustive_binary_test_f32_f64.cc90 BINARY_TEST_FLOAT_32(DISABLED_ON_CPU(Mul), {
96 BINARY_TEST_FLOAT_32(DISABLED_ON_CPU(Div), {
116 BINARY_TEST_FLOAT_32(DISABLED_ON_CPU(AbsComplex), {
184 BINARY_TEST_FLOAT_64(DISABLED_ON_CPU(Mul), {
190 BINARY_TEST_FLOAT_64(DISABLED_ON_CPU(Div), {
204 BINARY_TEST_FLOAT_64(DISABLED_ON_CPU(AbsComplex), {
Dtest_macros.h31 #define DISABLED_ON_CPU(X) X macro
47 # undef DISABLED_ON_CPU
48 # define DISABLED_ON_CPU(X) XLA_TEST_PASTE(DISABLED_, X) macro
Dexhaustive_binary_16_bit_test.cc106 BINARY_TEST_16BIT(DISABLED_ON_CPU(Mul), {
112 BINARY_TEST_16BIT(DISABLED_ON_CPU(Div), {
126 BINARY_TEST_16BIT(DISABLED_ON_GPU(DISABLED_ON_CPU(Pow)), {
138 BINARY_TEST_16BIT(DISABLED_ON_CPU(Atan2),
Dcollective_ops_test.cc231 DISABLED_ON_CPU(AllReduceTwoReplicasOneOperand_bfloat16)) { in XLA_TEST_F()
235 XLA_TEST_F(CollectiveOpsTest, DISABLED_ON_CPU(AllReduce_sum_complex64)) { in XLA_TEST_F()
242 XLA_TEST_F(CollectiveOpsTest, DISABLED_ON_CPU(AllReduce_sum_complex128)) { in XLA_TEST_F()
364 XLA_TEST_F(CollectiveOpsTest, DISABLED_ON_CPU(AllReduce_NcclChannelCaching)) { in XLA_TEST_F()
590 XLA_TEST_F(CollectiveOpsTest, DISABLED_ON_CPU(AsyncAllReduce)) { in XLA_TEST_F()
622 XLA_TEST_F(CollectiveOpsTest, DISABLED_ON_CPU(AsyncAllReduceTwoOperands)) { in XLA_TEST_F()
930 XLA_TEST_F(CollectiveOpsTest, DISABLED_ON_CPU(AllToAll_SplitDimension)) { in XLA_TEST_F()
1064 XLA_TEST_F(CollectiveOpsTest, DISABLED_ON_CPU(AllGatherMixedTypes)) { in XLA_TEST_F()
1170 XLA_TEST_F(CollectiveOpsTest, DISABLED_ON_CPU(AllReduceReassociate)) { in XLA_TEST_F()
1215 DISABLED_ON_CPU(AllGatherBroadcastReorder_NonUniform)) { in XLA_TEST_F()
[all …]
Dprng_test.cc92 DISABLED_ON_INTERPRETER(DISABLED_ON_GPU(DISABLED_ON_CPU(DoIt)))) { in XLA_TEST_P()
118 DISABLED_ON_CPU(ScalarBF16CountTests)))) { in XLA_TEST_F()
194 XLA_TEST_F(PrngTest, DISABLED_ON_GPU(DISABLED_ON_CPU(MapUsingRng))) { in XLA_TEST_F()
Dparams_test.cc259 DISABLED_ON_CPU(DISABLED_ON_GPU(ThreeThousandParameters))) { in XLA_TEST_F()
289 XLA_TEST_F(ParamsTest, DISABLED_ON_CPU(DISABLED_ON_GPU( in XLA_TEST_F()
349 DISABLED_ON_CPU(DISABLED_ON_GPU(ManyParametersIntoWhileLoop))) { in XLA_TEST_F()
Dexhaustive_unary_test_complex.cc99 UNARY_TEST_COMPLEX_64(DISABLED_ON_CPU(Log), {
Dlocal_client_execute_test.cc677 DISABLED_ON_CPU(RunOnStreamForWrongPlatform)) { in XLA_TEST_F()
697 DISABLED_ON_CPU(AllocatorDoesNotMatchPlatform)) { in XLA_TEST_F()
Dcpu_gpu_fusion_test.cc604 XLA_TEST_F(CpuGpuFusionTest, DISABLED_ON_CPU(Reduce)) { in XLA_TEST_F()
645 XLA_TEST_F(CpuGpuFusionTest, DISABLED_ON_CPU(ReduceWindow)) { in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsvd_test.cc244 XLA_TEST_F(SVDTest, DISABLED_ON_CPU(DISABLED_ON_INTERPRETER( in XLA_TEST_F()
258 XLA_TEST_F(SVDTest, DISABLED_ON_GPU(DISABLED_ON_CPU(DISABLED_ON_INTERPRETER( in XLA_TEST_F()