Searched refs:original_dim (Results 1 – 2 of 2) sorted by relevance
241 const int original_dim = sparsity.traversal_order()->Get(i); in VerifyAndCountElements() local244 if (dim_metadata->dense_size() != dim_sizes[original_dim]) { in VerifyAndCountElements()279 GetValueOfIndicesAt(dim_metadata, j) >= dim_sizes[original_dim]) { in VerifyAndCountElements()
553 const int64 original_dim = input_indices_t(i, j); in Split() local556 ? GetDimensionInSlice(original_dim, split_size, residual) in Split()557 : original_dim; in Split()