Home
last modified time | relevance | path

Searched refs:parsing_ops (Results 1 – 25 of 65) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/io_ops/
H A Dparsing_ops_test.py37 from tensorflow.python.ops import parsing_ops
85 self.evaluate(parsing_ops.parse_example(**kwargs))
88 parsing_ops.parse_example(**kwargs)
91 out = parsing_ops.parse_example(**kwargs)
101 if isinstance(f, parsing_ops.FixedLenFeature) and f.shape is not None:
103 elif isinstance(f, parsing_ops.VarLenFeature):
140 parsing_ops.VarLenFeature(dtypes.int64),
142 parsing_ops.FixedLenFeature(
145 parsing_ops.FixedLenFeature(
148 parsing_ops.FixedLenFeature(
[all …]
H A Dparse_single_example_op_test.py29 from tensorflow.python.ops import parsing_ops
93 out = parsing_ops.parse_single_example(**kwargs)
98 out = parsing_ops.parse_single_example(**kwargs)
102 out_with_example_name = parsing_ops.parse_single_example(
112 if isinstance(f, parsing_ops.FixedLenFeature) and f.shape is not None:
114 elif isinstance(f, parsing_ops.VarLenFeature):
147 parsing_ops.VarLenFeature(dtypes.int64),
149 parsing_ops.FixedLenFeature(
152 parsing_ops.FixedLenFeature(
155 parsing_ops.FixedLenFeature(
[all …]
H A DBUILD40 "//tensorflow/python:parsing_ops",
68 "//tensorflow/python:parsing_ops",
85 "//tensorflow/python:parsing_ops",
H A Ddecode_csv_op_test.py22 from tensorflow.python.ops import parsing_ops
30 decode = parsing_ops.decode_csv(**args)
40 decode = parsing_ops.decode_csv(**args)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Dparse_example_dataset_test.py24 from tensorflow.python.data.experimental.ops import parsing_ops as contrib_parsing_ops
36 from tensorflow.python.ops import parsing_ops
102 if isinstance(f, parsing_ops.FixedLenFeature) and f.shape is not None:
106 elif isinstance(f, parsing_ops.VarLenFeature):
135 parsing_ops.VarLenFeature(dtypes.int64),
137 parsing_ops.FixedLenFeature(
140 parsing_ops.FixedLenFeature(
143 parsing_ops.FixedLenFeature(
153 parsing_ops.VarLenFeature(dtypes.int64),
155 parsing_ops.FixedLenFeature(
[all …]
H A Dmake_batched_features_dataset_test.py30 from tensorflow.python.ops import parsing_ops
90 "file": parsing_ops.FixedLenFeature([], dtypes.int64),
91 "record": parsing_ops.FixedLenFeature([], dtypes.int64),
95 .map(lambda x: parsing_ops.parse_single_example(x, features))
241 "file": parsing_ops.FixedLenFeature([], dtypes.int64),
242 "record": parsing_ops.FixedLenFeature([], dtypes.int64),
243 "keywords": parsing_ops.VarLenFeature(dtypes.string),
244 "label": parsing_ops.FixedLenFeature([], dtypes.string),
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/image_ops/
H A Ddecode_raw_op_test.py22 from tensorflow.python.ops import parsing_ops
36 decode = parsing_ops.decode_raw(in_bytes, dtype)
42 parsing_ops.decode_raw(["A", "a"], dtypes.uint8))
46 parsing_ops.decode_raw(["wer", "XYZ"], dtypes.uint8))
51 self.evaluate(parsing_ops.decode_raw(["short", "longer"], dtypes.uint8))
56 parsing_ops.decode_raw(["AaBC"], dtypes.uint16))
61 self.evaluate(parsing_ops.decode_raw(["123", "456"], dtypes.int16))
66 parsing_ops.decode_raw(
70 parsing_ops.decode_raw(
73 parsing_ops.decode_raw([b"\x00\x00\x80?\x00\x00\x00@"],
[all …]
H A Ddecode_compressed_op_test.py24 from tensorflow.python.ops import parsing_ops
46 decompressed = parsing_ops.decode_compressed(
55 return parsing_ops.decode_compressed(
68 decompressed = parsing_ops.decode_compressed(in_bytes,
70 return parsing_ops.decode_raw(decompressed, out_type=dtypes.int16)
H A DBUILD44 "//tensorflow/python:parsing_ops",
97 "//tensorflow/python:parsing_ops",
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dparsing_ops.py22 from tensorflow.python.ops import parsing_ops
46 self._features = parsing_ops._prepend_none_dimension(features)
48 params = parsing_ops._ParseOpParams.from_features(self._features, [
49 parsing_ops.VarLenFeature, parsing_ops.SparseFeature,
50 parsing_ops.FixedLenFeature, parsing_ops.FixedLenSequenceFeature,
51 parsing_ops.RaggedFeature
149 isinstance(feature, parsing_ops.SparseFeature) or
150 isinstance(feature, parsing_ops.RaggedFeature)
155 lambda x: parsing_ops._construct_tensors_for_composite_features(
H A DBUILD235 name = "parsing_ops",
236 srcs = ["parsing_ops.py"],
242 "//tensorflow/python:parsing_ops",
301 ":parsing_ops",
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dfeature_column_test.py27 from tensorflow.python.ops import parsing_ops
55 'aaa': parsing_ops.VarLenFeature(dtypes.int64)
80 self.assertEqual({'aaa': parsing_ops.VarLenFeature(dtypes.int64)},
99 'aaa': parsing_ops.VarLenFeature(dtypes.int64)
197 'aaa': parsing_ops.VarLenFeature(dtypes.int64)
200 'bbb': parsing_ops.VarLenFeature(dtypes.int64)
235 'aaa': parsing_ops.VarLenFeature(dtypes.int64)
238 'bbb': parsing_ops.VarLenFeature(dtypes.int64)
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dexample_parser_configuration_test.py23 from tensorflow.python.ops import parsing_ops
83 'x': parsing_ops.FixedLenFeature([1], dtypes.float32, 33.0),
84 'y': parsing_ops.VarLenFeature(dtypes.string)
86 result = parsing_ops.parse_example(examples, feature_to_type)
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/
H A DBUILD36 "//tensorflow/python:parsing_ops",
73 "//tensorflow/python:parsing_ops",
141 "//tensorflow/python:parsing_ops",
185 "//tensorflow/python:parsing_ops",
222 "//tensorflow/python:parsing_ops",
240 "//tensorflow/python:parsing_ops",
H A Dfeature_column_v2_test.py41 from tensorflow.python.ops import parsing_ops
335 'aaa': parsing_ops.FixedLenFeature((2, 3), dtype=dtypes.int32)
347 features = parsing_ops.parse_example(
370 features = parsing_ops.parse_example(
515 'aaa': parsing_ops.FixedLenFeature((2,), dtype=dtypes.int32)
540 features = parsing_ops.parse_example(
819 'aaa': parsing_ops.VarLenFeature(dtypes.string)
825 'aaa': parsing_ops.VarLenFeature(dtypes.int32)
838 features = parsing_ops.parse_example(
1073 'a': parsing_ops.FixedLenFeature((2,), dtype=dtypes.int32),
[all …]
H A Dfeature_column_test.py47 from tensorflow.python.ops import parsing_ops
260 'aaa': parsing_ops.FixedLenFeature((2, 3), dtype=dtypes.int32)
271 features = parsing_ops.parse_example(
292 features = parsing_ops.parse_example(
413 'aaa': parsing_ops.FixedLenFeature((2,), dtype=dtypes.int32)
437 features = parsing_ops.parse_example(
686 'aaa': parsing_ops.VarLenFeature(dtypes.string)
692 'aaa': parsing_ops.VarLenFeature(dtypes.int32)
703 features = parsing_ops.parse_example(
948 'a': parsing_ops.FixedLenFeature((2,), dtype=dtypes.int32),
[all …]
H A Dsequence_feature_column_integration_test.py25 from tensorflow.python.ops import parsing_ops
77 context, seq_features = parsing_ops.parse_single_sequence_example(
H A Dfeature_column.py152 from tensorflow.python.ops import parsing_ops
2362 parsing_ops.FixedLenFeature(self.shape, self.dtype,
2777 return {self.key: parsing_ops.VarLenFeature(self.dtype)}
2828 return {self.key: parsing_ops.VarLenFeature(self.dtype)}
2880 return {self.key: parsing_ops.VarLenFeature(self.dtype)}
2932 return {self.key: parsing_ops.VarLenFeature(dtypes.int64)}
2989 config[self.weight_feature_key] = parsing_ops.VarLenFeature(self.dtype)
3044 config.update({key: parsing_ops.VarLenFeature(dtypes.string)})
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dtf_record_test_base.py26 from tensorflow.python.ops import parsing_ops
57 "file": parsing_ops.FixedLenFeature([], dtypes.int64),
58 "record": parsing_ops.FixedLenFeature([], dtypes.int64),
59 "keywords": parsing_ops.VarLenFeature(dtypes.string),
60 "label": parsing_ops.FixedLenFeature([], dtypes.string),
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
H A Dgrappler_test.py29 from tensorflow.python.ops import parsing_ops
41 features = {"x": parsing_ops.VarLenFeature(dtypes.int64)}
42 parsed = parsing_ops.parse_single_example(serialized, features)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/
H A Dcomposite_tensor_ops_test.py28 from tensorflow.python.ops import parsing_ops
89 proto = parsing_ops.SerializeTensor(tensor=encoded)
90 parsed = parsing_ops.ParseTensor(serialized=proto, out_type=dtypes.variant)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
H A Dcsv_dataset_benchmark.py24 from tensorflow.python.ops import parsing_ops
74 …dataset = dataset.map(lambda l: parsing_ops.decode_csv(l, **kwargs)) # pylint: disable=cell-var-f…
88 …dataset = dataset.map(lambda l: parsing_ops.decode_csv(l, **kwargs)) # pylint: disable=cell-var-f…
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
H A Dstring_to_number_op_test.py20 from tensorflow.python.ops import parsing_ops
32 output = parsing_ops.string_to_number(
/aosp_15_r20/external/tensorflow/tensorflow/python/tools/
H A Dfreeze_graph_test.py34 from tensorflow.python.ops import parsing_ops
130 feature_name: parsing_ops.FixedLenFeature(shape=[],
133 features = parsing_ops.parse_example(examples, feature_configs)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor_test_ops.py23 from tensorflow.python.ops import parsing_ops
120 parsing_ops.decode_compressed,

123