Searched refs:ConvertToScalarBuffer (Results 1 – 1 of 1) sorted by relevance
358 std::unique_ptr<xla::PjRtBuffer> ConvertToScalarBuffer( in ConvertToScalarBuffer() function572 return DevicePutResult(ConvertToScalarBuffer<bool, py::bool_>( in HandleBool()580 return DevicePutResult(ConvertToScalarBuffer<xla::int64, py::int_>( in HandleInt()584 return DevicePutResult(ConvertToScalarBuffer<int, py::int_>( in HandleInt()596 return DevicePutResult(ConvertToScalarBuffer<double, py::float_>( in HandleFloat()600 return DevicePutResult(ConvertToScalarBuffer<float, py::float_>( in HandleFloat()677 return DevicePutResult(ConvertToScalarBuffer<xla::int64, py::int_>( in HandleNpBool()681 return DevicePutResult(ConvertToScalarBuffer<int, py::int_>( in HandleNpBool()