| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/ |
| D | split_with_size_aclnn_kernel.cc | 64 std::vector<KernelTensor *> split_outputs = GetSplitRealOutputs(outputs); in GetWorkSpaceInfo() local
74 std::vector<KernelTensor *> split_outputs = GetSplitRealOutputs(outputs); in Launch() local
|
| D | split_tensor_aclnn_kernel.cc | 65 std::vector<KernelTensor *> split_outputs = GetSplitRealOutputs(outputs); in GetWorkSpaceInfo() local
74 std::vector<KernelTensor *> split_outputs = GetSplitRealOutputs(outputs); in Launch() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/ |
| D | split_inputs_for_reduce_scatter.cc | 30 std::vector<AnfNodePtr> split_outputs; in InsertSplitForInput() local
116 std::vector<AnfNodePtr> split_outputs = InsertSplitForInput(func_graph, cnode, rank_size); in Process() local
|
| /third_party/mesa3d/src/compiler/nir/ |
| D | nir_lower_io_arrays_to_elements.c | 358 struct hash_table *split_outputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements_no_indirects() local
397 struct hash_table *split_outputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements() local
|
| D | nir_lower_io_to_scalar.c | 388 struct hash_table *split_outputs) in lower_load_to_scalar_early()
441 struct hash_table *split_outputs) in lower_store_output_to_scalar_early()
484 struct hash_table *split_inputs, *split_outputs; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | all_to_all_unify_mindir.cc | 108 std::vector<AnfNodePtr> split_outputs; in CreateAllToAllvNode() local
180 std::vector<AnfNodePtr> split_outputs; in CreateAllToAllvNode() local
|
| D | neighbor_exchange_v2_unify_mindir.cc | 166 std::vector<AnfNodePtr> CreateAllToAllvInput(const std::vector<std::vector<AnfNodePtr>> &split_outp… in CreateAllToAllvInput()
219 … const std::vector<std::vector<AnfNodePtr>> &split_outputs, in CreateAllToAllvInputForGrad()
294 std::vector<std::vector<AnfNodePtr>> split_outputs; in CreateAllToAllvNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | depthwise_conv2d_info.cc | 342 … std::vector<AnfNodePtr> *split_outputs, size_t split_dim, in CreateOutputsOfSplit()
426 int DepthwiseConv2DInfo::CreateConstantOutputsOfSplit(std::vector<AnfNodePtr> *split_outputs, int i… in CreateConstantOutputsOfSplit()
|
| D | multi_conv_info.cc | 153 std::vector<AnfNodePtr> split_outputs; in MultiConvNHSplit() local
|
| 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/plugin/device/gpu/optimizer/ |
| D | alltoall_fusion.cc | 108 std::vector<AnfNodePtr> split_outputs; in CreateAllToAllvNode() local
|
| D | neighbor_exchange_v2_fusion.cc | 245 std::vector<AnfNodePtr> CreateAllToAllvInput(const std::vector<std::vector<AnfNodePtr>> &split_outp… in CreateAllToAllvInput()
301 … const std::vector<std::vector<AnfNodePtr>> &split_outputs, in CreateAllToAllvInputForGrad()
373 std::vector<std::vector<AnfNodePtr>> split_outputs; in CreateAllToAllvNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | all_to_all_v_for_ge.cc | 300 AnfNodePtrList split_outputs; in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | fisson_util.cc | 386 … std::vector<AnfNodePtr> *split_outputs, const SplitInfo &split_info, in CreateOutputsOfSplitWithOverlap()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | dynamic_gru_v2_grad_fission.cc | 387 std::vector<AnfNodePtr> split_outputs; in CreateDgateXConcatDNode() local
|
| D | dynamic_rnn_grad_fission_v2.cc | 254 std::vector<AnfNodePtr> split_outputs; in CreateTLoopNodeWithEdge() local
|
| /third_party/mindspore/mindspore-src/source/tests/mindspore_test_framework/utils/ |
| D | block_util.py | 341 …create_func(block, num_outputs, rand_func, desc_const, const_first, add_fake_input, split_outputs): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 3327 std::vector<std::string> split_outputs = {input_weights_name, hidden_weights_name}; in ExportPrimDynamicRNN() local
3451 std::vector<std::string> split_outputs = {input_weights_name, hidden_weights_name}; in ExportLSTMWeights() local
|