Searched refs:sparse_configs_ (Results 1 – 1 of 1) sorted by relevance
766 ParseConfig(context, kSparseConfigName, &sparse_configs_); in QuantileBucketsOp()767 OP_REQUIRES(context, sparse_configs_.size() == num_sparse_features_, in QuantileBucketsOp()805 if (i < sparse_configs_.size()) { in Compute()811 QuantileStream stream(sparse_configs_[sparse_index].eps(), in Compute()823 stream, sparse_configs_[sparse_index].num_quantiles()); in Compute()828 const int64 dense_index = i - sparse_configs_.size(); in Compute()847 const int64 num_features = sparse_configs_.size() + dense_configs_.size(); in Compute()858 std::vector<QuantileConfig> sparse_configs_; member in tensorflow::QuantileBucketsOp