Searched refs:rank_a (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 521 int64 rank_a, rank_b; in __anonc06300730602() local
523 rank_a = context->Rank(input_a); in __anonc06300730602()
553 if (rank_a != lhs_rhs_split[0].size()) { in __anonc06300730602()
555 "Expected equation[0] with size: ", rank_a, " Got '", in __anonc06300730602()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensordot_op_test.py | 225 for rank_a 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 | 208 const OutputType rank_a = rank_[index_a]; in do_union() local
213 } else if (rank_a < rank_b) { in do_union()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 3650 rank_a = array_ops.rank(a)
3652 axes = array_ops.where(axes >= 0, axes, axes + rank_a)
3653 free, _ = array_ops.setdiff1d(range(rank_a), axes)
|