Lines Matching refs:tensor_strides
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()
173 int64_t stride1 = tensor_strides[dim1]; in infer_dense_strides()