Searched refs:num_sparse_tensors (Results 1 – 3 of 3) sorted by relevance
65 int num_sparse_tensors = 1; in Compute() local67 num_sparse_tensors *= serialized_sparse.shape().dim_size(i); in Compute()71 context, num_sparse_tensors > 0, in Compute()77 if (num_sparse_tensors == 1 && ndims == 1) { in Compute()99 indices.reserve(num_sparse_tensors); in Compute()100 values.reserve(num_sparse_tensors); in Compute()104 for (int i = 0; i < num_sparse_tensors; ++i) { in Compute()168 tensors.reserve(num_sparse_tensors); in Compute()169 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()
497 int num_sparse_tensors = num_tensors() / 3; in get_tensor() local498 if (index < num_sparse_tensors) { in get_tensor()500 } else if (index < 2 * num_sparse_tensors) { in get_tensor()501 return values()->tensor(index % num_sparse_tensors); in get_tensor()503 return dense_shapes()->tensor(index % num_sparse_tensors); in get_tensor()