Searched defs:logit_backward (Results 1 – 2 of 2) sorted by relevance
306 TORCH_META_FUNC(logit_backward) (const Tensor& grad_output, const Tensor& input, std::optional<doub… in TORCH_META_FUNC() argument
1153 def logit_backward( function