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_show.h | 42 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader) in TFShow() argument 43 : ckpt_reader_(ckpt_reader) {} in TFShow()
|
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 | 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()
|