| /external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| D | api.py | 97 tensor_properties = TensorProperties(
101 sizes, tensor_properties=tensor_properties
150 tensor_properties = sharded_tensor_metadata.tensor_properties
155 if tensor_properties.layout != torch.strided:
167 dtype=tensor_properties.dtype,
168 layout=tensor_properties.layout,
169 pin_memory=tensor_properties.pin_memory,
170 requires_grad=tensor_properties.requires_grad,
268 self._metadata.tensor_properties.memory_format = memory_format
280 tensor_properties=self._metadata.tensor_properties,
[all …]
|
| D | utils.py | 196 tensor_properties=local_tensor_properties,
226 global_sharded_tensor_metadata.tensor_properties.dtype,
227 rank_metadata.tensor_properties.dtype,
234 global_sharded_tensor_metadata.tensor_properties.requires_grad,
235 rank_metadata.tensor_properties.requires_grad,
242 global_sharded_tensor_metadata.tensor_properties.pin_memory,
243 rank_metadata.tensor_properties.pin_memory,
|
| D | metadata.py | 95 tensor_properties: TensorProperties = field(default_factory=TensorProperties) variable in ShardedTensorMetadata
|
| /external/pytorch/torch/utils/benchmark/examples/ |
| D | fuzzer.py | 45 for i, (tensors, tensor_properties, _) in enumerate(add_fuzzer.take(n=n)):
46 x, x_order = tensors["x"], str(tensor_properties["x"]["order"])
47 y, y_order = tensors["y"], str(tensor_properties["y"]["order"])
|
| /external/pytorch/torch/distributed/_shard/sharding_spec/ |
| D | chunk_sharding_spec.py | 79 tensor_properties: sharded_tensor_meta.TensorProperties,
107 shards_metadata, tensor_sizes, tensor_properties
122 tensor_properties = sharded_tensor_meta.TensorProperties(
131 tensor_meta = self.build_metadata(tensor.size(), tensor_properties)
|
| D | api.py | 60 tensor_properties: sharded_tensor_meta.TensorProperties,
171 tensor_properties: sharded_tensor_meta.TensorProperties,
176 self.shards, tensor_sizes, tensor_properties
|
| /external/tensorflow/tensorflow/c/experimental/grappler/ |
| D | grappler.cc | 310 const std::vector<tensorflow::OpInfo::TensorProperties>& tensor_properties = in TF_GetInputPropertiesList() local
314 std::min(num_values, static_cast<int>(tensor_properties.size())); in TF_GetInputPropertiesList()
317 tensorflow::MessageToBuffer(tensor_properties[i], properties[i]); in TF_GetInputPropertiesList()
329 const std::vector<tensorflow::OpInfo::TensorProperties>& tensor_properties = in TF_GetOutputPropertiesList() local
333 std::min(num_values, static_cast<int>(tensor_properties.size())); in TF_GetOutputPropertiesList()
336 tensorflow::MessageToBuffer(tensor_properties[i], properties[i]); in TF_GetOutputPropertiesList()
|
| /external/pytorch/torch/utils/benchmark/examples/sparse/ |
| D | fuzzer.py | 59 for i, (tensors, tensor_properties, _) in enumerate(add_fuzzer.take(n=n)):
63 x_tensor_properties = tensor_properties["x"]
|
| /external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/ |
| D | tensor_ops.py | 106 st_meta.tensor_properties.dtype = tensor.dtype
186 new_metadata.tensor_properties.requires_grad = False
217 self_st._metadata.tensor_properties.requires_grad = requires_grad
|
| /external/pytorch/torch/csrc/jit/frontend/ |
| D | builtin_functions.cpp | 50 auto tensor_properties = variable
172 loadSource(tensor_properties, "prim"); in loadBuiltinFunctions()
|
| /external/pytorch/test/distributed/_shard/sharding_spec/ |
| D | test_sharding_spec.py | 508 tensor_properties: TensorProperties,
537 tensor_properties=tensor_properties,
557 tensor_properties = TensorProperties(
565 meta = grid_spec.build_metadata(torch.Size((8, 8)), tensor_properties)
|
| /external/pytorch/torch/distributed/tensor/parallel/ |
| D | fsdp.py | 102 tensor_properties=TensorProperties(
170 st_meta.tensor_properties.requires_grad = False
200 st_meta.tensor_properties.requires_grad = False
|
| /external/pytorch/torch/utils/benchmark/utils/ |
| D | fuzzer.py | 404 tensor_properties = {}
408 tensor_properties[t.name] = properties
409 yield tensors, tensor_properties, params
|
| /external/pytorch/torch/distributed/fsdp/ |
| D | _shard_utils.py | 79 tensor_properties=TensorProperties(
|
| /external/pytorch/test/distributed/_shard/sharded_tensor/ |
| D | test_sharded_tensor.py | 152 tensor_properties = TensorProperties(
161 5, 10, local_device=local_device, tensor_properties=tensor_properties
2497 tensor_properties = TensorProperties(
2508 tensor_properties=tensor_properties,
2568 tensor_properties = TensorProperties(
2579 tensor_properties=tensor_properties,
2903 tensor_properties = TensorProperties(
2914 tensor_properties=tensor_properties,
|
| /external/pytorch/test/distributed/checkpoint/ |
| D | test_utils.py | 53 tensor_properties=TensorProperties.create_from_tensor(torch.zeros(1)),
|
| D | test_planner.py | 79 tensor_properties=TensorProperties_Shard.create_from_tensor(torch.zeros(1)),
|
| /external/pytorch/torch/distributed/checkpoint/ |
| D | planner_helpers.py | 57 shard_properties = sharded_tensor.metadata().tensor_properties
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2776 auto tensor_properties = GetOutputProperties(node->name()); in AnnotateOutputShapes() local
2777 for (const auto& tensor_property : tensor_properties) { in AnnotateOutputShapes()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 931 const auto& tensor_properties = output_params.at(output_port); in ConvertToTensorOrWeights() local
932 const DataType dtype = tensor_properties.dtype(); in ConvertToTensorOrWeights()
933 const PartialTensorShape shape = tensor_properties.shape(); in ConvertToTensorOrWeights()
|