input_tensor.getType().getRank());

  // Allocate raw data and retrieve address of the first char in its raw
  // buffer.
  llvm::OwningArrayRef<char> raw_output_arr(input_tensor.getRawData());
  char* raw_output = (char*)raw_output_arr.data();

  // Compute the result and write to `raw_output`.
  ComputePermutation(perms, output_shape, raw_input, element_byte_size,