Searched refs:SimpleTFDeviceAssignmentPass (Results 1 – 2 of 2) sorted by relevance
30 class SimpleTFDeviceAssignmentPass class31 : public SimpleTFDeviceAssignmentPassBase<SimpleTFDeviceAssignmentPass> {33 SimpleTFDeviceAssignmentPass() = default;34 SimpleTFDeviceAssignmentPass(const SimpleTFDeviceAssignmentPass&) {} in SimpleTFDeviceAssignmentPass() function in mlir::TF::__anon0aeee71c0111::SimpleTFDeviceAssignmentPass35 explicit SimpleTFDeviceAssignmentPass(llvm::StringRef default_device) { in SimpleTFDeviceAssignmentPass() function in mlir::TF::__anon0aeee71c0111::SimpleTFDeviceAssignmentPass99 return std::make_unique<SimpleTFDeviceAssignmentPass>(default_device); in CreateSimpleTFDeviceAssignmentPass()
2039 def SimpleTFDeviceAssignmentPass : Pass<"tf-simple-device-assignment", "mlir::func::FuncOp"> {