Searched refs:total_dims (Results 1 – 2 of 2) sorted by relevance
299 const int total_dims = sparsity->traversal_order()->size(); in VerifyAndCountSparseElements() local302 if (total_dims < original_rank || sparsity_dim_metadata_size != total_dims) { in VerifyAndCountSparseElements()306 const int block_rank = total_dims - original_rank; in VerifyAndCountSparseElements()321 std::vector<int> traversal_order(total_dims); in VerifyAndCountSparseElements()322 for (int i = 0; i < total_dims; i++) { in VerifyAndCountSparseElements()334 for (int i = original_rank; i < total_dims; i++) { in VerifyAndCountSparseElements()346 expanded_dim_sizes.resize(total_dims); in VerifyAndCountSparseElements()359 if (original_block_dim < 0 || original_block_dim >= total_dims) { in VerifyAndCountSparseElements()371 if (mapped_block_dim < 0 || mapped_block_dim >= total_dims) { in VerifyAndCountSparseElements()
426 int total_dims = 0; in SetOpAttrList() local441 total_dims += size; in SetOpAttrList()446 std::unique_ptr<int64_t[]> buffer(new int64_t[total_dims]); in SetOpAttrList()591 int total_dims = 0; in SetOpAttrListDefault() local594 total_dims += attr.default_value().list().shape(i).dim_size(); in SetOpAttrListDefault()598 std::unique_ptr<int64_t[]> buffer(new int64_t[total_dims]); in SetOpAttrListDefault()