Searched refs:HandleBufferFromPyval (Results 1 – 1 of 1) sorted by relevance
663 DevicePutResult HandleBufferFromPyval(py::handle h, xla::PjRtDevice* to_device, in HandleBufferFromPyval() function767 (*p)[sda.ptr()] = HandleBufferFromPyval; in DevicePut()773 (*p)[dtypes.np_bool.ptr()] = HandleBufferFromPyval; in DevicePut()774 (*p)[dtypes.np_int8.ptr()] = HandleBufferFromPyval; in DevicePut()775 (*p)[dtypes.np_int16.ptr()] = HandleBufferFromPyval; in DevicePut()776 (*p)[dtypes.np_int32.ptr()] = HandleBufferFromPyval; in DevicePut()778 (*p)[dtypes.np_uint8.ptr()] = HandleBufferFromPyval; in DevicePut()779 (*p)[dtypes.np_uint16.ptr()] = HandleBufferFromPyval; in DevicePut()780 (*p)[dtypes.np_uint32.ptr()] = HandleBufferFromPyval; in DevicePut()782 (*p)[dtypes.np_float16.ptr()] = HandleBufferFromPyval; in DevicePut()[all …]