Searched defs:grad_input_reshaped (Results 1 – 2 of 2) sorted by relevance
609 Tensor grad_input_reshaped; in batch_norm_backward_cuda_template() local846 auto grad_input_reshaped = at::empty_like(input_reshaped, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in batch_norm_backward_elemt_cuda_template() local897 auto grad_input_reshaped = at::empty_like(input_reshaped, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in batch_norm_backward_elemt_cuda_template() local
550 auto grad_input_reshaped = grad_input; in max_unpooling3d_backward_out_cuda() local