Searched defs:split_outputs (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_inputs_for_reduce_scatter.cc | 27 std::vector<AnfNodePtr> split_outputs; in InsertSplitForInput() local
109 std::vector<AnfNodePtr> split_outputs = InsertSplitForInput(func_graph, cnode, rank_size); in Process() local
|
| /third_party/mesa3d/src/compiler/nir/ |
| D | nir_lower_io_to_scalar.c | 179 struct hash_table *split_outputs) in lower_load_to_scalar_early()
242 struct hash_table *split_outputs) in lower_store_output_to_scalar_early()
289 struct hash_table *split_inputs, *split_outputs; member
|
| D | nir_lower_io_arrays_to_elements.c | 368 struct hash_table *split_outputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements_no_indirects() local
406 struct hash_table *split_outputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | depthwise_conv2d_info.cc | 332 … std::vector<AnfNodePtr> *split_outputs, size_t split_num, in CreateOutputsOfSplit()
415 int DepthwiseConv2DInfo::CreateConstantOutputsOfSplit(std::vector<AnfNodePtr> *split_outputs, int i… in CreateConstantOutputsOfSplit()
|
| D | conv2d_info.cc | 145 … std::vector<AnfNodePtr> *split_outputs, size_t split_dim, size_t split_num, in CreateOutputsOfSplit()
|
| D | multi_conv_info.cc | 152 std::vector<AnfNodePtr> split_outputs; in MultiConvNHSplit() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 99 std::vector<AnfNodePtr> split_outputs; in CreateAllToAllvNode() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | fisson_util.cc | 359 std::vector<AnfNodePtr> *split_outputs, SplitInfo *split_info, in CreateOutputsOfSplitWithOverlap()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | dynamic_gru_v2_grad_fission.cc | 338 std::vector<AnfNodePtr> split_outputs; in CreateDgateXConcatDNode() local
|
| D | dynamic_rnn_grad_fission_v2.cc | 244 std::vector<AnfNodePtr> split_outputs; in AddLSTMInputGradNode() local
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | block_util.py | 329 …create_func(block, num_outputs, rand_func, desc_const, const_first, add_fake_input, split_outputs): argument
|