Searched defs:argmax (Results 1 – 10 of 10) sorted by relevance
116 xla::XlaOp argmax; in Compile() local
127 public static int argmax(double[] a) { in argmax() method in Utils
974 Tensor* argmax = nullptr; in Compute() local1087 const Tensor& argmax = context->input(2); in Compute() local1147 const Tensor& argmax = context->input(2); in Compute() local
747 argmax() const { in argmax() function773 argmax(const Index return_dim) const { in argmax() function
167 XlaOp argmax = GetTupleElement(max_argmax, 1); in ArgMinMax() local
488 def argmax(x): function
629 def argmax(x): function
1214 def argmax(a, axis=None): function
262 def argmax(input, function
2501 def argmax(x, axis=-1): function