Home
last modified time | relevance | path

Searched refs:trace_backward_decomp (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/functorch/
DPyTorchOperatorHacks.cpp111 Tensor trace_backward_decomp(const Tensor& grad, IntArrayRef sizes) { in trace_backward_decomp() function
262 m.impl("trace_backward", trace_backward_decomp); in TORCH_LIBRARY_IMPL()