Searched refs:VariantTensorDataReader (Results 1 – 13 of 13) sorted by relevance
151 VariantTensorDataReader::VariantTensorDataReader( in VariantTensorDataReader() function in tensorflow::data::VariantTensorDataReader162 Status VariantTensorDataReader::ReadScalar(StringPiece key, int64* val) { in ReadScalar()166 Status VariantTensorDataReader::ReadScalar(StringPiece key, string* val) { in ReadScalar()170 Status VariantTensorDataReader::ReadTensor(StringPiece key, Tensor* val) { in ReadTensor()174 bool VariantTensorDataReader::Contains(StringPiece key) { in Contains()179 Status VariantTensorDataReader::ReadScalarInternal(StringPiece key, T* val) { in ReadScalarInternal()187 Status VariantTensorDataReader::ReadTensorInternal(StringPiece key, in ReadTensorInternal()
61 class VariantTensorDataReader : public IteratorStateReader {63 explicit VariantTensorDataReader(const VariantTensorData* data);
57 VariantTensorDataReader reader(&data); in TEST()71 VariantTensorDataReader reader(&data); in TEST()
323 VariantTensorDataReader reader(&data); in TEST_P()
322 VariantTensorDataReader reader(&data); in TEST_P()
401 VariantTensorDataReader reader(&data); in TEST_P()
359 std::unique_ptr<VariantTensorDataReader> reader = in Decode()360 absl::make_unique<VariantTensorDataReader>(tensor_data.get()); in Decode()
502 VariantTensorDataReader reader(&data); in TEST_F()
494 VariantTensorDataReader reader(&data); in TEST_P()
602 VariantTensorDataReader reader(&data); in TEST_F()
520 VariantTensorDataReader reader(&data); in TEST_P()
513 VariantTensorDataReader reader(&data); in TEST_P()
528 VariantTensorDataReader reader(&data); in TEST_P()