TensorFlow C API能否按文件而非目录加载含有效信息的计算图？

1. 为什么加载frozen_graph.pb找不到指定Operation？ #

核心原因是**StatefulPartitionedCall是SavedModel格式特有的入口节点**，而frozen_graph.pb只是把训练变量转为常量的计算图，没有SavedModel的签名定义（SignatureDef）结构，自然不会生成这个节点。它的入口节点都是你训练时定义的原始节点（比如自定义的input、output，或是predict这类业务节点）。

你当前的加载流程本身没问题，只是找错了节点名称。可以用两种方式确认frozen_graph里的节点：

• 用TensorBoard加载pb文件查看完整节点结构；
• 在代码里遍历所有Operation打印名称排查：

// 遍历图中所有节点并打印名称
size_t num_ops = TF_GraphNumOperations(graph);
for (size_t i = 0; i < num_ops; i++) {
    const TF_Operation* op = TF_GraphGetOperation(graph, i);
    printf("节点名称: %s\n", TF_OperationName(op));
}

2. TensorFlow C API支持加载frozen_graph.pb吗？ #

完全支持，你的加载逻辑（读文件到TF_Buffer→导入图定义→创建会话）是标准流程，问题出在节点名称的差异，而非API不支持。

给你贴一段关键步骤的正确示例：

// 读取frozen_graph.pb到TF_Buffer
TF_Buffer* graph_def = read_file_to_tf_buffer("frozen_graph.pb");
TF_Graph* graph = TF_NewGraph();
TF_Status* status = TF_NewStatus();
TF_ImportGraphDefOptions* opts = TF_NewImportGraphDefOptions();

// 导入图定义
TF_GraphImportGraphDef(graph, graph_def, opts, status);
if (TF_GetCode(status) != TF_OK) {
    fprintf(stderr, "导入图失败: %s\n", TF_Message(status));
    // 错误处理
}

// 创建会话
TF_SessionOptions* sess_opts = TF_NewSessionOptions();
TF_Session* sess = TF_NewSession(graph, sess_opts, status);
if (TF_GetCode(status) != TF_OK) {
    fprintf(stderr, "创建会话失败: %s\n", TF_Message(status));
    // 错误处理
}

// 用正确的节点名称获取Operation
const TF_Operation* input_op = TF_GraphOperationByName(graph, "你的输入节点名");
const TF_Operation* output_op = TF_GraphOperationByName(graph, "你的输出节点名");

3. 要保留StatefulPartitionedCall的话，C++ API可行吗？ #

如果一定要用这类SavedModel特有的入口节点，直接用C++ API加载SavedModel目录更合适，它对SavedModel的签名支持更完善，可以直接通过签名获取目标节点：

示例代码片段：

#include <tensorflow/cc/saved_model/loader.h>
#include <tensorflow/cc/saved_model/tag_constants.h>

// 加载SavedModel目录
tensorflow::SavedModelBundle bundle;
tensorflow::Status status = tensorflow::LoadSavedModel(
    tensorflow::SessionOptions(), tensorflow::RunOptions(),
    "/path/to/saved_model_dir", {tensorflow::kSavedModelTagServe}, &bundle);

if (status.ok()) {
    // 通过签名获取StatefulPartitionedCall对应的输入输出节点
    auto signature_def = bundle.meta_graph_def.signature_def()["serving_default"];
    const std::string& input_name = signature_def.inputs["input"].name();
    const std::string& output_name = signature_def.outputs["output"].name();
    
    // 执行推理示例
    tensorflow::Tensor input_tensor(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, 28, 28, 1}));
    // 填充输入数据...
    std::vector<tensorflow::Tensor> outputs;
    status = bundle.session->Run({{input_name, input_tensor}}, {output_name}, {}, &outputs);
}

如果必须加载单个frozen_graph.pb文件，C++ API同样能处理，但还是得用图中的原始节点名称，和C API的情况一致。

内容的提问来源于stack exchange，提问作者Xyt