| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 347 if params.ragged_keys and serialized.shape.ndims is None:
355 ragged_keys=params.ragged_keys,
374 zip(params.sparse_keys + params.dense_keys + params.ragged_keys,
566 if context_params.ragged_keys:
569 if feature_list_params.ragged_keys:
616 has_ragged = context.ragged_keys or feature_list.ragged_keys
631 context_ragged_keys=context.ragged_keys,
634 feature_list_ragged_keys=feature_list.ragged_keys,
684 context.sparse_keys + context.dense_keys + context.ragged_keys,
690 feature_list.ragged_keys, feature_list_sparse_tensors +
[all …]
|
| D | parsing_config.py | 398 ragged_keys=None, argument
411 ragged_keys = [] if ragged_keys is None else ragged_keys
422 self.ragged_keys = ragged_keys
463 return len(self.dense_keys) + len(self.sparse_keys) + len(self.ragged_keys)
599 if key in self.ragged_keys:
600 original_value_type = self.ragged_value_types[self.ragged_keys.index(key)]
601 original_split_type = self.ragged_split_types[self.ragged_keys.index(key)]
609 self.ragged_keys.append(key)
636 if len(self.ragged_value_types) != len(self.ragged_keys):
639 (len(self.ragged_value_types), len(self.ragged_keys)))
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | parsing_ops.py | 59 self._ragged_keys = params.ragged_keys
81 for (key, value_type, splits_type) in zip(params.ragged_keys,
97 ragged_keys=self._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 | 155 ragged_keys = fetched[2].tolist()
159 assert len(ragged_keys) == num_ragged
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ParseExampleDatasetV2.pbtxt | 78 name: "ragged_keys"
|
| D | ParseExampleDataset.pbtxt | 78 name: "ragged_keys"
|
| D | ParseExampleV2.pbtxt | 20 name: "ragged_keys"
|
| /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 | 4119 ragged_keys = pfor_input.unstacked_input(4)
4136 ragged_keys=ragged_keys,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 3576 …d(%serialized: tensor<32x!tf.string>, %names : tensor<32x!tf.string>, %ragged_keys : tensor<2x!tf.…
3578 …ExampleV2"(%serialized, %names, %empty_str_vector, %empty_str_vector, %ragged_keys) {dense_shapes …
3584 …s(%serialized: tensor<32x!tf.string>, %names : tensor<32x!tf.string>, %ragged_keys : tensor<2x!tf.…
3587 …ExampleV2"(%serialized, %names, %empty_str_vector, %empty_str_vector, %ragged_keys) {dense_shapes …
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 513 TF_StrTensor:$ragged_keys,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2845 …, \'dense_shapes\', \'output_types\', \'output_shapes\', \'sloppy\', \'ragged_keys\', \'ragged_val…
2849 …se_shapes\', \'output_types\', \'output_shapes\', \'deterministic\', \'ragged_keys\', \'ragged_val…
2853 …argspec: "args=[\'serialized\', \'names\', \'sparse_keys\', \'dense_keys\', \'ragged_keys\', \'den…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2845 …, \'dense_shapes\', \'output_types\', \'output_shapes\', \'sloppy\', \'ragged_keys\', \'ragged_val…
2849 …se_shapes\', \'output_types\', \'output_shapes\', \'deterministic\', \'ragged_keys\', \'ragged_val…
2853 …argspec: "args=[\'serialized\', \'names\', \'sparse_keys\', \'dense_keys\', \'ragged_keys\', \'den…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 28399 name: "ragged_keys"
28516 name: "ragged_keys"
28575 name: "ragged_keys"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 28080 ….Output, names tf.Output, sparse_keys tf.Output, dense_keys tf.Output, ragged_keys tf.Output, dens…
28088 serialized, names, sparse_keys, dense_keys, ragged_keys, tf.OutputList(dense_defaults),
|