Home
last modified time | relevance | path

Searched refs:axis_size (Results 1 – 25 of 30) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreduce_fp32.c26 int ReduceMean(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_dat… in ReduceMean() argument
36 const float *outer_src = src_data + j * axis_size * inner_size; in ReduceMean()
42 for (i = 0; i < axis_size; i++) { in ReduceMean()
45 *inner_dst = tmp / (float)axis_size; in ReduceMean()
51 int IntReduceMean(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data… in IntReduceMean() argument
53 if (axis_size == 0) { in IntReduceMean()
62 NNACL_CHECK_ZERO_RETURN_ERR(axis_size); in IntReduceMean()
69 const int *outer_src = src_data + j * axis_size * inner_size; in IntReduceMean()
77 for (i = 0; i < axis_size; i++) { in IntReduceMean()
80 tmp[0] /= axis_size; in IntReduceMean()
[all …]
Dreduce_fp32.h25 int ReduceMean(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_dat…
27 int IntReduceMean(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data…
29 int ReduceSum(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data…
31 int IntReduceSum(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data,…
33 int ReduceMax(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data…
35 int IntReduceMax(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data,…
37 int ReduceMin(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_data…
39 int IntReduceMin(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data,…
41 int ReduceProd(int outer_size, int inner_size, int axis_size, const float *src_data, float *dst_dat…
43 int IntReduceProd(int outer_size, int inner_size, int axis_size, const int *src_data, int *dst_data…
[all …]
Dscale_fp32.c22 int outer_end, int axis_size, int inner_size) { in ScaleInner() argument
24 int out_offset = out * axis_size * inner_size; in ScaleInner()
25 for (int i = 0; i < axis_size; i++) { in ScaleInner()
57 int outer_end, int axis_size) { in ScaleAxis() argument
59 int out_offset = out * axis_size; in ScaleAxis()
62 for (; index <= axis_size - C8NUM; index += C8NUM) { in ScaleAxis()
72 for (; index <= axis_size - C4NUM; index += C4NUM) { in ScaleAxis()
81 for (; index < axis_size; index++) { in ScaleAxis()
106 int outer_end, int axis_size, int inner_size) { in ScaleInnerRelu() argument
114 int out_offset = out * axis_size * inner_size; in ScaleInnerRelu()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dreduce_fp16.c21 int ReduceMeanFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float1… in ReduceMeanFp16() argument
26 if (axis_size == 0) { in ReduceMeanFp16()
31 const float16_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMeanFp16()
37 for (i = 0; i < axis_size; i++) { in ReduceMeanFp16()
40 *inner_dst = (float16_t)(tmp / axis_size); in ReduceMeanFp16()
46 int ReduceMaxFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float16… in ReduceMaxFp16() argument
53 const float16_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMaxFp16()
59 for (i = 0; i < axis_size; i++) { in ReduceMaxFp16()
68 int ReduceSumFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float16… in ReduceSumFp16() argument
78 int src_stride = axis_size * inner_size; in ReduceSumFp16()
[all …]
Dscale_fp16.c20 int outer_start, int outer_end, int axis_size, int inner_size) { in Fp16ScaleInner() argument
22 int out_offset = out * axis_size * inner_size; in Fp16ScaleInner()
23 for (int i = 0; i < axis_size; i++) { in Fp16ScaleInner()
46 int outer_start, int outer_end, int axis_size) { in Fp16ScaleAxis() argument
48 int out_offset = out * axis_size; in Fp16ScaleAxis()
51 for (; index < axis_size - 8; index += 8) { in Fp16ScaleAxis()
60 for (; index < axis_size; index++) { in Fp16ScaleAxis()
83 int outer_start, int outer_end, int axis_size, int inner_size) { in Fp16ScaleInnerRelu() argument
88 int out_offset = out * axis_size * inner_size; in Fp16ScaleInnerRelu()
89 for (int i = 0; i < axis_size; i++) { in Fp16ScaleInnerRelu()
[all …]
Dreduce_fp16.h25 int ReduceMeanFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float1…
27 int ReduceMaxFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float16…
29 int ReduceSumFp16(int outer_size, int inner_size, int axis_size, const float16_t *src_data, float16…
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dreverse_v2_impl.cu21 const int64_t* axis, size_t input_size, size_t axis_size) { in ReverseV2() argument
24 for (size_t i = 0; i < axis_size; i++) { in ReverseV2()
37 size_t input_size, size_t axis_size, cudaStream_t cuda_stream) { in CalReverseV2() argument
39 input_size, axis_size); in CalReverseV2()
44 … const int64_t* axis, size_t input_size, size_t axis_size, cudaStream_t cuda_stream);
47 … const int64_t* axis, size_t input_size, size_t axis_size, cudaStream_t cuda_stream);
50 … const int64_t* strides, const int64_t* axis, size_t input_size, size_t axis_size,
54 … const int64_t* strides, const int64_t* axis, size_t input_size, size_t axis_size,
58 … const int64_t* strides, const int64_t* axis, size_t input_size, size_t axis_size,
62 … const int64_t* strides, const int64_t* axis, size_t input_size, size_t axis_size,
Dreverse_v2_impl.cuh20 size_t input_size, size_t axis_size, cudaStream_t cuda_stream);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dreduce_int8.h43 int ReduceMeanInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *…
45 int ReduceMeanLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32…
47 int ReduceSumInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *s…
49 int ReduceSumLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32_…
51 int ReduceMaxInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *s…
53 int ReduceMaxLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32_…
55 int ReduceMinInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *s…
57 int ReduceMinLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32_…
59 int ReduceProdLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32…
61 int ReduceProdInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *…
[all …]
Dreduce_int8.c176 int ReduceMeanInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *… in ReduceMeanInt8() argument
183 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMeanInt8()
189 for (i = 0; i < axis_size; i++) { in ReduceMeanInt8()
210 int ReduceMeanLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32… in ReduceMeanLastAxis() argument
217 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMeanLastAxis()
223 for (i = 0; i < axis_size; i++) { in ReduceMeanLastAxis()
251 int ReduceSumInt8(const int outer_size, const int inner_size, const int axis_size, const int32_t *s… in ReduceSumInt8() argument
258 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceSumInt8()
264 for (i = 0; i < axis_size; i++) { in ReduceSumInt8()
283 int ReduceSumLastAxis(const int outer_size, const int inner_size, const int axis_size, const int32_… in ReduceSumLastAxis() argument
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dshift_cpu_kernel.cc89 const int64_t axis_size = SizeToLong(axisIterator_.AxisSize()); in Launch() local
93 if (std::abs(periods_) >= axis_size) { in Launch()
94 (void)std::fill_n(output, outer_size * axis_size * inner_size, fill_value); in Launch()
98 if (inputs[0]->size != outer_size * axis_size * inner_size * sizeof(T)) { in Launch()
119 (void)tasks.emplace_back([this, i, fill_value, axis_size, inner_size, input, output, outputs] { in Launch()
120 size_t offset = i * axis_size * inner_size; in Launch()
Drolling_cpu_kernel.cc90 int axis_size = SizeToInt(axisIterator_.AxisSize()); in RollingBoundsCalculate() local
91 for (int i = 0; i < axis_size; ++i) { in RollingBoundsCalculate()
94 ends_[i] = IntToSize(std::max(0, std::min(end, axis_size))); in RollingBoundsCalculate()
95 starts_[i] = IntToSize(std::max(0, std::min(start, axis_size))); in RollingBoundsCalculate()
Dl2_normalize_cpu_kernel.cc57 size_t axis_size = IntToSize(axis_); in CalcDenominator() local
59 if (i != axis_size) { in CalcDenominator()
66 axes[k] = axis_size; in CalcDenominator()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dreduce_fp32.h29 typedef int (*Reducer)(const int outer_size, const int inner_size, const int axis_size, const float…
31 typedef int (*IntReducer)(const int outer_size, const int inner_size, const int axis_size, const in…
33 typedef int (*BoolReducer)(const int outer_size, const int inner_size, const int axis_size, const b…
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dreduce_int8.h33 …typedef int (*Reducer)(const int outer_size, const int inner_size, const int axis_size, const int3…
35 …typedef int (*LastReducer)(const int outer_size, const int inner_size, const int axis_size, const …
Dreduce_int8.cc274 int axis_size = in_tensors_.at(0)->shape()[axes_[i]]; in CalculateQuantArgs() local
280 double prod_multiplier = pow(quant_arg_.in_scale_, axis_size - 1); in CalculateQuantArgs()
/third_party/flutter/skia/third_party/externals/freetype/src/sfnt/
Dsfobjs.c560 FT_UShort axis_size; in sfnt_init_face() local
579 FT_READ_USHORT( axis_size ) || in sfnt_init_face()
586 axis_size = 0; in sfnt_init_face()
593 axis_size != 20 || in sfnt_init_face()
602 axis_size * num_axes + in sfnt_init_face()
639 default_value_offset += axis_size; in sfnt_init_face()
643 instance_offset = array_start + axis_size * num_axes + 4; in sfnt_init_face()
/third_party/freetype/src/sfnt/
Dsfobjs.c585 FT_UShort axis_size; in sfnt_init_face() local
604 FT_READ_USHORT( axis_size ) || in sfnt_init_face()
611 axis_size = 0; in sfnt_init_face()
618 axis_size != 20 || in sfnt_init_face()
627 axis_size * num_axes + in sfnt_init_face()
664 default_value_offset += axis_size; in sfnt_init_face()
668 instance_offset = array_start + axis_size * num_axes + 4; in sfnt_init_face()
/third_party/skia/third_party/externals/freetype/src/sfnt/
Dsfobjs.c598 FT_UShort axis_size; in sfnt_init_face() local
617 FT_READ_USHORT( axis_size ) || in sfnt_init_face()
624 axis_size = 0; in sfnt_init_face()
631 axis_size != 20 || in sfnt_init_face()
640 axis_size * num_axes + in sfnt_init_face()
677 default_value_offset += axis_size; in sfnt_init_face()
681 instance_offset = array_start + axis_size * num_axes + 4; in sfnt_init_face()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dreduce_fp16.h29 …typedef int (*Reducer)(const int outer_size, const int inner_size, const int axis_size, const floa…
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dreduce_int8_coder.cc65 int axis_size = in_shape.at(axis); in CalculateQuantArgs() local
66 double prod_multiplier = std::pow(quant_arg_.in_scale_, axis_size - 1); in CalculateQuantArgs()
/third_party/boost/libs/histogram/benchmark/
DCMakeLists.txt34 add_benchmark(axis_size)
/third_party/mindspore/mindspore/numpy/
Dutils_const.py356 def _expanded_shape(ndim, axis_size, axis): argument
361 return tuple([axis_size if i == axis else 1 for i in range(ndim)])
/third_party/mindspore/mindspore/
D_checkparam.py790 def expanded_shape(ndim, axis_size, axis): argument
795 return tuple(axis_size if i == axis else 1 for i in range(ndim))
/third_party/flutter/skia/third_party/externals/freetype/src/truetype/
Dttgxvar.c2024 FT_Offset axis_size; in TT_Get_MM_Var() local
2141 axis_size = ALIGN_SIZE( num_axes * in TT_Get_MM_Var()
2154 axis_size + in TT_Get_MM_Var()
2181 (FT_Var_Named_Style*)( (char*)mmvar->axis + axis_size ); in TT_Get_MM_Var()
2429 (FT_Var_Named_Style*)( (char*)mmvar->axis+ axis_size ); in TT_Get_MM_Var()

12