3cdd0ddb974ba967e9d39e9b275bedc89badd73b,chiron/export_test.py,,input_output_list,#,26

Before Change

    logits, _ = inference(x, seq_length, training=training)
    predict = tf.nn.ctc_greedy_decoder(tf.transpose(logits, perm=[1, 0, 2]), seq_length, merge_repeated=True)
    input_dict = {"x": x, "seq_length": seq_length, "training": training}
    output_dict = {"decoded_indices": predict[0][].indices, "decoded_values": predict[0][0].values,
                   "neg_sum_logits": predict[1]}
    return input_dict, output_dict

After Change

                              full_sequence_len = FLAGS.segment_len,
                              configure = model_configure)
    ratio = tf.constant(ratio,dtype = tf.float32,shape = [])
    seq_length_r = tf.cast(tf.round(seq_length/ratio),tf.int32)
    prob_logits = path_prob(logits)
    predict = tf.nn.ctc_beam_search_decoder(tf.transpose(logits, perm=[1, 0, 2]), 
                                            seq_length_r, 
                                            merge_repeated=True,
                                            beam_width = FLAGS.beam_width)
    input_dict = {"x": x, "seq_length": seq_length, "training": training}
    output_dict = {"predict_sequences":predict,"logits":logits, "prob_logits":prob_logits}
    return input_dict, output_dict


def build_and_run_exports(job_dir, serving_input_fn):

Instances