| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_variant_op_test.cc | 44 const std::vector<VALUE_TYPE>& ragged_values, const bool batched) { in BuildEncodeRaggedTensorGraph() argument
62 AddInputFromArray<VALUE_TYPE>(ragged_values_shape, ragged_values); in BuildEncodeRaggedTensorGraph()
69 const std::vector<VALUE_TYPE>& ragged_values) { in CreateVariantFromRagged() argument
79 test::FillValues<VALUE_TYPE>(&values, ragged_values); in CreateVariantFromRagged()
87 const std::vector<VALUE_TYPE>& ragged_values) { in CreateVariantFromRagged() argument
88 int num_values = ragged_values.size(); in CreateVariantFromRagged()
89 return CreateVariantFromRagged(ragged_splits, {num_values}, ragged_values); in CreateVariantFromRagged()
|
| D | example_parsing_ops.cc | 274 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()
|
| D | ragged_tensor_from_variant_op_test.cc | 62 const std::vector<VALUE_TYPE>& ragged_values) { in CreateVariantFromRagged() argument
72 test::FillValues<VALUE_TYPE>(&values, ragged_values); in CreateVariantFromRagged()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_config.py | 881 ragged_values, argument
886 ragged_values = [
888 for (val, split) in zip(ragged_values, ragged_inner_splits)
891 return ragged_values
895 for (val, split) in zip(ragged_values, ragged_row_splits)
|
| D | parsing_ops.py | 364 ragged_values, ragged_row_splits) = outputs
367 serialized.shape, ragged_values, ragged_row_splits)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ParseExampleV2.pbtxt | 46 name: "ragged_values"
|
| D | RaggedCross.pbtxt | 4 name: "ragged_values"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ParseExampleV2.pbtxt | 46 name: "ragged_values"
|
| D | RaggedCross.pbtxt | 4 name: "ragged_values"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RaggedCross.pbtxt | 5 name: "ragged_values"
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.h | 123 std::vector<Tensor> ragged_values; member
|
| D | example_proto_fast_parsing.cc | 1243 result->ragged_values.reserve(config.ragged.size()); in FastParseExample()
1324 result->ragged_values.emplace_back(config.ragged[d].dtype, values_shape); in FastParseExample()
1325 Tensor* values = &result->ragged_values.back(); in FastParseExample()
1476 result->ragged_values.reserve(config.ragged.size()); in FastParseSingleExample()
1506 result->ragged_values.emplace_back(); in FastParseSingleExample()
1740 out = &result->ragged_values[d]; in FastParseSingleExample()
1807 result->ragged_values[d] = in FastParseSingleExample()
2419 context_result->ragged_values[t] = Tensor(allocator, dtype, values_shape); in ParseContextRaggedFeatures()
2422 Tensor& out_values = context_result->ragged_values[t]; in ParseContextRaggedFeatures()
2779 sequence_result->ragged_values[t] = Tensor(allocator, dtype, values_shape); in ParseSequenceRaggedFeatures()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_array_ops.py | 825 ragged_values=[rt.values for rt in ragged_inputs],
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 684 serialized_ragged.set_values(example_result.ragged_values[d]); in ParseExample()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 526 Variadic<TensorOf<[TF_Float32, TF_Int64, TF_Str]>>:$ragged_values, // len(ragged_value_types)
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 27759 func RaggedCross(scope *Scope, ragged_values []tf.Output, ragged_row_splits []tf.Output, sparse_ind…
27767 …tf.OutputList(ragged_values), tf.OutputList(ragged_row_splits), tf.OutputList(sparse_indices), tf.…
28080 …ues []tf.Output, sparse_shapes []tf.Output, dense_values []tf.Output, ragged_values []tf.Output, r…
28114 if ragged_values, idx, err = makeOutputList(op, idx, "ragged_values"); err != nil {
28122 return sparse_indices, sparse_values, sparse_shapes, dense_values, ragged_values, ragged_row_splits
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3213 …argspec: "args=[\'ragged_values\', \'ragged_row_splits\', \'sparse_indices\', \'sparse_values\', \…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3213 …argspec: "args=[\'ragged_values\', \'ragged_row_splits\', \'sparse_indices\', \'sparse_values\', \…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 28601 name: "ragged_values"
34738 name: "ragged_values"
|