Home
last modified time | relevance | path

Searched refs:out_num (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dunstack_fp32.cc76 size_t out_num = out_tensors_.size(); in Run() local
77 for (size_t i = 0; i < out_num; i++) { in Run()
82 para->num_ = static_cast<int>(out_num); in Run()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddetection_post_process_fp32.c141 int out_num = 0; in DetectionPostProcessFast() local
152 *((BboxCorner *)(output_boxes) + out_num) = *box; in DetectionPostProcessFast()
153 output_scores[out_num] = input_scores[indexes[j]]; in DetectionPostProcessFast()
155 output_classes[out_num++] = (float)(indexes[j] % num_classes_with_bg - first_class_index); in DetectionPostProcessFast()
158 *output_num = (float)out_num; in DetectionPostProcessFast()
159 for (int i = out_num; i < param->max_detections_ * param->max_classes_per_detection_; ++i) { in DetectionPostProcessFast()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Deliminate_concat_split.cc94 int out_num = it->second.size(); in ConcatSplitEliminate() local
95 if (out_num != prim->get_number_split()) { in ConcatSplitEliminate()
100 for (int i = 0; i < out_num; i++) { in ConcatSplitEliminate()
Dfisson_util.cc89 int64_t split_num = split_info->out_num; in CalSplitOutputShape()
134 int64_t split_num = split_info->out_num; in CalSplitInShape()
328 if (nodes_num != static_cast<int32_t>(split_info->out_num)) { in CreateOutputsOfConcat()
369 split_prim->set_number_split(split_info->out_num); in CreateOutputsOfSplitWithOverlap()
388 if (!GetMultipleOutputsOfAnfNode(func_graph, split_cnode, split_info->out_num, split_outputs)) { in CreateOutputsOfSplitWithOverlap()
394 for (int64_t i = 0; i < split_info->out_num; i++) { in CreateOutputsOfSplitWithOverlap()
Dfisson_util.h38 int64_t out_num{0};
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dmulti_conv_info.cc29 split_info_.out_num = this->strategy_.dev_num; in GenSplitInfo()
42 std::vector<int64_t> tmp(split_info_.out_num, 0); in GenSplitInfo()
79 for (int64_t i = 0; i < split_info_.out_num; ++i) { in CheckSplitValid()
183 …for (int output_conv_index = 0; output_conv_index < static_cast<int>(split_info_.out_num); output_… in SplitSingleConv()
287 } else if (output_conv_index == static_cast<int>(split_info_.out_num - 1)) { in AdJustConvPrim()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_transpose_for_basiclstm_op.cc62 size_t out_num = AnfAlgo::GetOutputTensorNum(cnode); in Insert() local
63 for (size_t output_idx = 0; output_idx < out_num; output_idx++) { in Insert()
Dinsert_cast.cc47 size_t out_num = AnfAlgo::GetOutputTensorNum(cnode); in InsertCastForMultipleOutput() local
48 for (size_t output_idx = 0; output_idx < out_num; ++output_idx) { in InsertCastForMultipleOutput()
/third_party/mindspore/mindspore/core/ops/
Dneighborexchange.cc173 auto out_num = shapes_value.size(); in InferType() local
176 std::vector<TypePtr> type_vec(out_num, recv_type); in InferType()
/third_party/glib/glib/
Dgstrfuncs.h361 gint64 *out_num,
369 guint64 *out_num,
Dgstrfuncs.c3322 gint64 *out_num, in g_ascii_string_to_signed() argument
3378 if (out_num != NULL) in g_ascii_string_to_signed()
3379 *out_num = number; in g_ascii_string_to_signed()
3424 guint64 *out_num, in g_ascii_string_to_unsigned() argument
3482 if (out_num != NULL) in g_ascii_string_to_unsigned()
3483 *out_num = number; in g_ascii_string_to_unsigned()
/third_party/gstreamer/gstplugins_bad/sys/winks/
Dksvideohelpers.c235 compress_fraction (gint64 in_num, gint64 in_den, gint64 * out_num, in compress_fraction() argument
256 *out_num = cur_n; in compress_fraction()
263 *out_num = in_num; in compress_fraction()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc168 size_t out_num = AnfAlgo::GetOutputTensorNum(node); in InsertTransOpForMultipleOutput() local
169 for (size_t output_idx = 0; output_idx < out_num; ++output_idx) { in InsertTransOpForMultipleOutput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc514 size_t out_num = AnfAlgo::GetOutputTensorNum(out_node); in ReplaceFusionOp() local
515 for (size_t idx = 0; idx < out_num; ++idx) { in ReplaceFusionOp()