Searched refs:max_it (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/cpu/kernels/softmax/generic/neon/ |
| D | impl.cpp | 105 Iterator max_it(max, window); in neon_softmax_logits_1d_quantized() local
122 const auto max_val = *reinterpret_cast<const T *>(max_it.ptr()); in neon_softmax_logits_1d_quantized()
260 in_it, max_it, out_it); in neon_softmax_logits_1d_quantized()
275 Iterator max_it(max, window); in neon_softmax_logits_1d_float() local
297 const auto max_val = *reinterpret_cast<const T *>(max_it.ptr()); in neon_softmax_logits_1d_float()
390 in_it, max_it, out_it); in neon_softmax_logits_1d_float()
|
| /external/ComputeLibrary/src/cpu/kernels/softmax/generic/sve/ |
| D | impl.cpp | 80 Iterator max_it(max, window); in sve_softmax_logits_1d_float() local
97 const auto max_val = *reinterpret_cast<const ScalarType *>(max_it.ptr()); in sve_softmax_logits_1d_float()
166 in_it, max_it, out_it); in sve_softmax_logits_1d_float()
|
| /external/ComputeLibrary/src/cpu/kernels/softmax/generic/sve2/ |
| D | impl.cpp | 44 Iterator max_it(max, window); in sve2_softmax_logits_1d_quantized() local
65 const auto max_val = *reinterpret_cast<const ScalarType *>(max_it.ptr()); in sve2_softmax_logits_1d_quantized()
203 in_it, max_it, out_it); in sve2_softmax_logits_1d_quantized()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 1384 max_it = finfo.nmant + 1
1444 math_ops.less(i, max_it),
1540 max_it = 5 # Taken from LAPACK xSTEIN.
1547 math_ops.less(i, max_it),
|