Home
last modified time | relevance | path

Searched refs:a_rank (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py806 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 …]
Dnp_math_ops.py1145 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/
Dhlo_ops.cc2742 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/
Darray_ops.py2393 a_rank = rank(a)
2395 (gen_math_ops._range(0, a_rank - 2, 1), [a_rank - 1, a_rank - 2]), 0)