Searched defs:argmax (Results 1 – 10 of 10) sorted by relevance
113 xla::XlaOp argmax; in Compile() local
127 public static int argmax(double[] a) { in argmax() method in Utils
614 argmax() const { in argmax() function640 argmax(const int return_dim) const { in argmax() function
966 Tensor* argmax = nullptr; in Compute() local1069 const Tensor& argmax = context->input(2); in Compute() local1127 const Tensor& argmax = context->input(2); in Compute() local
167 XlaOp argmax = GetTupleElement(max_argmax, 1); in ArgMinMax() local
413 def argmax(x): function
614 def argmax(x): function
1213 def argmax(a, axis=None): function
261 def argmax(input, function
2499 def argmax(x, axis=-1): function