Status TFDataTypeToNGraphElementType()

in ngraph_bridge/tf_utils.cc [99:152]

• 54 lines of code
• 16 McCabe index (conditional complexity)

Status TFDataTypeToNGraphElementType(DataType tf_dt,
                                     ngraph::element::Type* ng_et) {
  switch (tf_dt) {
    case DataType::DT_FLOAT:
      *ng_et = ngraph::element::f32;
      break;
    case DataType::DT_DOUBLE:
      *ng_et = ngraph::element::f64;
      break;
    case DataType::DT_INT32:
      *ng_et = ngraph::element::i32;
      break;
    case DataType::DT_UINT8:
      *ng_et = ngraph::element::u8;
      break;
    case DataType::DT_INT8:
      *ng_et = ngraph::element::i8;
      break;
    case DataType::DT_UINT16:
      *ng_et = ngraph::element::u16;
      break;
    case DataType::DT_INT64:
      *ng_et = ngraph::element::i64;
      break;
    case DataType::DT_UINT32:
      *ng_et = ngraph::element::u32;
      break;
    case DataType::DT_UINT64:
      *ng_et = ngraph::element::u64;
      break;
    case DataType::DT_BOOL:
      *ng_et = ngraph::element::boolean;
      break;
    case DataType::DT_QINT8:
      *ng_et = ngraph::element::i8;
      break;
    case DataType::DT_QUINT8:
      *ng_et = ngraph::element::u8;
      break;
    case DataType::DT_QINT32:
      *ng_et = ngraph::element::i32;
      break;
    case DataType::DT_BFLOAT16:
      *ng_et = ngraph::element::bf16;
      break;
    case DataType::DT_HALF:
      *ng_et = ngraph::element::f16;
      break;
    default:
      return errors::Unimplemented("Unsupported TensorFlow data type: ",
                                   DataType_Name(tf_dt));
  }
  return Status::OK();
}