Searched refs:is_nested_tensor (Results 1 – 5 of 5) sorted by relevance
91 bool is_nested_tensor() const { in is_nested_tensor() function
200 "is_nested_tensor", &torch::autograd::InputMetadata::is_nested_tensor) in THPAutograd_initExtension()
104 out_is_nested = out_metadata.is_nested_tensor
399 TORCH_CHECK(!t.is_nested_tensor(), "NestedTensor not implemented"); in collect()
1076 self.assertTrue(nt_metadata.is_nested_tensor)