Searched refs:tensor_argmin (Results 1 – 1 of 1) sorted by relevance
147 Tensor<DenseIndex, 0, DataLayout> tensor_argmin; in test_simple_argmin() local149 tensor_argmin = tensor.argmin(); in test_simple_argmin()151 VERIFY_IS_EQUAL(tensor_argmin(0), 0); in test_simple_argmin()155 tensor_argmin = tensor.argmin(); in test_simple_argmin()157 VERIFY_IS_EQUAL(tensor_argmin(0), 2*3*5*7 - 1); in test_simple_argmin()228 Tensor<DenseIndex, 3, DataLayout> tensor_argmin; in test_argmin_dim() local243 tensor_argmin = tensor.argmin(dim); in test_argmin_dim()245 VERIFY_IS_EQUAL(tensor_argmin.size(), in test_argmin_dim()247 for (ptrdiff_t n = 0; n < tensor_argmin.size(); ++n) { in test_argmin_dim()249 VERIFY_IS_EQUAL(tensor_argmin.data()[n], 0); in test_argmin_dim()[all …]