Searched refs:THIRD_INPUT (Results 1 – 18 of 18) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | fused_batchnorm_fp32.cc | 48 auto offset = in_tensors_.at(THIRD_INPUT); in InitConstTensor()
80 float *offset = static_cast<float *>(in_tensors_.at(THIRD_INPUT)->data()); in Run()
96 CHECK_NULL_RETURN(out_tensors_.at(THIRD_INPUT)->data()); in Run()
100 memcpy(out_tensors_.at(THIRD_INPUT)->data(), offset, out_tensors_.at(THIRD_INPUT)->Size()); in Run()
106 memcpy(offset_, offset, in_tensors_.at(THIRD_INPUT)->Size()); in Run()
123 float *bias = static_cast<float *>(in_tensors_.at(THIRD_INPUT)->data()); in Eval()
131 memcpy(offset_, bias, in_tensors_.at(THIRD_INPUT)->Size()); in Eval()
|
| D | space_to_batch_fp32.cc | 43 auto padding_data = in_tensors_.at(THIRD_INPUT)->data(); in ProcessInput()
46 for (int i = 0; i < in_tensors_.at(THIRD_INPUT)->ElementsNum(); i++) { in ProcessInput()
69 in_tensors_.at(THIRD_INPUT) != nullptr && in_tensors_.at(THIRD_INPUT)->IsConst()) { in ReSize()
98 if (!in_tensors_.at(SECOND_INPUT)->IsConst() || !in_tensors_.at(THIRD_INPUT)->IsConst()) { in Run()
|
| D | batchnorm_fp32.cc | 83 variance_ = malloc(in_tensors_.at(THIRD_INPUT)->Size()); in InitConstTensor()
90 auto in_tensor_var_data = in_tensors_.at(THIRD_INPUT)->MutableData(); in InitConstTensor()
96 memcpy(variance_, in_tensor_var_data, in_tensors_.at(THIRD_INPUT)->Size()); in InitConstTensor()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | pooling_grad.cc | 44 CHECK_NULL_RETURN(in_tensors_.at(THIRD_INPUT)); in ReSize()
52 out_shape = in_tensors_.at(THIRD_INPUT)->shape(); in ReSize()
91 auto dy_ptr = reinterpret_cast<float *>(in_tensors_.at(THIRD_INPUT)->data()); in Execute()
95 input_ptr = reinterpret_cast<float *>(in_tensors_.at(THIRD_INPUT)->data()); in Execute()
|
| D | dropout_grad.cc | 34 CHECK_LESS_RETURN(in_tensors_.size(), THIRD_INPUT); in Init()
|
| D | arithmetic_self_grad.cc | 66 CHECK_LESS_RETURN(in_tensors_.size(), THIRD_INPUT); in DoArithmeticSelfGrad()
|
| D | smooth_l1_loss.cc | 86 CHECK_LESS_RETURN(in_tensors_.size(), THIRD_INPUT); in Init()
|
| D | arithmetic_grad.cc | 35 CHECK_NULL_RETURN(in_tensors_.at(THIRD_INPUT)); in Init()
36 CHECK_LESS_RETURN(out_tensors_.size(), THIRD_INPUT); in Init()
|
| D | convolution_grad_filter.cc | 38 CHECK_LESS_RETURN(in_tensors_.size(), THIRD_INPUT); in ReSize()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_batch_to_space_parser.cc | 45 …if (TransTfliteDataToVec2D(tflite_op->inputs.at(THIRD_INPUT), tflite_subgraph->tensors, tflite_mod… in Parse()
|
| D | tflite_range_parser.cc | 45 …ret = GetTfliteData(tflite_op->inputs.at(THIRD_INPUT), tflite_subgraph->tensors, tflite_model->buf… in Parse()
|
| D | tflite_pad_parser.cc | 66 …auto ret = GetTfliteData(tflite_op->inputs.at(THIRD_INPUT), tflite_subgraph->tensors, tflite_model… in Parse()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | dropout_fp16_grad.cc | 54 CHECK_LESS_RETURN(in_tensors_.size(), THIRD_INPUT); in Init()
|
| D | arithmetic_fp16_grad.cc | 35 CHECK_LESS_RETURN(out_tensors_.size(), THIRD_INPUT); in Init()
|
| D | convolution_fp16_grad_filter.cc | 37 CHECK_LESS_RETURN(in_tensors_.size(), THIRD_INPUT); in ReSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | op_base.h | 76 #define THIRD_INPUT 2 macro
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | norm_fusion.cc | 389 if (prim_type == schema::PrimitiveType_Reshape && i == THIRD_INPUT && in ShapeSizeInfer()
399 if (prim_type == schema::PrimitiveType_Reshape && i == THIRD_INPUT) { in ShapeSizeInfer()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | full_quant_quantizer.cc | 629 quant_param_holder->set_input_quant_param(THIRD_INPUT, quant_params); in DoBiasQuant()
801 auto index_node = cnode->input(THIRD_INPUT); in QuantNode()
1115 auto bias_quant_params = input_quant_params.at(THIRD_INPUT); in BiasCorrection()
|