Grows the input tensor by replicating the X and Y dimensions. The batch and depth/feature_map dimensions are unchanged. Currently, the input tensor is limited to X=1 and Y=1. For example, the input has dimensions x=2, y=3, and replicate_x=3, replicate_y=2. The diagonal elements of the output would be: [x0y0, x1y1, x0y2, x1y0, x0y1, x1y2]. Here is the example as a picture. input: AB CD EF broadcas
| 2401 | // output_data: un-owned device memory region in which to place the |
| 2402 | // padded result. |
| 2403 | virtual bool DoXYBroadcast(Stream* stream, |
| 2404 | const dnn::BatchDescriptor& dimensions, |
| 2405 | const DeviceMemory<float>& input_data, |
| 2406 | int64 replicate_x, int64 replicate_y, |
| 2407 | DeviceMemory<float>* output_data) { |
| 2408 | return false; |
| 2409 | } |
| 2410 | |
| 2411 | // Enqueues an asynchronous memcpy of the *quantized* output of a layer (that |
| 2412 | // is, bytes instead of scaled floats) into 'host_dst' if they are available |