DataFlow构图接口参考(Python)
2026/9/10 6:10:28 网站建设 项目流程

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),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询