Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc128 llvm::SmallVector<Device, 8> host_tpu_devices; in GetTPUDevices() local
140 auto host_tpu_devices = lookup(device); in GetTPUDevices() local
147 auto host_tpu_devices = lookup(device_spec); in GetTPUDevices() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_helpers.cc176 std::vector<Device*> host_tpu_devices; in GetTPUDevices() local