aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/boosted_trees/lib/utils/batch_features.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/boosted_trees/lib/utils/batch_features.h')
-rw-r--r--tensorflow/contrib/boosted_trees/lib/utils/batch_features.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/boosted_trees/lib/utils/batch_features.h b/tensorflow/contrib/boosted_trees/lib/utils/batch_features.h
index da5e744851..7815fa049a 100644
--- a/tensorflow/contrib/boosted_trees/lib/utils/batch_features.h
+++ b/tensorflow/contrib/boosted_trees/lib/utils/batch_features.h
@@ -48,9 +48,9 @@ class BatchFeatures {
Status GetFeatureColumnSizes(int64* const num_dense_float_features,
int64* const num_sparse_float_features,
int64* const num_sparse_int_features) const {
- QCHECK_NE(num_dense_float_features, nullptr);
- QCHECK_NE(num_sparse_float_features, nullptr);
- QCHECK_NE(num_sparse_int_features, nullptr);
+ QCHECK_NE(num_dense_float_features, (int64*) nullptr);
+ QCHECK_NE(num_sparse_float_features, (int64*) nullptr);
+ QCHECK_NE(num_sparse_int_features, (int64*) nullptr);
*num_dense_float_features = dense_float_feature_columns_.size();
*num_sparse_float_features = sparse_float_feature_columns_.size();
*num_sparse_int_features = sparse_int_feature_columns_.size();