Searched refs:LayoutAssignmentPass (Results 1 – 2 of 2) sorted by relevance
72 class LayoutAssignmentPass class73 : public LayoutAssignmentPassBase<LayoutAssignmentPass> {75 LayoutAssignmentPass() = default;76 explicit LayoutAssignmentPass(const std::string& force_data_format) { in LayoutAssignmentPass() function in mlir::TF::__anon0e7f28700111::LayoutAssignmentPass80 LayoutAssignmentPass(const LayoutAssignmentPass& pass) {} in LayoutAssignmentPass() function in mlir::TF::__anon0e7f28700111::LayoutAssignmentPass103 void LayoutAssignmentPass::runOnOperation() { in runOnOperation()479 pm.addPass(std::make_unique<LayoutAssignmentPass>(options.force_data_format)); in CreateLayoutOptimizationPipeline()498 return std::make_unique<LayoutAssignmentPass>(); in CreateLayoutAssignmentPass()
2073 def LayoutAssignmentPass : Pass<"tf-layout-assignment", "mlir::func::FuncOp"> {