| /external/tensorflow/tensorflow/python/kernel_tests/ |
| argmax_op_test.py | 60 # Check that argmin and argmax match numpy along the primary axis
61 self._testBothArg(math_ops.argmax, x, 0, x.argmax())
67 # Check that argmin and argmax match numpy along all axes
69 self._testBothArg(math_ops.argmax, x, axis, x.argmax(axis))
78 expected_values = x.argmax()
80 ans = math_ops.argmax(x, axis=0, output_type=dtypes.int32)
107 for op in math_ops.argmin, math_ops.argmax:
114 for op in math_ops.argmin, math_ops.argmax
[all...] |
| /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);
|
| index_ops_kernel_argmax_float_1d.cc | 41 out_eig = in_eig.argmax(0).cast<int64>();
46 // Implements argmax on CPU. This is called by an XLA custom call, set up by
|
| index_ops_kernel_argmax_float_2d.cc | 43 out_eig = in_eig.argmax(dim).cast<int64>();
48 // Implements argmax on CPU. This is called by an XLA custom call, set up by
|
| /cts/apps/CameraITS/tests/inprog/ |
| test_black_level.py | 76 ymodes.append(numpy.argmax(yhist))
78 umodes.append(numpy.argmax(uhist))
80 vmodes.append(numpy.argmax(vhist))
|
| test_param_black_level_lock.py | 59 modes.append(numpy.argmax(hist))
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| argminmax_test.py | 15 """Functional tests for ArgMin and ArgMax Ops."""
49 # Complex numbers do not support argmin/argmax.
53 lambda x: math_ops.argmax(x, axis=0, output_type=dtypes.int32),
57 lambda x: math_ops.argmax(x, axis=0, output_type=dtypes.int32),
61 lambda x: math_ops.argmax(x, axis=1, output_type=dtypes.int32),
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/ |
| metrics.py | 51 K.equal(K.argmax(y_true, axis=-1), K.argmax(y_pred, axis=-1)), K.floatx())
57 K.max(y_true, axis=-1), K.cast(K.argmax(y_pred, axis=-1),
63 return K.mean(K.in_top_k(y_pred, K.argmax(y_true, axis=-1), k), axis=-1)
|
| /cts/suite/audio_quality/test_description/processing/ |
| calc_thd.py | 31 np.argmax(fftData[baseI - iMargain /2: baseI + iMargain/2])
32 peakLoc = np.argmax(fftData[:fftLen])
|
| calc_delay.py | 57 return np.argmax(result)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| argmax_op.h | 29 struct ArgMax {
34 output.device(d) = input.argmax(dimension).template cast<Tout>(); \
|
| 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/tensorflow/tensorflow/python/keras/_impl/keras/utils/ |
| np_utils_test.py | 49 np.argmax(one_hot, -1).reshape(label.shape) == label))
|
| /external/eigen/unsupported/test/ |
| cxx11_tensor_argmax.cpp | 128 tensor_argmax = tensor.argmax();
134 tensor_argmax = tensor.argmax();
185 tensor_argmax = tensor.argmax(dim);
207 tensor_argmax = tensor.argmax(dim);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_helpers.h | 75 // Sets `argmax` to the argmax of `input` along `axis`. `input_shape` and
77 // `output_type` is the dtype to use for `argmax`.
78 static Status ArgMax(xla::ComputationBuilder* builder,
83 xla::ComputationDataHandle* argmax);
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/mnist/ |
| mnist.py | 112 tf.argmax(predictions, axis=1,
114 tf.argmax(labels, axis=1,
146 accuracy(tf.argmax(predictions, axis=1, output_type=tf.int64),
147 tf.argmax(labels, axis=1, output_type=tf.int64))
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| conditioning_utils_impl.py | 79 label_id = math_ops.argmax(one_hot, axis=1)
|
| /external/tensorflow/tensorflow/examples/tutorials/mnist/ |
| mnist_softmax.py | 67 correct_prediction = tf.equal(tf.argmax(y, 1), y_)
|
| mnist_softmax_xla.py | 88 correct_prediction = tf.equal(tf.argmax(y, 1), y_)
|
| /external/tensorflow/tensorflow/contrib/crf/python/ops/ |
| crf.py | 338 backpointers[t] = np.argmax(v, 0)
340 viterbi = [np.argmax(trellis[-1])]
395 backpointers = math_ops.argmax(transition_scores, 1)
461 # If max_seq_len is 1, we skip the algorithm and simply return the argmax tag
466 math_ops.argmax(squeezed_potentials, axis=1), 1)
494 initial_state = math_ops.cast(math_ops.argmax(last_score, axis=1), # [B]
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/nets/ |
| vgg_test.py | 115 predictions = math_ops.argmax(logits, 1)
138 predictions = math_ops.argmax(logits, 1)
249 predictions = math_ops.argmax(logits, 1)
272 predictions = math_ops.argmax(logits, 1)
390 predictions = math_ops.argmax(logits, 1)
413 predictions = math_ops.argmax(logits, 1)
|
| alexnet_test.py | 106 predictions = math_ops.argmax(logits, 1)
129 predictions = math_ops.argmax(logits, 1)
|
| overfeat_test.py | 106 predictions = math_ops.argmax(logits, 1)
129 predictions = math_ops.argmax(logits, 1)
|
| /external/autotest/server/brillo/ |
| audio_utils.py | 212 numpy.argmax(numpy.abs(fft_reference))]
214 freq_rec = fft_freqs_rec[numpy.argmax(abs_fft_rec)]
|
| /external/tensorflow/tensorflow/python/debug/examples/ |
| debug_mnist.py | 117 correct_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(y_, 1))
|
