Searched refs:order_with_batch_dim (Results 1 – 2 of 2) sorted by relevance
492 const std::vector<int>& order_with_batch_dim,
1155 const std::vector<int>& order_with_batch_dim, in TransposeTensor() argument1159 if (order_with_batch_dim.size() - 1 != size_t(dims.nbDims)) { in TransposeTensor()1163 if (order_with_batch_dim[0] != 0) { in TransposeTensor()1174 permutation.order[i] = order_with_batch_dim[i + 1] - 1; in TransposeTensor()