Commits on Apr 29, 2024

1. PR #10495: Enforce same input/output layout for offloading ops …

   Imported from GitHub PR openxla/xla#10495
   
   This patch makes sure that the host offloader will not introduce layout mismatches
   after removing the offloading custom calls by constraining the layout assignment
   to assign the same layout for the custom call's input and output.
   Copybara import of the project:
   
   --
   ce6aeac3ac5445ab014d36c92a181dbe7afc35e7 by Jaroslav Sevcik <jsevcik@nvidia.com>:
   
   Enforce same input/output layout for offloading ops
   
   Merging this change closes #10495
   
   PiperOrigin-RevId: 628970016

   

   jaro-sevcik authored and tensorflower-gardener committed Apr 29, 2024
   Configuration menu

   • View commit details
   • Copy full SHA for 3275525
   • Browse repository at this point

   Copy the full SHA

   3275525 View commit details

   Browse the repository at this point in the history