Searched defs:grad_input_bdim (Results 1 – 2 of 2) sorted by relevance
321 const Tensor& grad_input, std::optional<int64_t> grad_input_bdim, in diagonal_backward_batch_rule()393 const Tensor& grad_input, std::optional<int64_t> grad_input_bdim, in select_backward_batch_rule()406 const Tensor& grad_input, std::optional<int64_t> grad_input_bdim, in slice_backward_batch_rule()
621 grad_input_bdim, argument649 grad_input_bdim, argument