Searched refs:arg_max (Results 1 – 3 of 3) sorted by relevance
53 Unlike FractionalMaxPoolGrad, we don't need to find arg_max for
183 arg_max = deprecation.deprecated(None, "Use `argmax` instead")(arg_max) # pylint: disable=used-bef… variable201 gen_math_ops.arg_max.__doc__.replace("dimensions", "axes").replace(214 return gen_math_ops.arg_max(input, axis, name=name, output_type=output_type)
608 name: "arg_max"