| 172 | } |
| 173 | |
| 174 | Status Initialize() { |
| 175 | TF_RETURN_IF_ERROR(immutable_state_.Initialize()); |
| 176 | kernel_stats_.Initialize(immutable_state_.graph_view(), |
| 177 | immutable_state_.graph()); |
| 178 | if (immutable_state_.params().run_cost_model_executor) { |
| 179 | immutable_state_.InitializeScheduleInfo(&kernel_stats_); |
| 180 | } |
| 181 | return Status::OK(); |
| 182 | } |
| 183 | |
| 184 | void RunAsync(const Args& args, DoneCallback done) override; |
| 185 |
no test coverage detected