Searched defs:tensor_arg (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_argmax_gpu.cu | 81 Tensor<DenseIndex, 3, DataLayout> tensor_arg(out_shape); in test_gpu_argmax_dim() local
170 Tensor<DenseIndex, 3, DataLayout> tensor_arg(out_shape); in test_gpu_argmin_dim() local
|
| H A D | cxx11_tensor_argmax_sycl.cpp | 84 Tensor<DenseIndex, 3, DataLayout, DenseIndex> tensor_arg(out_shape); in test_sycl_argmax_dim() local
172 Tensor<DenseIndex, 3, DataLayout, DenseIndex> tensor_arg(out_shape); in test_sycl_argmin_dim() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Concat.cpp | 158 for (const at::Tensor& tensor_arg : tensors) { in cat_feature_mult4ch() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_torch.py | 10688 def make_neg_dim_test(name, tensor_arg, arg_constr, types, extra_dim=0): argument
|