Searched refs:numpy_type (Results 1 – 3 of 3) sorted by relevance
| /external/flatbuffers/python/flatbuffers/ |
| D | encode.py | 27 def GetVectorAsNumpy(numpy_type, buf, count, offset): argument
33 return np.frombuffer(buf, dtype=numpy_type, count=count, offset=offset)
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | bfloat16.cc | 476 bool RegisterBfloat16Cast(int numpy_type, bool cast_is_safe) { in RegisterBfloat16Cast() argument
477 if (PyArray_RegisterCastFunc(PyArray_DescrFromType(numpy_type), npy_bfloat16_, in RegisterBfloat16Cast()
481 if (PyArray_RegisterCastFunc(&NPyBfloat16_Descr, numpy_type, in RegisterBfloat16Cast()
485 if (cast_is_safe && PyArray_RegisterCanCast(&NPyBfloat16_Descr, numpy_type, in RegisterBfloat16Cast()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 4234 numpy_type = _TF_TYPE_INFO[parameters["dtype"]][0]
4235 input_value = np.array([[1, 0], [2, 1]], numpy_type)
|