Searched refs:ckpt_reader (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 157 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader, in ReadTensorFromCheckpoint() argument
159 if (ckpt_reader) { in ReadTensorFromCheckpoint()
172 TF_RETURN_IF_ERROR(ckpt_reader->LookupSlice( in ReadTensorFromCheckpoint()
176 ckpt_reader->Lookup(GetMonolithicTensorKey(tensor_name), tensor)); in ReadTensorFromCheckpoint()
184 std::unique_ptr<BundleReader>* ckpt_reader) { in InitializeCheckpointReader() argument
187 ckpt_reader->reset(new BundleReader(Env::Default(), input_checkpoint)); in InitializeCheckpointReader()
188 TF_RETURN_IF_ERROR((*ckpt_reader)->status()); in InitializeCheckpointReader()
228 const std::unique_ptr<BundleReader>& ckpt_reader, in SparsifyGatherInternal() argument
257 [&ckpt_reader, &any_match_found, &init_table_node_names, in SparsifyGatherInternal()
341 weights_node.name(), ckpt_reader, in SparsifyGatherInternal()
[all …]
|
| D | sparsify_gather_test.cc | 36 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader,
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_scope.h | 42 explicit TFScope(checkpoint::CheckpointReader* ckpt_reader) in TFScope() argument
43 : TFShow(ckpt_reader), root_(nullptr) {} in TFScope()
|
| D | tfprof_tensor_test.cc | 46 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfTensorTest() local
52 std::move(op_log_pb), std::move(ckpt_reader))); in TFProfTensorTest()
|
| D | tfprof_graph.h | 45 explicit TFGraph(checkpoint::CheckpointReader* ckpt_reader) in TFGraph() argument
46 : TFShow(ckpt_reader), root_(nullptr) {} in TFGraph()
|
| D | tfprof_stats.h | 57 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);
60 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);
|
| D | tfprof_stats.cc | 52 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats() argument
55 ckpt_reader_(std::move(ckpt_reader)) { in TFStats()
76 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats() argument
79 ckpt_reader_(std::move(ckpt_reader)) { in TFStats()
|
| D | tfprof_show_test.cc | 68 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfShowTest() local
74 std::move(op_log_pb), std::move(ckpt_reader))); in TFProfShowTest()
|
| D | tfprof_stats_test.cc | 62 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfStatsTest() local
68 std::move(op_log_pb), std::move(ckpt_reader))); in TFProfStatsTest()
|
| D | tfprof_show.h | 42 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader) in TFShow() argument
43 : ckpt_reader_(ckpt_reader) {} in TFShow()
|
| D | print_model_analysis.cc | 195 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in PrintModelAnalysis() local
198 std::move(op_log_ptr), std::move(ckpt_reader)); in PrintModelAnalysis()
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | profiler.cc | 187 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in Run() local
190 ckpt_reader.reset( in Run()
202 tf_stat.reset(new TFStats(FLAGS_profile_path, std::move(ckpt_reader))); in Run()
232 std::move(ckpt_reader))); in Run()
|