Home
last modified time | relevance | path

Searched defs:tpu_system (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_tpu_kernels.cc82 TpuSystemInterface* tpu_system = GetPreferredTpuSystem(); in Compute() local
247 TpuSystemInterface* tpu_system = GetPreferredTpuSystem(); in Compute() local
H A Dtpu_system_interface.cc27 void SetPreferredTpuSystem(TpuSystemInterface* tpu_system) { in SetPreferredTpuSystem()
H A Ddtensor_device.cc185 TpuSystemInterface* tpu_system = GetPreferredTpuSystem(); in TPUCoreIDsToLocations() local
219 TpuSystemInterface* tpu_system = GetPreferredTpuSystem(); in TPUCoreLocationsToIDs() local