Home
last modified time | relevance | path

Searched refs:num_sparse_feature_groups (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsdca_ops_test.cc122 void GetGraphs(const int32 num_examples, const int32 num_sparse_feature_groups, in GetGraphs() argument
134 VarVector(g, num_sparse_feature_groups, sparse_features_per_group); in GetGraphs()
156 VarVector(g, num_sparse_feature_groups, sparse_features_per_group); in GetGraphs()
176 sparse_example_indices.reserve(num_sparse_feature_groups); in GetGraphs()
177 for (int i = 0; i < num_sparse_feature_groups; ++i) { in GetGraphs()
181 sparse_feature_indices.reserve(num_sparse_feature_groups); in GetGraphs()
182 for (int i = 0; i < num_sparse_feature_groups; ++i) { in GetGraphs()
186 sparse_values.reserve(num_sparse_feature_groups); in GetGraphs()
187 for (int i = 0; i < num_sparse_feature_groups; ++i) { in GetGraphs()
207 .Attr("num_sparse_features", num_sparse_feature_groups) in GetGraphs()
[all …]