Searched defs:tensor_arg (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| 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
|
| 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
|