| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 343 if params.ragged_keys and serialized.shape.ndims is None:
351 ragged_keys=params.ragged_keys,
370 zip(params.sparse_keys + params.dense_keys + params.ragged_keys,
563 if context_params.ragged_keys:
566 if feature_list_params.ragged_keys:
613 has_ragged = context.ragged_keys or feature_list.ragged_keys
628 context_ragged_keys=context.ragged_keys,
631 feature_list_ragged_keys=feature_list.ragged_keys,
681 context.sparse_keys + context.dense_keys + context.ragged_keys,
687 feature_list.ragged_keys, feature_list_sparse_tensors +
[all …]
|
| D | parsing_config.py | 397 ragged_keys=None, argument
410 ragged_keys = [] if ragged_keys is None else ragged_keys
421 self.ragged_keys = ragged_keys
462 return len(self.dense_keys) + len(self.sparse_keys) + len(self.ragged_keys)
605 if key in self.ragged_keys:
606 original_value_type = self.ragged_value_types[self.ragged_keys.index(key)]
607 original_split_type = self.ragged_split_types[self.ragged_keys.index(key)]
615 self.ragged_keys.append(key)
639 if len(self.ragged_value_types) != len(self.ragged_keys):
643 if len(self.ragged_split_types) != len(self.ragged_keys):
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | parsing_ops.py | 56 self._ragged_keys = params.ragged_keys
78 for (key, value_type, splits_type) in zip(params.ragged_keys,
94 ragged_keys=self._ragged_keys,
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| D | derived_attrs.mlir | 9 %ragged_keys: tensor<0x!tf_type.string>,
13 … "tf.ParseExampleV2"(%serialized, %names, %sparse_keys, %dense_keys, %ragged_keys, %dense_default)
|
| D | fallback.mlir | 24 …%ragged_keys = "tf.Const"() {device = "/device:CPU:0", dtype = !tf_type.string, value = dense<""> …
38 …%outputs:8 = "tf.ParseExampleV2"(%serialized, %names, %sparse_keys, %dense_keys, %ragged_keys, %de…
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
| D | parse_example_v2.pbtxt | 136 name: "ParseExample/ParseExampleV2/ragged_keys"
164 input: "ParseExample/ParseExampleV2/ragged_keys"
|
| D | test13.pbtxt | 187 name: "ParseExample/ParseExampleV2/ragged_keys"
215 input: "ParseExample/ParseExampleV2/ragged_keys"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ParseExampleDatasetV2.pbtxt | 78 name: "ragged_keys"
202 name: "ragged_keys"
339 name: "ragged_keys"
|
| D | ParseExampleDataset.pbtxt | 78 name: "ragged_keys"
202 name: "ragged_keys"
339 name: "ragged_keys"
|
| D | ParseExampleV2.pbtxt | 20 name: "ragged_keys"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ParseExampleV2.pbtxt | 36 name: "ragged_keys"
76 given in ragged_keys (where `num_ragged = sparse_keys.size()`).
85 given in ragged_keys (where `num_ragged = sparse_keys.size()`).
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | parse_example_v2.pbtxt | 142 name: "ParseExample/ParseExampleV2/ragged_keys"
170 input: "ParseExample/ParseExampleV2/ragged_keys"
|
| /external/tensorflow/tensorflow/python/util/ |
| D | example_parser_configuration.py | 153 ragged_keys = fetched[2].tolist()
157 assert len(ragged_keys) == num_ragged
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ParseExampleDatasetV2.pbtxt | 78 name: "ragged_keys"
|
| D | ParseExampleV2.pbtxt | 20 name: "ragged_keys"
|
| D | ParseExampleDataset.pbtxt | 155 name: "ragged_keys"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 570 const Tensor* ragged_keys, in MakeContextConfig() argument
585 ragged_keys in MakeContextConfig()
586 ? gtl::ArraySlice<tstring>(ragged_keys->flat<tstring>().data(), in MakeContextConfig()
629 const Tensor* ragged_keys, in MakeFeatureListConfig() argument
644 ragged_keys in MakeFeatureListConfig()
645 ? gtl::ArraySlice<tstring>(ragged_keys->flat<tstring>().data(), in MakeFeatureListConfig()
|
| D | example_parsing_ops_test.cc | 259 auto& ragged_keys = (bm_type == kRagged) ? keys_t : empty_keys; in ParseExampleV2() local
268 .Input(test::graph::Constant(g, ragged_keys)) in ParseExampleV2()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 223 std::vector<string> ragged_keys, in Dataset() argument
231 ragged_keys_(std::move(ragged_keys)), in Dataset()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 4436 ragged_keys = pfor_input.unstacked_input(4)
4453 ragged_keys=ragged_keys,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 3680 …zed: tensor<32x!tf_type.string>, %names : tensor<32x!tf_type.string>, %ragged_keys : tensor<2x!tf_…
3682 …ExampleV2"(%serialized, %names, %empty_str_vector, %empty_str_vector, %ragged_keys) {dense_shapes …
3688 …zed: tensor<32x!tf_type.string>, %names : tensor<32x!tf_type.string>, %ragged_keys : tensor<2x!tf_…
3691 …ExampleV2"(%serialized, %names, %empty_str_vector, %empty_str_vector, %ragged_keys) {dense_shapes …
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 471 TF_StrTensor:$ragged_keys,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2917 …, \'dense_shapes\', \'output_types\', \'output_shapes\', \'sloppy\', \'ragged_keys\', \'ragged_val…
2921 …se_shapes\', \'output_types\', \'output_shapes\', \'deterministic\', \'ragged_keys\', \'ragged_val…
2925 …argspec: "args=[\'serialized\', \'names\', \'sparse_keys\', \'dense_keys\', \'ragged_keys\', \'den…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2917 …, \'dense_shapes\', \'output_types\', \'output_shapes\', \'sloppy\', \'ragged_keys\', \'ragged_val…
2921 …se_shapes\', \'output_types\', \'output_shapes\', \'deterministic\', \'ragged_keys\', \'ragged_val…
2925 …argspec: "args=[\'serialized\', \'names\', \'sparse_keys\', \'dense_keys\', \'ragged_keys\', \'den…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 31469 name: "ragged_keys"
31606 name: "ragged_keys"
31665 name: "ragged_keys"
|