Searched refs:VariantTensorDataReader (Results 1 – 15 of 15) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | serialization_utils.cc | 150 VariantTensorDataReader::VariantTensorDataReader( in VariantTensorDataReader() function in tensorflow::data::VariantTensorDataReader
165 Status VariantTensorDataReader::ReadScalar(StringPiece key, in ReadScalar()
172 Status VariantTensorDataReader::ReadScalar(StringPiece name, StringPiece key, in ReadScalar()
177 Status VariantTensorDataReader::ReadScalar(StringPiece key, in ReadScalar()
184 Status VariantTensorDataReader::ReadScalar(StringPiece name, StringPiece key, in ReadScalar()
189 Status VariantTensorDataReader::ReadTensor(StringPiece key, Tensor* val) const { in ReadTensor()
195 Status VariantTensorDataReader::ReadTensor(FunctionLibraryRuntime* flr, in ReadTensor()
202 Status VariantTensorDataReader::ReadTensor(StringPiece name, StringPiece key, in ReadTensor()
207 Status VariantTensorDataReader::ReadTensor(FunctionLibraryRuntime* flr, in ReadTensor()
213 bool VariantTensorDataReader::Contains(StringPiece key) const { in Contains()
[all …]
|
| H A D | serialization_utils_test.cc | 108 VariantTensorDataReader reader(data); in TEST()
137 VariantTensorDataReader reader(data); in TEST()
153 VariantTensorDataReader reader(reader_data); in TEST()
174 VariantTensorDataReader reader(data); in TEST()
190 VariantTensorDataReader reader(reader_data); in TEST()
|
| H A D | serialization_utils.h | 42 class VariantTensorDataReader : public IteratorStateReader {
44 explicit VariantTensorDataReader(
|
| H A D | split_utils_test.cc | 38 VariantTensorDataReader reader(variants); in SaveAndRestore()
|
| H A D | dataset_test_base.cc | 756 VariantTensorDataReader reader(data); in CheckIteratorSaveAndRestore()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | io_ops_test.cc | 272 VariantTensorDataReader reader(data); in TEST_P()
|
| H A D | list_dataset_op_test.cc | 408 VariantTensorDataReader reader(data); in TEST_P()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | cache_dataset_ops_test.cc | 343 VariantTensorDataReader reader(data); in TEST_P()
|
| H A D | repeat_dataset_op_test.cc | 330 VariantTensorDataReader reader(data); in TEST_P()
|
| H A D | tensor_dataset_op_test.cc | 245 VariantTensorDataReader reader(data); in TEST_P()
|
| H A D | tensor_slice_dataset_op_test.cc | 319 VariantTensorDataReader reader(data); in TEST_P()
|
| H A D | sparse_tensor_slice_dataset_op_test.cc | 409 VariantTensorDataReader reader(data); in TEST_P()
|
| H A D | shuffle_dataset_op_test.cc | 535 VariantTensorDataReader reader(data); in TEST_P()
|
| H A D | window_dataset_op_test.cc | 485 VariantTensorDataReader reader(data); in TEST_P()
|
| H A D | iterator_ops.cc | 396 reader_ = std::make_unique<VariantTensorDataReader>(data); in InitFromTensor()
|