Searched refs:rank_b (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | tensordot_op_test.py | 221 for rank_b in 1, 2, 4, 5:
222 for num_dims in range(0, min(rank_a, rank_b) + 1):
224 for testcase in _get_tensordot_tests(dtype, rank_a, rank_b,
227 rank_a, rank_b, num_dims,
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| H A D | segmentation_ops.h | 209 const OutputType rank_b = rank_[index_b]; local
213 } else if (rank_a < rank_b) {
|
Completed in 179 milliseconds