Searched defs:_create_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/test/distributed/fsdp/ | ||
| D | test_shard_utils.py | 23 def _create_tensor(self, *size): member in TestShardUtilsDistributed |
| /external/pytorch/test/ | ||
| D | test_subclass.py | 42 def _create_tensor(self, tensor_cls): member in TestSubclass |
| /external/pytorch/test/distributed/_tensor/ | ||
| D | test_dtensor.py | 826 def _create_tensor(self, size): member in TestDTensorPlacementTypes |
| /external/tensorflow/tensorflow/python/eager/ | ||
| D | tensor_test.py | 39 def _create_tensor(value, device=None, dtype=None): function |