Lines Matching defs:shuffle
1592 std::vector<int>* shuffle) {
1594 shuffle->resize(4);
1596 (*shuffle)[i] = i;
1601 shuffle->resize(2);
1602 (*shuffle)[0] = 1;
1603 (*shuffle)[1] = 0;
1608 (*shuffle)[0] = 1;
1609 (*shuffle)[1] = 2;
1610 (*shuffle)[2] = 3;
1611 (*shuffle)[3] = 0;
1616 (*shuffle)[0] = 3;
1617 (*shuffle)[1] = 0;
1618 (*shuffle)[2] = 1;
1619 (*shuffle)[3] = 2;
1621 LOG(FATAL) << "Bad shuffle";
1648 std::vector<int> shuffle;
1649 GetShuffleShape(input_axes_order, output_axes_order, &shuffle);
1653 (*output_dims)[i] = input_shape.dims(shuffle[i]);
1676 std::vector<int> shuffle;
1677 GetShuffleShape(input_axes_order, output_axes_order, &shuffle);
1678 CHECK(shuffle.size() >= dim);
1680 CHECK(shuffle[i] >= 0 && shuffle[i] < dim);
1681 CHECK(input_shape.dims(shuffle[i]) == output_shape.dims(i));
1688 ExtendShuffle(shuffle, 4, &extended_shuffle);