Function CreateDefaultValueNode

serving/processor/framework/graph_optimizer.cc:833–875

create 1-D const op, filled by zeros

// create 1-D const op, filled by zeros
Status CreateDefaultValueNode(Graph* graph, int dim_size,
                              Node** new_node, DataType type,
                              const std::string& name) {
  NodeDef const_def;
  const_def.set_name(name);
  const_def.set_op("Const");
  auto* attr = const_def.mutable_attr();
  (*attr)["dtype"].set_type(type);
  Tensor const_tensor(type, TensorShape({dim_size}));
  for (int i = 0; i < dim_size; ++i) {
    switch (type) {
      case DT_INT32: {
        const_tensor.vec<int>()(i) = 0;
        break;
      }
      case DT_INT64: {
        const_tensor.vec<int64>()(i) = 0;
        break;
      }
      case DT_FLOAT: {
        const_tensor.vec<float>()(i) = 0.0;
        break;
      }
      case DT_DOUBLE: {
        const_tensor.vec<double>()(i) = 0.0;
        break;
      }
      case DT_STRING: {
        const_tensor.vec<std::string>()(i) = "";
        break;
      }
      default: {
        LOG(FATAL) << "CreateDefaultValueNode not support type : "
                   << type;
      }
    }
  }
  const_tensor.AsProtoTensorContent((*attr)["value"].mutable_tensor());

  Status s_add_node;
  *new_node = graph->AddNode(const_def, &s_add_node);
  return s_add_node;
}

Status GetNodeAttr(const Node* node,
                   const std::string& attr_name,