Home
last modified time | relevance | path

Searched refs:inputs_addr (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.cc41 … const std::vector<uint8_t *> &inputs_addr) { in ObtainKernelInputs() argument
43 if (inputs_list.size() != inputs_addr.size()) { in ObtainKernelInputs()
45 << ", input_addr_ size: " << inputs_addr.size(); in ObtainKernelInputs()
51 input->addr = inputs_addr[i]; in ObtainKernelInputs()
77 void LaunchKernel::LaunchSingleKernel(const std::vector<uint8_t *> &inputs_addr) { in LaunchSingleKernel() argument
80 auto kernel_inputs = ObtainKernelInputs(kernel_mod_->GetInputSizeList(), inputs_addr); in LaunchSingleKernel()
Dlaunch_kernel.h31 void LaunchSingleKernel(const std::vector<uint8_t *> &inputs_addr);
52 const std::vector<uint8_t *> &inputs_addr);
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconcat_fp32.cc47 std::vector<void *> inputs_addr(input_num, nullptr); in DoConcat() local
52 inputs_addr[i] = in_tensors_[i]->data(); in DoConcat()
53 CHECK_NULL_RETURN(inputs_addr[i]); in DoConcat()
65 …Concat(inputs_addr.data(), input_num, concat_param_->axis_, inputs_output_shape.data(), output_sha… in DoConcat()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dctcloss_cpu_kernel.cc40 void InnerSoftMax(const T *inputs_addr, std::vector<std::vector<T>> *softmax_probs, const uint32_t … in InnerSoftMax() argument
47 if (inputs_addr[t * batch_size * num_class + b * num_class + c] > maxCoeff) { in InnerSoftMax()
48 maxCoeff = inputs_addr[t * batch_size * num_class + b * num_class + c]; in InnerSoftMax()
53 …sumCoeff += static_cast<T>(exp(inputs_addr[t * batch_size * num_class + b * num_class + c] - maxCo… in InnerSoftMax()
55 static_cast<T>(exp(inputs_addr[t * batch_size * num_class + b * num_class + c] - maxCoeff)); in InnerSoftMax()
260 const auto *inputs_addr = reinterpret_cast<T *>(inputs[0]->addr); in LaunchKernel() local
319 InnerSoftMax(inputs_addr, &y_b, sequence_length_addr[b], num_class_, batch_size_, b); in LaunchKernel()
Dpad_cpu_kernel.cc83 const auto *inputs_addr = reinterpret_cast<T *>(inputs[0]->addr); in LaunchKernel() local
116 …inputs_addr[(equiv_block_num * old_height + padded_h - pad_top) * old_width + padded_w - pad_left]; in LaunchKernel()
Dmirror_pad_cpu_kernel.cc123 auto inputs_addr = reinterpret_cast<T *>(inputs[0]->addr); in LaunchKernel() local
198 outputs_addr[pos] = inputs_addr[pos_index]; in LaunchKernel()
Dmirror_pad_grad_cpu_kernel.cc284 auto *inputs_addr = reinterpret_cast<T *>(inputs[0]->addr); in LaunchKernel() local
289 …MirrorPadGradBatchChannel(workspace_size_, inputs_addr, interim, output_shape_[0], output_shape_[1… in LaunchKernel()