Searched refs:TensorDevicePlacer (Results 1 – 1 of 1) sorted by relevance
412 class TensorDevicePlacer { class418 TensorDevicePlacer(int64 num_devices, const DataTypeVector& types, in TensorDevicePlacer() function in tensorflow::__anond2c82b040111::TensorDevicePlacer1968 TensorDevicePlacer args_device_selector(num_cores_per_replica, arg_types, in AssignArgsAndRetvalsToCores()2115 TensorDevicePlacer retvals_device_selector(num_cores_per_replica, in AssignArgsAndRetvalsToCores()