Searched refs:on_value_ (Results 1 – 4 of 4) sorted by relevance
121 …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()
47 float16_t on_value_ = 0.; variable50 float on_value_ = 0.; variable
83 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()
41 float on_value_{1.0f};