Searched refs:ragged_outer_splits (Results 1 – 3 of 3) sorted by relevance
125 std::vector<Tensor> ragged_outer_splits; // For SequenceExamples member
2782 sequence_result->ragged_outer_splits[t] = in ParseSequenceRaggedFeatures()2796 ? sequence_result->ragged_outer_splits[t].vec<int32>().data() in ParseSequenceRaggedFeatures()2800 ? sequence_result->ragged_outer_splits[t].vec<int64>().data() in ParseSequenceRaggedFeatures()2889 const auto& outer_splits = sequence_result->ragged_outer_splits[t]; in ParseSequenceRaggedFeatures()3029 context_result->ragged_outer_splits.resize(context_config.ragged.size()); in FastParseSequenceExample()3036 sequence_result->ragged_outer_splits.resize(sequence_config.ragged.size()); in FastParseSequenceExample()
759 d, feature_list_result.ragged_outer_splits[d]); in WriteOutput()