"cmake/git@developer.sourcefind.cn:OpenDAS/fast_rnnt.git" did not exist on "0818d487b6211d05531a3fd1706f78a46be0acca"
Commit 63447939 authored by Andrew Selle's avatar Andrew Selle
Browse files

Update concat_v2 to be concat to match 1.0 final

Fixes #1014
parent 596c9e23
...@@ -100,7 +100,7 @@ def build_input(dataset, data_path, batch_size, mode): ...@@ -100,7 +100,7 @@ def build_input(dataset, data_path, batch_size, mode):
labels = tf.reshape(labels, [batch_size, 1]) labels = tf.reshape(labels, [batch_size, 1])
indices = tf.reshape(tf.range(0, batch_size, 1), [batch_size, 1]) indices = tf.reshape(tf.range(0, batch_size, 1), [batch_size, 1])
labels = tf.sparse_to_dense( labels = tf.sparse_to_dense(
tf.concat_v2(values=[indices, labels], axis=1), tf.concat(values=[indices, labels], axis=1),
[batch_size, num_classes], 1.0, 0.0) [batch_size, num_classes], 1.0, 0.0)
assert len(images.get_shape()) == 4 assert len(images.get_shape()) == 4
......
...@@ -146,7 +146,7 @@ class PTBModel(object): ...@@ -146,7 +146,7 @@ class PTBModel(object):
(cell_output, state) = cell(inputs[:, time_step, :], state) (cell_output, state) = cell(inputs[:, time_step, :], state)
outputs.append(cell_output) outputs.append(cell_output)
output = tf.reshape(tf.concat_v2(outputs, 1), [-1, size]) output = tf.reshape(tf.concat(outputs, 1), [-1, size])
softmax_w = tf.get_variable( softmax_w = tf.get_variable(
"softmax_w", [size, vocab_size], dtype=data_type()) "softmax_w", [size, vocab_size], dtype=data_type())
softmax_b = tf.get_variable("softmax_b", [vocab_size], dtype=data_type()) softmax_b = tf.get_variable("softmax_b", [vocab_size], dtype=data_type())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment