Searched refs:bdyx_dims (Results 1 – 1 of 1) sorted by relevance
249 std::vector<int64> bdyx_dims(ndims() + 2); in full_dims() local250 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()