Searched defs:output_bdim (Results 1 – 1 of 1) sorted by relevance
261 const Tensor& output, std::optional<int64_t> output_bdim, in _softmax_backward_batch_rule()291 const Tensor& output, std::optional<int64_t> output_bdim, in _log_softmax_backward_batch_rule()