Fix kokoro build, add guard in IsotonicreRegressionOp benchmark
PiperOrigin-RevId: 327292455 Change-Id: Ic17fe951eff9fea374e1448b5cb1d4d1cba998cd
This commit is contained in:
parent
ad7e6583cd
commit
b8d0e867a2
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user