Searched refs:out_dev_matrix_shape_ (Results 1 – 3 of 3) sorted by relevance
408 out_dev_matrix_shape_.clear(); in InferDevMatrixShape()415 out_dev_matrix_shape_ = dev_matrix_shape_; in InferDevMatrixShape()422 out_dev_matrix_shape_ = dev_matrix_shape_; in InferDevMatrixShape()424 << ", out dev matrix is " << out_dev_matrix_shape_; in InferDevMatrixShape()440 out_dev_matrix_shape_.push_back(1); in InferDevMatrixShape()442 out_dev_matrix_shape_.push_back(param_strategy.at(i)); in InferDevMatrixShape()445 out_dev_matrix_shape_.push_back(param_strategy.at(0) * param_strategy.at(LongToSize(axis_))); in InferDevMatrixShape()447 out_dev_matrix_shape_ = dev_matrix_shape_; in InferDevMatrixShape()454 out_dev_matrix_shape_.push_back(repeated_calc_num); in InferDevMatrixShape()456 (void)out_dev_matrix_shape_.insert(out_dev_matrix_shape_.begin(), repeated_calc_num); in InferDevMatrixShape()[all …]
1614 if (out_dev_matrix_shape_.empty()) { in InferAsLossDivisor()1615 out_dev_matrix_shape_ = dev_matrix_shape_; in InferAsLossDivisor()1617 …as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(out_dev_matrix_shape_, outputs_tensor_map_[0]… in InferAsLossDivisor()1618 MS_LOG(INFO) << name_ << ": the dev matrix shape is " << ShapeToString(out_dev_matrix_shape_) in InferAsLossDivisor()
239 Shape out_dev_matrix_shape_; variable