Home
last modified time | relevance | path

Searched refs:weight_t (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dconv_1x1_int8_tests.cc84 …Tensor *weight_t = new Tensor(kNumberTypeInt8, {3, 1, 1, 4}, mindspore::NHWC, lite::Tensor::Catego… in Conv1x1Int8TestInit1_perchannel() local
85 weight_t->MallocData(); in Conv1x1Int8TestInit1_perchannel()
92 weight_t->AddQuantParam(*weight_quant_arg1); in Conv1x1Int8TestInit1_perchannel()
93 weight_t->AddQuantParam(*weight_quant_arg2); in Conv1x1Int8TestInit1_perchannel()
94 weight_t->AddQuantParam(*weight_quant_arg3); in Conv1x1Int8TestInit1_perchannel()
96 memcpy(weight_t->MutableData(), weight, weight_t->ElementsNum() * sizeof(int8_t)); in Conv1x1Int8TestInit1_perchannel()
97 inputs_->push_back(weight_t); in Conv1x1Int8TestInit1_perchannel()
155 …Tensor *weight_t = new Tensor(kNumberTypeInt8, {3, 1, 1, 4}, mindspore::NHWC, lite::Tensor::Catego… in Conv1x1Int8TestInit1() local
158 weight_t->AddQuantParam(*weight_quant_arg); in Conv1x1Int8TestInit1()
159 weight_t->MallocData(); in Conv1x1Int8TestInit1()
[all …]
Ddeconv_int8_tests.cc351 …auto *weight_t = new Tensor(kNumberTypeInt8, {3, 3, 3, 2}, mindspore::NHWC, lite::Tensor::Category… in DeConvInt8TestInit1() local
352 weight_t->MallocData(); in DeConvInt8TestInit1()
356 memcpy(weight_t->MutableData(), weight, sizeof(int8_t) * weight_t->ElementsNum()); in DeConvInt8TestInit1()
359 weight_t->AddQuantParam(*w_quant_arg); in DeConvInt8TestInit1()
360 inputs_->push_back(weight_t); in DeConvInt8TestInit1()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Ddeconvolution_fp32_tests.cc342 auto *weight_t = in DeConvTestInit1() local
344 weight_t->MallocData(); in DeConvTestInit1()
360 …PackNCHWToNHWCFp32(weight_nchw, weight_t->MutableData(), weight_t->Batch(), weight_t->Width() * we… in DeConvTestInit1()
361 weight_t->Channel(), 0, 0); in DeConvTestInit1()
362 inputs_->push_back(weight_t); in DeConvTestInit1()
509 auto *weight_t = in DeConvTestInit2() local
511 weight_t->MallocData(); in DeConvTestInit2()
520 memcpy(weight_t->MutableData(), weight, sizeof(float) * weight_t->ElementsNum()); in DeConvTestInit2()
521 inputs_->push_back(weight_t); in DeConvTestInit2()
581 auto *weight_t = in DeConvTestInit3() local
[all …]
Dfullconnection_fp32_tests.cc44 …auto *weight_t = new Tensor(kNumberTypeFloat, {3, 8}, mindspore::NHWC, lite::Tensor::Category::CON… in FcTestInit1() local
45 weight_t->MallocData(); in FcTestInit1()
50 memcpy(weight_t->MutableData(), weight, sizeof(float) * weight_t->ElementsNum()); in FcTestInit1()
51 inputs_->push_back(weight_t); in FcTestInit1()
115 …auto *weight_t = new Tensor(kNumberTypeFloat, {30, 80}, mindspore::NCHW, lite::Tensor::Category::C… in FcTestInit2() local
116 weight_t->MallocData(); in FcTestInit2()
119 memcpy(weight_t->MutableData(), w_data, buffer_size); in FcTestInit2()
120 inputs_->push_back(weight_t); in FcTestInit2()
182 …auto *weight_t = new Tensor(kNumberTypeFloat, {16, 20}, mindspore::NHWC, lite::Tensor::Category::C… in FcTestInit3() local
183 weight_t->MallocData(); in FcTestInit3()
[all …]
Dmatmul_fp32_tests.cc78 …auto weight_t = new lite::Tensor(kNumberTypeFloat, b_shape, mindspore::NHWC, lite::Tensor::Categor… in MMTestInit() local
79 weight_t->MallocData(); in MMTestInit()
80 memcpy(weight_t->MutableData(), b_ptr, sizeof(float) * weight_t->ElementsNum()); in MMTestInit()
81 inputs_->push_back(weight_t); in MMTestInit()
98 …auto weight_t = new lite::Tensor(kNumberTypeFloat, b_shape, mindspore::NHWC, lite::Tensor::Categor… in MMTestInit2() local
99 weight_t->MallocData(); in MMTestInit2()
100 memcpy(weight_t->MutableData(), b_ptr, sizeof(float) * weight_t->ElementsNum()); in MMTestInit2()
101 inputs_->push_back(weight_t); in MMTestInit2()
Dpower_fp32_tests.cc36 …auto weight_t = new lite::Tensor(kNumberTypeFloat, b_shape, mindspore::NHWC, lite::Tensor::Categor… in PowerTestInit() local
37 weight_t->MallocData(); in PowerTestInit()
38 memcpy(weight_t->MutableData(), b_ptr, sizeof(float) * weight_t->ElementsNum()); in PowerTestInit()
39 inputs_->push_back(weight_t); in PowerTestInit()
/third_party/boost/libs/graph/example/
Ddave.cpp77 typedef int weight_t; typedef
78 typedef property< edge_weight_t, weight_t > EProperty;
175 std::vector< weight_t > distance(N, (numeric_limits< weight_t >::max)()); in main()
/third_party/boost/libs/graph_parallel/test/
Dssca.cpp674 typedef weight_type weight_t; in run_non_distributed_graph_tests() typedef
675 weight_t unit_weight(1); in run_non_distributed_graph_tests()
686 … ref_property_map<graph_traits<seqGraph>::edge_descriptor, weight_t>(unit_weight), in run_non_distributed_graph_tests()
946 typedef weight_type weight_t; in run_distributed_graph_tests() typedef
947 weight_t unit_weight(1); in run_distributed_graph_tests()
955 … ref_property_map<typename graph_traits<Graph>::edge_descriptor, weight_t>(unit_weight), in run_distributed_graph_tests()