Searched refs:_rank (Results 1 – 12 of 12) sorted by relevance
94 self._rank = int(os.environ['OMPI_COMM_WORLD_RANK'])97 self._rank = int(os.environ['SLURM_PROCID'])173 if cluster_rank_offset_start <= self._rank < cluster_rank_offset_end:175 self.task_id = self._rank - cluster_rank_offset_start180 os.environ['CUDA_VISIBLE_DEVICES'] = self._gpu_allocation[self._rank]
323 … int _rank = -1; //rank -1 means the node was not inserted, but only used for structural purposes field in RegexLookup.RegexTree.RTNode330 _rank = rank; in RTNode()353 } else if (child._rank == -1) { in put()361 if (child._rank == -1) { in put()362 child._rank = node._rank; in put()370 … if (child._rank == -1 && node._finder.toString().startsWith(childFinderPattern)) { in put()387 } else if (!(finderPattern.equals("")) && this._rank == -1) { in put()425 if (child._rank != -1 && finder.equals(child._finder)) { in get()433 } else if (child._rank == -1) { in get()456 if (child._rank != -1) { in addToEntrySet()[all …]
55 def _rank(x): function74 ids_rank, ids_static = _rank(ids)75 params_rank, params_static = _rank(params)
112 concat_dim._numpy().item(0) % input_values[0]._rank()) # pylint: disable=protected-access
1526 if size_splits._rank() == 0:
90 if tensor._rank(): # pylint: disable=protected-access
19 _name = _rank = _serialNumber = 0;
402 self._rank = tensor_shape.Dimension(rank)443 return self._rank
39 return tensor._rank() # pylint: disable=protected-access
474 def _rank(self): member in Tensor796 def _rank(self): member in _EagerTensorBase
119 (context.executing_eagerly() and weights._rank() == 0 # pylint: disable=protected-access
9063 auto _rank = _o->rank;9067 _rank,