Searched refs:original_rank (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.cc | 150 int original_rank; member
169 dr->original_rank = i; in BuildDevRecs()
219 if (least_rank < 0 || it.second.original_rank < least_rank) { in OrderTaskDeviceMap()
220 least_rank = it.second.original_rank; in OrderTaskDeviceMap()
276 if (least_rank < 0 || it.second.original_rank < least_rank) { in OrderTaskDeviceMap()
277 least_rank = it.second.original_rank; in OrderTaskDeviceMap()
475 ir->shared.instance.device_names[dr.original_rank]; in CompleteDefaultRanking()
477 ir->shared.instance.task_names[dr.original_rank]; in CompleteDefaultRanking()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 765 original_rank = (
771 array_ops.slice(original_shape, [0], [original_rank - 1])),
772 array_ops.gather(original_shape, original_rank - 1)])
818 [original_rank - 1]),
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 129 original_rank = (
135 array_ops.slice(original_shape, [0], [original_rank - 1])),
136 array_ops.gather(original_shape, original_rank - 1)])
182 [original_rank - 1]),
|
| D | layers.py | 270 original_rank = original_shape.ndims
271 if original_rank is None:
273 elif original_rank not in [2, 4]:
276 original_rank))
277 if original_rank == 2:
|