Searched refs:repeated_num_in_dev_matrix_right_ (Results 1 – 10 of 10) sorted by relevance
308 repeated_num_in_dev_matrix_right_ = true; in SetAttribute()310 repeated_num_in_dev_matrix_right_ = false; in SetAttribute()312 …NFO) << "Set repeated_num_in_dev_matrix_right for gather to " << repeated_num_in_dev_matrix_right_; in SetAttribute()453 if (repeated_num_in_dev_matrix_right_) { in InferDevMatrixShape()620 if (repeated_num_in_dev_matrix_right_) { in InferBias()634 if (repeated_num_in_dev_matrix_right_) { in InferBias()688 if ((repeated_calc_num_ > 1) && !repeated_num_in_dev_matrix_right_) { in InferGroup()
157 } else if (!repeated_num_in_dev_matrix_right_) { in InferForwardCommunication()173 } else if (!repeated_num_in_dev_matrix_right_) { in InferForwardCommunication()
164 if ((dev_matrix_shape_.size() > size) && !repeated_num_in_dev_matrix_right_) { in InferForwardCommunication()178 if (repeated_num_in_dev_matrix_right_ && (repeated_calc_num_ > 1)) { in InferForwardCommunication()240 if ((dev_matrix_shape_.size() > size) && !repeated_num_in_dev_matrix_right_) { in InferForwardCommunication()255 if (repeated_num_in_dev_matrix_right_ && (repeated_calc_num_ > 1)) { in InferForwardCommunication()
88 repeated_num_in_dev_matrix_right_ = true; in InferDevMatrixShape()90 repeated_num_in_dev_matrix_right_ = false; in InferDevMatrixShape()
125 repeated_num_in_dev_matrix_right_ = false; in Init()
102 repeated_num_in_dev_matrix_right_ = false; in Init()
190 if (repeated_num_in_dev_matrix_right_) { in InferForwardCommunication()
290 bool repeated_num_in_dev_matrix_right_ = true; 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()
261 if (repeated_num_in_dev_matrix_right_) { in SetRepeatedCalcDevMatrix()271 if ((repeated_calc_num_ <= 1) || !repeated_num_in_dev_matrix_right_) { in ResetTensorMapIfRepeatedCalc()