Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4017 int64 depth_sum = 0; in DoDepthConcatenate() local
4033 output_host[index(batch, depth + depth_sum, yx, in DoDepthConcatenate()
4039 depth_sum += dimensions.feature_map_count(); in DoDepthConcatenate()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3913 int64 depth_sum = 0; in DoDepthConcatenate() local
3929 output_host[index(batch, depth + depth_sum, yx, in DoDepthConcatenate()
3935 depth_sum += dimensions.feature_map_count(); in DoDepthConcatenate()