Home
last modified time | relevance | path

Searched refs:output_host (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4015 std::vector<float> output_host(output_dimensions.ElementCount()); in DoDepthConcatenate() local
4033 output_host[index(batch, depth + depth_sum, yx, in DoDepthConcatenate()
4041 stream->ThenMemcpyH2D<float>(output_host, output_data); in DoDepthConcatenate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3911 std::vector<float> output_host(output_dimensions.ElementCount()); in DoDepthConcatenate() local
3929 output_host[index(batch, depth + depth_sum, yx, in DoDepthConcatenate()
3937 stream->ThenMemcpyH2D<float>(output_host, output_data); in DoDepthConcatenate()