transfer_manager_test.cc | 63 auto device_buffer = AllocateDeviceBuffer(shape); local 67 stream_executor_, *literal, *device_buffer)); 70 stream_executor_, *device_buffer)); 79 auto device_buffer = AllocateDeviceBuffer(shape); local 83 stream_executor_, *literal, *device_buffer)); 86 stream_executor_, *device_buffer)); 97 auto device_buffer = AllocateDeviceBuffer(shape); local 101 stream_executor_, *literal, *device_buffer)); 104 stream_executor_, *device_buffer)); 113 auto device_buffer = AllocateDeviceBuffer(shape) local 129 auto device_buffer = AllocateDeviceBuffer(shape); local 148 auto device_buffer = AllocateDeviceBuffer(ondevice_shape); local 169 auto device_buffer = AllocateDeviceBuffer(literal->shape()); local 183 auto device_buffer = AllocateDeviceBuffer(literal->shape()); local 203 auto device_buffer = AllocateDeviceBuffer(literal->shape()); local 218 auto device_buffer = AllocateDeviceBuffer(literal->shape()); local 237 auto device_buffer = AllocateDeviceBuffer(literal->shape()); local [all...] |