Searched refs:ComputeRepeatDeviceNumByTensorMap (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reluv2_info.cc | 117 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
|
| D | loss_info.cc | 101 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[1]); in InferAsLossDivisor()
|
| D | topk_info.cc | 98 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
|
| D | split_info.cc | 193 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
|
| D | batchnorm_info.cc | 243 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
|
| D | layer_norm_info.cc | 163 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
|
| D | operator_info.h | 312 int64_t ComputeRepeatDeviceNumByTensorMap(const Shape &dev_matrix_shape, const Shape &tensor_map);
|
| D | reduce_method_info.cc | 523 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
|
| D | activation_info.cc | 301 as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(dev_matrix_shape_, outputs_tensor_map_[0]); in InferAsLossDivisor()
|
| D | operator_info.cc | 1564 int64_t ComputeRepeatDeviceNumByTensorMap(const Shape &dev_matrix_shape, const Shape &tensor_map) { in ComputeRepeatDeviceNumByTensorMap() function
1617 …as_loss_divisor_ = ComputeRepeatDeviceNumByTensorMap(out_dev_matrix_shape_, outputs_tensor_map_[0]… in InferAsLossDivisor()
|