Lines Matching refs:DeviceMemory
52 se::port::StatusOr<se::DeviceMemory<uint8>> FftScratchAllocator::AllocateBytes(
74 return se::DeviceMemory<uint8>(allocated_buffer);
175 se::DeviceMemory<complex64> input_data(
177 se::DeviceMemory<complex64> output_data(
184 se::DeviceMemory<complex64> input_data(
186 se::DeviceMemory<complex64> output_data(
200 se::DeviceMemory<float> input_data(
202 se::DeviceMemory<complex64> output_data(
209 se::DeviceMemory<complex64> input_data(
211 se::DeviceMemory<float> output_data(
