Searched refs:self_sizes_batch_end (Results 1 – 2 of 2) sorted by relevance
83 const auto self_sizes_batch_end = self.sizes().end() - NUM_NON_BATCH_DIMS; in math_pixel_shuffle() local94 self.sizes().begin(), self_sizes_batch_end); in math_pixel_shuffle()100 std::vector<int64_t> permutation(self.sizes().begin(), self_sizes_batch_end); in math_pixel_shuffle()109 std::vector<int64_t> final_shape(self.sizes().begin(), self_sizes_batch_end); in math_pixel_shuffle()124 const auto self_sizes_batch_end = self.sizes().end() - NUM_NON_BATCH_DIMS; in math_pixel_unshuffle() local135 self.sizes().begin(), self_sizes_batch_end); in math_pixel_unshuffle()141 std::vector<int64_t> permutation(self.sizes().begin(), self_sizes_batch_end); in math_pixel_unshuffle()150 std::vector<int64_t> final_shape(self.sizes().begin(), self_sizes_batch_end); in math_pixel_unshuffle()
30 const auto self_sizes_batch_end = self.sizes().end() - NUM_NON_BATCH_DIMS;37 std::vector<int64_t> out_shape(self.sizes().begin(), self_sizes_batch_end);