Searched defs:grad_sizes (Results 1 – 2 of 2) sorted by relevance
30 const auto& grad_sizes = grad.sym_sizes(); in obeys_layout_contract() local
127 auto grad_sizes = nt_grad->get_nested_sizes(); in _nested_sum_backward_cpu() local