Searched refs:max_set_size (Results 1 – 1 of 1) sorted by relevance
443 int64 max_set_size = 0; in ComputeDenseToDense() local464 if (set_size > max_set_size) { in ComputeDenseToDense()465 max_set_size = set_size; in ComputeDenseToDense()473 output_shape.AddDim(max_set_size); in ComputeDenseToDense()499 int64 max_set_size = 0; in ComputeDenseToSparse() local546 if (set_size > max_set_size) { in ComputeDenseToSparse()547 max_set_size = set_size; in ComputeDenseToSparse()555 output_shape.AddDim(max_set_size); in ComputeDenseToSparse()620 int64 max_set_size = 0; in ComputeSparseToSparse() local669 if (set_size > max_set_size) { in ComputeSparseToSparse()[all …]