| /aosp_15_r20/external/pytorch/torch/distributed/tensor/examples/ |
| H A D | visualize_sharding_example.py | 19 dtensor = distribute_tensor(tensor, mesh, [Shard(dim=1)]) variable
30 dtensor = distribute_tensor(tensor, mesh, [Shard(dim=0)]) variable
44 dtensor = distribute_tensor(tensor, mesh, [Shard(dim=0), Replicate()]) variable
56 dtensor = distribute_tensor(tensor, mesh, [Replicate(), Shard(dim=0)]) variable
68 dtensor = distribute_tensor(tensor, mesh, [Replicate()]) variable
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_casts.cpp | 45 Tensor<double, 2> dtensor(20,30); in test_vectorized_cast() local
64 Tensor<double, 2> dtensor(20,30); in test_float_to_int_cast() local
81 Tensor<double, 2> dtensor(20, 30); in test_big_to_small_type_cast() local
98 Tensor<double, 2> dtensor(20, 30); in test_small_to_big_type_cast() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | list_ops.py | 232 def _TensorListStackGrad(unused_op, dtensor): argument
238 def _TensorListConcatGrad(op, dtensor, unused_dlengths): argument
324 def _TensorListGatherGrad(op, dtensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| H A D | collective_lowering.cc | 60 namespace dtensor { namespace
85 namespace dtensor { namespace
107 namespace dtensor { namespace
|
| H A D | dtensor_embedding_stub.cc | 20 namespace dtensor { namespace
|
| H A D | dtensor_embedding_v2_stub.cc | 20 namespace dtensor { namespace
|
| H A D | dtensor_mlir_passes_internal.cc | 22 namespace dtensor { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | spmd_expanders.cc | 58 namespace dtensor { namespace
|
| H A D | sparse_expanders.cc | 22 namespace dtensor { namespace
|
| H A D | dce.cc | 26 namespace dtensor { namespace
|
| H A D | device_utils.cc | 23 namespace dtensor { namespace
|
| H A D | function_renaming.cc | 29 namespace dtensor { namespace
|
| H A D | collectives_common.cc | 21 namespace dtensor { namespace
|
| H A D | op_utils.cc | 23 namespace dtensor { namespace
|
| H A D | sparse_expander.cc | 27 namespace dtensor { namespace
|
| H A D | undo_merge_const_across_mesh.cc | 32 namespace dtensor { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/ |
| H A D | matmul_sparse_expander.cc | 23 namespace dtensor { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | tpu_system_interface.cc | 19 namespace dtensor { namespace
|
| H A D | dtensor_tpu_ops.cc | 27 namespace dtensor { namespace
|
| H A D | dtensor_ops.cc | 29 namespace dtensor { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | tensorlist_setitem_spmd_expander.cc | 24 namespace dtensor { namespace
|
| H A D | tensorlist_getitem_spmd_expander.cc | 23 namespace dtensor { namespace
|
| H A D | range_spmd_expander.cc | 24 namespace dtensor { namespace
|
| H A D | sparse_to_dense_spmd_expander.cc | 23 namespace dtensor { namespace
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/ |
| H A D | _visualize_sharding.py | 136 def visualize_sharding(dtensor, header=""): argument
|