| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| categorical_op.cc | 81 xla::ComputationDataHandle argmax; variable
84 XlaHelpers::ArgMax(builder, ctx, softmax_entries, softmax_shape,
85 input_type(0), output_type(0), /*axis=*/2, &argmax));
87 ctx->SetOutput(0, argmax);
|
| /external/tensorflow/tensorflow/python/eager/ |
| backprop_test.py | 241 lambda x: array_ops.stop_gradient(math_ops.argmax(x)))
246 def argmax(x): function in function:BackpropTest.testArgmax
247 i = math_ops.argmax(x)
250 grad = backprop.gradients_function(argmax)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| maxpooling_op.cc | 907 Tensor* argmax = nullptr; variable
953 const Tensor& argmax = context->input(2); variable
1003 const Tensor& argmax = context->input(2); variable
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorBase.h | 614 argmax() const { function in class:Eigen::TensorBase
640 argmax(const int return_dim) const { function in class:Eigen::TensorBase
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| math_ops.py | 141 @@argmax
183 arg_max = deprecation.deprecated(None, "Use `argmax` instead")(arg_max) # pylint: disable=used-before-assignment
197 @tf_export("argmax")
203 def argmax(input, function
[all...] |
| /external/tensorflow/tensorflow/python/keras/_impl/keras/ |
| backend.py | 1665 def argmax(x, axis=-1): function
[all...] |
