Searched refs:ckpt_reader (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 156 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader, in ReadTensorFromCheckpoint() argument
158 if (ckpt_reader) { in ReadTensorFromCheckpoint()
171 TF_RETURN_IF_ERROR(ckpt_reader->LookupSlice( in ReadTensorFromCheckpoint()
175 ckpt_reader->Lookup(GetMonolithicTensorKey(tensor_name), tensor)); in ReadTensorFromCheckpoint()
183 std::unique_ptr<BundleReader>* ckpt_reader) { in InitializeCheckpointReader() argument
186 ckpt_reader->reset(new BundleReader(Env::Default(), input_checkpoint)); in InitializeCheckpointReader()
187 TF_RETURN_IF_ERROR((*ckpt_reader)->status()); in InitializeCheckpointReader()
227 const std::unique_ptr<BundleReader>& ckpt_reader, in SparsifyGatherInternal() argument
256 [&ckpt_reader, &any_match_found, &init_table_node_names, in SparsifyGatherInternal()
340 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 | 41 explicit TFScope(checkpoint::CheckpointReader* ckpt_reader) in TFScope() argument
42 : TFShow(ckpt_reader), root_(nullptr) {} in TFScope()
|
| D | tfprof_tensor_test.cc | 43 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfTensorTest() local
49 std::move(op_log_pb), std::move(ckpt_reader))); in TFProfTensorTest()
|
| D | tfprof_graph.h | 43 explicit TFGraph(checkpoint::CheckpointReader* ckpt_reader) in TFGraph() argument
44 : TFShow(ckpt_reader), root_(nullptr) {} in TFGraph()
|
| D | tfprof_stats.h | 56 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);
59 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);
|
| D | tfprof_stats.cc | 54 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats() argument
57 ckpt_reader_(std::move(ckpt_reader)) { in TFStats()
77 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats() argument
80 ckpt_reader_(std::move(ckpt_reader)) { in TFStats()
|
| D | tfprof_show_test.cc | 65 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfShowTest() local
71 std::move(op_log_pb), std::move(ckpt_reader))); in TFProfShowTest()
|
| D | tfprof_stats_test.cc | 58 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfStatsTest() local
64 std::move(op_log_pb), std::move(ckpt_reader))); in TFProfStatsTest()
|
| D | tfprof_show.h | 40 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader) in TFShow() argument
41 : ckpt_reader_(ckpt_reader) {} in TFShow()
|
| D | print_model_analysis.cc | 196 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in PrintModelAnalysis() local
199 std::move(op_log_ptr), std::move(ckpt_reader)); in PrintModelAnalysis()
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | profiler.cc | 184 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in Run() local
187 ckpt_reader.reset( in Run()
199 tf_stat.reset(new TFStats(FLAGS_profile_path, std::move(ckpt_reader))); in Run()
227 std::move(ckpt_reader))); in Run()
|