| 79 | } |
| 80 | |
| 81 | Node* SparseIndices(Graph* const g, const int sparse_features_per_group) { |
| 82 | Tensor data(DT_INT64, TensorShape({sparse_features_per_group})); |
| 83 | test::FillFn<int64>(&data, [&](const int i) { return i; }); |
| 84 | return test::graph::Constant(g, data); |
| 85 | } |
| 86 | |
| 87 | Node* SparseExampleIndices(Graph* const g, const int sparse_features_per_group, |
| 88 | const int num_examples) { |
no test coverage detected