Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dtensor_slice_writer_test.cc78 TensorSliceWriter writer(filename, CreateTableTensorSliceBuilder); in TEST()
292 TensorSliceWriter::SaveData(lo_data.data(), lo_data.size(), &ss)); in BytesPerElementHelper()
298 TensorSliceWriter::SaveData(hi_data.data(), hi_data.size(), &ss)); in BytesPerElementHelper()
305 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_BOOL), in TEST()
307 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_BOOL), in TEST()
309 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_FLOAT), in TEST()
311 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_DOUBLE), in TEST()
313 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_COMPLEX64), in TEST()
315 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_COMPLEX128), in TEST()
317 EXPECT_EQ(TensorSliceWriter::MaxBytesPerElement(DT_INT32), in TEST()
[all …]
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
104 Status TensorSliceWriter::Finish() { in Finish()
145 size_t TensorSliceWriter::MaxBytesPerElement(DataType dt) { in MaxBytesPerElement()
147 TensorSliceWriter::MaxBytesPerElementOrZero(dt); in MaxBytesPerElement()
155 size_t TensorSliceWriter::MaxBytesPerElementOrZero(DataType dt) { in MaxBytesPerElementOrZero()
200 Status TensorSliceWriter::SaveData(const tstring* data, int64_t num_elements, in SaveData()
Dtensor_slice_writer.h42 class TensorSliceWriter {
53 TensorSliceWriter(const string& filename,
55 virtual ~TensorSliceWriter() {} in ~TensorSliceWriter()
97 TF_DISALLOW_COPY_AND_ASSIGN(TensorSliceWriter);
101 Status TensorSliceWriter::Add(const string& name, const TensorShape& shape, in Add()
166 Status TensorSliceWriter::SaveData(const T* data, int64_t num_elements, in SaveData()
189 Status TensorSliceWriter::SaveData(const tstring* data, int64_t num_elements,
197 TensorSliceWriter::Builder** builder);
Dtensor_slice_reader_test.cc62 const TensorSliceWriter::CreateBuilderFunction& create_function, in SimpleFloatHelper()
76 TensorSliceWriter writer(fname, create_function); in SimpleFloatHelper()
98 TensorSliceWriter writer(fname, create_function); in SimpleFloatHelper()
186 const TensorSliceWriter::CreateBuilderFunction& create_function, in SimpleIntXHelper()
201 TensorSliceWriter writer(fname, create_function); in SimpleIntXHelper()
223 TensorSliceWriter writer(fname, create_function); in SimpleIntXHelper()
360 TensorSliceWriter writer(fname, CreateTableTensorSliceBuilder); in TEST()
388 TensorSliceWriter writer(fname, CreateTableTensorSliceBuilder); in TEST()
417 TensorSliceWriter writer(fname, CreateTableTensorSliceBuilder); in TEST()
442 TensorSliceWriter writer(fname, CreateTableTensorSliceBuilder); in TEST()
[all …]
/external/federated-compute/fcp/aggregation/tensorflow/
Dcheckpoint_writer.cc46 tf::Status AddTensorSlice(tf::checkpoint::TensorSliceWriter* writer, in AddTensorSlice()
55 tf::checkpoint::TensorSliceWriter* writer, const std::string& name, in AddTensorSlice()
73 tf::checkpoint::TensorSliceWriter::CreateBuilderFunction create_builder_fn) in CheckpointWriter()
Dcheckpoint_writer.h45 ::tensorflow::checkpoint::TensorSliceWriter::CreateBuilderFunction
55 ::tensorflow::checkpoint::TensorSliceWriter tensorflow_writer_;
/external/federated-compute/fcp/tensorflow/
Dappend_slices_op.cc81 tensorflow::checkpoint::TensorSliceWriter::CreateBuilderFunction in SaveTensors()
127 tensorflow::checkpoint::TensorSliceWriter writer(filename, in SaveTensors()
235 class TableBuilder : public tensorflow::checkpoint::TensorSliceWriter::Builder {
282 tensorflow::checkpoint::TensorSliceWriter::Builder** builder) { in CreateAppendingTensorSliceBuilder()
492 tensorflow::checkpoint::TensorSliceWriter::Builder* raw_builder; in LoadAndMergeAppendedSlices()
495 std::unique_ptr<tensorflow::checkpoint::TensorSliceWriter::Builder> builder( in LoadAndMergeAppendedSlices()
532 tensorflow::checkpoint::TensorSliceWriter::Builder** builder) { in Compute()
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.h37 checkpoint::TensorSliceWriter::CreateBuilderFunction builder_func,
Dsave_restore_tensor.cc45 checkpoint::TensorSliceWriter::CreateBuilderFunction builder_func, in SaveTensors()
87 checkpoint::TensorSliceWriter writer(filename_t.flat<tstring>()(0), in SaveTensors()
/external/federated-compute/fcp/client/engine/
Dexample_query_plan_engine.cc58 absl::Status WriteSlice(tf::checkpoint::TensorSliceWriter& slice_writer, in WriteSlice()
99 tf::checkpoint::TensorSliceWriter slice_writer( in WriteCheckpoint()