Home
last modified time | relevance | path

Searched refs:output_num (Results 1 – 25 of 250) sorted by relevance

12345678910

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dunpack.cu22 __global__ void Unpack(const size_t size, const size_t output_num, in Unpack() argument
25 size_t cur_input_index = pos / dims_after_axis % output_num; in Unpack()
26 size_t cycle_len = output_num * dims_after_axis; in Unpack()
34 void UnpackKernel(const size_t size, const size_t output_num, in UnpackKernel() argument
37 Unpack<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>(size, output_num, in UnpackKernel()
42 template void UnpackKernel(const size_t size, const size_t output_num,
45 template void UnpackKernel(const size_t size, const size_t output_num,
48 template void UnpackKernel(const size_t size, const size_t output_num,
51 template void UnpackKernel(const size_t size, const size_t output_num,
54 template void UnpackKernel(const size_t size, const size_t output_num,
[all …]
/third_party/ffmpeg/libavfilter/dnn/
Ddnn_backend_native_layer_dense.c35 dense_params->output_num = (int32_t)avio_rl32(model_file_context); in ff_dnn_load_layer_dense()
39 kernel_size = dense_params->input_num * dense_params->output_num; in ff_dnn_load_layer_dense()
42 dnn_size += dense_params->output_num * 4; in ff_dnn_load_layer_dense()
45 dense_params->output_num <= 0){ in ff_dnn_load_layer_dense()
61 dense_params->biases = av_malloc(dense_params->output_num * sizeof(float)); in ff_dnn_load_layer_dense()
67 for (int i = 0; i < dense_params->output_num; ++i){ in ff_dnn_load_layer_dense()
102 output_operand->dims[3] = dense_params->output_num; in ff_dnn_execute_layer_dense()
120 for (int n_filter = 0; n_filter < dense_params->output_num; ++n_filter) { in ff_dnn_execute_layer_dense()
147 output += dense_params->output_num; in ff_dnn_execute_layer_dense()
Ddnn_backend_native_layer_conv2d.c59 conv_params->output_num = (int32_t)avio_rl32(model_file_context); in ff_dnn_load_layer_conv2d()
64 kernel_size = conv_params->input_num * conv_params->output_num * in ff_dnn_load_layer_conv2d()
68 dnn_size += conv_params->output_num * 4; in ff_dnn_load_layer_conv2d()
71 conv_params->output_num <= 0 || conv_params->kernel_size <= 0){ in ff_dnn_load_layer_conv2d()
87 … conv_params->biases = av_malloc_array(conv_params->output_num, sizeof(*conv_params->biases)); in ff_dnn_load_layer_conv2d()
93 for (int i = 0; i < conv_params->output_num; ++i){ in ff_dnn_load_layer_conv2d()
131 …output += (conv_params->output_num) * (width - 2 * pad_size) * (thread_param->thread_start - pad_s… in dnn_execute_layer_conv2d_thread()
137 for (int n_filter = 0; n_filter < conv_params->output_num; ++n_filter) { in dnn_execute_layer_conv2d_thread()
180 output += conv_params->output_num; in dnn_execute_layer_conv2d_thread()
209 output_operand->dims[3] = conv_params->output_num; in ff_dnn_execute_layer_conv2d()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dkernel_select_cpu.cc64 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in GetOutputDtypes() local
65 for (size_t output_index = 0; output_index < output_num; ++output_index) { in GetOutputDtypes()
73 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in GetOutputFormat() local
74 for (size_t output_index = 0; output_index < output_num; ++output_index) { in GetOutputFormat()
103 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in GetOutputFormatsAndDtypes() local
104 for (size_t output_index = 0; output_index < output_num; ++output_index) { in GetOutputFormatsAndDtypes()
132 auto output_num = output_types.size(); in GetOutputDtypeMatchedNum() local
133 for (size_t i = 0; i < output_num; ++i) { in GetOutputDtypeMatchedNum()
172 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in ExpandKernelAttr() local
173 for (size_t i = 1; i < output_num; ++i) { in ExpandKernelAttr()
[all …]
Dcpu_simple_mem_plan.cc42 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel); in MemPlan() local
43 for (size_t i = 0; i < output_num; ++i) { in MemPlan()
87 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel); in MemAssign() local
88 for (size_t i = 0; i < output_num; ++i) { in MemAssign()
/third_party/mindspore/mindspore/core/ops/
Dsplit.cc24 void Split::Init(const int64_t axis, const int64_t output_num) { in Init() argument
26 this->set_output_num(output_num); in Init()
33 void Split::set_output_num(const int64_t output_num) { (void)this->AddAttr(kOutputNum, MakeValue(ou… in set_output_num() argument
Dcustom_predict.cc24 void CustomPredict::Init(const int64_t output_num, const float weight_threshold) { in Init() argument
25 this->set_output_num(output_num); in Init()
29 …ustomPredict::set_output_num(const int64_t output_num) { (void)this->AddAttr(kOutputNum, MakeValue… in set_output_num() argument
Dunstack.cc35 auto output_num = x_shape[LongToSize(axis)]; in UnstackInfer() local
36 (void)CheckAndConvertUtils::CheckInteger("output_num", output_num, kGreaterThan, 0, prim_name); in UnstackInfer()
37 auto output_valid_check = x_shape[LongToSize(axis)] - output_num; in UnstackInfer()
46 for (int64_t i = 0; i != output_num; i++) { in UnstackInfer()
Dunpack.cc38 auto output_num = x_shape[LongToSize(axis)]; in UnpackInfer() local
39 (void)CheckAndConvertUtils::CheckInteger("output_num", output_num, kGreaterThan, 0, prim_name); in UnpackInfer()
40 auto output_valid_check = x_shape[LongToSize(axis)] - output_num; in UnpackInfer()
49 for (int64_t i = 0; i != output_num; i++) { in UnpackInfer()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_build.cc83 size_t output_num = AnfAlgo::GetOutputTensorNum(anf_node); in SetIOSize() local
89 if (output_num == 1 && HasAbstractMonad(anf_node)) { in SetIOSize()
90 output_num = 0; in SetIOSize()
92 for (size_t i = 0; i < output_num; i++) { in SetIOSize()
222 size_t output_num = AnfAlgo::GetOutputTensorNum(anf_node); in SetNodeOutputs() local
223 if (output_num == 1 && HasAbstractMonad(anf_node)) { in SetNodeOutputs()
224 output_num = 0; in SetNodeOutputs()
226 if (output_num == 0) { in SetNodeOutputs()
231 for (size_t output_index = 0; output_index < output_num; output_index++) { in SetNodeOutputs()
335 const std::shared_ptr<AnfNode> &anf_node, size_t output_num) { in SetExtInfoOutputShapeType() argument
[all …]
/third_party/mindspore/mindspore/lite/src/c_api/
Dmodel_c.cc39 …nst OH_AI_TensorHandle *inputs, size_t input_num, OH_AI_TensorHandle **outputs, size_t *output_num,
43 MSTensor::Impl **GetOutputs(size_t *output_num);
130 … size_t *output_num, const OH_AI_KernelCallBack &before, const OH_AI_KernelCallBack &after) { in Predict() argument
180 *outputs = reinterpret_cast<OH_AI_TensorHandle *>(GetOutputs(output_num)); in Predict()
272 MSTensor::Impl **ModelC::GetOutputs(size_t *output_num) { in GetOutputs() argument
273 if (session_ == nullptr || output_num == nullptr) { in GetOutputs()
278 *output_num = outputs.size(); in GetOutputs()
279 if (outputs_.capacity() < *output_num) { in GetOutputs()
280 outputs_.reserve(*output_num); in GetOutputs()
397 size_t output_num; in OH_AI_ModelGetOutputs() local
[all …]
/third_party/mindspore/mindspore/lite/test/common/
Dcommon_test.h76 …static int CompareOutput(const float *output_data, size_t output_num, const std::string &file_path… in CompareOutput() argument
109 size_t output_num = output_size / sizeof(float); in CompareRelativeOutput() local
110 float error = CompareOutputRelativeData(output_data, ground_truth, output_num); in CompareRelativeOutput()
121 size_t output_num = output_size / sizeof(float); in RelativeOutputError() local
122 float error = CompareOutputRelativeData(output_data, ground_truth, output_num); in RelativeOutputError()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/
Dtrt_kernel.cc39 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in Init() local
40 for (size_t j = 0; j < output_num; j++) { in Init()
57 if (SizeToInt(input_num + output_num) != engine_->getNbBindings()) { in Init()
58 MS_LOG(EXCEPTION) << "Inputs and outputs num not match. Got: " << input_num + output_num in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dsplit_gpu_kernel.h142 int output_num = SizeToInt(AnfAlgo::GetOutputTensorNum(kernel_node)); in CheckParam() local
143 if (output_num <= 0) { in CheckParam()
144 MS_LOG(ERROR) << "Output number is " << output_num << ", must > 0."; in CheckParam()
163 if (output_num_ != output_num) { in CheckParam()
164 MS_LOG(ERROR) << "Output num is " << output_num << ", but need " << output_num_; in CheckParam()
Dselect_gpu_kernel.h86 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in CheckParam() local
87 if (output_num != 1) { in CheckParam()
88 MS_LOG(ERROR) << "Output number is " << output_num << ", but SelectGpuKernel needs 1 output."; in CheckParam()
Doneslike_gpu_kernel.h54 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in Init() local
55 if (output_num != 1) { in Init()
56 MS_LOG(ERROR) << "Output number is " << output_num << ", but oneslike needs 1 output."; in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddetection_post_process_fp32.c73 const int output_num = num_boxes < max_detections ? num_boxes : max_detections; in NmsSingleClass() local
83 …if (possible_candidate_num == 0 || selected_num >= output_num || scores[indexes[i]] < param->nms_s… in NmsSingleClass()
134 float *output_scores, float *output_num, in DetectionPostProcessFast() argument
138 output_scores == NULL || output_num == NULL || param == NULL || PartialArgSort == NULL) { in DetectionPostProcessFast()
158 *output_num = (float)out_num; in DetectionPostProcessFast()
171 … float *output_boxes, float *output_classes, float *output_scores, float *output_num, in DetectionPostProcessRegular() argument
175 output_num == NULL || param == NULL || PartialArgSort == NULL) { in DetectionPostProcessRegular()
233 *output_num = (float)all_classes_output_num; in DetectionPostProcessRegular()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadd_relu_v2_fusion.cc42 size_t output_num = AnfAlgo::GetOutputTensorNum(node); in GenerateKernelBuildInfo() local
43 for (size_t output_index = 0; output_index < output_num; ++output_index) { in GenerateKernelBuildInfo()
91 size_t output_num = AnfAlgo::GetOutputTensorNum(node); in Process() local
92 for (size_t i = 0; i < output_num; i++) { in Process()
Dinsert_cast_gpu.cc90 size_t output_num = AnfAlgo::GetOutputTensorNum(node); in Run() local
91 for (size_t i = 0; i < output_num; i++) { in Run()
110 auto output_types = std::vector<TypeId>(output_num, kNumberTypeFloat32); in Run()
112 for (size_t output_index = 0; output_index < output_num; ++output_index) { in Run()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_getnext.cc30 size_t output_num = AnfAlgo::GetOutputTensorNum(node); in InsertTensorMoveForGetNextOutputs() local
31 if (output_num == 0) { in InsertTensorMoveForGetNextOutputs()
40 for (size_t output_index = 0; output_index < output_num; ++output_index) { in InsertTensorMoveForGetNextOutputs()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dassign_add_gpu_kernel.h57 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in Init() local
58 if (output_num != 1) { in Init()
59 MS_LOG(ERROR) << "Output number is " << output_num << ", but cudnnAddTensor needs 1 output."; in Init()
Dequalcount_gpu_kernel.h57 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in Init() local
58 if (output_num != 1) { in Init()
59 MS_LOG(ERROR) << "Output number is " << output_num << ", but equalcount needs 1 output."; in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/
Dminmax_update_perchannel_gpu_kernel.cc43 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in Init() local
44 if (output_num != 2) { in Init()
45 …MS_LOG(EXCEPTION) << "Output number is " << output_num << ", but FakeQuant GpuKernel OP needs 1 ou… in Init()
Dminmax_update_perlayer_gpu_kernel.cc41 size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node); in Init() local
42 if (output_num != 2) { in Init()
43 …MS_LOG(EXCEPTION) << "Output number is " << output_num << ", but FakeQuant GpuKernel OP needs 1 ou… in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dmasked_select_cpu_kernel.cc72 size_t output_num = AnfAlgo::GetOutputTensorNum(node_); in Launch() local
73 std::vector<TypeId> dtypes(output_num); in Launch()
74 for (size_t i = 0; i < output_num; i++) { in Launch()

12345678910