Searched refs:CreateCopyWithNewLayout (Results 1 – 2 of 2) sorted by relevance
471 StatusOr<HloInstruction*> CreateCopyWithNewLayout(
799 StatusOr<HloInstruction*> LayoutAssignment::CreateCopyWithNewLayout( in CreateCopyWithNewLayout() function in xla::LayoutAssignment827 CreateCopyWithNewLayout(target_shape, gte)); in CreateCopyWithNewLayout()885 CreateCopyWithNewLayout(operand_layout.shape(), param)); in CopyOperandIfLayoutsDiffer()901 CreateCopyWithNewLayout(operand_layout.shape(), operand)); in CopyOperandIfLayoutsDiffer()1948 CreateCopyWithNewLayout(constraints.ResultLayout()->shape(), in RunOnComputation()