Searched defs:input_bdim (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 113 grid_sample_batch_rule(const Tensor& input, std::optional<int64_t> input_bdim, const Tensor& grid, … in grid_sample_batch_rule()
141 const Tensor& input, std::optional<int64_t> input_bdim, in grid_sample_backward_helper_in()
181 const Tensor& input, std::optional<int64_t> input_bdim, in grid_sample_backward_batch_rule()
201 const Tensor& input, std::optional<int64_t> input_bdim, in cudnn_grid_sample_backward_batch_rule()
|
| H A D | BatchRulesNorm.cpp | 20 std::optional<int64_t> input_bdim, in compute_stat_bdim()
47 const Tensor& input, std::optional<int64_t> input_bdim, in batch_norm_batch_rule()
129 const at::Tensor & input, std::optional<int64_t> input_bdim, in batch_norm_backward_no_weight_bias_batch_rule()
342 const at::Tensor & input, std::optional<int64_t> input_bdim, in group_norm_backward_no_weight_bias_batch_rule()
489 const Tensor& input, std::optional<int64_t> input_bdim, in native_layer_norm_batch_rule()
535 const at::Tensor & input, std::optional<int64_t> input_bdim, in native_layer_norm_backward_no_weight_bias_batch_rule()
|
| H A D | BatchRulesLinearAlgebra.cpp | 161 householder_product_batch_rule(const Tensor &input, std::optional<int64_t> input_bdim, in householder_product_batch_rule()
454 F Func, const Tensor& input, std::optional<int64_t> input_bdim, in atol_rtol_tensor_batch_rule()
483 const Tensor& input, std::optional<int64_t> input_bdim, const std::optional<Tensor>& atol, in pinv_batch_rule()
|
| H A D | BatchRulesConvolution.cpp | 170 const Tensor& input, std::optional<int64_t> input_bdim, in convolution_backward_input_batch_rule()
251 const Tensor& input, std::optional<int64_t> input_bdim, in convolution_backward_weight_batch_rule()
|
| H A D | BatchRulesBinaryOps.cpp | 164 …_out, std::optional<int64_t> grad_out_bdim, const Tensor& input, std::optional<int64_t> input_bdim, in gelu_backward_batch_rule()
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | autograd_function.py | 622 input_bdim, argument
650 input_bdim, argument
|