Searched defs:new_stride (Results 1 – 11 of 11) sorted by relevance
| /external/pytorch/aten/src/ATen/ |
| D | SparseTensorImpl.cpp | 62 void SparseTensorImpl::set_stride(int64_t dim, int64_t new_stride) { in set_stride()
|
| D | LegacyBatchedTensorImpl.cpp | 99 void BatchedTensorImpl::set_stride(int64_t dim, int64_t new_stride) { in set_stride()
|
| D | SparseCsrTensorImpl.cpp | 247 void SparseCsrTensorImpl::set_stride(int64_t dim, int64_t new_stride) { in set_stride()
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchedTensorImpl.cpp | 141 void BatchedTensorImpl::set_stride(int64_t dim, int64_t new_stride) { in set_stride()
|
| /external/e2fsprogs/lib/ext2fs/ |
| D | mkjournal.c | 196 int new_stride = num; in ext2fs_zero_blocks2() local
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorMath.cpp | 680 Tensor new_stride; in unsqueeze_nested() local
|
| /external/gemmlowp/meta/generators/ |
| D | neon_emitter.py | 711 def EmitColBlockStride(self, cols, stride, new_stride): argument
|
| D | neon_emitter_64.py | 1128 def EmitColBlockStride(self, cols, stride, new_stride): argument
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | shape_analysis.cpp | 2038 int64_t new_stride = dim >= static_cast<int64_t>(sizes.size()) in PropagateCompleteShapeOnNode() local
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_lower_io.c | 2558 unsigned new_stride = align(size, alignment); in nir_lower_vars_to_explicit_types_impl() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorShape.cpp | 3232 c10::SymInt new_stride = dim >= tensor.dim() ? 1 : result.sizes[dim] * result.strides[dim]; in inferUnsqueezeGeometry_symint() local
3242 int64_t new_stride = dim >= tensor.dim() ? 1 : result.sizes[dim] * result.strides[dim]; in inferUnsqueezeGeometry() local
|