# DataFlow构图接口参考（Python）
【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
- 简介
- DataFlow构图接口
DataFlow接口列表
构图接口
dataflow.FlowData
dataflow.FlowNode
- FlowNode
- add_process_point
- map_input
- map_output
- set_attr
- __call__
- set_balance_scatter
- set_balance_gather
- set_alias
dataflow.FlowFlag
dataflow.FlowGraph（构图接口）
- FlowGraph
- set_contains_n_mapping_node
- set_inputs_align_attrs
- set_exception_catch
dataflow.FlowOutput
dataflow.Framework
dataflow.FuncProcessPoint
- FuncProcessPoint
- set_init_param
- add_invoked_closure
dataflow.GraphProcessPoint
- GraphProcessPoint
- fnode
dataflow.FlowGraphProcessPoint
dataflow.Tensor
- Tensor
- numpy
dataflow.TensorDesc
dataflow.alloc_tensor
dataflow.utils.generate_deploy_template
dataflow.MsgTypeRegister
- register
- registered
- get_msg_type（dataflow）
- get_serialize_func
- get_deserialize_func
- get_size_func
- deserialize_from_file
pyflow
method
npu_model
dataflow.CountBatch
dataflow.TimeBatch
运行接口
dataflow.init
dataflow.FlowInfo
- FlowInfo
- set_user_data
- get_user_data（dataflow）
- user_data
- data_size
- start_time
- end_time
- flow_flags
- transaction_id
dataflow.FlowGraph（运行接口）
- feed_data
- feed
- fetch_data
- fetch
dataflow.finalize
dataflow.get_running_device_id
dataflow.get_running_instance_id
dataflow.get_running_instance_num
DataFlow错误码
UDF接口
UDF接口列表
FlowMsg类
- FlowMsg构造函数
- get_msg_type（UDF）
- get_tensor
- set_ret_code
- get_ret_code
- set_start_time
- get_start_time
- set_end_time
- get_end_time
- set_flow_flags
- get_flow_flags
- set_route_label
- get_transaction_id
- set_msg_type
- get_raw_data
- set_transaction_id
Tensor类
- Tensor构造函数
- get_shape
- get_data_type
- get_data_size
- get_element_cnt
- reshape
MetaParams类
- PyMetaParams构造函数
- get_name
- get_attr_int
- get_attr_bool_list
- get_attr_int_list
- get_attr_int_list_list
- get_attr_bool
- get_attr_float_list
- get_attr_tensor_dtype
- get_attr_tensor_dtype_list
- get_attr_str
- get_attr_str_list
- get_attr_float
- get_input_num
- get_output_num
- get_work_path
- get_running_device_id
MetaRunContext类
- MetaRunContext构造函数
- alloc_tensor_msg
- set_output
- set_multi_outputs
- run_flow_model
- alloc_empty_data_msg
- get_user_data（UDF）
- raise_exception
- get_exception
- alloc_raw_data_msg
- to_flow_msg
AffinityPolicy类
BalanceConfig类
- BalanceConfig构造函数
- set_data_pos
- get_inner_config
FlowMsgQueue类
- FlowMsgQueue构造函数
- get
- get_nowait
- full
- empty
- qsize
UDF日志接口
- UDF日志接口简介
- FlowFuncLogger构造函数
- get_log_header
- is_log_enable
- 运行日志Error级别日志宏
- 运行日志Info级别日志宏
- 调试日志Error级别日志宏
- 调试日志Warn级别日志宏
- 调试日志Info级别日志宏
- 调试日志Debug级别日志宏
UDF错误码
【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
