Home
last modified time | relevance | path

Searched defs:weight_data (Results 1 – 25 of 34) sorted by relevance

12

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dconv2d_tests.cc39 void TestMain_Conv2D(const std::string &attr, float *input_data, float *weight_data, float *bias_da… in TestMain_Conv2D()
59 float weight_data[] = {1, 1, 1, 1, 1, 1, 1, 1}; in TEST_F() local
79 float weight_data[] = {1, 1, 1, 1, 1, 1, 1, 1}; in TEST_F() local
89 float weight_data[] = {1, 2, 3, 4, 5, 6, 7, 8}; in TEST_F() local
100 float weight_data[] = {1, 1, 1, 1, 1, 1, 1, 1}; in TEST_F() local
111 float weight_data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}; in TEST_F() local
122 float weight_data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}; in TEST_F() local
264 auto weight_data = std::get<3>(case_).data(); in TEST_F() local
Dmatmul_tests.cc42 float weight_data[] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}; variable
62 float weight_data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, variable
82 float weight_data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, variable
102 float weight_data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, in TEST_F() local
122 float weight_data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, in TEST_F() local
Dfullconnection_tests.cc59 float weight_data[] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}; variable
81 float weight_data[] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}; variable
104 float weight_data[] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}; variable
127 float weight_data[] = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1}; variable
Dconv2d_transpose_tests.cc63 float weight_data[] = {0, 2, 4, 6, 1, 3, 5, 7}; in TEST_F() local
90 float weight_data[] = {0, 2, 4, 6, 1, 3, 5, 7}; in TEST_F() local
117 float weight_data[] = {0.0, 2.0, 4.0, 6.0, 8.0, 10.0, 12.0, 14.0, 16.0, in TEST_F() local
146 float weight_data[] = {0, 2, 4, 6, 1, 3, 5, 7}; in TEST_F() local
175 float weight_data[] = {0, 2, 4, 6, 1, 3, 5, 7}; in TEST_F() local
204 float weight_data[] = {0.0, 2.0, 4.0, 6.0, 8.0, 10.0, 12.0, 14.0, 16.0, in TEST_F() local
Ddepthwise_conv2d_tests.cc75 …float weight_data[] = {0.19658236, 0.36872518, 0.82099324, 0.09710128, 0.8379449, 0.09609841, 0.9… in TEST_F() local
116 …float weight_data[] = {0.67063785, 0.21038257, 0.12892629, 0.31542835, 0.36371076, 0.57019675, 0.4… in TEST_F() local
166 float weight_data[] = {0.19658236, 0.36872517, 0.82099323, 0.09710128, 0.83794491, 0.09609841, in TEST_F() local
219 …float weight_data[] = {0.1494483, 0.86812606, 0.16249293, 0.61555956, 0.12381998, 0.84800823, 0.8… in TEST_F() local
Dcommon.cc164 auto *weight_data = std::get<1>(input_infos[i]); in TestMain() local
198 auto *weight_data = std::get<1>(input_infos[i]); in TestMain() local
329 auto *weight_data = std::get<1>(input_infos[i]); in TestMain() local
361 auto *weight_data = std::get<1>(input_infos[i]); in TestMain() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dconvolution_winograd_fp32_coder.cc78 int ConvolutionWinogradFP32Coder::WinogradFilterTransform(const float *weight_data, float *matrix_g, in WinogradFilterTransform()
152 auto weight_data = reinterpret_cast<float *>(filter_tensor_->data()); in InitWeightBias() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/string/
Dextract_feature.cc59 auto weight_data = out_tensors_.at(1)->MutableData(); in Run() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_scale_fusion.cc72 auto weight_data = reinterpret_cast<const float *>(weight_value->data_c()); in InitTransParam() local
Dconv_transform_fusion.cc75 auto weight_data = reinterpret_cast<float *>(weight_tensor->data_c()); in GenerateNewWeightConv2DTranspose() local
290 auto weight_data = reinterpret_cast<float *>(weight_tensor->data_c()); in CalNewWeightTensor() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/
Dget_model_kernel.h81 float *weight_data = const_cast<float *>(feature_map->Get(i)->data()->data()); in Launch() local
Dfused_pull_weight_kernel.h187 float *weight_data = const_cast<float *>(fbs_feature_map->Get(i)->data()->data()); in ParseFeatureMap() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/
DTiledC4MatMulFp32.c48 __m128 weight_data[4]; in TiledC4MatmulFp32() local
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dmodel_store.cc127 auto weight_data = std::make_unique<char[]>(weight_size); in AssignNewModelMemory() local
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/
Dpush_weight_kernel.cc129 float *weight_data = const_cast<float *>(fbs_feature_map->Get(i)->data()->data()); in ParseFeatureMap() local
Dupdate_model_kernel.cc236 float *weight_data = const_cast<float *>(fbs_feature_map->Get(i)->data()->data()); in ParseFeatureMap() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_winograd_fp32.cc28 int ConvolutionWinogradCPUKernel::WinogradFilterTransform(const float *weight_data, float *matrix_g, in WinogradFilterTransform()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_winograd_fp16.cc23 int ConvolutionWinogradFP16CPUKernel::WinogradFilterTransformFp16(const float16_t *weight_data, con… in WinogradFilterTransformFp16()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_3x3_int8.cc105 auto weight_data = reinterpret_cast<int8_t *>(in_tensors_.at(kWeightIndex)->MutableData()); in InitWeightBias() local
Dmatmul_base_int8.cc208 auto weight_data = reinterpret_cast<int8_t *>(in_tensors_.at(1)->data()); in TransferB() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_fp32_tests.cc71 …auto weight_data = reinterpret_cast<float *>(mindspore::lite::ReadFile(weight_path.c_str(), &weigh… in InitConvDwCreator() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dfullconnection_int8_tests.cc57 int8_t *weight_data = new int8_t[weight->len]; in FcInt8TestInit() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv_depthwise_int8.c80 …nt8(int8_t *output_data, int32_t *row_buffer, const int8_t *input_data, const int16_t *weight_data, in ConvDwInt8()
297 …w3x3Int8(int8_t *output_data, int8_t *buffer, const int8_t *input_data, const int16_t *weight_data, in ConvDw3x3Int8()
413 void ConvDw3x3Int8Pad(int8_t *output_data, const int8_t *input_data, const int16_t *weight_data, in ConvDw3x3Int8Pad()
619 void ConvDwInt8SW(int8_t *output_data, const int8_t *input_data, const int16_t *weight_data, const … in ConvDwInt8SW()
773 …int8_t *output_data, int32_t *output_buffer, const int16_t *input_data, const int16_t *weight_data, in DeconvDwInt8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_depthwise_fp32.c35 int ConvDw(float *output_data, const float *input_data, const float *weight_data, const float *bias… in ConvDw()
305 void ConvDwSWFp32(float *output_data, const float *input_data, const float *weight_data, const floa… in ConvDwSWFp32()
673 void ConvDw3x3(float *output_data, float *buffer, const float *input_data, const float *weight_data, in ConvDw3x3()
830 void ConvDwIndirection(float *output_data, float **indirect_buffer, const float *weight_data, const… in ConvDwIndirection()
977 void DeconvDwSWFp32(float *output_data, const float *input_data, const float *weight_data, const fl… in DeconvDwSWFp32()
1049 void DepthwiseSWAvxFp32(float *output_data, const float *input_data, const float *weight_data, cons… in DepthwiseSWAvxFp32()
1144 __m256 weight_data[4]; in DepthwiseSWWxKKernel() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dconv_depthwise_fp16.c349 …oat16_t *output_data, float16_t *buffer, const float16_t *input_data, const float16_t *weight_data, in ConvDw3x3Fp16()
398 void ConvDwFp16(float16_t *output_data, const float16_t *input_data, const float16_t *weight_data, in ConvDwFp16()
571 void ConvDwC8Fp16(float16_t *output_data, const float16_t *input_data, const float16_t *weight_data, in ConvDwC8Fp16()
735 …d DeconvDwC8Fp16(float16_t *output_data, const float16_t *input_data, const float16_t *weight_data, in DeconvDwC8Fp16()

12