Searched refs:num_dense_float_features (Results 1 – 4 of 4) sorted by relevance
48 Status GetFeatureColumnSizes(int64* const num_dense_float_features, in GetFeatureColumnSizes() argument51 QCHECK_NE(num_dense_float_features, static_cast<int64*>(nullptr)); in GetFeatureColumnSizes()54 *num_dense_float_features = dense_float_feature_columns_.size(); in GetFeatureColumnSizes()57 if (*num_dense_float_features == 0 && *num_sparse_float_features == 0 && in GetFeatureColumnSizes()
34 auto num_dense_float_features = dense_float_features_list.size(); in Initialize() local37 QCHECK(num_dense_float_features + num_sparse_float_features + in Initialize()43 dense_float_feature_columns_.reserve(num_dense_float_features); in Initialize()44 for (uint32 dense_feat_idx = 0; dense_feat_idx < num_dense_float_features; in Initialize()
27 tensorflow::random::SimplePhilox* rng, uint32 num_dense_float_features, in RandomlyInitializeBatchFeatures() argument34 for (int i = 0; i < num_dense_float_features; ++i) { in RandomlyInitializeBatchFeatures()
37 tensorflow::random::SimplePhilox* rng, uint32 num_dense_float_features,