Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_topology.h36 explicit TpuCoreLocationExternal(SE_TpuTopology_Core* core_location) in TpuCoreLocationExternal()
43 SE_TpuTopology_Core* impl() const { return core_location_; } in impl()
46 SE_TpuTopology_Core* core_location_;
Dtpu_topology.cc53 std::vector<SE_TpuTopology_Core*> core_ptrs(num_cores); in Cores()
58 for (SE_TpuTopology_Core* ptr : core_ptrs) { in Cores()
111 std::vector<SE_TpuTopology_Core*> core_ptrs(num_cores); in cores()
116 for (SE_TpuTopology_Core* ptr : core_ptrs) { in cores()
Dtpu_executor_c_api.h68 SE_TpuTopology_Core* TpuExecutor_GetCoreLocation(SE_StreamExecutor* executor);
254 SE_TpuTopology_Core* TpuTopology_CoreForId(SE_TpuTopology* tpu_topology,
257 SE_TpuTopology_Core* TpuTopology_Core(SE_TpuTopology* tpu_topology,
265 SE_TpuTopology_Core** cores);
268 void TpuCoreLocation_ChipCoordinates(SE_TpuTopology_Core* tpu_core_location,
270 void TpuCoreLocation_HostCoordinates(SE_TpuTopology_Core* tpu_core_location,
272 int TpuCoreLocation_Index(SE_TpuTopology_Core* tpu_core_location);
273 int TpuCoreLocation_Id(SE_TpuTopology_Core* tpu_core_location);
281 SE_TpuTopology_Core** cores);
Dc_api_decl.h314 typedef struct SE_TpuTopology_Core SE_TpuTopology_Core; typedef
315 typedef struct SE_TpuTopology_Core SE_TpuTopology_Host;
Dc_api_defn.h70 struct SE_TpuTopology_Core {}; struct