tensorflow::DataType GetTfDataType()

in tensorflow_fold/llgtm/backend/tf_evaluator.cc [71:83]

• 13 lines of code
• 10 McCabe index (conditional complexity)

tensorflow::DataType GetTfDataType(TensorDataType dtype) {
  switch (dtype) {
    case kDTvoid: return tensorflow::DT_INVALID;
    case kDTbool: return tensorflow::DT_BOOL;
    case kDTint8: return tensorflow::DT_INT8;
    case kDTint16: return tensorflow::DT_INT16;
    case kDTint32: return tensorflow::DT_INT32;
    case kDTint64: return tensorflow::DT_INT64;
    case kDTfloat32: return tensorflow::DT_FLOAT;
    case kDTfloat64: return tensorflow::DT_DOUBLE;
    case kMaximumTensorDataType: return tensorflow::DT_INVALID;
  }
}