Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc249 std::vector<int64> bdyx_dims(ndims() + 2); in full_dims() local
250 bdyx_dims[0] = count(); in full_dims()
251 bdyx_dims[1] = feature_map_count(); in full_dims()
253 bdyx_dims.begin() + 2); in full_dims()
254 return ReorderDims(bdyx_dims, DataLayout::kBatchDepthYX, layout); in full_dims()