Searched defs:arg_max (Results 1 – 3 of 3) sorted by relevance
130 XlaOp arg_max = Select(cmp, lhs_index, rhs_index); in CreateMinMaxComputation() local
229 arg_max = deprecation.deprecated(None, "Use `tf.math.argmax` instead")(arg_max) # pylint: disable=… variable
3987 Output arg_max = ops::ArgMax(s.WithOpName("arg_max"), sqrt, 1); in TEST_F() local