Skip to content

Commit aca5cfa

Browse files
deqiangctensorflow-copybara
authored andcommitted
Define an option to specify different IFRT client.
PiperOrigin-RevId: 653499497
1 parent d1870ac commit aca5cfa

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tensorflow_serving/servables/tensorflow/tfrt_saved_model_source_adapter.proto

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,16 @@ message TfrtSavedModelConfig {
211211

212212
// PjRt memory kind to use for paged-out buffers.
213213
string paging_host_memory_kind = 2023;
214+
215+
// Specific IFRT client type.
216+
enum IfrtClientType {
217+
IFRT_CLIENT_UNSPECIFIED = 0;
218+
// PJRT IFRT client.
219+
IFRT_CLIENT_PJRT = 1;
220+
// Pathways IFRT client.
221+
IFRT_CLIENT_PATHWAYS = 2;
222+
}
223+
IfrtClientType ifrt_client_type = 2024;
214224
}
215225

216226
// Config proto for TfrtSavedModelSourceAdapter.

0 commit comments

Comments
 (0)