Home
last modified time | relevance | path

Searched refs:ragged_splits (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_variant_op_test.cc42 const std::vector<std::vector<SPLIT_TYPE>>& ragged_splits, in BuildEncodeRaggedTensorGraph() argument
47 int64 num_splits = ragged_splits.size(); in BuildEncodeRaggedTensorGraph()
58 for (const auto& splits : ragged_splits) { in BuildEncodeRaggedTensorGraph()
67 const std::vector<std::vector<SPLIT_TYPE>>& ragged_splits, in CreateVariantFromRagged() argument
71 for (auto ragged_split : ragged_splits) { in CreateVariantFromRagged()
86 const std::vector<std::vector<SPLIT_TYPE>>& ragged_splits, in CreateVariantFromRagged() argument
89 return CreateVariantFromRagged(ragged_splits, {num_values}, ragged_values); in CreateVariantFromRagged()
Dexample_parsing_ops.cc275 OpOutputList ragged_splits; in WriteOutput() local
277 TF_RETURN_IF_ERROR(ctx->output_list("ragged_row_splits", &ragged_splits)); in WriteOutput()
280 ragged_splits.set(d, result.ragged_splits[d]); in WriteOutput()
754 context_ragged_splits.set(d, context_result.ragged_splits[d]); in WriteOutput()
761 d, feature_list_result.ragged_splits[d]); in WriteOutput()
Dragged_tensor_from_variant_op_test.cc60 const std::vector<std::vector<SPLIT_TYPE>>& ragged_splits, in CreateVariantFromRagged() argument
64 for (auto ragged_split : ragged_splits) { in CreateVariantFromRagged()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc1244 result->ragged_splits.reserve(config.ragged.size()); in FastParseExample()
1313 result->ragged_splits.emplace_back(config.ragged[d].splits_dtype, in FastParseExample()
1315 Tensor* row_splits = &result->ragged_splits.back(); in FastParseExample()
1477 result->ragged_splits.reserve(config.ragged.size()); in FastParseSingleExample()
1509 result->ragged_splits.emplace_back(splits_dtype, TensorShape({0})); in FastParseSingleExample()
2420 context_result->ragged_splits[t] = in ParseContextRaggedFeatures()
2426 ? context_result->ragged_splits[t].vec<int32>().data() in ParseContextRaggedFeatures()
2430 ? context_result->ragged_splits[t].vec<int64>().data() in ParseContextRaggedFeatures()
2465 context_result->ragged_splits[t].vec<int32>().data() in ParseContextRaggedFeatures()
2467 context_result->ragged_splits[t].vec<int64>().data(); in ParseContextRaggedFeatures()
[all …]
Dexample_proto_fast_parsing.h124 std::vector<Tensor> ragged_splits; member
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc683 serialized_ragged.append_splits(example_result.ragged_splits[d]); in ParseExample()