Searched refs:step_ids_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| D | beam_search_ops.cc | 54 const TensorShape& step_ids_shape = step_ids.shape(); in Compute() local
56 ctx, step_ids_shape.dims() == 3, in Compute()
58 step_ids_shape.DebugString())); in Compute()
68 ctx, step_ids_shape == parent_ids.shape(), in Compute()
71 step_ids_shape.DebugString(), " and ", in Compute()
75 step_ids_shape.dim_size(1) == max_sequence_lengths.shape().dim_size(0), in Compute()
79 step_ids_shape.DebugString(), " and ", in Compute()
82 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, step_ids_shape, &beams)); in Compute()
|