Searched refs:sparse_features_per_group (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops_test.cc | 81 Node* SparseIndices(Graph* const g, const int sparse_features_per_group) { in SparseIndices() argument
82 Tensor data(DT_INT64, TensorShape({sparse_features_per_group})); in SparseIndices()
87 Node* SparseExampleIndices(Graph* const g, const int sparse_features_per_group, in SparseExampleIndices() argument
95 Node* SparseFeatureIndices(Graph* const g, const int sparse_features_per_group, in SparseFeatureIndices() argument
100 &data, [&](const int i) { return i % sparse_features_per_group; }); in SparseFeatureIndices()
123 const int32 sparse_features_per_group, in GetGraphs() argument
134 VarVector(g, num_sparse_feature_groups, sparse_features_per_group); in GetGraphs()
137 Node* const multi_zero = Zeros(g, sparse_features_per_group); in GetGraphs()
156 VarVector(g, num_sparse_feature_groups, sparse_features_per_group); in GetGraphs()
164 NodeBuilder::NodeOut(SparseIndices(g, sparse_features_per_group))); in GetGraphs()
[all …]
|