Home
last modified time | relevance | path

Searched refs:ragged_values (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_variant_op_test.h47 const std::vector<VALUE_TYPE>& ragged_values, const bool batched) { in BuildEncodeRaggedTensorGraph() argument
65 AddInputFromArray<VALUE_TYPE>(ragged_values_shape, ragged_values); in BuildEncodeRaggedTensorGraph()
72 const std::vector<VALUE_TYPE>& ragged_values) { in CreateVariantFromRagged() argument
82 test::FillValues<VALUE_TYPE>(&values, ragged_values); in CreateVariantFromRagged()
90 const std::vector<VALUE_TYPE>& ragged_values) { in CreateVariantFromRagged() argument
91 int num_values = ragged_values.size(); in CreateVariantFromRagged()
92 return CreateVariantFromRagged(ragged_splits, {num_values}, ragged_values); in CreateVariantFromRagged()
Dexample_parsing_ops.cc274 OpOutputList ragged_values; in WriteOutput() local
276 TF_RETURN_IF_ERROR(ctx->output_list("ragged_values", &ragged_values)); in WriteOutput()
279 ragged_values.set(d, result.ragged_values[d]); in WriteOutput()
753 context_ragged_values.set(d, context_result.ragged_values[d]); in WriteOutput()
757 feature_list_ragged_values.set(d, feature_list_result.ragged_values[d]); in WriteOutput()
Dragged_tensor_from_variant_op_test.cc62 const std::vector<VALUE_TYPE>& ragged_values) { in CreateVariantFromRagged() argument
72 test::FillValues<VALUE_TYPE>(&values, ragged_values); in CreateVariantFromRagged()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-021.md13 ragged_values = []
27 tf.raw_ops.RaggedCross(ragged_values=ragged_values,
Dtfsa-2021-183.md14 y = tf.raw_ops.RaggedCross(ragged_values=[],
/external/tensorflow/tensorflow/python/ops/
Dparsing_config.py887 ragged_values, argument
892 ragged_values = [
894 for (val, split) in zip(ragged_values, ragged_inner_splits)
897 return ragged_values
901 for (val, split) in zip(ragged_values, ragged_row_splits)
Dparsing_ops.py360 ragged_values, ragged_row_splits) = outputs
363 serialized.shape, ragged_values, ragged_row_splits)
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DParseExampleV2.pbtxt46 name: "ragged_values"
DRaggedCross.pbtxt4 name: "ragged_values"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DParseExampleV2.pbtxt46 name: "ragged_values"
DRaggedCross.pbtxt4 name: "ragged_values"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RaggedCross.pbtxt5 name: "ragged_values"
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.h123 std::vector<Tensor> ragged_values; member
Dexample_proto_fast_parsing.cc1259 result->ragged_values.reserve(config.ragged.size()); in FastParseExample()
1340 result->ragged_values.emplace_back(config.ragged[d].dtype, values_shape); in FastParseExample()
1341 Tensor* values = &result->ragged_values.back(); in FastParseExample()
1492 result->ragged_values.reserve(config.ragged.size()); in FastParseSingleExample()
1522 result->ragged_values.emplace_back(); in FastParseSingleExample()
1756 out = &result->ragged_values[d]; in FastParseSingleExample()
1823 result->ragged_values[d] = in FastParseSingleExample()
2436 context_result->ragged_values[t] = Tensor(allocator, dtype, values_shape); in ParseContextRaggedFeatures()
2439 Tensor& out_values = context_result->ragged_values[t]; in ParseContextRaggedFeatures()
2797 sequence_result->ragged_values[t] = Tensor(allocator, dtype, values_shape); in ParseSequenceRaggedFeatures()
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_cross_op_test.py428 ragged_values=[],
446 ragged_values=[['a']],
Dragged_array_ops.py854 ragged_values=[rt.values for rt in ragged_inputs],
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc695 serialized_ragged.set_values(example_result.ragged_values[d]); in ParseExample()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td484 Variadic<TensorOf<[TF_Float32, TF_Int64, TF_Str]>>:$ragged_values, // len(ragged_value_types)
/external/tensorflow/tensorflow/go/op/
Dwrappers.go30002 …ues []tf.Output, sparse_shapes []tf.Output, dense_values []tf.Output, ragged_values []tf.Output, r…
30036 if ragged_values, idx, err = makeOutputList(op, idx, "ragged_values"); err != nil {
30044 return sparse_indices, sparse_values, sparse_shapes, dense_values, ragged_values, ragged_row_splits
34291 func RaggedCross(scope *Scope, ragged_values []tf.Output, ragged_row_splits []tf.Output, sparse_ind…
34299 …tf.OutputList(ragged_values), tf.OutputList(ragged_row_splits), tf.OutputList(sparse_indices), tf.…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt3285 …argspec: "args=[\'ragged_values\', \'ragged_row_splits\', \'sparse_indices\', \'sparse_values\', \…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt3285 …argspec: "args=[\'ragged_values\', \'ragged_row_splits\', \'sparse_indices\', \'sparse_values\', \…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt31691 name: "ragged_values"
37875 name: "ragged_values"