Searched refs:a_rank (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
D | np_array_ops.py | 806 x = array_ops.where_v2(x < 0, np_utils.add(x, a_rank), x) 814 a_rank = a.shape.rank 815 axis1, axis2 = adjust_axes((axis1, axis2), a_rank) 816 perm = list(range(a_rank)) 820 a_rank = array_ops.rank(a) 821 axis1, axis2 = adjust_axes((axis1, axis2), a_rank) 822 perm = math_ops.range(a_rank) 844 a_rank = np_utils._maybe_static(array_ops.rank(a)) # pylint: disable=protected-access 851 source = tuple(_correct_axis(axis, a_rank) for axis in source) 852 destination = tuple(_correct_axis(axis, a_rank) for axis in destination) [all …]
|
D | np_math_ops.py | 1145 a_rank = array_ops.rank(a) 1148 reps, [[math_ops.maximum(a_rank - reps_size, 0), 0]], constant_values=1) 1150 array_ops.shape(a), [[math_ops.maximum(reps_size - a_rank, 0), 0]],
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
D | hlo_ops.cc | 2742 auto a_rank = a_type.getRank(); in Verify() local 2743 if (a_rank < 2) in Verify() 2748 if (a_type.getDimSize(a_rank - 2) != a_type.getDimSize(a_rank - 1)) in Verify() 2759 if (a_rank != b_rank) in Verify() 2764 if (a_type.getDimSize(a_rank - 1) != in Verify()
|
/external/tensorflow/tensorflow/python/ops/ |
D | array_ops.py | 2393 a_rank = rank(a) 2395 (gen_math_ops._range(0, a_rank - 2, 1), [a_rank - 1, a_rank - 2]), 0)
|