Searched refs:_nested_tensor_strides (Results 1 – 6 of 6) sorted by relevance
26 at::Tensor _nested_tensor_strides(const at::Tensor& self){ in _nested_tensor_strides() function
424 nt._nested_tensor_strides()
469 aten::_nested_tensor_strides470 aten::_nested_tensor_strides.out
411 self._nested_tensor_strides(),
1454 Tensor._nested_tensor_strides: lambda self: -1,
1362 self.assertEqual(nt_split._nested_tensor_strides(), dense_strides)2345 nt_strides = nt._nested_tensor_strides()2359 unsqueezed_stride = nt_unsqueezed._nested_tensor_strides()[:, size_idx]2369 self.assertEqual(nt_squeezed._nested_tensor_strides(), nt_strides)5139 self.assertEqual(nt1._nested_tensor_strides(), nt2._nested_tensor_strides())