Home
last modified time | relevance | path

Searched refs:numpy_array_handler (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_values.cc489 ToPyArgSignatureHandler numpy_array_handler = in PyArgSignatureOfValue() local
502 (*p)[dtypes.np_bool.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
503 (*p)[dtypes.np_int8.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
504 (*p)[dtypes.np_int16.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
505 (*p)[dtypes.np_int32.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
507 (*p)[dtypes.np_uint8.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
508 (*p)[dtypes.np_uint16.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
509 (*p)[dtypes.np_uint32.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
511 (*p)[dtypes.np_float16.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
512 (*p)[dtypes.np_bfloat16.ptr()] = numpy_array_handler; in PyArgSignatureOfValue()
[all …]