Searched refs:bdyx_dims (Results 1 – 1 of 1) sorted by relevance
280 std::vector<int64> bdyx_dims(ndims() + 2); in full_dims() local281 bdyx_dims[0] = count(); in full_dims()282 bdyx_dims[1] = feature_map_count(); in full_dims()284 bdyx_dims.begin() + 2); in full_dims()285 return ReorderDims(bdyx_dims, DataLayout::kBatchDepthYX, layout); in full_dims()