Searched defs:layout_or_status (Results 1 – 6 of 6) sorted by relevance
63 auto layout_or_status = Layout::FromString(layout_str); in PropagateDTensorLayoutForRelayout() local96 auto layout_or_status = Layout::FromString(layout_str); in PropagateFunctionArgAttrToLayoutOp() local175 auto layout_or_status = ExtractLayoutFromOp(op); in runOnOperation() local
160 const auto layout_or_status = ExtractSingleLayoutFromOp(first_reduction_op); in OptimizeAllReduceAndSum() local393 const auto layout_or_status = ExtractSingleLayoutFromOp(all_reduce); in ExtractAllReduceFromWhileOp() local
63 auto layout_or_status = ExtractLayoutFromOperand(block_arg); in ExtractMeshFromBlockArgument() local408 auto layout_or_status = in AnnotateFunctionReturnValuesWithMeshInformation() local
79 auto layout_or_status = ExtractLayoutFromOperand( in AttachRetvalLayouts() local
197 auto layout_or_status = Layout::FromString(layout_attr); in LowerToSendRecv() local
144 auto layout_or_status = ExtractLayoutFromOperand(resource_arg); in UpdateResourceArgumentType() local