Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dargmax_op_test.py62 self._testBothArg(math_ops.argmin, x, 0, x.argmin())
70 self._testBothArg(math_ops.argmin, x, axis, x.argmin(axis))
86 expected_values = x.argmin()
88 ans = math_ops.argmin(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:
/external/tensorflow/tensorflow/compiler/tests/
Dargminmax_test.py66 lambda x: math_ops.argmin(x, axis=0, output_type=dtypes.int32),
70 lambda x: math_ops.argmin(x, axis=0, output_type=dtypes.int32),
74 lambda x: math_ops.argmin(x, axis=1, output_type=dtypes.int32),
/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/lisa/tools/scripts/power/
Dpower_average.py52 end_idx = np.abs(df.index - end).argmin()
57 start_idx = np.abs(df.index - start).argmin()
/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 +
56 \\(X = \mathrm{argmin}_{Z \in \mathbb{C}^{n \times k} } ||Z||_F^2 \\),
/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/tools/compatibility/testdata/
Dtest_file_v0_11.py103 tf.argmin([[1, 2, 3], [4, 1, 0]], dimension=1).eval(),
106 tf.argmin([[1, 2, 3], [4, 1, 0]], dimension=0).eval(),
/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/compiler/tf2xla/
Dxla_helpers.cc197 xla::ComputationDataHandle* argmin) { in ArgMin() argument
199 axis, /*is_min=*/true, argmin); in ArgMin()
Dxla_helpers.h93 xla::ComputationDataHandle* argmin);
/external/tensorflow/tensorflow/docs_src/api_guides/python/
Dmath_ops.md193 * @{tf.argmin}
/external/tensorflow/tensorflow/contrib/keras/api/keras/backend/
D__init__.py27 from tensorflow.python.keras._impl.keras.backend import argmin
/external/tensorflow/tensorflow/python/keras/backend/
D__init__.py27 from tensorflow.python.keras._impl.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/_impl/keras/
Dbackend_test.py335 (keras.backend.argmin, np.argmin),
/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.py853 min_time_batch = math_ops.cast(math_ops.argmin(times[:, 0]), dtypes.int32)
/external/tensorflow/tensorflow/docs_src/install/
Dmigration.md215 * `tf.argmin`
/external/tensorflow/tensorflow/tools/api/golden/
Dtensorflow.keras.backend.pbtxt28 name: "argmin"

12