tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
// Updates func argument type to have the updated input shape.
Updates func argument type to have the updated input shape.
Updates func argument type to have the updated input shape .
void UpdateFuncType(FuncOp func) {
  auto arg_types = func.front().getArgumentTypes();
  auto result_types = func.front().getTerminator()->getOperandTypes();
  func.setType(FunctionType::get(func.getContext(), arg_types, result_types));
}
void UpdateFuncType ( FuncOp func ) { auto arg_types = func . front ( ) . getArgumentTypes ( ) ; auto result_types = func . front ( ) . getTerminator ( ) -> getOperandTypes ( ) ; func . setType ( FunctionType :: get ( func . getContext ( ) , arg_types , result_types ) ) ; }
Updates func argument type to have the updated input shape.
Updates func argument type to have the updated input shape .