| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader.cc | 38 TensorSliceReader::Table::~Table() {} in ~Table()
41 class TensorSliceReaderTable : public TensorSliceReader::Table {
71 TensorSliceReader::Table** result) { in OpenTableTensorSliceReader()
99 TensorSliceReader::TensorSliceReader(const string& filepattern) in TensorSliceReader() function in tensorflow::checkpoint::TensorSliceReader
100 : TensorSliceReader(filepattern, OpenTableTensorSliceReader, in TensorSliceReader()
103 TensorSliceReader::TensorSliceReader(const string& filepattern, in TensorSliceReader() function in tensorflow::checkpoint::TensorSliceReader
105 : TensorSliceReader(filepattern, std::move(open_function), kLoadAllShards) { in TensorSliceReader()
108 TensorSliceReader::TensorSliceReader(const string& filepattern, in TensorSliceReader() function in tensorflow::checkpoint::TensorSliceReader
141 void TensorSliceReader::LoadShard(int shard) const { in LoadShard()
181 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 | 57 TensorSliceReader::OpenTableFunction open_function) { in SimpleFloatHelper()
116 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleFloatHelper()
181 TensorSliceReader::OpenTableFunction open_function, in SimpleIntXHelper()
241 TensorSliceReader reader(filepattern, std::move(open_function)); in SimpleIntXHelper()
314 const TensorSliceReader::OpenTableFunction& open_function) { in TEST_SIMPLE_INT()
374 const TensorSliceReader* reader = cache.GetReader( in TEST_SIMPLE_INT()
375 filepattern, open_function, TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
390 const TensorSliceReader* reader2 = cache.GetReader( in TEST_SIMPLE_INT()
391 filepattern, open_function, TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
395 TensorSliceReader::kLoadAllShards); in TEST_SIMPLE_INT()
[all …]
|
| D | tensor_slice_writer_test.cc | 40 static void GetData(TensorSliceReader::Table* table, const string& name,
127 void TensorSliceWriteTestHelper::GetData(TensorSliceReader::Table* table, in GetData()
144 TensorSliceReader::Table* tptr; in CheckEntries()
146 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 | 145 checkpoint::TensorSliceReader::OpenTableFunction open_func, in RestoreTensor()
162 std::unique_ptr<checkpoint::TensorSliceReader> allocated_reader; in RestoreTensor()
164 const checkpoint::TensorSliceReader* reader = nullptr; in RestoreTensor()
171 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()
|