Searched refs:numpy_type (Results 1 – 4 of 4) sorted by relevance
| /external/flatbuffers/python/flatbuffers/ |
| D | encode.py | 29 def GetVectorAsNumpy(numpy_type, buf, count, offset): argument
35 return np.frombuffer(buf, dtype=numpy_type, count=count, offset=offset)
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | placeholder_with_default.py | 47 numpy_type = TF_TYPE_INFO[parameters["dtype"]][0]
48 input_value = np.array([[1, 0], [2, 1]], numpy_type)
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 33 char const* numpy_type_name(int numpy_type) { in numpy_type_name() argument
34 switch (numpy_type) { in numpy_type_name()
|
| D | bfloat16.cc | 702 bool RegisterBfloat16Cast(int numpy_type) { in RegisterBfloat16Cast() argument
703 PyArray_Descr* descr = PyArray_DescrFromType(numpy_type); in RegisterBfloat16Cast()
707 if (PyArray_RegisterCastFunc(&NPyBfloat16_Descr, numpy_type, in RegisterBfloat16Cast()
|