Searched refs:CheckpointReader (Results 1 – 13 of 13) sorted by relevance
105 tensorflow::checkpoint::CheckpointReader* reader, in CheckpointReader_GetTensor()124 tensorflow::checkpoint::CheckpointReader* reader,132 %unignore tensorflow::checkpoint::CheckpointReader;133 %unignore tensorflow::checkpoint::CheckpointReader::CheckpointReader;134 %unignore tensorflow::checkpoint::CheckpointReader::~CheckpointReader;135 %rename("debug_string") tensorflow::checkpoint::CheckpointReader::DebugString;136 %rename("get_variable_to_shape_map") tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMa…137 %rename("_GetVariableToDataTypeMap") tensorflow::checkpoint::CheckpointReader::GetVariableToDataTyp…138 %rename("_HasTensor") tensorflow::checkpoint::CheckpointReader::HasTensor;141 %extend tensorflow::checkpoint::CheckpointReader {[all …]
32 CheckpointReader::CheckpointReader(const string& filename, in CheckpointReader() function in tensorflow::checkpoint::CheckpointReader65 bool CheckpointReader::HasTensor(const string& name) const { in HasTensor()73 CheckpointReader::GetVariableToShapeMap() const { in GetVariableToShapeMap()79 CheckpointReader::GetVariableToDataTypeMap() const { in GetVariableToDataTypeMap()84 const string CheckpointReader::DebugString() const { in DebugString()89 void CheckpointReader::GetTensor( in GetTensor()112 CheckpointReader::BuildV2VarMaps() { in BuildV2VarMaps()
40 class CheckpointReader {42 CheckpointReader(const string& filepattern, TF_Status* out_status);76 TF_DISALLOW_COPY_AND_ASSIGN(CheckpointReader);
57 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);60 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);112 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader_;
46 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfTensorTest()47 new checkpoint::CheckpointReader(ckpt_path, status)); in TFProfTensorTest()
68 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfShowTest()69 new checkpoint::CheckpointReader(ckpt_path, status)); in TFProfShowTest()
62 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfStatsTest()63 new checkpoint::CheckpointReader(ckpt_path, status)); in TFProfStatsTest()
42 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader) in TFShow()121 checkpoint::CheckpointReader* ckpt_reader_;
42 explicit TFScope(checkpoint::CheckpointReader* ckpt_reader) in TFScope()
45 explicit TFGraph(checkpoint::CheckpointReader* ckpt_reader) in TFGraph()
52 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()76 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()
195 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in PrintModelAnalysis()
187 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in Run()191 new checkpoint::CheckpointReader(FLAGS_checkpoint_path, status)); in Run()