Searched refs:InitializeTypedBuffer (Results 1 – 1 of 1) sorted by relevance
251 static void InitializeTypedBuffer(se::Stream* stream, in InitializeTypedBuffer() function308 return InitializeTypedBuffer<Eigen::half>(stream, buffer, rng_state); in InitializeBuffer()311 return InitializeTypedBuffer<float>(stream, buffer, rng_state); in InitializeBuffer()314 return InitializeTypedBuffer<double>(stream, buffer, rng_state); in InitializeBuffer()316 return InitializeTypedBuffer<int8>(stream, buffer, rng_state); in InitializeBuffer()