Searched refs:CheckpointReader (Results 1 – 16 of 16) sorted by relevance
23 from tensorflow.python.util._pywrap_checkpoint_reader import CheckpointReader57 CheckpointReader.get_variable_to_dtype_map = get_variable_to_dtype_map63 CheckpointReader.has_tensor = has_tensor69 return CheckpointReader.CheckpointReader_GetTensor(77 CheckpointReader.get_tensor = get_tensor95 return CheckpointReader(compat.as_bytes(filepattern))
103 tensorflow::checkpoint::CheckpointReader* reader, const string& name) { in CheckpointReader_GetTensor()124 py::class_<tensorflow::checkpoint::CheckpointReader> checkpoint_reader_class( in PYBIND11_MODULE()134 std::make_unique<tensorflow::checkpoint::CheckpointReader>( in PYBIND11_MODULE()140 [](tensorflow::checkpoint::CheckpointReader& self) { in PYBIND11_MODULE()144 &tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap) in PYBIND11_MODULE()146 &tensorflow::checkpoint::CheckpointReader::GetVariableToDataTypeMap) in PYBIND11_MODULE()147 .def("_HasTensor", &tensorflow::checkpoint::CheckpointReader::HasTensor) in PYBIND11_MODULE()
32 CheckpointReader::CheckpointReader(const string& filename, TF_Status* status) in CheckpointReader() function in tensorflow::checkpoint::CheckpointReader64 bool CheckpointReader::HasTensor(const string& name) const { in HasTensor()72 CheckpointReader::GetVariableToShapeMap() const { in GetVariableToShapeMap()78 CheckpointReader::GetVariableToDataTypeMap() const { in GetVariableToDataTypeMap()83 const string CheckpointReader::DebugString() const { in DebugString()88 void CheckpointReader::GetTensor( in GetTensor()111 CheckpointReader::BuildV2VarMaps() { in BuildV2VarMaps()
40 class CheckpointReader {42 CheckpointReader(const string& filename, TF_Status* status);76 TF_DISALLOW_COPY_AND_ASSIGN(CheckpointReader);
336 struct TF_CheckpointReader : public tensorflow::checkpoint::CheckpointReader {337 using tensorflow::checkpoint::CheckpointReader::CheckpointReader;
56 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);59 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader);111 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()
41 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader) in TFShow()120 checkpoint::CheckpointReader* ckpt_reader_;
62 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfStatsTest()63 new checkpoint::CheckpointReader(ckpt_path, status)); in TFProfStatsTest()
42 explicit TFScope(checkpoint::CheckpointReader* ckpt_reader) in TFScope()
45 explicit TFGraph(checkpoint::CheckpointReader* ckpt_reader) in TFGraph()
55 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()78 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()
200 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in PrintModelAnalysis()
120 tensorflow::checkpoint::CheckpointReader121 tensorflow::checkpoint::CheckpointReader::Init122 tensorflow::checkpoint::CheckpointReader::DebugString123 tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap124 tensorflow::checkpoint::CheckpointReader::GetVariableToDataTypeMap125 tensorflow::checkpoint::CheckpointReader::GetTensor126 tensorflow::checkpoint::CheckpointReader::HasTensor
189 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in Run()193 new checkpoint::CheckpointReader(FLAGS_checkpoint_path, status)); in Run()