static Status ValidateNode()

in source/neuropod/backends/tensorflow/saved_model/loader.cc [53:81]

• 29 lines of code
• 5 McCabe index (conditional complexity)

static Status ValidateNode(const NodeDef &node)
{
    const auto node_iterator = node.attr().find("value");
    if (node_iterator != node.attr().end())
    {
        AttrValue node_value = node_iterator->second;
        if (node_value.has_tensor())
        {
            const PartialTensorShape node_shape(node_value.tensor().tensor_shape());
            if (node_shape.num_elements() < 0)
            {
                return errors::FailedPrecondition("Saved model contains node \"",
                                                  node.name(),
                                                  "\" (op \"",
                                                  node.op(),
                                                  "\") which initializes from a tensor with ",
                                                  node_shape.num_elements(),
                                                  " elements");
            }
        }
    }
    else if (node.op() == "Const")
    {
        return errors::FailedPrecondition("Saved model contains node \"",
                                          node.name(),
                                          "\" which is a constant tensor but no value has been provided");
    }
    return Status::OK();
}