Searched refs:CreateCopyWithNewLayout (Results 1 – 2 of 2) sorted by relevance
989 StatusOr<HloInstruction*> LayoutAssignment::CreateCopyWithNewLayout( in CreateCopyWithNewLayout() function in xla::LayoutAssignment1017 CreateCopyWithNewLayout(target_shape, gte)); in CreateCopyWithNewLayout()1076 CreateCopyWithNewLayout(operand_layout.shape(), param)); in CopyOperandIfLayoutsDiffer()1100 CreateCopyWithNewLayout(operand_layout.shape(), operand)); in CopyOperandIfLayoutsDiffer()2049 CreateCopyWithNewLayout(constraints.ResultLayout()->shape(), in AssignLayouts()
575 StatusOr<HloInstruction*> CreateCopyWithNewLayout(