Searched refs:out_strides_ (Results 1 – 6 of 6) sorted by relevance
105 out_strides_ = out_.strides().Permute(out_axis_map); in Permute()118 CHAINERX_ASSERT(out_shape_.ndim() == out_strides_.ndim()); in Squash()126 std::tuple<Shape, Axes> out_squashed_result = SquashShape(out_shape_, out_strides_); in Squash()129 Strides out_squashed_strides = GetSquashedStrides(out_strides_, out_keep_axes); in Squash()141 out_strides_ = out_squashed_strides; in Squash()
47 const Strides& out_strides() const { return out_strides_; } in out_strides()58 Strides out_strides_{};
45 std::vector<size_t> out_strides_; variable
315 blender.out_strides_.push_back(output_->PixelsPerRow()); in PrepareRect()329 blender.out_strides_.push_back((*output_extra_channels_)[c].PixelsPerRow()); in PrepareRect()464 out_row_ptrs_[c] = out_ptrs_[c] + y * out_strides_[c]; in DoBlending()
321 blender.out_strides_.push_back(output_->PixelsPerRow()); in PrepareRect()335 blender.out_strides_.push_back((*output_extra_channels_)[c].PixelsPerRow()); in PrepareRect()472 out_row_ptrs_[c] = out_ptrs_[c] + y * out_strides_[c]; in DoBlending()