Searched refs:sparse_keys_t (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 59 std::vector<StringPiece> sparse_keys_t; in Compute() local
68 OP_REQUIRES_OK(ctx, GetTensorKeys(ctx, "sparse_keys", &sparse_keys_t)); in Compute()
72 OP_REQUIRES_OK(ctx, GetInputListKeys(ctx, "sparse_keys", &sparse_keys_t)); in Compute()
74 absl::call_once(flag_, [&dense_keys_t, &sparse_keys_t, &ragged_keys_t]() { in Compute()
76 metrics::RecordParseSparseFeature(sparse_keys_t.size()); in Compute()
84 sparse_keys_t, ragged_keys_t)); in Compute()
87 MakeConfig(dense_keys_t, sparse_keys_t, ragged_keys_t, dense_defaults); in Compute()
129 const std::vector<StringPiece>& sparse_keys_t, in CheckInputShapes() argument
156 if (sparse_keys_t.size() != attrs_.num_sparse) { in CheckInputShapes()
159 sparse_keys_t.size(), " vs. ", attrs_.num_sparse); in CheckInputShapes()
[all …]
|