Searched refs:trace_backward_decomp (Results 1 – 1 of 1) sorted by relevance
111 Tensor trace_backward_decomp(const Tensor& grad, IntArrayRef sizes) { in trace_backward_decomp() function262 m.impl("trace_backward", trace_backward_decomp); in TORCH_LIBRARY_IMPL()