Searched defs:max_nnz (Results 1 – 2 of 2) sorted by relevance
314 void GenerateRandomSparseTensor(int64_t max_nnz, const TensorShape& shape, in GenerateRandomSparseTensor()
441 int64_t t_nnz = t._nnz(), s_nnz = src._nnz(), max_nnz = t_nnz + s_nnz; in add_out_sparse_contiguous() local1104 int64_t max_nnz = std::min(t_nnz, s_nnz); // multiply by zero is zero, and can be dropped in mul_out_sparse_cpu() local