Searched refs:as_loss_divisor_ (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reluv2_info.cc | 102 as_loss_divisor_ = 1; in InferAsLossDivisor()
112 as_loss_divisor_ = stage_device_size_; in InferAsLossDivisor()
113 …MS_LOG(INFO) << name_ << ": The output is a scalar, use the dev size " << as_loss_divisor_ << ", l… in InferAsLossDivisor()
117 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
120 << as_loss_divisor_; in InferAsLossDivisor()
|
| D | split_info.cc | 178 as_loss_divisor_ = 1; in InferAsLossDivisor()
188 as_loss_divisor_ = stage_device_size_; in InferAsLossDivisor()
189 …MS_LOG(INFO) << name_ << ": The output is a scalar, use the dev size " << as_loss_divisor_ << ", l… in InferAsLossDivisor()
193 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
196 << as_loss_divisor_; in InferAsLossDivisor()
|
| D | topk_info.cc | 93 as_loss_divisor_ = stage_device_size_; in InferAsLossDivisor()
94 …MS_LOG(INFO) << name_ << ": The output is a scalar, use the dev size" << as_loss_divisor_ << " as … in InferAsLossDivisor()
98 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
103 << ", " << output_tensor_map_str << ", " << as_loss_divisor_; in InferAsLossDivisor()
|
| D | loss_info.cc | 101 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[1]); in InferAsLossDivisor()
104 << as_loss_divisor_; in InferAsLossDivisor()
|
| D | batchnorm_info.cc | 243 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
246 << ", as_loss_divisor_ is " << as_loss_divisor_; in InferAsLossDivisor()
|
| D | reduce_method_info.cc | 518 as_loss_divisor_ = stage_device_size_; in InferAsLossDivisor()
519 …MS_LOG(INFO) << name_ << ": The output is a scalar, use the dev size" << as_loss_divisor_ << " as … in InferAsLossDivisor()
523 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
528 << ", " << output_tensor_map_str << ", " << as_loss_divisor_; in InferAsLossDivisor()
|
| D | layer_norm_info.cc | 163 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
166 << ", as_loss_divisor_ is " << as_loss_divisor_; in InferAsLossDivisor()
|
| D | operator_info.cc | 1593 as_loss_divisor_ = 1; in InferAsLossDivisor()
1609 as_loss_divisor_ = stage_device_size_; in InferAsLossDivisor()
1610 …MS_LOG(INFO) << name_ << ": The output is a scalar, use the dev size " << as_loss_divisor_ << ", l… in InferAsLossDivisor()
1617 …as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(out_dev_matrix_shape_, outputs_tensor_map_[0]… in InferAsLossDivisor()
1620 << as_loss_divisor_; in InferAsLossDivisor()
1631 if (as_loss_divisor_ <= 0) { in InferVirtualDivOps()
1632 MS_LOG(ERROR) << name_ << ": Invalid loss divisor: " << as_loss_divisor_; in InferVirtualDivOps()
1634 } else if (as_loss_divisor_ == 1) { in InferVirtualDivOps()
1641 Operator op = CreateVirtualDivOp(as_loss_divisor_); in InferVirtualDivOps()
|
| D | virtual_dataset_info.cc | 192 as_loss_divisor_ = 1; in InferAsLossDivisor()
|
| D | batch_parallel_info.cc | 175 as_loss_divisor_ = 1; in InferAsLossDivisor()
|
| D | uniform_candidate_sampler_info.cc | 146 as_loss_divisor_ = 1; in InferAsLossDivisor()
|
| D | activation_info.cc | 301 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
304 << ", as_loss_divisor_ is " << as_loss_divisor_; in InferAsLossDivisor()
|
| D | operator_info.h | 241 int64_t as_loss_divisor_ = 1; variable
|