054095d922edda5134e520522bc82a1b95cc5bd4,onnx_tf/backend.py,TensorflowBackend,run_node,#Any#Any#Any#Any#Any#,167

Before Change

      with tf.compat.v1.Session() as sess:
        with tf.device(device_option):
          sess.run(tf.compat.v1.global_variables_initializer())
          output_vals = sess.run(ops)

    return namedtupledict("Outputs", node.outputs)(*output_vals)

  @classmethod

After Change

    module = TFModule(node)

    output_vals = module(**input_dict)
    output_vals = [val.numpy() if isinstance(val, tf.Tensor) else val for val in output_vals]

    return namedtupledict("Outputs", node.outputs)(*output_vals)

Instances