| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesViews.cpp | 97 auto self_ = moveBatchDimToFront(self, self_bdim); in unsqueeze_batch_rule()
111 auto self_ = moveBatchDimToFront(self, self_bdim); in repeat_batch_rule()
123 auto self_ = moveBatchDimToFront(self, self_bdim); in _unsafe_view_batch_rule()
141 auto self_ = moveBatchDimToFront(self, self_bdim); in flip_batch_rule()
175 self_value = moveBatchDimToFront(self_value, self_bdim); in resize__plumbing()
265 auto _self = moveBatchDimToFront(self, bdim); in select_batching_rule()
275 auto self_ = moveBatchDimToFront(self, bdim); in _reshape_alias_batch_rule()
285 auto self_ = moveBatchDimToFront(self, bdim); in roll_batch_rule()
313 auto self_ = moveBatchDimToFront(self, self_bdim); in diagonal_batching_rule()
324 auto grad_input_ = moveBatchDimToFront(grad_input, grad_input_bdim); in diagonal_backward_batch_rule()
[all …]
|
| D | BatchRulesLinearAlgebra.cpp | 27 auto A_ = moveBatchDimToFront(A, A_bdim); in dot_batch_rule()
28 auto B_ = moveBatchDimToFront(B, B_bdim); in dot_batch_rule()
49 auto other_ = moveBatchDimToFront(other, other_bdim); in tv_batch_rule()
57 auto self_ = moveBatchDimToFront(self, self_bdim); in tv_batch_rule()
92 auto self_ = moveBatchDimToFront(self, self_bdim); in mm_batch_rule()
93 auto other_ = moveBatchDimToFront(other, other_bdim); in mm_batch_rule()
107 auto self_ = moveBatchDimToFront(self, self_bdim); in bmm_batch_rule()
108 auto other_ = moveBatchDimToFront(other, other_bdim); in bmm_batch_rule()
155 auto info_ = moveBatchDimToFront(info, info_bdim); in _linalg_check_errors_batch_rule()
164 auto input_ = moveBatchDimToFront(input, input_bdim); in householder_product_batch_rule()
[all …]
|
| D | BatchRulesLoss.cpp | 17 auto result = moveBatchDimToFront(tensor, bdim); in flatten_logical()
42 moveBatchDimToFront(self, self_bdim) : moveBatchDimToFront(target, target_bdim); in loss_batch_rule_helper()
108 auto self_ = moveBatchDimToFront(self_value, self_bdim); in binary_cross_entropy_plumbing()
109 auto target_ = moveBatchDimToFront(target_value, target_bdim); in binary_cross_entropy_plumbing()
147 auto grad_ = moveBatchDimToFront(grad_value, grad_bdim); in binary_cross_entropy_backward_plumbing()
148 auto input_ = moveBatchDimToFront(input_value, input_bdim); in binary_cross_entropy_backward_plumbing()
149 auto target_ = moveBatchDimToFront(target_value, target_bdim); in binary_cross_entropy_backward_plumbing()
|
| D | BatchRulesActivation.cpp | 24 const auto self_ = moveBatchDimToFront(self, self_bdim); in glu_batch_rule()
44 …const auto grad_output_ = ensure_has_bdim(moveBatchDimToFront(grad_output, grad_output_bdim), grad… in glu_backward_batch_rule()
45 …const auto self_ = ensure_has_bdim(moveBatchDimToFront(self, self_bdim), self_bdim.has_value(), ba… in glu_backward_batch_rule()
|
| D | BatchRulesBinaryOps.cpp | 109 auto tensor_ = moveBatchDimToFront(tensor, tensor_batch_dim); in binary_pointwise_inplace_batch_rule()
110 auto other_ = moveBatchDimToFront(other, other_batch_dim); in binary_pointwise_inplace_batch_rule()
131 auto tensor_ = moveBatchDimToFront(tensor, tensor_batch_dim); in comparison_pointwise_batch_rule()
132 auto other_ = moveBatchDimToFront(other, other_batch_dim); in comparison_pointwise_batch_rule()
153 auto condition_ = moveBatchDimToFront(condition, condition_bdim); in where_self_batch_rule()
154 auto self_ = moveBatchDimToFront(self, self_bdim); in where_self_batch_rule()
155 auto other_ = moveBatchDimToFront(other, other_bdim); in where_self_batch_rule()
188 auto self_ = moveBatchDimToFront(self, self_bdim); in masked_select_batch_rule()
208 auto self_ = moveBatchDimToFront(self, self_bdim); in masked_select_backward_batch_rule()
209 auto grad_ = moveBatchDimToFront(grad, grad_bdim); in masked_select_backward_batch_rule()
[all …]
|
| D | BatchRulesReduceOps.cpp | 128 self = moveBatchDimToFront(self, self_bdim); in boxed_reduction_batch_rule()
268 auto grad_output_ = moveBatchDimToFront(grad_output, grad_output_bdim); in _softmax_backward_batch_rule()
269 auto output_ = moveBatchDimToFront(output, output_bdim); in _softmax_backward_batch_rule()
299 auto grad_output_ = moveBatchDimToFront(grad_output, grad_output_bdim); in _log_softmax_backward_batch_rule()
300 auto output_ = moveBatchDimToFront(output, output_bdim); in _log_softmax_backward_batch_rule()
333 auto buckets = moveBatchDimToFront(sorted_sequence, sorted_sequence_bdim); in searchsorted_batch_rule()
341 auto sorter__ = moveBatchDimToFront(*sorter, sorter_bdim); in searchsorted_batch_rule()
359 auto self_ = moveBatchDimToFront(self, self_bdim); in searchsorted_batch_rule()
365 auto self_ = moveBatchDimToFront(self, self_bdim); in searchsorted_batch_rule()
383 auto self_ = moveBatchDimToFront(self, self_bdim); in searchsorted_batch_rule()
|
| D | BatchRulesScatterOps.cpp | 94 …indices_.emplace_back(maybePadToLogicalRank(moveBatchDimToFront(index.value(), idx_bdim), idx_bdim… in batchIndices()
233 auto self_ = moveBatchDimToFront(self, self_bdim); in index_batch_rule()
411 Tensor self_ = moveBatchDimToFront(self, self_bdim);
412 Tensor values_ = moveBatchDimToFront(values, values_bdim);
684 auto self_ = moveBatchDimToFront(self, self_bdim); in scatter_batch_rule()
685 auto index_ = moveBatchDimToFront(index, index_bdim); in scatter_batch_rule()
717 auto self_ = moveBatchDimToFront(self, self_bdim); in scatter_batch_rule()
718 auto index_ = moveBatchDimToFront(index, index_bdim); in scatter_batch_rule()
719 auto src_ = moveBatchDimToFront(src, src_bdim); in scatter_batch_rule()
801 auto self_ = moveBatchDimToFront(self, self_bdim); in gather_batch_rule()
[all …]
|
| D | BatchRulesNorm.cpp | 79 auto input_ = moveBatchDimToFront(input, input_bdim); in batch_norm_batch_rule()
86 running_mean_ = moveBatchDimToFront(running_mean, running_mean_bdim); in batch_norm_batch_rule()
91 running_var_ = moveBatchDimToFront(running_var, running_var_bdim); in batch_norm_batch_rule()
110 auto weight_ = moveBatchDimToFront(weight, weight_bdim); in batch_norm_batch_rule()
118 auto bias_ = moveBatchDimToFront(bias, bias_bdim); in batch_norm_batch_rule()
150 auto grad_out_ = moveBatchDimToFront(grad_out, grad_out_bdim); in batch_norm_backward_no_weight_bias_batch_rule()
151 auto input_ = moveBatchDimToFront(input, input_bdim); in batch_norm_backward_no_weight_bias_batch_rule()
152 auto mean_ = moveBatchDimToFront(mean, mean_bdim); in batch_norm_backward_no_weight_bias_batch_rule()
153 auto rstd_ = moveBatchDimToFront(rstd, rstd_bdim); in batch_norm_backward_no_weight_bias_batch_rule()
165 running_mean_ = moveBatchDimToFront(running_mean, running_mean_bdim); in batch_norm_backward_no_weight_bias_batch_rule()
[all …]
|
| D | BatchRulesModules.cpp | 45 auto indices_ = moveBatchDimToFront(indices, indices_bdim); in embedding_batch_rule()
69 indices = moveBatchDimToFront(indices, indices_bdim); in embedding_dense_backward_batch_rule()
70 grad = moveBatchDimToFront(grad, grad_bdim); in embedding_dense_backward_batch_rule()
147 auto grad_output_ = moveBatchDimToFront(grad_output, grad_output_bdim); in grid_sample_backward_helper_in()
151 auto input_ = moveBatchDimToFront(input, input_bdim); in grid_sample_backward_helper_in()
155 auto grid_ = moveBatchDimToFront(grid, grid_bdim); in grid_sample_backward_helper_in()
|
| D | BatchRulesHelper.cpp | 12 Tensor moveBatchDimToFront(const Tensor& tensor, std::optional<int64_t> maybe_batch_dim) { in moveBatchDimToFront() function
180 auto tensor_ = moveBatchDimToFront(tensor, tensor_batch_dim); in _binary_pointwise_helper()
181 auto other_ = moveBatchDimToFront(other, other_batch_dim); in _binary_pointwise_helper()
|
| D | BatchRulesRandomness.cpp | 43 self_value = moveBatchDimToFront(self_value, self_bdim); in random_inplace_batching_rule()
82 auto self_ = moveBatchDimToFront(self_value, self_bdim); in bernoulli_inplace_Tensor_batching_rule()
83 auto other_ = moveBatchDimToFront(other_value, other_bdim); in bernoulli_inplace_Tensor_batching_rule()
132 tensor_value = moveBatchDimToFront(tensor_value, tensor_bdim); in unary_pointwise_random_batch_rule()
160 tensor_value = moveBatchDimToFront(tensor_value, tensor_bdim); in tensor_like_random_batch_rule()
183 tensor_value = moveBatchDimToFront(tensor_value, tensor_bdim); in native_dropout_batching_rule()
222 self_value = moveBatchDimToFront(self_value, self_bdim); in multinomial_batching_rule()
|
| D | BatchRulesFactory.cpp | 84 auto base_ = moveBatchDimToFront(base, base_bdim); in _new_zeros_with_same_feature_meta_batch_rule()
101 auto tangent_ = moveBatchDimToFront(tangent, tangent_bdim); in _new_zeros_with_same_feature_meta_batch_rule()
119 start_ = moveBatchDimToFront(start_, start_bdim); in linspace_logspace_batch_rule_helper()
120 end_ = moveBatchDimToFront(end_, end_bdim); in linspace_logspace_batch_rule_helper()
|
| D | BatchRulesUnaryOps.cpp | 41 auto self_ = moveBatchDimToFront(self, self_bdim); in clone_batch_rule()
57 auto self_ = moveBatchDimToFront(self, self_bdim); in view_as_complex_batch_rule()
|
| D | BatchRulesHelper.h | 33 Tensor moveBatchDimToFront(const Tensor& tensor, std::optional<int64_t> maybe_batch_dim);
103 auto tensor_ = moveBatchDimToFront(tensor, batch_dim);
184 tensor_input.first = moveBatchDimToFront(tensor_input.first, tensor_input.second);
197 tensor_input.first = moveBatchDimToFront(tensor_input.first, tensor_input.second);
338 value_ = moveBatchDimToFront(value_, bdim);
|
| D | BatchRulesConvolution.cpp | 113 A = moveBatchDimToFront(A, A_batch_dim); in convolution_batch_rule()
114 B = moveBatchDimToFront(B, B_batch_dim); in convolution_batch_rule()
223 auto weight_ = moveBatchDimToFront(weight, weight_bdim); // B(GI)O in convolution_backward_input_batch_rule()
283 auto grad_output_ = moveBatchDimToFront(grad_output, grad_output_bdim); // BN(GO) in convolution_backward_weight_batch_rule()
324 auto input_ = moveBatchDimToFront(input, input_bdim); // BN(GI) in convolution_backward_weight_batch_rule()
|
| D | BatchRulesPooling.cpp | 25 auto self_ = moveBatchDimToFront(self, self_bdim); in max_pool_with_indices_batch_rule_helper()
|
| D | LegacyBatchingRegistrations.cpp | 572 tensors_to_cat.emplace_back(moveBatchDimToFront(batched->value(), batched->bdim())); in cat_batching_rule()
|
| /external/pytorch/functorch/ |
| D | writing_batching_rules.md | 25 auto self_ = moveBatchDimToFront(self, self_bdim);
41 auto self_ = moveBatchDimToFront(self, self_bdim);
|