Searched defs:TensorDevicePlacer (Results 1 – 1 of 1) sorted by relevance
425 class TensorDevicePlacer { class431 TensorDevicePlacer(int64_t num_devices, const DataTypeVector& types, in TensorDevicePlacer() function in tensorflow::__anonf6cf56690111::TensorDevicePlacer