Searched refs:max_seq_len (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_loss_calculator.h | 144 auto max_seq_len = seq_len(0); in CalculateLoss() local
152 max_seq_len = std::max(seq_len(b), max_seq_len); in CalculateLoss()
262 max_seq_len * num_classes * in CalculateLoss()
264 max_seq_len * 2 * (2 * num_classes + 1) * in CalculateLoss()
266 max_seq_len * in CalculateLoss()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 71 const int64 max_seq_len = input_shape.dim_size(seq_dim_); in Compile() local
78 xla::ShapeUtil::MakeShape(seq_lens_type, {batch_size, max_seq_len, 1}), in Compile()
82 xla::ShapeUtil::MakeShape(seq_lens_type, {batch_size, max_seq_len, 1}), in Compile()
|
| /external/tensorflow/tensorflow/contrib/crf/python/ops/ |
| D | crf.py | 276 max_seq_len = array_ops.shape(inputs)[1]
282 math_ops.range(batch_size) * max_seq_len * num_tags, 1)
283 offsets += array_ops.expand_dims(math_ops.range(max_seq_len) * num_tags, 0)
291 [batch_size, max_seq_len])
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | lstm_ops.py | 694 max_seq_len = math_ops.cast(time_len, dtypes.int64)
696 max_seq_len = math_ops.cast(math_ops.reduce_max(sequence_length),
700 seq_len_max=max_seq_len,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2745 int max_seq_len = dims[seq_dim]; in TEST_F() local
2747 std::uniform_int_distribution<int32> d(0, max_seq_len); in TEST_F()
|