Searched refs:bdyx_dims (Results 1 – 1 of 1) sorted by relevance
473 std::vector<int64_t> bdyx_dims(ndims() + 2); in full_dims() local474 bdyx_dims[0] = count(); in full_dims()475 bdyx_dims[1] = feature_map_count(); in full_dims()477 bdyx_dims.begin() + 2); in full_dims()478 return ReorderDims(bdyx_dims, DataLayout::kBatchDepthYX, layout); in full_dims()495 std::vector<int64_t> bdyx_dims = full_dims(dnn::DataLayout::kBatchDepthYX); in vectorized_dims() local497 bdyx_dims[vector_dim] /= vector_size; in vectorized_dims()499 return dnn::ReorderDims(bdyx_dims, dnn::DataLayout::kBatchDepthYX, layout); in vectorized_dims()