Home
last modified time | relevance | path

Searched refs:out_buffer_ (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dstack_fp16.cc55 out_buffer_ = nullptr; in MallocAssignBuffer()
56out_buffer_ = MallocOutputFp16(out_tensors_.at(0), static_cast<const lite::InnerContext *>(this->m… in MallocAssignBuffer()
57 if (out_buffer_ == nullptr) { in MallocAssignBuffer()
70 if (malloc_out_ && out_buffer_ != nullptr) { in FreeBuffer()
71 ms_context_->allocator->Free(out_buffer_); in FreeBuffer()
72 out_buffer_ = nullptr; in FreeBuffer()
88 void *output_data = reinterpret_cast<void *>(out_buffer_); in DoExecute()
127 …Float16ToFloat32(out_buffer_, reinterpret_cast<float *>(out_tensor->data()), out_tensor->ElementsN… in Run()
Dstack_fp16.h42 float16_t *out_buffer_ = nullptr; variable