Function ConvertTensorShapeToString

src/armnnUtils/PrototxtConversions.cpp:32–43

Converts an TensorShape into Prototxt representation

/// Converts an TensorShape into Prototxt representation
std::string ConvertTensorShapeToString(const armnn::TensorShape& shape)
{
    std::stringstream ss;
    for (unsigned int i = 0 ; i < shape.GetNumDimensions() ; i++)
    {
        ss << "dim {\n";
        ss << "size: " << std::to_string(shape[i]) << "\n";
        ss << "}\n";
    }
    return ss.str();

}
} // namespace armnnUtils