Home
last modified time | relevance | path

Searched refs:SE_TpuTopology (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor_c_api.h41 SE_TpuTopology* TpuPlatform_GetTopologyPtr(SE_Platform* platform);
243 int TpuTopology_LogicalDevicesPerHost(SE_TpuTopology* tpu_topology,
245 int TpuTopology_LogicalDevicesPerChip(SE_TpuTopology* tpu_topology,
247 int TpuTopology_HostCount(SE_TpuTopology* tpu_topology);
248 int TpuTopology_ChipsPerHost(SE_TpuTopology* tpu_topology);
250 int TpuTopology_ChipBounds_X(SE_TpuTopology* tpu_topology);
251 int TpuTopology_ChipBounds_Y(SE_TpuTopology* tpu_topology);
252 int TpuTopology_ChipBounds_Z(SE_TpuTopology* tpu_topology);
253 bool TpuTopology_HasChip(SE_TpuTopology* tpu_topology, int x, int y, int z);
254 SE_TpuTopology_Core* TpuTopology_CoreForId(SE_TpuTopology* tpu_topology,
[all …]
Dtpu_topology.h70 explicit TpuTopologyExternal(SE_TpuTopology* topology) in TpuTopologyExternal()
86 SE_TpuTopology* topology_;
Dc_api_decl.h313 typedef struct SE_TpuTopology SE_TpuTopology; typedef
Dtpu_platform_interface.h29 typedef SE_TpuTopology* TpuTopologyPtr;
Dc_api_defn.h69 struct SE_TpuTopology {}; struct