Searched defs:relayout (Results 1 – 5 of 5) 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
403 CJS_Result CJX_LayoutPseudoModel::relayout( in relayout() function in CJX_LayoutPseudoModel
5670 auto relayout = [&](size_t lines, bool ellipsis, in UNIX_ONLY_TEST() local