Searched defs:argmax (Results 1 – 10 of 10) sorted by relevance
114 xla::XlaOp argmax = xla::ArgMax(softmax_entries, xla_output_type, in Compile() local
127 public static int argmax(double[] a) { in argmax() method in Utils
160 XlaOp argmax = GetTupleElement(max_argmax, 1); in ArgMinMax() local
1005 Tensor* argmax = nullptr; in Compute() local1118 const Tensor& argmax = context->input(2); in Compute() local1186 const Tensor& argmax = context->input(2); in Compute() local
747 argmax() const { in argmax() function773 argmax(const Index return_dim) const { in argmax() function
492 def argmax(x): function
649 def argmax(x): function
1225 def argmax(a, axis=None): function
257 def argmax(input, function
2501 def argmax(x, axis=-1): function