Searched refs:tensor_arg (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_argmax_cuda.cu | 82 Tensor<DenseIndex, 3, DataLayout> tensor_arg(out_shape); in test_cuda_argmax_dim() local
99 std::size_t out_bytes = tensor_arg.size() * sizeof(DenseIndex); in test_cuda_argmax_dim()
116 …assert(cudaMemcpyAsync(tensor_arg.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, gpu_device.str… in test_cuda_argmax_dim()
119 VERIFY_IS_EQUAL(tensor_arg.size(), in test_cuda_argmax_dim()
122 for (DenseIndex n = 0; n < tensor_arg.size(); ++n) { in test_cuda_argmax_dim()
124 VERIFY_IS_EQUAL(tensor_arg.data()[n], 0); in test_cuda_argmax_dim()
144 …assert(cudaMemcpyAsync(tensor_arg.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, gpu_device.str… in test_cuda_argmax_dim()
147 for (DenseIndex n = 0; n < tensor_arg.size(); ++n) { in test_cuda_argmax_dim()
149 VERIFY_IS_EQUAL(tensor_arg.data()[n], tensor.dimension(dim) - 1); in test_cuda_argmax_dim()
171 Tensor<DenseIndex, 3, DataLayout> tensor_arg(out_shape); in test_cuda_argmin_dim() local
[all …]
|