Searched refs:CreateCopyWithNewLayout (Results 1 – 2 of 2) sorted by relevance
463 StatusOr<HloInstruction*> CreateCopyWithNewLayout(
772 StatusOr<HloInstruction*> LayoutAssignment::CreateCopyWithNewLayout( in CreateCopyWithNewLayout() function in xla::LayoutAssignment799 CreateCopyWithNewLayout(target_shape, gte)); in CreateCopyWithNewLayout()848 CreateCopyWithNewLayout(operand_layout.shape(), operand)); in CopyOperandIfLayoutsDiffer()1801 CreateCopyWithNewLayout(constraints.ResultLayout()->shape(), in RunOnComputation()