| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader.cc | 39 TensorSliceReader::Table::~Table() {} in ~Table()
42 class TensorSliceReaderTable : public TensorSliceReader::Table {
72 TensorSliceReader::Table** result) { in OpenTableTensorSliceReader()
100 TensorSliceReader::TensorSliceReader(const string& filepattern) in TensorSliceReader() function in tensorflow::checkpoint::TensorSliceReader
101 : TensorSliceReader(filepattern, OpenTableTensorSliceReader, in TensorSliceReader()
104 TensorSliceReader::TensorSliceReader(const string& filepattern, in TensorSliceReader() function in tensorflow::checkpoint::TensorSliceReader
106 : TensorSliceReader(filepattern, std::move(open_function), kLoadAllShards) { in TensorSliceReader()
109 TensorSliceReader::TensorSliceReader(const string& filepattern, in TensorSliceReader() function in tensorflow::checkpoint::TensorSliceReader
142 void TensorSliceReader::LoadShard(int shard) const { in LoadShard()
186 void TensorSliceReader::LoadAllShards() const { in LoadAllShards()
[all …]
|
| D | tensor_slice_reader_cache.h | 42 const TensorSliceReader* GetReader(
44 TensorSliceReader::OpenTableFunction open_function,
61 const TensorSliceReader* GetReader(
63 TensorSliceReader::OpenTableFunction open_function, int preferred_shard);
68 typedef Status (*OpenFuncType)(const string&, TensorSliceReader::Table**);
74 std::unordered_map<string, std::pair<OpenFuncType, TensorSliceReader*>>
|
| D | tensor_slice_reader_cache.cc | 32 const TensorSliceReader* TensorSliceReaderCacheWrapper::GetReader( in GetReader()
34 TensorSliceReader::OpenTableFunction open_function, in GetReader()
52 const TensorSliceReader* TensorSliceReaderCache::GetReader( in GetReader()
54 TensorSliceReader::OpenTableFunction open_function, int preferred_shard) { in GetReader()
80 TensorSliceReader* reader = nullptr; in GetReader()
87 TensorSliceReader* tmp_reader( in GetReader()
88 new TensorSliceReader(filepattern, open_function, preferred_shard)); in GetReader()
|
| D | tensor_slice_reader.h | 54 class TensorSliceReader {
65 TensorSliceReader(const string& filepattern);
66 TensorSliceReader(const string& filepattern, OpenTableFunction open_function);
67 TensorSliceReader(const string& filepattern, OpenTableFunction open_function,
69 virtual ~TensorSliceReader();
138 TF_DISALLOW_COPY_AND_ASSIGN(TensorSliceReader);
142 TensorSliceReader::Table** table);
145 bool TensorSliceReader::CopySliceData(const string& name, in CopySliceData()
|
| D | tensor_slice_reader_test.cc | 63 TensorSliceReader::OpenTableFunction open_function) { in SimpleFloatHelper()
122 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleFloatHelper()
187 TensorSliceReader::OpenTableFunction open_function, in SimpleIntXHelper()
247 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleIntXHelper()
376 TensorSliceReader reader(fname, OpenTableTensorSliceReader); in TEST()
404 TensorSliceReader reader(fname, OpenTableTensorSliceReader); in TEST()
434 TensorSliceReader reader(fname, OpenTableTensorSliceReader); in TEST()
457 TensorSliceReader reader(fname, OpenTableTensorSliceReader); in TEST()
479 TensorSliceReader reader(fname, OpenTableTensorSliceReader); in TEST()
491 const TensorSliceReader::OpenTableFunction& open_function) { in CachedTensorSliceReaderTesterHelper()
[all …]
|
| D | tensor_slice_writer_test.cc | 42 static void GetData(TensorSliceReader::Table* table, const string& name,
129 void TensorSliceWriteTestHelper::GetData(TensorSliceReader::Table* table, in GetData()
146 TensorSliceReader::Table* tptr; in CheckEntries()
148 std::unique_ptr<TensorSliceReader::Table> table(tptr); in CheckEntries()
|
| /external/tensorflow/tensorflow/c/ |
| D | checkpoint_reader.h | 32 class TensorSliceReader; variable
49 const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;
53 const TensorSliceReader::VarToDataTypeMap& GetVariableToDataTypeMap() const;
65 std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>,
66 std::unique_ptr<TensorSliceReader::VarToDataTypeMap> >
70 std::unique_ptr<TensorSliceReader> reader_;
73 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map_;
74 std::unique_ptr<TensorSliceReader::VarToDataTypeMap> var_to_data_type_map_;
|
| D | checkpoint_reader.cc | 30 class TensorSliceReader;
52 reader_.reset(new TensorSliceReader(filename)); in CheckpointReader()
58 new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap())); in CheckpointReader()
59 var_to_data_type_map_.reset(new TensorSliceReader::VarToDataTypeMap( in CheckpointReader()
71 const TensorSliceReader::VarToShapeMap&
77 const TensorSliceReader::VarToDataTypeMap&
109 std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>,
110 std::unique_ptr<TensorSliceReader::VarToDataTypeMap>>
134 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map( in BuildV2VarMaps()
135 new TensorSliceReader::VarToShapeMap); in BuildV2VarMaps()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | restore_op.cc | 34 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards; in RestoreOp()
60 preferred_shard_ = checkpoint::TensorSliceReader::kLoadAllShards; in RestoreSliceOp()
|
| D | save_restore_tensor.cc | 151 checkpoint::TensorSliceReader::OpenTableFunction open_func, in RestoreTensor()
175 std::unique_ptr<checkpoint::TensorSliceReader> allocated_reader; in RestoreTensor()
177 const checkpoint::TensorSliceReader* reader = nullptr; in RestoreTensor()
184 allocated_reader.reset(new checkpoint::TensorSliceReader( in RestoreTensor()
|
| D | save_restore_tensor.h | 53 checkpoint::TensorSliceReader::OpenTableFunction open_func,
|
| D | save_op_test.cc | 134 checkpoint::TensorSliceReader reader(filename, in TEST_F()
467 checkpoint::TensorSliceReader reader(filename, in TEST_F()
611 checkpoint::TensorSliceReader reader(filename, in TEST_F()
|