experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use XLA_TEST_P instead of TEST_P in iota_test

Browse Source 
This address github PR 30518.

PiperOrigin-RevId: 257513086
This commit is contained in:
Sanjoy Das 2019-07-10 17:08:29 -07:00 committed by TensorFlower Gardener
parent 2246b38305
commit 6e03f9430a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/xla/tests/iota_test.cc
View File

@ -34,7 +34,7 @@ class IotaR1Test
: public ClientLibraryTestBase, : public ClientLibraryTestBase,
public ::testing::WithParamInterface<std::tuple<PrimitiveType, int>> {}; public ::testing::WithParamInterface<std::tuple<PrimitiveType, int>> {};
TEST_P(IotaR1Test, DoIt) { XLA_TEST_P(IotaR1Test, DoIt) {
const auto& spec = GetParam(); const auto& spec = GetParam();
const auto element_type = std::get<0>(spec); const auto element_type = std::get<0>(spec);
const int64 num_elements = std::get<1>(spec); const int64 num_elements = std::get<1>(spec);
@ -63,7 +63,7 @@ class IotaR2Test : public ClientLibraryTestBase,
public ::testing::WithParamInterface< public ::testing::WithParamInterface<
std::tuple<PrimitiveType, int, int>> {}; std::tuple<PrimitiveType, int, int>> {};
TEST_P(IotaR2Test, DoIt) { XLA_TEST_P(IotaR2Test, DoIt) {
const auto& spec = GetParam(); const auto& spec = GetParam();
const auto element_type = std::get<0>(spec); const auto element_type = std::get<0>(spec);
const int64 num_elements = std::get<1>(spec); const int64 num_elements = std::get<1>(spec);
@ -95,7 +95,7 @@ class IotaR3Test : public ClientLibraryTestBase,
public ::testing::WithParamInterface< public ::testing::WithParamInterface<
std::tuple<PrimitiveType, int, int>> {}; std::tuple<PrimitiveType, int, int>> {};
TEST_P(IotaR3Test, DoIt) { XLA_TEST_P(IotaR3Test, DoIt) {
const auto& spec = GetParam(); const auto& spec = GetParam();
const auto element_type = std::get<0>(spec); const auto element_type = std::get<0>(spec);
const int64 num_elements = std::get<1>(spec); const int64 num_elements = std::get<1>(spec);