接口名称 |
功能说明 |
---|---|
IsBasicType(const DataType dtype) |
判断给定dype是否是基础类型。 |
IsNumberType(const DataType dtype) |
判断给定dype是否是数值类型。 |
IsRealNumberType(const DataType dtype) |
判断给定dype是否是实数类型。 |
TypeSize(DataType dataType) |
获取给定dtype的大小。 |
ToDataType(const std::string &dataTypeStr) |
将给定的字符串转换为op::DataType。 |
ToOpDataType(aclDataType type) |
将aclDataType转换为op::DataType。 |
ToAclDataType(DataType type) |
将op::DataType转换为aclDataType。 |
CalcShapeBytes(int64_t size, DataType dataType, bool ceil=false) |
用给定的shape大小和dtype计算总大小。 |