Searched refs:repeated_calc_num_ (Results 1 – 11 of 11) sorted by relevance
154 if (repeated_calc_num_ == 1) { in InferForwardCommunication()170 if (repeated_calc_num_ == 1) { in InferForwardCommunication()
619 if (repeated_calc_num_ > 1) { in InferBias()621 rank = rank / repeated_calc_num_; in InferBias()633 if (repeated_calc_num_ > 1) { in InferBias()635 rank = rank / repeated_calc_num_; in InferBias()688 if ((repeated_calc_num_ > 1) && !repeated_num_in_dev_matrix_right_) { in InferGroup()
189 if (repeated_calc_num_ > 1) { in InferForwardCommunication()279 if (repeated_calc_num_ > 1) { in InferTensorMap()
241 repeated_calc_num_ = 1; in InferRepeatedCalcInfo()243 repeated_calc_num_ = ((int64_t)(g_dev_list_size / dev_matrix_size)); in InferRepeatedCalcInfo()258 if (repeated_calc_num_ <= 1) { in SetRepeatedCalcDevMatrix()262 dev_matrix_shape_.push_back(repeated_calc_num_); in SetRepeatedCalcDevMatrix()264 (void)dev_matrix_shape_.insert(dev_matrix_shape_.begin(), repeated_calc_num_); in SetRepeatedCalcDevMatrix()271 if ((repeated_calc_num_ <= 1) || !repeated_num_in_dev_matrix_right_) { in ResetTensorMapIfRepeatedCalc()275 …MS_LOG(DEBUG) << name_ << ": the repeated calc num is " << repeated_calc_num_ << ", and reset the … in ResetTensorMapIfRepeatedCalc()
178 if (repeated_num_in_dev_matrix_right_ && (repeated_calc_num_ > 1)) { in InferForwardCommunication()255 if (repeated_num_in_dev_matrix_right_ && (repeated_calc_num_ > 1)) { in InferForwardCommunication()
247 if ((seed_0 == 0) && (seed_1 == 0) && (repeated_calc_num_ > 1)) { in GetDropoutGenMaskReplaceOp()
215 if (repeated_calc_num_ > 1) { in InferForwardCommunication()
240 int64_t repeated_calc_num_ = 1; variable
371 if (repeated_calc_num_ > 1 && !repeated_num_in_dev_matrix_right_) { in InferRankBias()503 if (repeated_calc_num_ > 1 && !repeated_num_in_dev_matrix_right_) { in InferForwardCommunication()
250 if (repeated_calc_num_ > 1) { in InferTensorMap()
309 if ((seed0_ != 0) || (seed1_ != 0) || (repeated_calc_num_ == 1)) { in InferReplaceOps()