Home
last modified time | relevance | path

Searched refs:SE_Platform (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor_c_api.h28 SE_Platform* TpuPlatform_New();
29 void TpuPlatform_Free(SE_Platform* platform);
30 void TpuPlatform_Initialize(SE_Platform* platform, size_t options_size,
33 bool TpuPlatform_Initialized(SE_Platform* platform);
34 SE_StreamExecutor* TpuPlatform_GetExecutor(SE_Platform* platform,
37 SE_PlatformId TpuPlatform_Id(SE_Platform* platform);
38 int64_t TpuPlatform_VisibleDeviceCount(SE_Platform* platform);
39 int64_t TpuPlatform_TpuMemoryLimit(SE_Platform* platform);
40 bool TpuPlatform_ShouldRegisterTpuDeviceToDeviceCopy(SE_Platform* platform);
41 SE_TpuTopology* TpuPlatform_GetTopologyPtr(SE_Platform* platform);
[all …]
Dc_api_decl.h53 typedef struct SE_Platform SE_Platform; typedef
111 SE_Platform* platform;
Dtpu_platform.h132 SE_Platform* se_platform() const { return platform_; } in se_platform()
143 mutable SE_Platform* platform_;
Dc_api_defn.h34 struct SE_Platform { struct