Home
last modified time | relevance | path

Searched refs:stride_size (Results 1 – 11 of 11) sorted by relevance

/third_party/boost/libs/range/test/adaptor_test/
Dstrided.cpp34 void strided_test_impl( Container& c, int stride_size ) in strided_test_impl() argument
61 for (int i = 0; (it != last) && (i < stride_size); ++i) in strided_test_impl()
67 boost::push_back( test, c | strided(stride_size) ); in strided_test_impl()
73 boost::push_back( test2, adaptors::stride(c, stride_size) ); in strided_test_impl()
81 boost::push_back( test3, cc | strided(stride_size) ); in strided_test_impl()
87 boost::push_back( test4, adaptors::stride(cc, stride_size) ); in strided_test_impl()
94 void strided_test_impl(int stride_size) in strided_test_impl() argument
101 strided_test_impl(c, stride_size); in strided_test_impl()
105 strided_test_impl(c, stride_size); in strided_test_impl()
109 strided_test_impl(c, stride_size); in strided_test_impl()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dcommon_func_fp16.c47 size_t stride_size = oc_stride * sizeof(float16_t); in PostConvFuncFp16C8() local
48 PostFuncBiasReluC8Fp16(nhwc_out, c8_out, bias, oc8div, oc8mod, plane, stride_size, act_type); in PostConvFuncFp16C8()
59 size_t stride_size = (plane_stride - plane) * C4NUM * sizeof(float16_t); in PostConvFuncFp16C4() local
60 PostFuncBiasReluC4Fp16(nhwc_out, c4_out, bias, oc4div, oc4mod, plane, stride_size, act_type); in PostConvFuncFp16C4()
/third_party/mindspore/mindspore/core/ops/
Daudio_spectrogram.cc62 auto stride_size = GetValue<int64_t>(primitive->GetAttr(kStride)); in AudioSpectrogramInferShape() local
63 if (stride_size < 1) { in AudioSpectrogramInferShape()
64 MS_LOG(ERROR) << "stride must be positive, now is " << stride_size; in AudioSpectrogramInferShape()
69 infer_shape.push_back(sample_sub_window < 0 ? 0 : 1 + sample_sub_window / stride_size); in AudioSpectrogramInferShape()
Dconv2d_transpose.cc65 const int64_t stride_size = 2; in set_stride() local
66 …(void)CheckAndConvertUtils::CheckInteger(kStride, SizeToLong(stride.size()), kEqual, stride_size, … in set_stride()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dsoftmax_int8.cc105 int stride_size = stride * task_id * inner_size; in DoSoftmax() local
107 …auto error_code = SoftmaxInt8(input_ptr + stride_size, output_ptr + stride_size, count, exp_data_ … in DoSoftmax()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dcommon_func_fp32.c49 size_t stride_size = stride * sizeof(float); in PostConvFuncFp32C8()
50 …PostFuncBiasReluC8(out_ptr, c8_out_ptr, bias_ptr, oc8div, oc8mod, plane_size, stride_size, relu_ty… in PostConvFuncFp32C8()
60 size_t stride_size = (plane_stride - plane_size) * C4NUM * sizeof(float); in PostConvFuncFp32C4() local
61 …PostFuncBiasReluC4(out_ptr, c4_out_ptr, bias_ptr, oc4div, oc4mod, plane_size, stride_size, relu_ty… in PostConvFuncFp32C4()
/third_party/mindspore/mindspore/nn/layer/
Dconv.py872 def _deconv_output_length(is_valid, is_same, is_pad, input_length, filter_size, stride_size, dilati… argument
877 if filter_size - stride_size > 0:
878 length = input_length * stride_size + filter_size - stride_size
880 length = input_length * stride_size
882 length = input_length * stride_size
884 length = input_length * stride_size - padding + filter_size - stride_size
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dstridedslice_cpu_kernel.cc134 int stride_size = LongToInt(stride[i]); in InitSliceParam() local
135 if (stride_size == 0) { in InitSliceParam()
139 slice_param_.strides_[i] = stride_size; in InitSliceParam()
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_conv_base_parser.cc74 if (convParam.stride_size() != 0) { in ParseStrides()
85 const int num_stride_dims = convParam.stride_size(); in ParseStrides()
/third_party/boost/libs/range/doc/
Dupgrade.qbk36 # __range_adaptors_strided__ no longer requires `distance(rng) % stride_size == 0` or `stride_size
/third_party/mindspore/mindspore/ops/operations/
Dnn_ops.py8594 def _deconv_output_length(input_length, kernel_size, stride_size, dilation_size): argument
8596 if filter_size - stride_size > 0:
8597 length = input_length * stride_size + filter_size - stride_size
8599 length = input_length * stride_size