Home
last modified time | relevance | path

Searched refs:original_rank (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc300 const int original_rank = tensor.shape()->size(); in VerifyAndCountSparseElements() local
302 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()
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/
Dformat_converter.cc241 int original_rank = shape.size(); in FormatConverter() local
244 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()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc280 int original_rank; member
299 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()
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py883 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]),