Searched refs:dense_feature_lengths (Results 1 – 3 of 3) sorted by relevance
122 std::vector<Tensor>* dense_feature_lengths);
1762 Result* feature_list_result, std::vector<Tensor>* dense_feature_lengths) { in FastParseSequenceExample() argument1766 DCHECK(dense_feature_lengths != nullptr); in FastParseSequenceExample()1999 dense_feature_lengths->resize(feature_list_config.dense.size()); in FastParseSequenceExample()2216 (*dense_feature_lengths)[t] = Tensor(DT_INT64, dense_length_shape); in FastParseSequenceExample()2217 int64* out_lengths = (*dense_feature_lengths)[t].flat<int64>().data(); in FastParseSequenceExample()
324 std::vector<Tensor> dense_feature_lengths; in Compute() local366 &context_result, &feature_list_result, &dense_feature_lengths)); in Compute()405 feature_list_dense_lengths.set(d, dense_feature_lengths[d]); in Compute()