| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_depthwise_fp16.cc | 28 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
38 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
62 if (op_parameter_->is_train_session_) { in Init()
|
| D | convolution_depthwise_3x3_fp16.cc | 32 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
43 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
68 if (op_parameter_->is_train_session_) { in Init()
|
| D | convolution_fp16.cc | 35 void *weight_origin = (op_parameter_->is_train_session_) ? filter_tensor->data() : origin_weight_; in PackWeight()
52 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
99 if (op_parameter_->is_train_session_) { in Init()
|
| D | convolution_depthwise_slidewindow_fp16.cc | 59 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
69 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
98 if (op_parameter_->is_train_session_) { in Init()
|
| D | deconvolution_depthwise_fp16.cc | 76 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
96 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
108 if (op_parameter_->is_train_session_) { in Init()
|
| D | convolution_winograd_fp16.cc | 44 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
83 void *weight_origin = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
150 if (op_parameter_->is_train_session_) { in Init()
|
| D | deconvolution_fp16.cc | 62 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
75 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
190 if (op_parameter_->is_train_session_) { in Init()
|
| D | convolution_1x1_fp16.cc | 89 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
118 void *weight_origin = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
150 if (op_parameter_->is_train_session_) { in Init()
|
| D | biasadd_fp16.cc | 55 if (op_parameter_->is_train_session_) { in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_depthwise_fp32.cc | 28 if (op_parameter_->is_train_session_) { in Init()
102 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
116 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | convolution_depthwise_3x3_fp32.cc | 30 if (op_parameter_->is_train_session_) { in Init()
118 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
128 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | matmul_fp32_base.cc | 48 if (op_parameter_->is_train_session_) { in InitParameter()
83 if (!op_parameter_->is_train_session_) { in InitBufferA()
107 if (op_parameter_->is_train_session_) { in InitBufferB()
236 if (!op_parameter_->is_train_session_) { in FreeResizeBufA()
258 if (!op_parameter_->is_train_session_) { in FreeResizeBufB()
386 if (op_parameter_->is_train_session_) { in ReSize()
|
| D | convolution_depthwise_slidewindow_fp32.cc | 62 if (op_parameter_->is_train_session_) { in Init()
165 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
175 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | convolution_depthwise_indirect_fp32.cc | 39 if (op_parameter_->is_train_session_) { in Init()
178 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
198 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | convolution_fp32.cc | 80 if (op_parameter_->is_train_session_) { in Init()
187 void *origin_weight = (op_parameter_->is_train_session_) ? filter_tensor->data() : origin_weight_; in PackWeight()
210 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | convolution_winograd_fp32.cc | 106 if (op_parameter_->is_train_session_) { in Init()
206 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
247 void *origin_weight = (op_parameter_->is_train_session_) ? weight_tensor->data() : origin_weight_; in PackWeight()
|
| D | convolution_slidewindow_fp32.cc | 39 if (op_parameter_->is_train_session_) { in Init()
195 void *origin_weight = (op_parameter_->is_train_session_) ? filter_tensor->data() : origin_weight_; in PackWeight()
212 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | convolution_depthwise_slidewindow_x86_fp32.cc | 66 if (op_parameter_->is_train_session_) { in Init()
181 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | convolution_1x1_fp32.cc | 119 if (op_parameter_->is_train_session_) { in Init()
284 void *origin_weight = (op_parameter_->is_train_session_) ? filter_tensor->data() : origin_weight_; in PackWeight()
303 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | deconvolution_depthwise_fp32.cc | 80 if (op_parameter_->is_train_session_) { in Init()
193 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
|
| D | deconvolution_fp32.cc | 64 if (!op_parameter_->is_train_session_) { in MallocWeightBiasData()
176 if (op_parameter_->is_train_session_) { in Init()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | convolution_base.cc | 51 } else if (!op_parameter_->is_train_session_) { in ~ConvolutionBaseCPUKernel()
126 if (op_parameter_->is_train_session_) { in InitConvWeightBias()
146 if (!op_parameter_->is_train_session_) { in InitConvWeightBias()
163 if (IsRepack() || (op_parameter_->is_train_session_)) { in RepackWeight()
164 if (op_parameter_->is_train_session_) { in RepackWeight()
|
| D | reshape_base.cc | 53 op_parameter_->is_train_session_) { in Run()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | scheduler.h | 53 is_train_session_(is_train_session), in context_()
147 bool is_train_session_ = false; variable
|
| D | scheduler.cc | 194 if (is_train_session_ || kernel->type() == schema::PrimitiveType_Custom || in HandleBuildinCpuKernelWeight()
215 if (is_train_session_) { in InitKernels()
892 if (is_train_session_) { in FindCpuKernel()
905 if (is_train_session_) { in FindCpuKernel()
1039 op_parameter->is_train_session_ = is_train_session_; in FindBackendKernel()
1068 ((is_train_session_ == false) || (sched_cb_ && sched_cb_->SchedFp16Kernel(node)))) { in FindBackendKernel()
|