Searched defs:XlaSplitNDOpTest (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | sharding_util_ops_test.cc | 210 using XlaSplitNDOpTest = ::testing::TestWithParam<XlaSplitNDTestParam>; typedef
212 TEST_P(XlaSplitNDOpTest, SplitDimensionZero) { in TEST_P() argument
230 TEST_P(XlaSplitNDOpTest, SplitDimensionNegative) { in TEST_P() argument
248 TEST_P(XlaSplitNDOpTest, NumOutputsMismatch) { in TEST_P() argument
266 TEST_P(XlaSplitNDOpTest, PaddingsLengthMismatch) { in TEST_P() argument
283 TEST_P(XlaSplitNDOpTest, PaddingsNegative) { in TEST_P() argument
301 TEST_P(XlaSplitNDOpTest, InputRank0) { in TEST_P() argument
318 TEST_P(XlaSplitNDOpTest, InputRank9) { in TEST_P() argument
335 TEST_P(XlaSplitNDOpTest, InputRankSplitMismatch) { in TEST_P() argument
353 TEST_P(XlaSplitNDOpTest, DimNotEvenlySplit) { in TEST_P() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sharding_util_ops_test.py | 62 class XlaSplitNDOpTest(xla_test.XLATestCase, parameterized.TestCase): class
|