下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

接口列表

本文档主要描述UDF(User Define Function)模块对外提供的接口,用户可以调用这些接口进行自定义处理函数的开发,然后通过DataFlow构图在CPU上执行该处理函数。

您可以在CANN软件安装后文件存储路径下的“python/site-packages/dataflow/flow_func/flow_func.py”查看对应接口的实现。接口列表如下。

FlowMsg类

表1 FlowMsg类接口

接口名称

简介

FlowMsg构造函数

FlowMsg的构造函数。

get_msg_type

获取FlowMsg的消息类型。

get_tensor

获取FlowMsg中的tensor对象。

set_ret_code

设置FlowMsg消息中的错误码。

get_ret_code

获取输入FlowMsg消息中的错误码。

set_start_time

设置FlowMsg消息头中的开始时间戳。

get_start_time

获取FlowMsg消息中的开始时间戳。

set_end_time

设置FlowMsg消息头中的结束时间戳。

get_end_time

获取FlowMsg消息中的结束时间戳。

set_flow_flags

设置FlowMsg消息头中的flags。

get_flow_flags

获取FlowMsg消息头中的flags。

set_route_label

设置路由的标签

get_transaction_id

获取FlowMsg消息中的事务ID,事务Id从1开始计数,每feed一批数据,事务Id会加一,可用于识别哪一批数据

Tensor类

表2 Tensor类接口

接口名称

简介

Tensor构造函数和析构函数

Tensor构造函数和析构函数。

get_shape

获取Tensor的Shape。

get_data_type

获取Tensor中的数据类型。

get_data_size

获取Tensor中的数据大小。

get_element_cnt

获取Tensor中的元素的个数。

reshape

对tensor进行Reshape操作,不改变tensor的内容。

MetaParams类

表3 MetaParams类接口

接口名称

简介

PyMetaParams构造函数

PyMetaParams构造函数。

get_name

获取Flowfunc的实例名。

get_attr_int

获取指定名称的int类型属性值。

get_attr_bool_list

获取指定名称的bool数组类型属性值。

get_attr_int_list

获取指定名称的int数组类型属性值。

get_attr_int_list_list

获取指定名称的int二维数组类型属性值。

get_attr_bool

获取指定名称的bool类型属性值。

get_attr_float_list

获取指定名称的float数组类型属性值。

get_attr_tensor_dtype

获取指定名称的numpy dtype类型的属性值。

get_attr_tensor_dtype_list

获取指定名称的numpy dtype数组类型的属性值。

get_attr_str

获取指定名称的string类型的属性值。

get_attr_str_list

获取指定名称的string数组类型的属性值。

get_attr_float

获取指定名称的float类型属性值。

get_input_num

获取Flowfunc的输入个数。

get_output_num

获取Flowfunc的输出个数。

get_work_path

获取Flowfunc的工作路径。

get_running_device_id

获取正在运行的设备ID。

MetaRunContext类

表4 MetaRunContext类接口

接口名称

简介

MetaRunContext构造函数

MetaRunContext构造函数。

alloc_tensor_msg

根据shape、data type以及对齐大小申请tensor类型的FlowMsg。

set_output

设置指定index的output的tensor。

set_multi_outputs

批量设置指定index的output的tensor。

run_flow_model

同步执行指定的模型。

alloc_empty_data_msg

申请空数据的MsgType类型的message。

get_user_data

获取用户定义数据。

raise_exception

UDF主动上报异常。

get_exception

UDF获取其他UDF节点上报的异常。

AffinityPolicy类

表5 AffinityPolicy类接口

接口名称

简介

AffinityPolicy类

亲和策略枚举定义。

BalanceConfig类

表6 BalanceConfig类接口

接口名称

简介

BalanceConfig构造函数

BalanceConfig构造函数。

set_data_pos

设置输出数据对应权重矩阵中的位置。

get_inner_config

获取内部配置对象,被set_output或者set_multi_outputs调用。

UDF日志接口

表7 UDF日志接口

接口名称

简介

FlowFuncLogger构造函数

FlowFuncLogger构造函数。

get_log_header

获取日志扩展头信息。

is_log_enable

查询对应级别和类型的日志是否开启。

运行日志Error级别日志宏

运行日志Error级别日志宏。

运行日志Info级别日志宏

运行日志Info级别日志宏。

调试日志Error级别日志宏

调试日志Error级别日志宏。

调试日志Warn级别日志宏

调试日志Warn级别日志宏。

调试日志Info级别日志宏

调试日志Info级别日志宏。

调试日志Debug级别日志宏

调试日志Debug级别日志宏。

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

当前产品无相关内容

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