Searched refs:_get_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 467 batch_size = _get_dim(label_seq, 0)
468 num_labels = _get_dim(label_seq, 1)
488 values = array_ops.ones([_get_dim(indices, 0)])
524 batch_size = _get_dim(seq_lengths, 0)
553 num_label_states = _get_dim(labels, 1)
569 num_label_states = _get_dim(labels, 1) + 1
588 num_label_states = _get_dim(labels, 1) + 1
653 num_labels = _get_dim(logits, 2)
654 max_label_seq_length = _get_dim(labels, 1)
679 max_logit_length = _get_dim(logits, 0)
[all …]
|
| D | image_ops_impl.py | 924 def _get_dim(tensor, idx): function
933 img_h, dynamic_h = _get_dim(image, 0)
934 img_w, dynamic_w = _get_dim(image, 1)
938 img_h, dynamic_h = _get_dim(image, 1)
939 img_w, dynamic_w = _get_dim(image, 2)
|