Searched defs:normalized_values (Results 1 – 1 of 1) sorted by relevance
349 const Tensor& normalized_values, in spmm_reduce_normalize_values_kernel_impl()501 const Tensor& normalized_values, in spmm_reduce_normalize_values_kernel()530 Tensor normalized_values = at::empty(values.sizes(), values.options()); in spmm_reduce_backward_other_kernel() local