| /external/tensorflow/tensorflow/python/kernel_tests/ |
| argmax_op_test.py | 62 # Check that argmin and argmax match numpy along the primary axis
64 self._testBothArg(math_ops.argmin, x, 0, x.argmin())
69 # Check that argmin and argmax match numpy along all axes
72 self._testBothArg(math_ops.argmin, x, axis, x.argmin(axis))
88 expected_values = x.argmin()
90 ans = math_ops.argmin(x, axis=0, output_type=dtypes.int32)
109 for op in math_ops.argmin, math_ops.argmax:
117 for op in math_ops.argmin, math_ops.argmax
[all...] |
| control_flow_ops_py_test.py | [all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| argminmax_test.py | 15 """Functional tests for ArgMin and ArgMax Ops."""
37 op: argmin or argmax operator to test.
52 # Complex numbers do not support argmin/argmax.
79 math_ops.argmin,
85 math_ops.argmin,
91 math_ops.argmin,
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| device_setter.py | 100 task = np.argmin(self._ps_loads)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| argmax_op.h | 47 struct ArgMin {
52 output.device(d) = input.argmin(dimension).template cast<Tout>(); \
|
| /external/walt/android/WALT/app/src/test/java/org/chromium/latency/walt/ |
| UtilsTest.java | 130 assertThat(Utils.argmin(new double[]{5, 2, 1, -10, -20, 5, 19, 100}), is(4));
135 assertThat(Utils.argmin(new double[]{}), is(0));
|
| /external/walt/pywalt/pywalt/ |
| minimization.py | 69 best_shift0 = shifts[numpy.argmin(residuals0)]
70 best_shift1 = shifts[numpy.argmin(residuals1)]
|
| /external/eigen/unsupported/test/ |
| cxx11_tensor_argmax.cpp | 149 tensor_argmin = tensor.argmin();
155 tensor_argmin = tensor.argmin();
243 tensor_argmin = tensor.argmin(dim);
265 tensor_argmin = tensor.argmin(dim);
|
| /external/tensorflow/tensorflow/tools/compatibility/testdata/ |
| test_file_v1_12.py | 66 tf.argmin([[1, 3, 2]], name='abc', dimension=1))
|
| test_file_v0_11.py | 105 tf.argmin([[1, 2, 3], [4, 1, 0]], dimension=1).eval(),
108 tf.argmin([[1, 2, 3], [4, 1, 0]], dimension=0).eval(),
|
| /external/walt/android/WALT/app/src/main/java/org/chromium/latency/walt/ |
| Utils.java | 124 public static int argmin(double[] a) { method in class:Utils
156 bestShift = argmin(stddevs) * shiftStep + bestShift - shiftStep * steps / 2;
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| grow_stats.cc | 622 int argmin(const std::unordered_map<int, float>& m) { function in namespace:tensorflow::tensorforest
639 smallest_weight_class_ = argmin(class_weights_);
649 smallest_weight_class_ = argmin(class_weights_);
662 smallest_weight_class_ = argmin(class_weights_);
710 smallest_weight_class_ = argmin(class_weights_);
[all...] |
| /external/tensorflow/tensorflow/contrib/keras/api/keras/backend/ |
| __init__.py | 27 from tensorflow.python.keras.backend import argmin
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorBase.h | 627 argmin() const { function in class:Eigen::TensorBase
653 argmin(const int return_dim) const { function in class:Eigen::TensorBase
[all...] |
| /external/tensorflow/tensorflow/python/keras/ |
| metrics.py | [all...] |
| backend_test.py | 306 (keras.backend.argmin, np.argmin),
[all...] |
| backend.py | 1833 def argmin(x, axis=-1): function
[all...] |
| /external/tensorflow/tensorflow/contrib/losses/python/metric_learning/ |
| metric_loss_ops.py | 533 predictions = math_ops.argmin(
[all...] |
| metric_loss_ops_test.py | 510 cluster_ics = np.argmin(pdists[subset, :], axis=0)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| math_utils.py | [all...] |
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| metric_ops.py | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| math_ops.py | 105 arg_min = deprecation.deprecated(None, "Use `tf.math.argmin` instead")(arg_min) # pylint: disable=used-before-assignment
169 @tf_export(v1=["math.argmin", "argmin"])
175 def argmin(input, function
185 @tf_export("math.argmin", "argmin", v1=[])
[all...] |
| metrics_impl.py | [all...] |
