Home
last modified time | relevance | path

Searched refs:on_value_ (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Done_hot_base.cc121 …auto ret = OneHotToFp32(indices_data, on_value_, off_value_, reinterpret_cast<float *>(output_data… in OneHotImpl()
126 …auto ret = OneHotToFp16(indices_data, on_value_, off_value_, reinterpret_cast<float16_t *>(output_… in OneHotImpl()
189 this->on_value_ = *on_value; in InitOnOffValueForFourInputs()
196 this->on_value_ = *on_value; in InitOnOffValueForFourInputs()
243 this->on_value_ = off_on_values[1]; in InitOnOffValueForThreeInputs()
251 this->on_value_ = off_on_values[1]; in InitOnOffValueForThreeInputs()
Done_hot_base.h47 float16_t on_value_ = 0.; variable
50 float on_value_ = 0.; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Done_hot.cc83 on_value_ = static_cast<float *>(in_tensors_[2]->data())[1]; in InitWeights()
87 on_value_ = static_cast<float *>(in_tensors_[2]->data())[0]; in InitWeights()
92 MS_ASSERT(on_value_); in InitWeights()
113 if (ocl_runtime_->SetKernelArg(kernel_, arg_idx++, on_value_) != CL_SUCCESS) { in SetConstArgs()
Done_hot.h41 float on_value_{1.0f};