| 173 | } |
| 174 | |
| 175 | const Eigen::ThreadPoolDevice* GetThreadPoolDevice(TfLiteContext* context) { |
| 176 | auto* ptr = GetEigenContext(context); |
| 177 | if (ptr == nullptr) { |
| 178 | TF_LITE_FATAL( |
| 179 | "Call to GetFromContext() not preceded by IncrementUsageCounter()"); |
| 180 | } |
| 181 | return ptr->thread_pool_holder->GetThreadPoolDevice(); |
| 182 | } |
| 183 | |
| 184 | } // namespace eigen_support |
| 185 | } // namespace tflite |