Searched refs:use_cudnn_batchnorm (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc53 public ::testing::WithParamInterface<bool /*use_cudnn_batchnorm*/> {
367 bool use_cudnn_batchnorm; member in struct:xla::__anon25717::BatchNormTestParam
376 // Don't print use_cudnn_batchnorm when it's false, because most backends
378 if (p.use_cudnn_batchnorm) {
379 os << ", use_cudnn_batchnorm=true";
392 GetParam().use_cudnn_batchnorm);
402 random_value_var, /*use_cudnn_batchnorm=*/false};
407 p.use_cudnn_batchnorm = true;

Completed in 148 milliseconds