Searched defs:layout_op (Results 1 – 12 of 12) sorted by relevance
48 mlir::TF::DTensorLayout layout_op = in runOnOperation() local63 for (auto& layout_op : unique_layout_ops) { in runOnOperation() local
43 if (auto layout_op = in ExtractGlobalInputShape() local80 if (auto layout_op = mlir::dyn_cast<mlir::TF::DTensorLayout>(user)) { in ExtractGlobalOutputShape() local250 auto layout_op = mlir::cast<mlir::TF::DTensorLayout>(value.getDefiningOp()); in GetGlobalShapeOfValueFromDTensorLayout() local254 auto layout_op = in GetGlobalShapeOfValueFromDTensorLayout() local
51 auto layout_op = builder->create<mlir::TF::DTensorLayout>( in CreateDTensorLayoutOp() local193 auto layout_op = builder.create<mlir::TF::DTensorLayout>( in runOnOperation() local
56 if (auto layout_op = llvm::dyn_cast<mlir::TF::DTensorLayout>(op)) { in WrapDeviceCluster() local
80 if (auto layout_op = mlir::dyn_cast<mlir::TF::DTensorLayout>( in RemoveUnusedSparseToDenseOps() local
329 if (auto layout_op = llvm::dyn_cast<mlir::TF::DTensorLayout>(op)) in ConductSPMDExpansion() local362 for (auto layout_op : layout_ops) RemoveDTensorLayoutOp(layout_op); in RemoveDTensorLayoutOps() local
167 if (auto layout_op = llvm::dyn_cast<mlir::TF::DTensorLayout>(op)) { in ExtractLayoutFromOperand() local
49 if (mlir::TF::DTensorLayout layout_op = in GetForwardedInput() local
99 auto layout_op = builder.create<mlir::TF::DTensorLayout>( in CloneOpToCluster() local
208 if (auto layout_op = in InferMeshFromInputs() local
322 auto layout_op = in GetForwardedDTensorLayoutInput() local
699 auto layout_op = builder.create<mlir::TF::DTensorLayout>( in InsertDTensorLayoutOps() local1024 for (auto layout_op : llvm::reverse(layout_ops)) { in ReplaceAuxiliaryDTensorLayoutOpsWithIdentity() local