Add resource location hints to base RunInference Implementation
Open
core
ml
P2
python
run-inference
sub-task
- Dominant language
- Java
- Stars
- 8.7k
- Forks
- 4.7k
- Avg merge
- 1d 20h
- Merged PRs (30d)
- 196
Description
Our generic version of RunInference should add resource location hints, similar to what TFX-BSL does here.
[https://github.com/tensorflow/tfx-bsl/blob/182918aaefd5287e7669d87bfc818155470315aa/tfx_bsl/beam/run_inference.py#L117](https://github.com/tensorflow/tfx-bsl/blob/182918aaefd5287e7669d87bfc818155470315aa/tfx_bsl/beam/run_inference.py#L117)
There's no reason that these hints shouldn't be just as relevant to sk-learn or pytorch large models hosted on GC or wherever these hints are supported.
Imported from Jira [BEAM-14218](https://issues.apache.org/jira/browse/BEAM-14218). Original Jira may contain additional context.
Reported by: Ryan.Thompson.
Subtask of issue #21435
Contributor guide
Assessment
This issue has not been assessed yet.