| /external/pytorch/aten/src/ATen/ |
| D | ExpandUtils.cpp | 64 IntArrayRef tensor_strides, in inferExpandGeometryImpl() argument
82 int64_t stride = (dim >= 0) ? tensor_strides[dim] in inferExpandGeometryImpl()
118 IntArrayRef tensor_strides, in inferExpandGeometry() argument
121 tensor_sizes, tensor_strides, sizes); in inferExpandGeometry()
127 IntArrayRef tensor_strides, in inferExpandGeometry_dimvector() argument
130 tensor_sizes, tensor_strides, sizes); in inferExpandGeometry_dimvector()
148 std::vector<int64_t> infer_dense_strides(IntArrayRef tensor_sizes, IntArrayRef tensor_strides) { in infer_dense_strides() argument
150 TORCH_CHECK(tensor_sizes.size() == tensor_strides.size(), in infer_dense_strides()
151 … and strides should have same size but got ", tensor_sizes.size(), " and ", tensor_strides.size()); in infer_dense_strides()
172 int64_t stride0 = tensor_strides[dim0]; in infer_dense_strides()
[all …]
|
| D | ExpandUtils.h | 45 IntArrayRef tensor_strides,
50 IntArrayRef tensor_strides,
55 IntArrayRef tensor_strides);
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Blas.cpp | 63 IntArrayRef tensor_strides = tensor.strides(); in prepare_matrix_for_cublas() local
65 if ((tensor_strides[0] == 1) && (tensor_strides[1] >= std::max<int64_t>(1, tensor_sizes[0]))) { in prepare_matrix_for_cublas()
68 …} else if ((tensor_strides[1] == 1) && (tensor_strides[0] >= std::max<int64_t>(1, tensor_sizes[1])… in prepare_matrix_for_cublas()
82 IntArrayRef tensor_strides = tensor.strides(); in prepare_matrix_for_cublas() local
84 if ((tensor_strides[0] == 1) && (tensor_strides[1] >= std::max<int64_t>(1, tensor_sizes[0]))) { in prepare_matrix_for_cublas()
87 …} else if ((tensor_strides[1] == 1) && (tensor_strides[0] >= std::max<int64_t>(1, tensor_sizes[1])… in prepare_matrix_for_cublas()
128 IntArrayRef tensor_strides = tensor.strides(); in prepare_batch_matrix_for_cublas() local
133 if (tensor_strides[fast_dim] == 1 && in prepare_batch_matrix_for_cublas()
134 (tensor_strides[leading_dim] >= std::max<int64_t>(1, m))) { in prepare_batch_matrix_for_cublas()
138 } else if ((tensor_strides[leading_dim] == 1) && in prepare_batch_matrix_for_cublas()
[all …]
|
| /external/pytorch/aten/src/ATen/native/nested/cuda/ |
| D | NestedTensorTransformerUtils.cpp | 89 const Tensor& tensor_strides = tensor->get_nested_strides(); in is_safe_to_get_storage_as_tensor() local
91 const int64_t n_tensors = tensor_strides.size(0); in is_safe_to_get_storage_as_tensor()
95 const int64_t tensor_stride_0 = tensor_strides.stride(0); in is_safe_to_get_storage_as_tensor()
101 int64_t* previous_tensor_stride = tensor_strides.data_ptr<int64_t>(); in is_safe_to_get_storage_as_tensor()
135 const int64_t* tensor_stride_ptr = tensor_strides.const_data_ptr<int64_t>(); in is_safe_to_get_storage_as_tensor()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NamedTensor.cpp | 195 const auto tensor_strides = tensor.strides(); in align_to() local
243 new_strides[out_dim] = tensor_strides[tensor_dim]; in align_to()
285 auto tensor_strides = tensor.strides(); in align_to() local
300 new_strides[new_idx] = tensor_strides[idx]; in align_to()
|
| D | TensorShape.cpp | 3224 InferUnsqueezeGeometryResult(ArrayRef<T> tensor_sizes, ArrayRef<T> tensor_strides) in InferUnsqueezeGeometryResult()
3226 , strides(tensor_strides.begin(), tensor_strides.end()) {} in InferUnsqueezeGeometryResult()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_io.cpp | 230 auto tensor_strides = internal::strides<Layout>(tensor_dims); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local
246 IOSrc src(tensor_strides, tensor_data, 0); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze()
278 auto tensor_strides = internal::strides<Layout>(tensor_dims); in test_block_io_copy_using_reordered_dimensions_squeeze() local
294 IOSrc src(tensor_strides, tensor_data, 0); in test_block_io_copy_using_reordered_dimensions_squeeze()
|
| D | cxx11_tensor_block_access.cpp | 104 const DSizes<Index, NumDims>& tensor_strides, in UpdateCoeffSet() argument
110 if (tensor_strides[dim_index] == 1) { in UpdateCoeffSet()
116 UpdateCoeffSet<NumDims, Layout>(tensor_strides, block, first_coeff_index, in UpdateCoeffSet()
118 first_coeff_index += tensor_strides[dim_index]; in UpdateCoeffSet()
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorUtils.h | 194 std::vector<IntArrayRef> tensor_strides = in get_stride_for_index() local
196 return tensor_strides[i]; in get_stride_for_index()
|
| /external/pytorch/torch/_inductor/ |
| D | graph.py | 1144 def significant_strides_equal(shape, meta_strides, tensor_strides): argument
1145 for dim, s1, s2 in zip(shape, meta_strides, tensor_strides):
|