Home
last modified time | relevance | path

Searched refs:dense_feature_lengths (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.h158 std::vector<Tensor>* dense_feature_lengths, bool is_batch = true);
Dexample_proto_fast_parsing.cc2484 std::vector<Tensor>* dense_feature_lengths) { in ParseSequenceDenseFeatures() argument
2517 (*dense_feature_lengths)[t] = in ParseSequenceDenseFeatures()
2519 int64* out_lengths = (*dense_feature_lengths)[t].flat<int64>().data(); in ParseSequenceDenseFeatures()
2913 std::vector<Tensor>* dense_feature_lengths, in FastParseSequenceExample() argument
2918 DCHECK(dense_feature_lengths != nullptr); in FastParseSequenceExample()
3037 dense_feature_lengths->resize(sequence_config.dense.size()); in FastParseSequenceExample()
3054 allocator, sequence_result, dense_feature_lengths)); in FastParseSequenceExample()
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops.cc474 std::vector<Tensor> dense_feature_lengths; in Compute() local
479 &context_result, &feature_list_result, &dense_feature_lengths, in Compute()
483 dense_feature_lengths, ctx)); in Compute()
687 const std::vector<Tensor>& dense_feature_lengths, in WriteOutput() argument
724 feature_list_dense_lengths.set(d, dense_feature_lengths[d]); in WriteOutput()