Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_kernel.h370 std::vector<size_t> tensor_max_size_hints_; variable
H A Dnnapi_delegate.cc4524 tensor_max_size_hints_.resize(context->tensors_size, 0); in Init()
4532 tensor_max_size_hints_[tensor_index] = max_size_hint; in Init()
4883 if (tensor_max_size_hints_[i] == 0) { in Invoke()
4891 tensor_size = std::max(tensor_size, tensor_max_size_hints_[i]); in Invoke()
5079 std::max(tensor->bytes, tensor_max_size_hints_[output_index]); in Invoke()
5188 std::max(tensor->bytes, tensor_max_size_hints_[output_index]); in Invoke()
6171 std::max(context->tensors[i].bytes, tensor_max_size_hints_[i]); in BuildGraph()
6195 std::max(context->tensors[i].bytes, tensor_max_size_hints_[i]); in BuildGraph()