message: set PacketAccess to false for TENSORFLOW_USE_ROCM in cwise_ops_gradients.h
diff: @@ -94,7 +94,11 @@ template <typename T>
  struct functor_traits<scalar_inverse_gradient_op<T>> {
  enum {
  Cost = NumTraits<T>::AddCost + 2 * NumTraits<T>::MulCost,
  +#if TENSORFLOW_USE_ROCM
  + PacketAccess = false,
  +#else
  PacketAccess = packet_traits<T>::HasMul,
  +#endif
  };
  };
  @@ -123,7 +127,11 @@ struct scalar_sqrt_gradient_op {
  template <typename T>
  struct functor_traits<scalar_sqrt_gradient_op<T>> {
  enum {
  +#if TENSORFLOW_USE_ROCM
  + PacketAccess = false,
  +#else
  PacketAccess = packet_traits<T>::HasMul & packet_traits<T>::HasDiv,
  +#endif
  Cost = NumTraits<T>::MulCost + scalar_div_cost<T, PacketAccess>::value,
  };
  };
  @@ -158,7 +166,11 @@ template <typename T>
  struct functor_traits<scalar_rsqrt_gradient_op<T>> {
  enum {
  Cost = 4 * NumTraits<T>::MulCost,
  +#if TENSORFLOW_USE_ROCM
  + PacketAccess = false,
  +#else
  PacketAccess = packet_traits<T>::HasMul,
  +#endif
  };
  };
  