Home
last modified time | relevance | path

Searched refs:TFTPU_SET_FN (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/
Dtpu_executor_init_fns.inc6 TFTPU_SET_FN(executor_fn, TpuPlatform_New);
7 TFTPU_SET_FN(executor_fn, TpuPlatform_Free);
8 TFTPU_SET_FN(executor_fn, TpuPlatform_Initialize);
9 TFTPU_SET_FN(executor_fn, TpuPlatform_Initialized);
10 TFTPU_SET_FN(executor_fn, TpuPlatform_GetExecutor);
11 TFTPU_SET_FN(executor_fn, TpuPlatform_Id);
12 TFTPU_SET_FN(executor_fn, TpuPlatform_VisibleDeviceCount);
13 TFTPU_SET_FN(executor_fn, TpuPlatform_TpuMemoryLimit);
14 TFTPU_SET_FN(executor_fn, TpuPlatform_ShouldRegisterTpuDeviceToDeviceCopy);
15 TFTPU_SET_FN(executor_fn, TpuPlatform_GetTopologyPtr);
[all …]
Dtpu_library_init_fns.inc14 TFTPU_SET_FN(ops_api_fn, ConfigureDistributedTpuOp_DoWork);
15 TFTPU_SET_FN(ops_api_fn, WaitForDistributedTpuOp_DoWork);
16 TFTPU_SET_FN(ops_api_fn, InitializeHostForDistributedTpuOp_DoWork);
17 TFTPU_SET_FN(ops_api_fn, SetGlobalTPUArrayOp_DoWork);
18 TFTPU_SET_FN(ops_api_fn, DisconnectDistributedTpuChipsOp_DoWork);
19 TFTPU_SET_FN(ops_api_fn, TpuConfigurationApi_FreeCharArray);
20 TFTPU_SET_FN(ops_api_fn, TpuConfigurationApi_FreeInt32Array);
21 TFTPU_SET_FN(ops_api_fn, TpuConfigurationApi_HasTPUPodState);
22 TFTPU_SET_FN(ops_api_fn, TpuConfigurationApi_TpusPerHost);
23 TFTPU_SET_FN(ops_api_fn, TpuConfigurationApi_TpuMemoryLimit);
[all …]
Dtpu_api_dlsym_set_fn.h19 #define TFTPU_SET_FN(Struct, FnName) \ macro