Searched refs:num_sparse_float_features (Results 1 – 4 of 4) sorted by relevance
35 auto num_sparse_float_features = sparse_float_feature_indices_list.size(); in Initialize() local37 QCHECK(num_dense_float_features + num_sparse_float_features + in Initialize()64 sparse_float_feature_columns_.reserve(num_sparse_float_features); in Initialize()66 sparse_float_feature_values_list.size() == num_sparse_float_features && in Initialize()67 sparse_float_feature_shapes_list.size() == num_sparse_float_features, in Initialize()69 for (uint32 sparse_feat_idx = 0; sparse_feat_idx < num_sparse_float_features; in Initialize()
49 int64* const num_sparse_float_features, in GetFeatureColumnSizes() argument52 QCHECK_NE(num_sparse_float_features, static_cast<int64*>(nullptr)); in GetFeatureColumnSizes()55 *num_sparse_float_features = sparse_float_feature_columns_.size(); in GetFeatureColumnSizes()57 if (*num_dense_float_features == 0 && *num_sparse_float_features == 0 && in GetFeatureColumnSizes()
28 uint32 num_sparse_float_features, double sparsity_lo, double sparsity_hi, in RandomlyInitializeBatchFeatures() argument48 for (int i = 0; i < num_sparse_float_features; ++i) { in RandomlyInitializeBatchFeatures()
38 uint32 num_sparse_float_features, double sparsity_lo, double sparsity_hi,