| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | index_add_impl.cu | 22 …exAddAtomic(T *dst, const int *index, const T *src, const size_t src_size, const size_t outer_size, in IndexAddAtomic()
37 …id IndexAdd(T *dst, const int *index, const T *src, const size_t src_size, const size_t outer_size, in IndexAdd()
52 void CalIndexAdd(T *dst, const int *index, const T *src, const size_t outer_size, const size_t src_… in CalIndexAdd()
|
| D | argmax_impl.cu | 21 __global__ void Argmax(const T *input, const S bound, const size_t outer_size, in Argmax()
42 void CalArgmax(const T *input, const S bound, const size_t outer_size, const size_t inner_size, in CalArgmax()
|
| D | unsorted_segment_max.cu | 21 …rtedSegmentMax(const T *input, const S *segment_ids, const int64_t num_segments, size_t outer_size, in UnsortedSegmentMax()
60 …rtedSegmentMax(const T *input, const S *segment_ids, const int64_t num_segments, size_t outer_size, in CalUnsortedSegmentMax()
|
| D | unsorted_segment_min.cu | 32 size_t outer_size, size_t inner_size, T init_K, T *output) { in UnsortedSegmentMin()
65 …edSegmentMin(const T *input, const int *segment_ids, const int64_t num_segments, size_t outer_size, in CalUnsortedSegmentMin()
|
| D | general_reduction_impl.cu | 47 __global__ void ThreadReduction(bool small, size_t outer_size, size_t bound, size_t inner_size, con… in ThreadReduction()
78 __global__ void WarpReduction(bool small, size_t outer_size, size_t bound, size_t inner_size, const… in WarpReduction()
126 __global__ void Warp4Reduction(bool small, size_t outer_size, size_t bound, size_t inner_size, cons… in Warp4Reduction()
215 __global__ void BlockReduction(bool small, size_t outer_size, size_t bound, size_t inner_size, cons… in BlockReduction()
299 void GeneralReduction(bool small, size_t outer_size, size_t bound, size_t inner_size, const T *inpu… in GeneralReduction()
|
| D | topk_impl.cu | 100 inline __device__ void TopKStep(const int &outer_size, const int &inner_size, const T *input, T *ou… in TopKStep()
178 __global__ void TopKBlock(int outer_size, int inner_size, const T *input, T *output, S *output_inde… in TopKBlock()
207 void FastTopK(const int outer_size, const int inner_size, const T *input, S k_cut, T *output, S *ou… in FastTopK()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | reduce_fp32.c | 26 int ReduceMean(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_dat… in ReduceMean()
51 int IntReduceMean(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data… in IntReduceMean()
100 int ReduceSum(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data… in ReduceSum()
141 int IntReduceSum(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data,… in IntReduceSum()
182 int ReduceMax(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data… in ReduceMax()
207 int IntReduceMax(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data,… in IntReduceMax()
232 int ReduceMin(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data… in ReduceMin()
257 int IntReduceMin(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data,… in IntReduceMin()
282 int ReduceAll(int outer_size, int inner_size, int axis_size, const bool *src_data, bool *dst_data, … in ReduceAll()
307 int ReduceProd(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_dat… in ReduceProd()
[all …]
|
| D | one_hot_fp32.c | 29 int outer_size = one_hot_param->outer_size_; in OneHotToFp32() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | reduce_fp16.c | 21 int ReduceMeanFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float1… in ReduceMeanFp16()
46 int ReduceMaxFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float16… in ReduceMaxFp16()
68 int ReduceSumFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float16… in ReduceSumFp16()
|
| D | one_hot_fp16.c | 28 int outer_size = one_hot_param->outer_size_; in OneHotToFp16() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | reduce_int8.c | 176 int ReduceMeanInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *… in ReduceMeanInt8()
210 int ReduceMeanLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32… in ReduceMeanLastAxis()
251 int ReduceSumInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *s… in ReduceSumInt8()
283 int ReduceSumLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32_… in ReduceSumLastAxis()
323 int ReduceMaxLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32_… in ReduceMaxLastAxis()
359 int ReduceMaxInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *s… in ReduceMaxInt8()
382 int ReduceMinLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32_… in ReduceMinLastAxis()
420 int ReduceMinInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *s… in ReduceMinInt8()
442 int ReduceProdLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32… in ReduceProdLastAxis()
485 int ReduceProdInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *… in ReduceProdInt8()
[all …]
|
| D | gather_int8.c | 22 int GatherInt8(int8_t *in_data, int8_t *out_data, int outer_size, int inner_size, int limit, const … in GatherInt8()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | gather_base.c | 19 int Gather(const void *input, int outer_size, int inner_size, int limit, const int *indices, int in… in Gather()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | gather_fp32_coder.cc | 53 int outer_size = 1, inner_size = 1; in DoCode() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | l2_norm_int8.cc | 75 int outer_size = input_tensor->ElementsNum() / input_tensor->shape().back(); in DoExecute() local
|
| D | gather_int8.cc | 76 int outer_size = 1; in DoGather() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | local_response_norm_fp32.cc | 52 int outer_size = batch * width * height; in DoLocalResponseNorm() local
|
| D | gather_fp32.cc | 56 int outer_size = 1, inner_size = 1; in DoGather() local
|
| D | l2_norm_fp32.cc | 120 int outer_size = input->ElementsNum() / input->shape().back(); in CalcL2NormTrailingAxis() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/ |
| D | reduce_base_coder.cc | 105 int outer_size = 1; in CalculateInnerOuterSize() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | shift_cpu_kernel.cc | 88 const int64_t outer_size = SizeToLong(axisIterator_.OuterSize()); in Launch() local
|
| D | gather_cpu_kernel.cc | 76 size_t outer_size = 1, inner_size = 1; in ParallelRun() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | reduce_base.cc | 141 auto outer_size = 1; in CalculateInnerOuterSize() local
|
| D | stack_base.cc | 51 int outer_size = 1; in GetOuterSize() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | gather_fp16.cc | 78 int outer_size = 1, inner_size = 1; in DoGather() local
|