| /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() argument
191 chan_vars = get_channel_variables(split_outputs, var); in lower_load_to_scalar_early()
242 struct hash_table *split_outputs) in lower_store_output_to_scalar_early() argument
248 nir_variable **chan_vars = get_channel_variables(split_outputs, var); in lower_store_output_to_scalar_early()
289 struct hash_table *split_inputs, *split_outputs; member
355 state->split_outputs); in nir_lower_io_to_scalar_early_instr()
362 lower_store_output_to_scalar_early(b, intr, var, state->split_outputs); in nir_lower_io_to_scalar_early_instr()
382 .split_outputs = _mesa_pointer_hash_table_create(NULL), in nir_lower_io_to_scalar_early()
401 hash_table_foreach(state.split_outputs, entry) { in nir_lower_io_to_scalar_early()
409 _mesa_hash_table_destroy(state.split_outputs, NULL); in nir_lower_io_to_scalar_early()
|
| 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
373 indirects, split_outputs, true); in nir_lower_io_arrays_to_elements_no_indirects()
389 hash_table_foreach(split_outputs, entry) { in nir_lower_io_arrays_to_elements_no_indirects()
397 _mesa_hash_table_destroy(split_outputs, NULL); in nir_lower_io_arrays_to_elements_no_indirects()
406 struct hash_table *split_outputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements() local
414 indirects, split_outputs, false); in nir_lower_io_arrays_to_elements()
428 hash_table_foreach(split_outputs, entry) { in nir_lower_io_arrays_to_elements()
436 _mesa_hash_table_destroy(split_outputs, NULL); in nir_lower_io_arrays_to_elements()
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | facade_util.py | 59 …sampling_times, reduce_output, init_param_with, split_outputs, exception, error_keywords = get_fun…
64 split_outputs, exception, error_keywords = get_function_config(block_config[-1])
83 keyword.split_outputs: split_outputs,
|
| D | config_util.py | 91 split_outputs = function.get(keyword.split_outputs, True)
95 reduce_output, init_param_with, split_outputs, exception, error_keywords
|
| D | block_util.py | 329 …create_func(block, num_outputs, rand_func, desc_const, const_first, add_fake_input, split_outputs): argument
343 if split_outputs:
382 split_outputs = config.get(keyword.split_outputs, True)
384 const_first, add_fake_input, split_outputs)
|
| D | keyword.py | 42 keyword.split_outputs = "split_outputs"
|
| /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
51 split_outputs.push_back(new_outputs[j]); in InsertSplitForInput()
54 return split_outputs; in InsertSplitForInput()
109 std::vector<AnfNodePtr> split_outputs = InsertSplitForInput(func_graph, cnode, rank_size); in Process() local
110 return RearrangeInputsForReduceScatter(func_graph, node, split_outputs, rank_size); in Process()
|
| /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
100 CreateMultipleOutputsOfAnfNode(graph, split, static_cast<size_t>(split_count), &split_outputs); in CreateAllToAllvNode()
101 if (split_outputs.empty()) { in CreateAllToAllvNode()
105 …(void)all_to_all_v_input.insert(all_to_all_v_input.end(), split_outputs.begin(), split_outputs.end… in CreateAllToAllvNode()
108 auto single_shape = AnfAlgo::GetOutputInferShape(split_outputs[0], 0); in CreateAllToAllvNode()
109 auto single_type = AnfAlgo::GetOutputInferDataType(split_outputs[0], 0); in CreateAllToAllvNode()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | multi_conv_info.cc | 152 std::vector<AnfNodePtr> split_outputs; in MultiConvNHSplit() local
153 …sOfSplitWithOverlap(func_graph_, conv_nodes_[conv_nodes_.size() - 1], &split_outputs, &split_info_, in MultiConvNHSplit()
162 if (!SplitSingleConv(conv_nodes_[i], split_outputs, &outputs_node)) { in MultiConvNHSplit()
166 split_outputs.clear(); in MultiConvNHSplit()
167 std::copy(outputs_node.begin(), outputs_node.end(), std::back_inserter(split_outputs)); in MultiConvNHSplit()
171 …auto concat_node = CreateOutputsOfConcat(func_graph_, node, split_outputs, &split_info_, conv_cnod… in MultiConvNHSplit()
172 split_outputs.clear(); in MultiConvNHSplit()
|
| D | depthwise_conv2d_info.h | 41 …tputsOfSplit(const CNodePtr &orig_node, size_t input_index, std::vector<AnfNodePtr> *split_outputs,
43 int CreateConstantOutputsOfSplit(std::vector<AnfNodePtr> *split_outputs, int input_index);
|
| D | depthwise_conv2d_info.cc | 332 … std::vector<AnfNodePtr> *split_outputs, size_t split_num, in CreateOutputsOfSplit() argument
334 MS_ASSERT(orig_node != nullptr && split_outputs != nullptr); in CreateOutputsOfSplit()
390 if (CreateMultipleOutputsOfAnfNode(split_cnode, split_num, split_outputs) != RET_OK) { in CreateOutputsOfSplit()
415 int DepthwiseConv2DInfo::CreateConstantOutputsOfSplit(std::vector<AnfNodePtr> *split_outputs, int i… in CreateConstantOutputsOfSplit() argument
417 MS_ASSERT(split_outputs != nullptr); in CreateConstantOutputsOfSplit()
439 split_outputs->push_back(parameter_node); in CreateConstantOutputsOfSplit()
441 if (split_outputs->empty()) { in CreateConstantOutputsOfSplit()
|
| D | conv2d_info.h | 44 …tputsOfSplit(const CNodePtr &orig_node, size_t input_index, std::vector<AnfNodePtr> *split_outputs,
|
| D | conv2d_info.cc | 145 … std::vector<AnfNodePtr> *split_outputs, size_t split_dim, size_t split_num, in CreateOutputsOfSplit() argument
147 MS_ASSERT(orig_node != nullptr && split_outputs != nullptr); in CreateOutputsOfSplit()
199 if (CreateMultipleOutputsOfAnfNode(split_cnode, split_num, split_outputs) != RET_OK) { in CreateOutputsOfSplit()
|
| D | operator_info.h | 75 … std::vector<AnfNodePtr> *split_outputs, size_t split_dim, size_t split_num,
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | fisson_util.cc | 359 std::vector<AnfNodePtr> *split_outputs, SplitInfo *split_info, in CreateOutputsOfSplitWithOverlap() argument
363 MS_CHECK_TRUE_MSG(split_outputs != nullptr, false, "input split_outputs is nullptr"); in CreateOutputsOfSplitWithOverlap()
388 if (!GetMultipleOutputsOfAnfNode(func_graph, split_cnode, split_info->out_num, split_outputs)) { in CreateOutputsOfSplitWithOverlap()
395 auto node = (*split_outputs)[i]; in CreateOutputsOfSplitWithOverlap()
|
| D | fisson_util.h | 67 std::vector<AnfNodePtr> *split_outputs, SplitInfo *split_info,
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | dynamic_rnn_grad_fission_v2.cc | 244 std::vector<AnfNodePtr> split_outputs; in AddLSTMInputGradNode() local
245 CreateMultipleOutputsOfAnfNode(func_graph, split_v, kSplitVOutputNum, &split_outputs); in AddLSTMInputGradNode()
246 pre_split_outputs = split_outputs; in AddLSTMInputGradNode()
248 lstm_x_concat_input[idx + 1] = split_outputs[0]; in AddLSTMInputGradNode()
|
| D | dynamic_gru_v2_grad_fission.cc | 338 std::vector<AnfNodePtr> split_outputs; in CreateDgateXConcatDNode() local
339 CreateMultipleOutputsOfAnfNode(func_graph, split, kSplitVOutputNum, &split_outputs); in CreateDgateXConcatDNode()
341 split_outputs[kIndex0]}; in CreateDgateXConcatDNode()
|