Searched defs:tensor_arg (Results 1 – 1 of 1) sorted by relevance
82 Tensor<DenseIndex, 3, DataLayout> tensor_arg(out_shape); in test_cuda_argmax_dim() local171 Tensor<DenseIndex, 3, DataLayout> tensor_arg(out_shape); in test_cuda_argmin_dim() local