Searched defs:relayout (Results 1 – 6 of 6) sorted by relevance
61 mlir::MLIRContext& c, mlir::TF::RelayoutOp relayout) { in PropagateDTensorLayoutForRelayout()169 if (auto relayout = llvm::dyn_cast<mlir::TF::RelayoutOp>(op)) { in runOnOperation() local
147 mlir::TF::RelayoutOp relayout = llvm::cast<mlir::TF::RelayoutOp>(op); in ComputeRelayoutLayout() local175 mlir::TF::RelayoutOp relayout = mlir::cast<mlir::TF::RelayoutOp>(op); in ExpandOp() local
373 def relayout(tensor: ops.Tensor, layout: layout_lib.Layout) -> ops.Tensor: function
399 CJS_Result CJX_LayoutPseudoModel::relayout( in relayout() function in CJX_LayoutPseudoModel
486 protected int relayout(Object[] args) { in relayout() method in ShadowWindowManagerGlobal.WindowSessionDelegate
5980 auto relayout = [&](size_t lines, bool ellipsis, in UNIX_ONLY_TEST() local