message: Use type erased unranked descriptors also for unary operations.
diff: @@ -201,7 +201,7 @@ class MlirUnrankedOp : public OpKernel {
  #define GENERATE_UNARY_KERNEL2(tf_op, mlir_type, tf_data_type, data_type, \
  input_data_type) \
  - extern "C" ::UnrankedMemRefType<data_type> MLIR_FUNCTION(tf_op, mlir_type)( \
  + extern "C" UntypedUnrankedMemRefType MLIR_FUNCTION(tf_op, mlir_type)( \
  tensorflow::OpKernelContext * ctx, \
  const ::UnrankedMemRefType<input_data_type>* arg); \
  \
  @@ -216,7 +216,8 @@ class MlirUnrankedOp : public OpKernel {
  static ::UnrankedMemRefType<data_type> Invoke( \
  OpKernelContext* ctx, \
  llvm::ArrayRef<::UnrankedMemRefType<input_data_type>> args) { \
  - return MLIR_FUNCTION(tf_op, mlir_type)(ctx, &args[0]); \
  + return ConvertToTyped<data_type>( \
  + MLIR_FUNCTION(tf_op, mlir_type)(ctx, &args[0])); \
  } \
  }; \
  }
  