| /aosp_15_r20/external/executorch/examples/mediatek/executor_runner/llama_runner/ |
| H A D | Utils.h | 70 static uint64_t argmax(const void* logits_buffer, const size_t vocab_size) { in argmax() function
83 static uint64_t argmax( in argmax() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AdaptiveMaxPooling2d.cu | 87 int argmax = istartH * isizeW + istartW; in adaptivemaxpool() local
146 int argmax = (*ptr_ind); in adaptivemaxgradinput() local
193 int argmax = (*ptr_ind); in atomicadaptivemaxgradinput() local
|
| H A D | AdaptiveMaxPooling3d.cu | 99 int64_t argmax = istartT*isizeH*isizeW + istartH*isizeW + istartW; in adaptivemaxpool() local
196 int argmax = (*ptr_ind); in adaptivemaxgradinput() local
272 int64_t argmax = (*ptr_ind); in atomicadaptivemaxgradinput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op.cc | 1005 Tensor* argmax = nullptr; in Compute() local
1118 const Tensor& argmax = context->input(2); in Compute() local
1186 const Tensor& argmax = context->input(2); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | categorical_op.cc | 114 xla::XlaOp argmax = xla::ArgMax(softmax_entries, xla_output_type, in Compile() local
|
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorBase.h | 747 argmax() const { in argmax() function
773 argmax(const Index return_dim) const { in argmax() function
|
| /aosp_15_r20/external/pytorch/torch/profiler/ |
| H A D | _utils.py | 360 def argmax(seq, key=lambda x: x, start=0, end=None): function
|
| /aosp_15_r20/external/walt/android/WALT/app/src/main/java/org/chromium/latency/walt/ |
| H A D | Utils.java | 127 public static int argmax(double[] a) { in argmax() method in Utils
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | arithmetic.cc | 160 XlaOp argmax = GetTupleElement(max_argmax, 1); in ArgMinMax() local
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset12.py | 271 def argmax( function
|
| H A D | symbolic_opset9.py | 5046 def argmax( function
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _reductions_impl.py | 72 def argmax( function
|
| /aosp_15_r20/external/pytorch/torch/masked/ |
| H A D | _ops.py | 1309 def argmax( function
|
| /aosp_15_r20/external/liblc3/src/ |
| H A D | ltpf.c | 474 LC3_HOT static int argmax(const float *x, int n, float *x_max) in argmax() function
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _shape_functions.py | 1070 def argmax( function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | def_function_xla_jit_test.py | 492 def argmax(x): function
|
| H A D | backprop_test.py | 649 def argmax(x): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| H A D | np_math_ops.py | 1225 def argmax(a, axis=None): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | math_ops.py | 257 def argmax(input, function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | backend.py | 2501 def argmax(x, axis=-1): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ReduceOps.cpp | 231 TORCH_META_FUNC(argmax) in TORCH_META_FUNC() argument
|