Home
last modified time | relevance | path

Searched refs:is_regulated_ (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dembedding_lookup_fp32.cc80 …param_->is_regulated_ = reinterpret_cast<bool *>(ms_context_->allocator->Malloc(sizeof(bool) * par… in Run()
81 if (input_addr_ == nullptr || param_->is_regulated_ == nullptr) { in Run()
87 param_->is_regulated_[i] = param_->max_norm_ == 0; in Run()
111 ms_context_->allocator->Free(param_->is_regulated_); in FreeRunBuff()
112 param_->is_regulated_ = nullptr; in FreeRunBuff()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dembedding_lookup_fp32.c41 if (!parameter->is_regulated_[ids[num]]) { in CopyData()
43 parameter->is_regulated_[ids[num]] = true; in CopyData()
Dembedding_lookup_fp32.h28 bool *is_regulated_; member