MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / DoXYBroadcast

Method DoXYBroadcast

tensorflow/stream_executor/dnn.h:2403–2409  ·  view source on GitHub ↗

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

Source from the content-addressed store, hash-verified

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

Callers 1

stream.ccFile · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected