experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Register HostCount and ChipsPerHost API to executor_fn.

Browse Source 
PiperOrigin-RevId: 336166509
Change-Id: Ib70ad53a95ba52ef0c9ed75d9c279405b9253d0f
This commit is contained in:
A. Unique TensorFlower 2020-10-08 14:33:04 -07:00 committed by TensorFlower Gardener
parent 913e0999ec
commit 3eed7f3438
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/tpu/tpu_executor_init_fns.inc
View File

@ -112,6 +112,8 @@ tensorflow::Status SetExecutorStructFn(void* library_handle) {
TFTPU_SET_FN(executor_fn, TpuTopology_LogicalDevicesPerHost);
TFTPU_SET_FN(executor_fn, TpuTopology_LogicalDevicesPerChip);
TFTPU_SET_FN(executor_fn, TpuTopology_HostCount);
TFTPU_SET_FN(executor_fn, TpuTopology_ChipsPerHost);
TFTPU_SET_FN(executor_fn, TpuTopology_ChipBounds_X);
TFTPU_SET_FN(executor_fn, TpuTopology_ChipBounds_Y);
TFTPU_SET_FN(executor_fn, TpuTopology_ChipBounds_Z);