Searched refs:_get_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 372 batch_size = _get_dim(label_seq, 0)
373 num_labels = _get_dim(label_seq, 1)
393 values = array_ops.ones([_get_dim(indices, 0)])
429 batch_size = _get_dim(seq_lengths, 0)
458 num_label_states = _get_dim(labels, 1)
474 num_label_states = _get_dim(labels, 1) + 1
492 num_label_states = _get_dim(labels, 1) + 1
550 num_labels = _get_dim(logits, 2)
551 max_label_seq_length = _get_dim(labels, 1)
642 blank_index += _get_dim(logits, 2)
[all …]
|
| D | image_ops_impl.py | 596 def _get_dim(tensor, idx): function
605 img_h, dynamic_h = _get_dim(image, 0)
606 img_w, dynamic_w = _get_dim(image, 1)
610 img_h, dynamic_h = _get_dim(image, 1)
611 img_w, dynamic_w = _get_dim(image, 2)
|