Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dimplementation_selector_test.cc41 constexpr char TpuDevice[] = "/device:TPU_REPLICATED_CORE"; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.cc41 TpuDevice::TpuDevice(int id, int process_index, in TpuDevice() function in xla::TpuDevice