Searched refs:is_cpp_nested_tensor (Results 1 – 5 of 5) sorted by relevance
22 bool is_cpp_nested_tensor(const at::Tensor& tensor) { in is_cpp_nested_tensor() function67 if (is_nested_ || is_cpp_nested_tensor() || grad.is_nested() || in maybe_reduce()68 ::torch::autograd::is_cpp_nested_tensor(grad)) { in maybe_reduce()121 if (is_cpp_nested_tensor()) { in is_same_shape()145 if (::torch::autograd::is_cpp_nested_tensor(grad)) { in incompatible_shape_error_message()151 if (is_cpp_nested_tensor()) { in incompatible_shape_error_message()159 bool InputMetadata::is_cpp_nested_tensor() const { in is_cpp_nested_tensor() function in torch::autograd::InputMetadata178 ::torch::autograd::is_cpp_nested_tensor(grad) == is_cpp_nested_tensor()); in is_nestedness_same()
89 bool is_cpp_nested_tensor() const;
203 &torch::autograd::InputMetadata::is_cpp_nested_tensor); in THPAutograd_initExtension()
105 if out_metadata.is_cpp_nested_tensor:
1077 self.assertFalse(nt_metadata.is_cpp_nested_tensor)