Searched defs:d_out (Results 1 – 12 of 12) sorted by relevance
82 float* d_out; in test_gpu_elementwise_small() local135 float* d_out; in test_gpu_elementwise() local181 bool* d_out; in test_gpu_props() local219 float* d_out; in test_gpu_reduction() local334 float* d_out; in test_gpu_convolution_1d() local387 float* d_out; in test_gpu_convolution_inner_dim_col_major_1d() local440 float* d_out; in test_gpu_convolution_inner_dim_row_major_1d() local494 float* d_out; in test_gpu_convolution_2d() local558 float* d_out; in test_gpu_convolution_3d() local635 Scalar* d_out; in test_gpu_lgamma() local[all …]
28 float* d_out; in test_gpu_random_uniform() local53 float* d_out; in test_gpu_random_normal() local
35 DataType* d_out = static_cast<DataType*>(sycl_device.allocate(out_bytes)); in test_sycl_random_uniform() local64 DataType* d_out = static_cast<DataType*>(sycl_device.allocate(out_bytes)); in test_sycl_random_normal() local
101 DenseIndex* d_out; in test_gpu_argmax_dim() local190 DenseIndex* d_out; in test_gpu_argmin_dim() local
107 DenseIndex* d_out = static_cast<DenseIndex*>(sycl_device.allocate(out_bytes)); in test_sycl_argmax_dim() local194 DenseIndex* d_out = static_cast<DenseIndex*>(sycl_device.allocate(out_bytes)); in test_sycl_argmin_dim() local
26 std::complex<T>* d_out; in test_cuda_complex_cwise_ops() local
284 float* d_out; in test_gpu() local
46 typename Output::Scalar* d_out; in run_on_gpu() local
109 def getLinear(d_in, d_out): argument
1032 FT_Pos d_in, d_out, d_hypot; in ft_corner_is_flat() local
433 d_out = 300 variable in TestModels._test_snli.Config
463 cl_mem d_out = NULL; in doTest() local