Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.cc513 ToArgSignatureHandler numpy_array_handler = in ArgSignatureOfValue() local
526 (*p)[dtypes.np_bool.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
527 (*p)[dtypes.np_int8.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
528 (*p)[dtypes.np_int16.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
529 (*p)[dtypes.np_int32.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
531 (*p)[dtypes.np_uint8.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
532 (*p)[dtypes.np_uint16.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
533 (*p)[dtypes.np_uint32.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
535 (*p)[dtypes.np_float16.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
536 (*p)[dtypes.np_float32.ptr()] = numpy_array_handler; in ArgSignatureOfValue()
[all …]