Searched defs:d_out (Results 1 – 8 of 8) sorted by relevance
81 float* d_out; in test_cuda_elementwise_small() local134 float* d_out; in test_cuda_elementwise() local180 bool* d_out; in test_cuda_props() local218 float* d_out; in test_cuda_reduction() local333 float* d_out; in test_cuda_convolution_1d() local386 float* d_out; in test_cuda_convolution_inner_dim_col_major_1d() local439 float* d_out; in test_cuda_convolution_inner_dim_row_major_1d() local493 float* d_out; in test_cuda_convolution_2d() local557 float* d_out; in test_cuda_convolution_3d() local633 Scalar* d_out; in test_cuda_lgamma() local[all …]
30 float* d_out; in test_cuda_random_uniform() local55 float* d_out; in test_cuda_random_normal() local
102 DenseIndex* d_out; in test_cuda_argmax_dim() local191 DenseIndex* d_out; in test_cuda_argmin_dim() local
29 std::complex<T>* d_out; in test_cuda_complex_cwise_ops() local
285 float* d_out; in test_gpu() local
37 typename Output::Scalar* d_out; in run_on_cuda() local
70 def _test_spinn_config(d_embed, d_out, logdir=None, inference_sentences=None): argument
983 FT_Pos d_in, d_out, d_hypot; in ft_corner_is_flat() local