Home
last modified time | relevance | path

Searched refs:VariantTensorDataWriter (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc196 Status VariantTensorDataWriter::WriteScalar(StringPiece key, const int64 val) { in WriteScalar()
200 Status VariantTensorDataWriter::WriteScalar(StringPiece key, in WriteScalar()
205 Status VariantTensorDataWriter::WriteTensor(StringPiece key, in WriteTensor()
210 Status VariantTensorDataWriter::Flush() { in Flush()
220 Status VariantTensorDataWriter::WriteScalarInternal(StringPiece key, in WriteScalarInternal()
227 Status VariantTensorDataWriter::WriteTensorInternal(StringPiece key, in WriteTensorInternal()
Ddataset_utils.h81 class VariantTensorDataWriter : public IteratorStateWriter {
84 explicit VariantTensorDataWriter(VariantTensorData* data) : data_(data) {} in VariantTensorDataWriter() function
Dconcatenate_dataset_op_test.cc270 VariantTensorDataWriter writer(&data); in TEST_F()
320 VariantTensorDataWriter writer(&data); in TEST_P()
Dzip_dataset_op_test.cc254 VariantTensorDataWriter writer(&data); in TEST_F()
319 VariantTensorDataWriter writer(&data); in TEST_P()
Ddataset_utils_test.cc50 VariantTensorDataWriter writer(&data); in TEST()
Drange_dataset_op_test.cc253 VariantTensorDataWriter writer(&data); in TEST_F()
398 VariantTensorDataWriter writer(&data); in TEST_P()
Dtensor_dataset_op_test.cc345 VariantTensorDataWriter writer(&data); in TEST_P()
491 VariantTensorDataWriter writer(&data); in TEST_P()
Dsparse_tensor_slice_dataset_op_test.cc322 VariantTensorDataWriter writer(&data); in TEST_F()
499 VariantTensorDataWriter writer(&data); in TEST_F()
Dtake_dataset_op_test.cc359 VariantTensorDataWriter writer(&data); in TEST_F()
525 VariantTensorDataWriter writer(&data); in TEST_P()
Drepeat_dataset_op_test.cc352 VariantTensorDataWriter writer(&data); in TEST_F()
517 VariantTensorDataWriter writer(&data); in TEST_P()
Dmap_dataset_op_test.cc335 VariantTensorDataWriter writer(&data); in TEST_F()
510 VariantTensorDataWriter writer(&data); in TEST_P()
Dtensor_slice_dataset_op_test.cc381 VariantTensorDataWriter writer(&data); in TEST_F()
599 VariantTensorDataWriter writer(&data); in TEST_F()
Diterator_ops.cc345 VariantTensorDataWriter writer(data_.get()); in InitializeFromIterator()