Fix kokoro build, add guard in IsotonicreRegressionOp benchmark

PiperOrigin-RevId: 327292455
Change-Id: Ic17fe951eff9fea374e1448b5cb1d4d1cba998cd
This commit is contained in:
Josip Djolonga 2020-08-18 13:14:58 -07:00 committed by TensorFlower Gardener
parent ad7e6583cd
commit b8d0e867a2

View File

@ -105,6 +105,8 @@ TEST_F(IsotonicRegressionOpTest, Decreasing) {
test::ExpectTensorEqual<int>(expected_ord, *GetOutput((1)));
}
#ifdef PLATFORM_GOOGLE
static void BM_IncreasingSequence(benchmark::State& state) {
int batch_size = state.range(0);
int input_size = state.range(1);
@ -135,5 +137,7 @@ BENCHMARK(BM_IncreasingSequence)
->Args({1 << 9, 1 << 10})
->Args({1 << 10, 1 << 10});
#endif // PLATFORM_GOOGLE
} // namespace
} // namespace tensorflow