Searched defs:grad_output_bdim (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesModules.cpp | 140 const Tensor& grad_output, std::optional<int64_t> grad_output_bdim, in grid_sample_backward_helper_in()
180 const Tensor& grad_output, std::optional<int64_t> grad_output_bdim, in grid_sample_backward_batch_rule()
203 const Tensor& grad_output, std::optional<int64_t> grad_output_bdim) { in cudnn_grid_sample_backward_batch_rule()
|
| D | BatchRulesActivation.cpp | 31 const Tensor& grad_output, std::optional<int64_t> grad_output_bdim, in glu_backward_batch_rule()
|
| D | BatchRulesReduceOps.cpp | 260 const Tensor& grad_output, std::optional<int64_t> grad_output_bdim, in _softmax_backward_batch_rule()
290 const Tensor& grad_output, std::optional<int64_t> grad_output_bdim, in _log_softmax_backward_batch_rule()
|
| D | BatchRulesConvolution.cpp | 169 const Tensor& grad_output, std::optional<int64_t> grad_output_bdim, in convolution_backward_input_batch_rule()
250 const Tensor& grad_output, std::optional<int64_t> grad_output_bdim, in convolution_backward_weight_batch_rule()
|