sess.run模式下的session配置
在昇腾AI处理器上通过sess.run模式进行训练或在线推理时,支持如下配置项。

如果表中未列出,表示该参数预留或适用于其他昇腾AI处理器版本,用户无需关注。
调用示例
import tensorflow as tf from npu_bridge.npu_init import * config = tf.ConfigProto() custom_op = config.graph_options.rewrite_options.custom_optimizers.add() custom_op.name = "NpuOptimizer" custom_op.parameter_map["use_off_line"].b = True config.graph_options.rewrite_options.remapping = RewriterConfig.OFF config.graph_options.rewrite_options.memory_optimization = RewriterConfig.OFF with tf.Session(config=config) as sess: sess.run(cost)
父主题: 参考