set_attr【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况产品是否支持Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√函数功能设置FlowNode的属性。函数原型set_attr(attr_name, attr_value)参数说明参数名称数据类型取值说明attr_namestr属性名称当前支持如下三种- _flow_attrBOOL类型属性可以配置为true或falsetrue表示存在flow属性false表示不存在flow属性。-_flow_attr_depthINT类型属性指定了队列的深度范围大于等于2若不配置默认深度为128。- _flow_attr_enqueue_policySTRING类型属性指定的入队的策略仅支持FIFO和OVERWRITEFIFO表示队列数据顺序入队队列满的时候会等待dequeueOVERWRITE表示入队不会等待数据会循环覆盖若不配置默认策略为FIFO。attr_valueUnion[bool, str, int]属性值。返回值正常场景下返回None。返回“TypeError”表示参数类型不正确。调用示例import dataflow as df pp df.FuncProcessPoint(...) flow_node df.FlowNode(...) flow_node.set_attr(_flow_attr,True) flow_node.set_attr(_flow_attr_depth,5) flow_node.set_attr(_flow_attr_enqueue_policy,FIFO)约束说明通过_flow_attr_depth和_flow_attr_enqueue_policy设置depth和policy需要先使能_flow_attr否则不会生效。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考