Searched defs:input_layout_op (Results 1 – 1 of 1) sorted by relevance
1026 if (auto input_layout_op = in ReplaceAuxiliaryDTensorLayoutOpsWithIdentity() local1153 mlir::Operation* input_layout_op = argument.getUses().begin().getUser(); in InsertRelayoutForWhileLoops() local