Searched refs:arg_max (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FractionalAvgPoolGrad.pbtxt | 53 Unlike FractionalMaxPoolGrad, we don't need to find arg_max for
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 104 arg_max = deprecation.deprecated(None, "Use `tf.math.argmax` instead")(arg_max) # pylint: disable=… variable
106 tf_export(v1=["arg_max"])(arg_max)
128 gen_math_ops.arg_max.__doc__.replace("dimensions", "axes").replace(
166 return gen_math_ops.arg_max(input, axis, name=name, output_type=output_type)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 3401 Output arg_max = ops::ArgMax(s.WithOpName("arg_max"), sqrt, 1); in TEST_F() local
3402 Output final_out = ops::Identity(s.WithOpName("final_out"), arg_max); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 3138 out = tf.arg_max(input_value, axis, output_type=parameters["output_type"])
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 736 name: "arg_max"
|