Searched refs:ops_api_fn (Results 1 – 2 of 2) sorted by relevance
12 auto* ops_api_fn = const_cast<TfTpu_OpsApiFn*>(tensorflow::tpu::OpsApiFn());14 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);[all …]
27 static TfTpu_OpsApiFn ops_api_fn; in OpsApiFn() local28 return &ops_api_fn; in OpsApiFn()