Searched refs:CreateCopyWithNewLayout (Results 1 – 2 of 2) sorted by relevance
886 StatusOr<HloInstruction*> LayoutAssignment::CreateCopyWithNewLayout( in CreateCopyWithNewLayout() function in xla::LayoutAssignment914 CreateCopyWithNewLayout(target_shape, gte)); in CreateCopyWithNewLayout()973 CreateCopyWithNewLayout(operand_layout.shape(), param)); in CopyOperandIfLayoutsDiffer()989 CreateCopyWithNewLayout(operand_layout.shape(), operand)); in CopyOperandIfLayoutsDiffer()2090 CreateCopyWithNewLayout(constraints.ResultLayout()->shape(), in RunOnComputation()
492 StatusOr<HloInstruction*> CreateCopyWithNewLayout(