Home
last modified time | relevance | path

Searched defs:core_location (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_topology.h36 explicit TpuCoreLocationExternal(SE_TpuTopology_Core* core_location) in TpuCoreLocationExternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc1694 tpu::TpuCoreLocationExternal core_location = in ParseDeviceAssignmentAttr() local
1750 const tpu::TpuCoreLocationExternal& core_location = in BuildGeneralDeviceAssignment() local