Home
last modified time | relevance | path

Searched refs:argmin (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dargmax_op_test.py64 self._testBothArg(math_ops.argmin, x, 0, x.argmin())
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:
124 for op in math_ops.argmin, math_ops.argmax:
/external/tensorflow/tensorflow/compiler/tests/
Dargminmax_test.py79 math_ops.argmin,
85 math_ops.argmin,
91 math_ops.argmin,
/external/eigen/unsupported/test/
Dcxx11_tensor_argmax.cpp149 tensor_argmin = tensor.argmin(); in test_simple_argmin()
155 tensor_argmin = tensor.argmin(); in test_simple_argmin()
243 tensor_argmin = tensor.argmin(dim); in test_argmin_dim()
265 tensor_argmin = tensor.argmin(dim); in test_argmin_dim()
Dcxx11_tensor_argmax_cuda.cu54 gpu_out_min.device(gpu_device) = gpu_in.argmin(); in test_cuda_simple_argmax()
203 gpu_out.device(gpu_device) = gpu_in.argmin(dim); in test_cuda_argmin_dim()
231 gpu_out.device(gpu_device) = gpu_in.argmin(dim); in test_cuda_argmin_dim()
/external/walt/android/WALT/app/src/test/java/org/chromium/latency/walt/
DUtilsTest.java130 assertThat(Utils.argmin(new double[]{5, 2, 1, -10, -20, 5, 19, 100}), is(4)); in testArgmin()
135 assertThat(Utils.argmin(new double[]{}), is(0)); in testArgmin_empty()
/external/walt/pywalt/pywalt/
Dminimization.py69 best_shift0 = shifts[numpy.argmin(residuals0)]
70 best_shift1 = shifts[numpy.argmin(residuals1)]
/external/walt/android/WALT/app/src/main/java/org/chromium/latency/walt/
DUtils.java124 public static int argmin(double[] a) { in argmin() method in Utils
156 bestShift = argmin(stddevs) * shiftStep + bestShift - shiftStep * steps / 2; in findBestShift()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_MatrixSolveLs.pbtxt52 problem \\(X = \mathrm{argmin}_{Z \in \Re^{n \times k} } ||A Z - B||_F^2 + \lambda ||Z||_F^2\\).
56 \\(X = \mathrm{argmin}_{Z \in \mathbb{C}^{n \times k} } ||Z||_F^2 \\),
/external/tensorflow/tensorflow/core/kernels/
Dargmax_op.h52 output.device(d) = input.argmin(dimension).template cast<Tout>(); \
/external/tensorflow/tensorflow/contrib/training/python/training/
Ddevice_setter.py100 task = np.argmin(self._ps_loads)
/external/tensorflow/tensorflow/tools/compatibility/testdata/
Dtest_file_v0_11.py105 tf.argmin([[1, 2, 3], [4, 1, 0]], dimension=1).eval(),
108 tf.argmin([[1, 2, 3], [4, 1, 0]], dimension=0).eval(),
Dtest_file_v1_12.py66 tf.argmin([[1, 3, 2]], name='abc', dimension=1))
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats.cc622 int argmin(const std::unordered_map<int, float>& m) { in argmin() function
639 smallest_weight_class_ = argmin(class_weights_); in accumulate()
649 smallest_weight_class_ = argmin(class_weights_); in accumulate()
662 smallest_weight_class_ = argmin(class_weights_); in accumulate()
710 smallest_weight_class_ = argmin(class_weights_); in ExtractFromProto()
/external/tensorflow/tensorflow/contrib/keras/api/keras/backend/
D__init__.py27 from tensorflow.python.keras.backend import argmin
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h627 argmin() const { in argmin() function
653 argmin(const int return_dim) const { in argmin() function
/external/tensorflow/tensorflow/python/keras/
Dmetrics.py1457 min_index = math_ops.argmin(
1538 min_index = math_ops.argmin(
Dbackend_test.py306 (keras.backend.argmin, np.argmin),
Dbackend.py1833 def argmin(x, axis=-1): function
1843 return math_ops.argmin(x, axis)
/external/tensorflow/tensorflow/contrib/losses/python/metric_learning/
Dmetric_loss_ops_test.py510 cluster_ics = np.argmin(pdists[subset, :], axis=0)
Dmetric_loss_ops.py533 predictions = math_ops.argmin(
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dmath_utils.py886 min_time_batch = math_ops.cast(math_ops.argmin(times[:, 0]), dtypes.int32)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt36 name: "argmin"
Dtensorflow.keras.backend.pbtxt28 name: "argmin"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt36 name: "argmin"
Dtensorflow.keras.backend.pbtxt28 name: "argmin"

12