Untitled

template <typename T1> auto SspaddmmCpuKernel::return_type_addr(T1 *output) {
  DataType dt = static_cast<DataType>(output->GetDataType());
  switch (dt): {
    case DT_INT8:
      int8_t *addr = reinterpret_cast<int8_t *>(scalar->GetData());
      return addr;
    case DT_UINT8:
      uint8_t *addr = reinterpret_cast<uint8_t *>(scalar->GetData());
      return addr;
    case DT_INT16:
      int16_t *addr = reinterpret_cast<int16_t *>(scalar->GetData());
      return addr;
    case DT_UINT16:
      uint16_t *addr = reinterpret_cast<uint16_t *>(scalar->GetData());
      return addr;
    case DT_INT32:
      int32_t *addr = reinterpret_cast<int32_t *>(scalar->GetData());
      return addr;
    case DT_INT64:
      int32_t *addr = reinterpret_cast<int64_t *>(scalar->GetData());
      return addr;
    case DT_FLOAT16:
      Eigen::half *addr = reinterpret_cast<Eigen::half *>(scalar->GetData());
      return addr;
    case DT_FLOAT:
      float *addr = reinterpret_cast<float *>(scalar->GetData());
      return addr;
    case DT_BOOL:
      bool *addr = reinterpret_cast<bool *>(scalar->GetData());
      return addr;
    case DT_DOUBLE:
      double *addr = reinterpret_cast<double *>(scalar->GetData());
      return addr;
    }
  return nullptr;
}