CANN/ge DataFlow Python API参考
# DataFlow构图接口参考Python【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge简介DataFlow构图接口DataFlow接口列表构图接口dataflow.FlowDatadataflow.FlowNodeFlowNodeadd_process_pointmap_inputmap_outputset_attr__call__set_balance_scatterset_balance_gatherset_aliasdataflow.FlowFlagdataflow.FlowGraph构图接口FlowGraphset_contains_n_mapping_nodeset_inputs_align_attrsset_exception_catchdataflow.FlowOutputdataflow.Frameworkdataflow.FuncProcessPointFuncProcessPointset_init_paramadd_invoked_closuredataflow.GraphProcessPointGraphProcessPointfnodedataflow.FlowGraphProcessPointdataflow.TensorTensornumpydataflow.TensorDescdataflow.alloc_tensordataflow.utils.generate_deploy_templatedataflow.MsgTypeRegisterregisterregisteredget_msg_typedataflowget_serialize_funcget_deserialize_funcget_size_funcdeserialize_from_filepyflowmethodnpu_modeldataflow.CountBatchdataflow.TimeBatch运行接口dataflow.initdataflow.FlowInfoFlowInfoset_user_dataget_user_datadataflowuser_datadata_sizestart_timeend_timeflow_flagstransaction_iddataflow.FlowGraph运行接口feed_datafeedfetch_datafetchdataflow.finalizedataflow.get_running_device_iddataflow.get_running_instance_iddataflow.get_running_instance_numDataFlow错误码UDF接口UDF接口列表FlowMsg类FlowMsg构造函数get_msg_typeUDFget_tensorset_ret_codeget_ret_codeset_start_timeget_start_timeset_end_timeget_end_timeset_flow_flagsget_flow_flagsset_route_labelget_transaction_idset_msg_typeget_raw_dataset_transaction_idTensor类Tensor构造函数get_shapeget_data_typeget_data_sizeget_element_cntreshapeMetaParams类PyMetaParams构造函数get_nameget_attr_intget_attr_bool_listget_attr_int_listget_attr_int_list_listget_attr_boolget_attr_float_listget_attr_tensor_dtypeget_attr_tensor_dtype_listget_attr_strget_attr_str_listget_attr_floatget_input_numget_output_numget_work_pathget_running_device_idMetaRunContext类MetaRunContext构造函数alloc_tensor_msgset_outputset_multi_outputsrun_flow_modelalloc_empty_data_msgget_user_dataUDFraise_exceptionget_exceptionalloc_raw_data_msgto_flow_msgAffinityPolicy类BalanceConfig类BalanceConfig构造函数set_data_posget_inner_configFlowMsgQueue类FlowMsgQueue构造函数getget_nowaitfullemptyqsizeUDF日志接口UDF日志接口简介FlowFuncLogger构造函数get_log_headeris_log_enable运行日志Error级别日志宏运行日志Info级别日志宏调试日志Error级别日志宏调试日志Warn级别日志宏调试日志Info级别日志宏调试日志Debug级别日志宏UDF错误码【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考