Searched defs:value_bdim (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesLinearAlgebra.cpp | 493 const Tensor& value, std::optional<int64_t> value_bdim, in _scaled_dot_product_flash_attention_batch_rule()
541 const Tensor& value, optional<int64_t> value_bdim, in _scaled_dot_product_efficient_attention_batch_rule()
582 const Tensor& value, std::optional<int64_t> value_bdim, in _scaled_dot_product_cudnn_attention_batch_rule()
|
| H A D | BatchRulesScatterOps.cpp | 1143 const Tensor & value, std::optional<int64_t> value_bdim, in index_fill_int_tensor_batch_rule_impl()
1206 const Tensor & value, std::optional<int64_t> value_bdim) { in index_fill__int_tensor_batch_rule()
1223 const Tensor & value, std::optional<int64_t> value_bdim) { in index_fill_int_tensor_batch_rule()
|