Searched defs:gemm_in_ptr (Results 1 – 5 of 5) sorted by relevance
200 auto gemm_in_ptr = requires_columns ? in slow_conv2d_forward() local356 auto gemm_in_ptr = requires_columns ? in slow_conv2d_grad_weight() local
454 auto gemm_in_ptr = need_columns ? grad_columns.const_data_ptr<scalar_t>() in slow_conv_transpose2d_backward_out_cuda_template() local651 auto gemm_in_ptr = need_columns ? columns.const_data_ptr<scalar_t>() in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
574 auto gemm_in_ptr = need_columns ? grad_columns.const_data_ptr<scalar_t>() in slow_conv_transpose3d_backward_out_cuda_template() local802 …auto gemm_in_ptr = need_columns ? columns.const_data_ptr<scalar_t>() : grad_output_n.const_data_pt… in slow_conv_transpose3d_acc_grad_parameters_cuda() local
532 auto gemm_in_ptr = need_columns ? grad_columns.const_data_ptr<scalar_t>() in slow_conv_transpose2d_backward_out_cpu_template() local730 auto gemm_in_ptr = need_columns ? columns.const_data_ptr<scalar_t>() in slow_conv_transpose2d_acc_grad_parameters_cpu() local
565 auto gemm_in_ptr = need_columns ? grad_columns.const_data_ptr<scalar_t>() in slow_conv_transpose3d_backward_out_cpu_template() local785 auto gemm_in_ptr = need_columns ? columns.const_data_ptr<scalar_t>() in slow_conv_transpose3d_acc_grad_parameters_cpu() local