Searched refs:argmin (Results 1 – 3 of 3) sorted by relevance
149 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()
54 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()
627 argmin() const { in argmin() function653 argmin(const int return_dim) const { in argmin() function