Searched refs:dense_configs_ (Results 1 – 1 of 1) sorted by relevance
762 ParseConfig(context, kDenseConfigName, &dense_configs_); in QuantileBucketsOp()763 OP_REQUIRES(context, dense_configs_.size() == num_dense_features_, in QuantileBucketsOp()831 QuantileStream stream(dense_configs_[dense_index].eps(), batch_size); in Compute()839 stream, dense_configs_[dense_index].num_quantiles()); in Compute()847 const int64 num_features = sparse_configs_.size() + dense_configs_.size(); in Compute()857 std::vector<QuantileConfig> dense_configs_; member in tensorflow::QuantileBucketsOp