Searched refs:numpy_type (Results 1 – 6 of 6) 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 | 44 numpy_type = MAP_TF_TO_NUMPY_TYPE[parameters["dtype"]]
45 input_value = np.array([[1, 0], [2, 1]], numpy_type)
|
| D | while_loop.py | 79 numpy_type = zip_test_utils.MAP_TF_TO_NUMPY_TYPE[parameters["dtype"]]
82 np.array(parameters["increment_value"], dtype=numpy_type)
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | while_loop.py | 83 numpy_type = zip_test_utils.TF_TYPE_INFO[parameters["dtype"]][0]
86 np.array(parameters["increment_value"], dtype=numpy_type)
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 34 char const* numpy_type_name(int numpy_type) { in numpy_type_name() argument
35 switch (numpy_type) { in numpy_type_name()
|
| D | bfloat16.cc | 799 bool RegisterCustomFloatCast(int numpy_type = TypeDescriptor<OtherT>::Dtype()) { in RegisterCustomFloatCast() argument
800 PyArray_Descr* descr = PyArray_DescrFromType(numpy_type); in RegisterCustomFloatCast()
806 numpy_type, NPyCast<T, OtherT>) < 0) { in RegisterCustomFloatCast()
|