Searched defs:src_bdim (Results 1 – 1 of 1) sorted by relevance
711 const Tensor& src, std::optional<int64_t> src_bdim, Args... args) { in scatter_batch_rule()758 const Tensor& src, std::optional<int64_t> src_bdim) { in scatter_src_batch_rule()767 const Tensor& src, std::optional<int64_t> src_bdim) { in scatter_add_batch_rule()776 const Tensor& src, std::optional<int64_t> src_bdim, in scatter_reduce_batch_rule()907 const Tensor &src, std::optional<int64_t> src_bdim, in diagonal_scatter_batch_rule()