Searched refs:nested_tensors (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | function.py | 776 nested_tensors = _unflatten(flat_output, self._nested_output)
777 return nested_tensors
801 nested_tensors = _map_tensor_data(self._nested_input)
802 result = self.forward_extended(*nested_tensors) # type: ignore[func-returns-value]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_transformers.py | 1092 …ef _test_fastpath(model, key_padding_mask, mock_return_value, attn_mask=None, nested_tensors=True): argument
1102 self.assertEqual(call_args[0].is_nested, nested_tensors)
1116 … _test_fastpath(model, aligned_key_padding_mask, nested_tensor_return_value, nested_tensors=True)
1119 … _test_fastpath(model, not_aligned_key_padding_mask, tensor_return_value, nested_tensors=False)
1125 _test_fastpath(model, aligned_key_padding_mask, tensor_return_value, nested_tensors=False)
1126 … _test_fastpath(model, not_aligned_key_padding_mask, tensor_return_value, nested_tensors=False)
1128 …th(model, aligned_key_padding_mask, tensor_return_value, attn_mask=attn_mask, nested_tensors=False)
1134 … _test_fastpath(model, aligned_key_padding_mask, nested_tensor_return_value, nested_tensors=True)
1135 …test_fastpath(model, not_aligned_key_padding_mask, nested_tensor_return_value, nested_tensors=True)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/util/ |
| H A D | structure_test.py | 849 nested_tensors = (int64_t, (float32_t, string_t))
851 tensor_list = structure.to_tensor_list(s, nested_tensors)
|