diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2016-12-12 13:57:21 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-12 14:07:41 -0800 |
| commit | a46b6d211eac423c72d3a57a177daf2f64db8642 (patch) | |
| tree | 8b9633ba87fdd4677994a0fec7c61a71862ca412 /tensorflow/contrib/grid_rnn | |
| parent | 55735379ccda8a64e49717e95e9e0915e7b8dc8e (diff) | |
Major intent of this CL is to rename split_v -> split in the python API.
Requires:
1) Add name arguments to tf.split calls introduced since major Rosie CL cleaning
this up across the codebase.
2) Change uses of array_ops.split to use named arguments, which was not covered
in the Rosie CL.
3) Rename split_v calls to split.
Change: 141806936
Diffstat (limited to 'tensorflow/contrib/grid_rnn')
| -rw-r--r-- | tensorflow/contrib/grid_rnn/python/ops/grid_rnn_cell.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/grid_rnn/python/ops/grid_rnn_cell.py b/tensorflow/contrib/grid_rnn/python/ops/grid_rnn_cell.py index 6247cad380..aef4ce3fdb 100644 --- a/tensorflow/contrib/grid_rnn/python/ops/grid_rnn_cell.py +++ b/tensorflow/contrib/grid_rnn/python/ops/grid_rnn_cell.py @@ -170,7 +170,8 @@ class GridRNNCell(rnn.RNNCell): # project input if inputs is not None and sum(inputs.get_shape().as_list()) > 0 and len( conf.inputs) > 0: - input_splits = array_ops.split(1, len(conf.inputs), inputs) + input_splits = array_ops.split( + value=inputs, num_or_size_splits=len(conf.inputs), axis=1) input_sz = input_splits[0].get_shape().as_list()[1] for i, j in enumerate(conf.inputs): |