Home
last modified time | relevance | path

Searched refs:TPUDeviceAssignment (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.h61 struct TPUDeviceAssignment { struct
62 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() argument
67 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() argument
234 StatusOr<TPUDeviceAssignment> GetTPUCompilationAndExecutionDevices(
Dtpu_rewrite_device_util.cc449 StatusOr<TPUDeviceAssignment> GetTPUCompilationAndExecutionDevices( in GetTPUCompilationAndExecutionDevices()
472 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()
480 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()