Searched refs:GetTensorKeys (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 67 OP_REQUIRES_OK(ctx, GetTensorKeys(ctx, "dense_keys", &dense_keys_t)); in Compute()
68 OP_REQUIRES_OK(ctx, GetTensorKeys(ctx, "sparse_keys", &sparse_keys_t)); in Compute()
69 OP_REQUIRES_OK(ctx, GetTensorKeys(ctx, "ragged_keys", &ragged_keys_t)); in Compute()
101 Status GetTensorKeys(OpKernelContext* ctx, StringPiece input_name, in GetTensorKeys() function in tensorflow::ParseExampleOp
|