| 518 | } |
| 519 | |
| 520 | TRITONTF_Error* |
| 521 | ModelImpl::MakeCallable(const tensorflow::CallableOptions& opts) |
| 522 | { |
| 523 | if (has_callable_) { |
| 524 | session_->ReleaseCallable(callable_).IgnoreError(); |
| 525 | has_callable_ = false; |
| 526 | } |
| 527 | |
| 528 | RETURN_IF_TF_ERROR(session_->MakeCallable(opts, &callable_)); |
| 529 | for (int idx = 0; idx < opts.fetch_size(); idx++) { |
| 530 | output_index_map_[opts.fetch(idx)] = idx; |
| 531 | } |
| 532 | has_callable_ = true; |
| 533 | return nullptr; |
| 534 | } |
| 535 | |
| 536 | TRITONTF_Error* |
| 537 | ModelImpl::Run( |
no test coverage detected