Searched refs:dense_keys_t (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 58 std::vector<StringPiece> dense_keys_t; in Compute() local
67 OP_REQUIRES_OK(ctx, GetTensorKeys(ctx, "dense_keys", &dense_keys_t)); in Compute()
71 OP_REQUIRES_OK(ctx, GetInputListKeys(ctx, "dense_keys", &dense_keys_t)); in Compute()
74 absl::call_once(flag_, [&dense_keys_t, &sparse_keys_t, &ragged_keys_t]() { in Compute()
75 metrics::RecordParseDenseFeature(dense_keys_t.size()); in Compute()
83 ctx, CheckInputShapes(serialized, names, dense_defaults, dense_keys_t, in Compute()
87 MakeConfig(dense_keys_t, sparse_keys_t, ragged_keys_t, dense_defaults); in Compute()
128 const std::vector<StringPiece>& dense_keys_t, in CheckInputShapes() argument
151 if (dense_keys_t.size() != attrs_.num_dense) { in CheckInputShapes()
154 dense_keys_t.size(), " vs. ", attrs_.num_dense); in CheckInputShapes()
[all …]
|