Searched refs:original_rank (Results 1 – 4 of 4) sorted by relevance
300 const int original_rank = tensor.shape()->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()326 std::sort(traversal_order.begin(), traversal_order.begin() + original_rank); in VerifyAndCountSparseElements()327 for (int i = 0; i < original_rank; i++) { in VerifyAndCountSparseElements()333 std::sort(traversal_order.begin() + original_rank, traversal_order.end()); in VerifyAndCountSparseElements()334 for (int i = original_rank; i < total_dims; i++) { in VerifyAndCountSparseElements()348 for (int i = 0; i < original_rank; i++) { in VerifyAndCountSparseElements()358 sparsity->traversal_order()->Get(i + original_rank); in VerifyAndCountSparseElements()363 sparsity->dim_metadata()->Get(i + original_rank)->dense_size(); in VerifyAndCountSparseElements()
241 int original_rank = shape.size(); in FormatConverter() local244 blocked_shape_.resize(original_rank); in FormatConverter()246 for (int i = 0; i < original_rank; i++) { in FormatConverter()248 int orig_dim = traversal_order_[original_rank + block_dim]; in FormatConverter()
280 int original_rank; member299 dr->original_rank = i; in BuildDevRecs()352 if (least_rank < 0 || it.second.original_rank < least_rank) { in OrderTaskDeviceMap()353 least_rank = it.second.original_rank; in OrderTaskDeviceMap()409 if (least_rank < 0 || it.second.original_rank < least_rank) { in OrderTaskDeviceMap()410 least_rank = it.second.original_rank; in OrderTaskDeviceMap()584 new_device_names[dr.global_rank] = gp->device_names[dr.original_rank]; in CompleteDefaultRanking()585 new_task_names[dr.global_rank] = gp->task_names[dr.original_rank]; in CompleteDefaultRanking()
883 original_rank = (888 array_ops.slice(original_shape, [0], [original_rank - 1])),889 array_ops.gather(original_shape, original_rank - 1)933 [original_rank - 1]),