| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/ |
| D | pytorch_split_parser.cc | 31 int64_t split_num = 0; in Parse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_split_parser.cc | 31 int64_t split_num = 0; in Parse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | embedding_look_up_comm_grad_cpu_kernel.cc | 26 T split_num = static_cast<T *>(inputs[kIndex1]->device_ptr())[0]; in InitSplitNum() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | fisson_util.cc | 97 int64_t split_num = split_info->out_num; in CalSplitOutputShape() local
142 int64_t split_num = split_info->out_num; in CalSplitInShape() local
260 size_t split_num = split_info->size_splits.size(); in UpdateSplitInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | bounding_box_encode_info.cc | 102 Status BoundingBoxEncodeInfo::PrepareStrategy(int64_t stage_id, int64_t split_num, size_t dev_num, in PrepareStrategy()
|
| D | flash_attention_score_info.cc | 105 …auto split_num = std::accumulate(tensor_map.begin(), tensor_map.end(), 1, [&dev_matrix](int64_t a,… in GetSplitNumByTensorMap() local
944 … const int64_t split_num) { in GetAttentionMaskAttrs()
1135 void FlashAttentionScoreInfo::GetFlashAttentionScoreOpNode(int64_t split_id, int64_t split_num, con… in GetFlashAttentionScoreOpNode()
|
| D | prompt_flash_attention_info.cc | 217 … const int64_t split_num) { in GetAttenionMaskAttrs()
|
| D | fused_infer_attention_score_info.cc | 383 … const int64_t split_num) { in GetAttentionMaskAttrs()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | yes_no_op.cc | 81 Status YesNoOp::Split(const std::string &line, std::vector<int32_t> *split_num) { in Split()
|
| D | usps_op.cc | 242 int32_t split_num = 0; in ParseLine() local
|
| D | wider_face_op.cc | 222 Status WIDERFaceOp::Split(const std::string &line, std::vector<int32_t> *split_num) { in Split()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | grouped_pairwise_exchange_alltoall.cc | 179 CNodePtr NewSplitNode(const AnfNodePtr &input_node, size_t split_dim, size_t split_num) { in NewSplitNode()
200 CNodePtr NewSplitNode(const AnfNodePtr &input_node, size_t split_dim, size_t split_num, const Shape… in NewSplitNode()
284 void NewTupleGetItemNodes(const AnfNodePtr &input_node, size_t split_num, std::vector<AnfNodePtr> *… in NewTupleGetItemNodes()
505 size_t split_num = LongToSize(gpea_info->GetGroupNum()); in CreateReplaceGraph() local
|
| D | flash_sp.cc | 133 CNodePtr NewSplitNode(const AnfNodePtr &input_node, size_t split_dim, size_t split_num) { in NewSplitNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | neighbor_exchange_v2_fusion.cc | 270 … const std::vector<CNodePtr> &split_nodes, const std::vector<int64_t> &split_num, in GetCenter()
359 … const std::vector<CNodePtr> &split_nodes, const std::vector<int64_t> &split_num, in CreateAllToAllvNode()
1007 std::vector<int64_t> split_num; in Process() local
1023 std::vector<int64_t> split_num; in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | neighbor_exchange_v2_unify_mindir.cc | 188 … const std::vector<CNodePtr> &split_nodes, const std::vector<int64_t> &split_num, in GetCenter()
277 … const std::vector<CNodePtr> &split_nodes, const std::vector<int64_t> &split_num, in CreateAllToAllvNode()
957 std::vector<int64_t> split_num; in Process() local
980 std::vector<int64_t> split_num; in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | depthwise_conv2d_info.cc | 67 int64_t split_num = static_cast<int64_t>(splits.size()); in CreateSplitConstantTensors() local
343 … size_t split_num, const std::vector<int64_t> &splits) { in CreateOutputsOfSplit()
|
| D | conv2d_info.cc | 148 … std::vector<AnfNodePtr> *split_outputs, size_t split_dim, size_t split_num, in CreateOutputsOfSplit()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/ |
| D | parallel_op_combine.cc | 339 size_t split_num) { in NewSplitNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/ |
| D | op_api_convert.h | 662 auto split_num = shape[kIndex0]; variable
|
| D | acl_convert.cc | 164 AddressPtr UpdateKernelTensorAddress(KernelTensor *tensor, size_t split_num, size_t index) { in UpdateKernelTensorAddress()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | _grad_ops.py | 1818 def __infer__(self, dy, split_num): argument
|