Searched refs:backend_grad_input (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Convolution.cpp | 2049 Tensor backend_grad_input, backend_grad_weight, backend_grad_bias; in convolution_backward() local
2055 std::tie(backend_grad_input, backend_grad_weight) = in convolution_backward()
2062 std::tie(backend_grad_input, backend_grad_weight, backend_grad_bias) = in convolution_backward()
2071 std::tie(backend_grad_input, backend_grad_weight) = cudnn_convolution_backward_stub( in convolution_backward()
2084 std::tie(backend_grad_input, backend_grad_weight, backend_grad_bias) = in convolution_backward()
2097 std::tie(backend_grad_input, backend_grad_weight) = at::mps_convolution_transpose_backward( in convolution_backward()
2111 std::tie(backend_grad_input, backend_grad_weight) = cudnn_convolution_transpose_backward_stub( in convolution_backward()
2122 backend_grad_input = at::zeros_like(input); in convolution_backward()
2135 …backend_grad_input = empty_mkldnn(input.sizes(), optTypeMetaToScalarType(input.options().dtype_opt… in convolution_backward()
2137 backend_grad_input.zero_(); in convolution_backward()
[all …]
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 3113 backend_grad_input = None
3118 backend_grad_input = grad_output_.new_empty(input_.size())
3124 return (backend_grad_input, backend_grad_weight, backend_grad_bias)
|