Searched refs:tensor_max_size_hints_ (Results 1 – 2 of 2) sorted by relevance
370 std::vector<size_t> tensor_max_size_hints_; variable
4524 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()