Home
last modified time | relevance | path

Searched refs:TensorSliceWriter (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_slice_writer_test.cc76 TensorSliceWriter writer(filename, CreateTableTensorSliceBuilder); in TEST()
278 TensorSliceWriter::SaveData(lo_data.data(), lo_data.size(), &ss)); in BytesPerElementHelper()
284 TensorSliceWriter::SaveData(hi_data.data(), hi_data.size(), &ss)); in BytesPerElementHelper()
291 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_BOOL), in TEST()
293 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_BOOL), in TEST()
295 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_FLOAT), in TEST()
297 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_DOUBLE), in TEST()
299 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_COMPLEX64), in TEST()
301 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_COMPLEX128), in TEST()
303 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_INT32), in TEST()
[all …]
Dtensor_slice_writer.h42 class TensorSliceWriter {
53 TensorSliceWriter(const string& filename,
55 virtual ~TensorSliceWriter() {} in ~TensorSliceWriter()
94 TF_DISALLOW_COPY_AND_ASSIGN(TensorSliceWriter);
98 Status TensorSliceWriter::Add(const string& name, const TensorShape& shape, in Add()
163 Status TensorSliceWriter::SaveData(const T* data, int64 num_elements, in SaveData()
180 Status TensorSliceWriter::SaveData(const tstring* data, int64 num_elements,
188 TensorSliceWriter::Builder** builder);
Dtensor_slice_writer.cc36 class TableBuilder : public TensorSliceWriter::Builder {
72 TensorSliceWriter::Builder** builder) { in CreateTableTensorSliceBuilder()
84 TensorSliceWriter::TensorSliceWriter(const string& filename, in TensorSliceWriter() function in tensorflow::checkpoint::TensorSliceWriter
95 Status TensorSliceWriter::Finish() { in Finish()
133 size_t TensorSliceWriter::MaxBytesPerElement(DataType dt) { in MaxBytesPerElement()
179 Status TensorSliceWriter::SaveData(const tstring* data, int64 num_elements, in SaveData()
Dtensor_slice_reader_test.cc56 const TensorSliceWriter::CreateBuilderFunction& create_function, in SimpleFloatHelper()
70 TensorSliceWriter writer(fname, create_function); in SimpleFloatHelper()
92 TensorSliceWriter writer(fname, create_function); in SimpleFloatHelper()
180 const TensorSliceWriter::CreateBuilderFunction& create_function, in SimpleIntXHelper()
195 TensorSliceWriter writer(fname, create_function); in SimpleIntXHelper()
217 TensorSliceWriter writer(fname, create_function); in SimpleIntXHelper()
313 const TensorSliceWriter::CreateBuilderFunction& create_function, in TEST_SIMPLE_INT()
327 TensorSliceWriter writer(fname, create_function); in TEST_SIMPLE_INT()
349 TensorSliceWriter writer(fname, create_function); in TEST_SIMPLE_INT()
415 TensorSliceWriter::Builder* builder; in VersionTest()
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.h37 checkpoint::TensorSliceWriter::CreateBuilderFunction builder_func,
Dsave_restore_tensor.cc42 checkpoint::TensorSliceWriter::CreateBuilderFunction builder_func, in SaveTensors()
84 checkpoint::TensorSliceWriter writer(filename_t.flat<tstring>()(0), in SaveTensors()