Searched refs:rank_b (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 521 int64 rank_a, rank_b; in __anonc06300730602() local
528 rank_b = context->Rank(input_b); in __anonc06300730602()
559 if (rank_b != lhs_rhs_split[1].size()) { in __anonc06300730602()
561 "Expected equation[1] with size: ", rank_b, " Got '", in __anonc06300730602()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensordot_op_test.py | 226 for rank_b in 1, 2, 4, 5:
227 for num_dims in range(0, min(rank_a, rank_b) + 1):
230 for testcase in _get_tensordot_tests(dtype, rank_a, rank_b,
233 rank_a, rank_b, num_dims,
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| D | segmentation_ops.h | 209 const OutputType rank_b = rank_[index_b]; in do_union() local
213 } else if (rank_a < rank_b) { in do_union()
|