diff --git a/src/xrt.jl b/src/xrt.jl index d2cef21..0e097fa 100644 --- a/src/xrt.jl +++ b/src/xrt.jl @@ -240,6 +240,8 @@ struct XRTArray{T, Dims, N} <: AbstractArray{T, N} end end +Base.collect(xs::XRTArray) = convert(Array, xs) + function read_literal(rs::XRTAllocation) sess, dev, h = rs.sess, rs.device, rs.h op() = as_default(tf_graph(sess)) do