Searched refs:num_sparse_tensors (Results 1 – 2 of 2) sorted by relevance
63 int num_sparse_tensors = 1; in Compute() local65 num_sparse_tensors *= serialized_sparse.shape().dim_size(i); in Compute()69 context, num_sparse_tensors > 0, in Compute()75 if (num_sparse_tensors == 1 && ndims == 1) { in Compute()97 indices.reserve(num_sparse_tensors); in Compute()98 values.reserve(num_sparse_tensors); in Compute()102 for (int i = 0; i < num_sparse_tensors; ++i) { in Compute()166 tensors.reserve(num_sparse_tensors); in Compute()167 for (int i = 0; i < num_sparse_tensors; ++i) { in Compute()
50 int num_sparse_tensors = 1; in Compute() local52 num_sparse_tensors *= input.shape().dim_size(i); in Compute()55 if (num_sparse_tensors == 1 && input_dims_to_stack == 0) { in Compute()78 context, num_sparse_tensors > 0, in Compute()112 output_shape_vec(0) = num_sparse_tensors; in Compute()128 for (int i = 1; i < num_sparse_tensors; ++i) { in Compute()172 for (int i = 0; i < num_sparse_tensors; ++i) { in Compute()