Searched defs:reduce_max (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| D | softmax.h | 54 __device__ __forceinline__ void reduce_max(Tensor<Engine0, Layout0> const& tensor, Tensor<Engine1, … in reduce_max() function
|
| /external/libaom/av1/encoder/x86/ |
| D | ml_sse3.c | 264 static inline __m128 reduce_max(__m128 reg) { in reduce_max() function
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 656 def reduce_max(input_tensor: ragged_tensor.Ragged, function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 3946 Output reduce_max = ops::Max(s.WithOpName("reduce_max"), sqrt, {0}); in TEST_F() local
4029 Output reduce_max = ops::Max(s.WithOpName("reduce_max"), sqrt, {0}); in TEST_F() local
4127 Output reduce_max = ops::Max(s.WithOpName("reduce_max"), neg, {0}); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 3061 def reduce_max(input_tensor, axis=None, keepdims=False, name=None): function
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 5723 def reduce_max(x, dim=None, keepdim=False): function
|