| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesScatterOps.cpp | 62 std::optional<int64_t> self_bdim, in batchIndices()
176 std::optional<int64_t> self_bdim, in index_batch_rule()
404 std::optional<int64_t> self_bdim, in index_put_batch_rule_helper()
480 std::optional<int64_t> self_bdim, in index_put__batch_rule()
519 std::optional<int64_t> self_bdim, in _index_put_impl__batch_rule()
607 std::optional<int64_t> self_bdim, in index_put_batch_rule()
676 const Tensor& self, std::optional<int64_t> self_bdim, in scatter_batch_rule()
708 const Tensor& self, std::optional<int64_t> self_bdim, in scatter_batch_rule()
746 const Tensor& self, std::optional<int64_t> self_bdim, in scatter_value_batch_rule()
755 const Tensor& self, std::optional<int64_t> self_bdim, in scatter_src_batch_rule()
[all …]
|
| H A D | BatchRulesViews.cpp | 95 std::optional<int64_t> self_bdim, in unsqueeze_batch_rule()
106 std::optional<int64_t> self_bdim, in repeat_batch_rule()
121 std::optional<int64_t> self_bdim, in _unsafe_view_batch_rule()
140 …std::optional<int64_t>> flip_batch_rule(const Tensor& self, std::optional<int64_t> self_bdim, IntA… in flip_batch_rule()
309 const Tensor &self, std::optional<int64_t> self_bdim, in diagonal_batching_rule()
336 std::optional<int64_t> self_bdim, in slice_batch_rule()
355 std::optional<int64_t> self_bdim, in transpose_int_batch_rule()
375 const Tensor &self, std::optional<int64_t> self_bdim, IntArrayRef dims) in permute_batching_rule()
419 const Tensor &self, std::optional<int64_t> self_bdim, SymIntArrayRef sym_size) in view_batching_rule()
432 std::optional<int64_t> self_bdim, in view_copy_batch_rule()
[all …]
|
| H A D | BatchRulesLoss.cpp | 31 loss_batch_rule_helper(const at::Tensor& self, std::optional<int64_t> self_bdim, const at::Tensor& … in loss_batch_rule_helper()
53 mse_loss_batch_rule(const at::Tensor& self, std::optional<int64_t> self_bdim, const at::Tensor& tar… in mse_loss_batch_rule()
62 huber_loss_batch_rule(const at::Tensor& self, std::optional<int64_t> self_bdim, const at::Tensor& t… in huber_loss_batch_rule()
71 smooth_l1_loss_batch_rule(const at::Tensor& self, std::optional<int64_t> self_bdim, const at::Tenso… in smooth_l1_loss_batch_rule()
|
| H A D | BatchRulesLinearAlgebra.cpp | 43 const Tensor& self, std::optional<int64_t> self_bdim, in tv_batch_rule()
70 const Tensor& self, std::optional<int64_t> self_bdim, in mv_batch_rule()
83 const Tensor& self, std::optional<int64_t> self_bdim, in mm_batch_rule()
98 const Tensor& self, std::optional<int64_t> self_bdim, in bmm_batch_rule()
333 const Tensor& self, std::optional<int64_t> self_bdim, in cholesky_solve_batch_rule()
355 oneOutput matrix_exp_batch_rule(const Tensor& self, std::optional<int64_t> self_bdim) { in matrix_exp_batch_rule()
403 oneOutput cross_batch_rule(const Tensor& self, std::optional<int64_t> self_bdim, in cross_batch_rule()
429 …const Tensor& self, std::optional<int64_t> self_bdim, const Tensor& b, std::optional<int64_t> b_bd… in linalg_lstsq_batch_rule()
|
| H A D | BatchRulesPooling.cpp | 17 const Tensor& self, std::optional<int64_t> self_bdim, in max_pool_with_indices_batch_rule_helper()
42 const Tensor& self, std::optional<int64_t> self_bdim, in max_pool3d_with_indices_batch_rule()
50 const Tensor& self, std::optional<int64_t> self_bdim, in max_pool2d_with_indices_batch_rule()
|
| H A D | BatchRulesBinaryOps.cpp | 147 …const Tensor& self, std::optional<int64_t> self_bdim, const Tensor& other, std::optional<int64_t> … in where_self_batch_rule()
181 const Tensor& self, std::optional<int64_t> self_bdim, in masked_select_batch_rule()
201 const Tensor& self, std::optional<int64_t> self_bdim, in masked_select_backward_batch_rule()
271 std::optional<int64_t> self_bdim, in fill__Tensor_batch_rule()
290 Tensor& self, std::optional<int64_t> self_bdim, in log_sigmoid_backward_batch_rule()
|
| H A D | BatchRulesUnaryOps.cpp | 16 std::optional<int64_t> self_bdim, in clone_batch_rule()
52 view_as_complex_batch_rule(const Tensor& self, std::optional<int64_t> self_bdim) { in view_as_complex_batch_rule()
|
| H A D | BatchRulesActivation.cpp | 15 glu_batch_rule(const Tensor& self, std::optional<int64_t> self_bdim, int64_t dim) { in glu_batch_rule()
32 const Tensor& self, std::optional<int64_t> self_bdim, int64_t dim) { in glu_backward_batch_rule()
|
| H A D | BatchRulesReduceOps.cpp | 26 const Tensor& self, std::optional<int64_t> self_bdim) { in _is_all_true_batch_rule()
31 const Tensor& self, std::optional<int64_t> self_bdim) { in _is_any_true_batch_rule()
321 std::optional<int64_t> self_bdim, in searchsorted_batch_rule()
|
| H A D | BatchRulesFactory.cpp | 66 const Tensor& self, std::optional<int64_t> self_bdim, in _new_zeros_with_same_feature_meta_batch_rule()
|