Searched defs:values_2 (Results 1 – 4 of 4) sorted by relevance
324 auto values_2 = values.view({nnz, nvalues}); in cpu_sparse_coo_softmax() local462 auto values_2 = values.view({nnz, nvalues}); in cpu_sparse_coo_softmax_backward() local
421 auto values_2 = values.view({nnz, nvalues}); in cuda_sparse_coo_softmax() local535 auto values_2 = values.view({nnz, nvalues}); in cuda_sparse_coo_softmax_backward() local
148 const std::vector<int> values_2 = {}; in TEST_F() local
5837 def grad_test_func(values_1, values_2, offsets): argument