// get list of input/output tensors (by name)
if args.signature_def_key:
input_tensors = [inputs_tensor_info[t].name for t in input_tensor_names]
output_tensors = [outputs_tensor_info[t].name for t in output_tensor_names]
else:
input_tensors = [t + ":0" for t in input_tensor_names]
output_tensors = [t + ":0" for t in output_tensor_names]
After Change
single_node_env(tf_args)
tf.compat.v1.disable_eager_execution()
logging.info("===== input_mapping: {}".format(args.input_mapping))
logging.info("===== output_mapping: {}".format(args.output_mapping))
input_tensor_names = [tensor for col, tensor in sorted(args.input_mapping.items())]
