Searched refs:order_with_batch_dim (Results 1 – 2 of 2) sorted by relevance
528 const std::vector<int>& order_with_batch_dim,
1581 const std::vector<int>& order_with_batch_dim, in TransposeTensor() argument1586 const int order_size = use_implicit_batch_ ? order_with_batch_dim.size() - 1 in TransposeTensor()1587 : order_with_batch_dim.size(); in TransposeTensor()1592 if (use_implicit_batch_ && order_with_batch_dim[0] != 0) { in TransposeTensor()1606 permutation.order[i] = order_with_batch_dim[i + 1] - 1; in TransposeTensor()1609 std::copy(order_with_batch_dim.begin(), order_with_batch_dim.end(), in TransposeTensor()