Searched refs:pyarray_type (Results 1 – 2 of 2) sorted by relevance
113 int pyarray_type = PyArray_TYPE(array); in PyArray_TYPE_to_TF_DataType() local115 switch (pyarray_type) { in PyArray_TYPE_to_TF_DataType()171 if (pyarray_type == Bfloat16NumpyType()) { in PyArray_TYPE_to_TF_DataType()174 } else if (pyarray_type == NPY_ULONGLONG) { in PyArray_TYPE_to_TF_DataType()179 } else if (pyarray_type == NPY_LONGLONG) { in PyArray_TYPE_to_TF_DataType()184 } else if (pyarray_type == NPY_INT) { in PyArray_TYPE_to_TF_DataType()189 } else if (pyarray_type == NPY_UINT) { in PyArray_TYPE_to_TF_DataType()196 numpy_type_name(pyarray_type)); in PyArray_TYPE_to_TF_DataType()
112 int pyarray_type = PyArray_TYPE(array); in TfLiteTypeFromPyArray() local113 return TfLiteTypeFromPyType(pyarray_type); in TfLiteTypeFromPyArray()