Searched refs:custom_allocations_ (Results 1 – 2 of 2) sorted by relevance
981 for (int i = 0; i < custom_allocations_.size(); ++i) { in PrepareOpsAndTensors()982 auto index_and_alloc = custom_allocations_[i]; in PrepareOpsAndTensors()1605 custom_allocations_.begin(), custom_allocations_.end(), in SetCustomAllocationForTensor()1610 if (alloc_it == custom_allocations_.end()) { in SetCustomAllocationForTensor()1611 custom_allocations_.emplace_back(tensor_index, allocation); in SetCustomAllocationForTensor()
713 std::vector<std::pair<int, TfLiteCustomAllocation>> custom_allocations_; variable