Searched refs:layout_sensitive_interface (Results 1 – 2 of 2) sorted by relevance
45 auto layout_sensitive_interface = cast<LayoutSensitiveInterface>(op); in VerifyLayoutSensitiveInterface() local46 return VerifyLayoutDependentArgsAndResults(op, layout_sensitive_interface); in VerifyLayoutSensitiveInterface()
135 func.walk([&](LayoutSensitiveInterface layout_sensitive_interface) { in runOnFunction() argument139 target_data_format = layout_sensitive_interface.GetOptimalLayout(devices); in runOnFunction()143 auto data_format = layout_sensitive_interface.data_format(); in runOnFunction()156 mlir::Operation* op = layout_sensitive_interface.getOperation(); in runOnFunction()166 if (failed(layout_sensitive_interface.UpdateDataFormat(target_data_format))) in runOnFunction()173 for (int64_t arg : layout_sensitive_interface.GetLayoutDependentArgs()) { in runOnFunction()182 for (int64_t res : layout_sensitive_interface.GetLayoutDependentResults()) { in runOnFunction()