Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_kernel.h344 std::unique_ptr<NNMemory> nn_output_memory_; variable
Dnnapi_delegate.cc3900 if (total_output_byte_size > nn_output_memory_->get_byte_size()) { in Invoke()
3901 nn_output_memory_.reset( in Invoke()
4048 nn_output_memory_->get_handle(), output_offset, tensor->bytes), in Invoke()
4104 nn_output_memory_->get_data_ptr() + output_offset); in Invoke()
4111 memcpy(tensor->data.raw, nn_output_memory_->get_data_ptr() + output_offset, in Invoke()
4768 nn_output_memory_.reset( in BuildGraph()