Searched defs:fgrad_input (Results 1 – 2 of 2) sorted by relevance
290 scalar_t *fgrad_input, in slow_conv2d_backward_update_grad_input_frame()413 auto fgrad_input = std::make_unique<scalar_t[]>(fgrad_input_size); in slow_conv2d_backward_out_cpu_template() local
316 TensorAccessor<scalar_t, 2> fgrad_input, in slow_conv3d_backward_update_grad_input_frame()427 Tensor fgrad_input = at::empty({batch_size, in slow_conv3d_backward_out_cpu_template() local