Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: d1bf0debc5
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'd1bf0debc5'
${ noResults }
mindspore/mindspore/lite/nnacl/fp32/reduce_fp32.h

53 lines
2.4 kB
Raw Blame History 

  1. /**

  2.  * Copyright 2020 Huawei Technologies Co., Ltd

  3.  *

  4.  * Licensed under the Apache License, Version 2.0 (the "License");

  5.  * you may not use this file except in compliance with the License.

  6.  * You may obtain a copy of the License at

  7.  *

  8.  * http://www.apache.org/licenses/LICENSE-2.0

  9.  *

  10.  * Unless required by applicable law or agreed to in writing, software

  11.  * distributed under the License is distributed on an "AS IS" BASIS,

  12.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  13.  * See the License for the specific language governing permissions and

  14.  * limitations under the License.

  15.  */

  16. 

  17. #ifndef MINDSPORE_LITE_NNACL_FP32_REDUCE_H_

  18. #define MINDSPORE_LITE_NNACL_FP32_REDUCE_H_

  19. #include "nnacl/op_base.h"

  20. #include "nnacl/reduce_parameter.h"

  21. 

  22. #ifdef __cplusplus

  23. extern "C" {

  24. #endif

  25. int ReduceMean(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data, int tid,

  26.                int thread_num);

  27. int ReduceSum(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data, int tid,

  28.               int thread_num);

  29. int ReduceMax(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data, int tid,

  30.               int thread_num);

  31. int ReduceMin(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data, int tid,

  32.               int thread_num);

  33. int IntReduceMin(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data, int tid,

  34.                  int thread_num);

  35. int ReduceProd(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data, int tid,

  36.                int thread_num);

  37. int IntReduceProd(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data, int tid,

  38.                   int thread_num);

  39. int ReduceSumSquare(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data, int tid,

  40.                     int thread_num);

  41. int ReduceAll(int outer_size, int inner_size, int axis_size, const bool *src_data, bool *dst_data, int tid,

  42.               int thread_num);

  43. 

  44. #ifdef ENABLE_NNACL_INFER_SHAPE

  45. int ReduceInferShape(int **in_shape, size_t *dim_size, int *out_shape, int *in_format, int *out_format,

  46.                      int *in_datatype, int *out_datatype, OpParameter *param);

  47. #endif

  48. #ifdef __cplusplus

  49. }

  50. #endif

  51. 

  52. #endif  // MINDSPORE_LITE_NNACL_FP32_REDUCE_H_