Searched refs:nchw_to_logical (Results 1 – 1 of 1) sorted by relevance
76 std::vector<int64> nchw_to_logical; in GetShapeInfo() local78 nchw_to_logical.push_back(n_dim); in GetShapeInfo()79 nchw_to_logical.push_back(c_dim); in GetShapeInfo()81 nchw_to_logical.push_back(dim); in GetShapeInfo()84 for (int64 dim : nchw_to_logical) { in GetShapeInfo()94 for (int64 dim : ComposePermutations(InversePermutation(nchw_to_logical), in GetShapeInfo()