文档
注册
评分
提单
论坛
小AI

NodeDefBuilder类

NodeDef是一个Protocol Buffer,用于框架UT/ST测试时,构造节点信息,包含了节点的名称、类型、输入和输出等信息。

本类提供了NodeDef的构造接口,包含添加NodeDef的Input,Output,Attr。开发者不直接调用下列接口。

表1 NodeDefBuilder类成员函数

接口名

接口功能

static std::shared_ptr<NodeDef> CreateNodeDef();

创建算子NodeDef,返回NodeDef类型指针。

NodeDefBuilder(NodeDef *nodeDef, std::string name, std::string opName);

NodeDef的构造函数,通过入参nodeDef指针,name算子名称以及opName算子类型构造算子的NodeDef。返回NodeDefBuilder类型对象。

NodeDefBuilder& Input(const InputOutputNode& input);

NodeDef添加input tensor。通过入参InputOutputNode中提供node的shape,type,format信息构造NodeDef信息。

NodeDefBuilder& Output(const InputOutputNode& output);

NodeDef添加output tensor。通过入参InputOutputNode中提供提供node的shape、type、format信息构造NodeDef信息。

NodeDefBuilder& Attr(std::string name, int32_t value);

NodeDef添加属性,属性类型是int32。

NodeDefBuilder& Attr(std::string name, int64_t value);

NodeDef添加属性,属性类型是int64。

NodeDefBuilder& Attr(std::string name, float value);

NodeDef添加属性,属性类型是float。

NodeDefBuilder& Attr(std::string name, double value);

NodeDef添加属性,属性类型是double 。

NodeDefBuilder& Attr(std::string name, boolvalue);

NodeDef添加属性,属性类型是bool。

NodeDefBuilder& Attr(std::string name, aicpu::DataType value);

NodeDef添加属性,属性类型是aicpu::DataType。

NodeDefBuilder& Attr(std::string name, const std::vector<bool> &value);

NodeDef添加属性,属性类型是std::vector<bool>。

NodeDefBuilder& Attr(std::string name, const std::string &value);

NodeDef添加属性,属性类型是std::string。

NodeDefBuilder& Attr(std::string name, const std::vector<std::string> &value);

NodeDef添加属性,属性类型是std::vector<std::string>。

NodeDefBuilder& Attr(std::string name, const std::vector<int64_t> &value);

NodeDef添加属性,属性类型是std::vector<int64_t>。

NodeDefBuilder& Attr(std::string name, const std::vector<float> &value);

NodeDef添加属性,属性类型是std::vector<float>。

NodeDefBuilder& Attr(std::string name, const std::vector<aicpu::DataType> &value);

NodeDef添加属性,属性类型是std::vector<aicpu::DataType>。

NodeDefBuilder& Attr(std::string name, const std::vector<int64_t> &shape, std::string type);

NodeDef添加属性,属性类型是std::vector<int64_t>,入参type需要设置为"shape"。

NodeDefBuilder& Attr(std::string name, const std::vector<std::vector<int64_t>> &shapeLists, std::string type);

NodeDef添加属性,属性类型是std::vector<std::vector<int64_t>>,入参type需要设置为"shape_list"。

NodeDefBuilder& Attr(std::string name, aicpu::Tensor *tensor);

NodeDef添加属性,属性类型是aicpu::Tensor。

NodeDefBuilder& Attr(std::string name, std::vector<aicpu::Tensor *> &tensors);

NodeDef添加属性,属性类型是std::vector<aicpu::Tensor *>。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词