Searched defs:XlaConcatNDOpTest (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | sharding_util_ops_test.cc | 888 using XlaConcatNDOpTest = ::testing::TestWithParam<XlaConcatNDTestParam>; typedef
890 TEST_P(XlaConcatNDOpTest, ConcatDimensionZero) { in TEST_P() argument
906 TEST_P(XlaConcatNDOpTest, ConcatDimensionNegative) { in TEST_P() argument
922 TEST_P(XlaConcatNDOpTest, NumInputsMismatch) { in TEST_P() argument
938 TEST_P(XlaConcatNDOpTest, PaddingsLengthMismatch) { in TEST_P() argument
953 TEST_P(XlaConcatNDOpTest, PaddingsNegative) { in TEST_P() argument
969 TEST_P(XlaConcatNDOpTest, InputRank0) { in TEST_P() argument
984 TEST_P(XlaConcatNDOpTest, InputRank9) { in TEST_P() argument
999 TEST_P(XlaConcatNDOpTest, InputRankConcatMismatch) { in TEST_P() argument
1015 TEST_P(XlaConcatNDOpTest, DifferentShapedInputs) { in TEST_P() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sharding_util_ops_test.py | 351 class XlaConcatNDOpTest(xla_test.XLATestCase, parameterized.TestCase): class
|