Searched defs:new_strides (Results 1 – 14 of 14) sorted by relevance
49 auto new_strides = computeStrideForViewAsReal(self.sym_strides()); in _view_as_real_physical() local88 const auto new_strides = computeStrideForViewAsComplex(self.sym_strides()); in view_as_complex() local
743 SymDimVector new_strides(ndim); in compute_strides_for_view_dtype_downsize() local760 SymDimVector new_strides(ndim); in compute_strides_for_view_dtype_upsize() local804 auto new_strides = compute_strides_for_view_dtype_downsize( in view_dtype() local833 auto new_strides = compute_strides_for_view_dtype_upsize( in view_dtype() local
238 std::vector<int64_t> new_strides(out_dim, 0); in align_to() local287 std::vector<int64_t> new_strides(names.size(), 0); in align_to() local
1427 auto new_strides = DimVector(is_strided_layout ? ndim : 0); in _permute_size_stride_estimation() local
59 const StridesType* new_strides, in increment_index_and_offset()93 StridesType new_strides[kTensorDimensionLimit]; in transpose_tensors() local
96 auto new_strides = strides.clone(); in chunk_nested_tensor() local153 auto new_strides = strides.clone(); in split_with_sizes_nested() local
451 auto new_strides = at::empty({ntensors, ndims-1}, TensorOptions().dtype(kLong)); in select_nested() local
197 index_t* new_strides = this->strides_ + 1; variable203 const index_t* new_strides = this->strides_ + 1; variable
23 std::vector<int64_t> new_strides(ndim); in infer_dense_strides_dim_last() local
47 int64_t new_strides[kTensorDimensionLimit]; in diagonal_copy_impl() local
49 c10::SymDimVector new_strides; in refreshTensorMetadata() local
2805 std::vector<int32> new_strides; in CopyFormatAttrsConv() local2857 std::vector<int32> new_strides; in CopyAttrsPooling() local
2034 auto new_strides = A.allocate<int64_t>(dims.size()); in _bind_dims_to_size() local
4524 std::vector<int64_t> new_strides = slice->slice_strides(); in TryToReorderSliceAndReverse() local