Home
last modified time | relevance | path

Searched defs:Placement (Results 1 – 25 of 50) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/
H A Dstyle.py75 input_layouts: Optional[Placement] = None,
76 output_layouts: Optional[Placement] = None,
181 input_layouts: Optional[Placement] = None,
182 output_layouts: Optional[Placement] = None,
382 input_layouts: Optional[Union[Placement, Tuple[Optional[Placement]]]] = None,
383 desired_input_layouts: Optional[Union[Placement, Tuple[Optional[Placement]]]] = None,
407 input_layout: Optional[Placement],
408 desired_layout: Optional[Placement]
503 output_layouts: Union[Placement, Tuple[Placement]],
504 desired_output_layouts: Union[Placement, Tuple[Placement]],
H A Dloss.py80 def _find_all_reduce_mesh_dim(placements: Tuple[Placement, ...], dim: int) -> int:
93 tensor, placements: Tuple[Placement, ...], mesh: DeviceMesh
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_api.py76 grad_placements: Optional[Sequence[Placement]],
125 placements: Tuple[Placement, ...],
350 placements: Optional[Sequence[Placement]] = None,
432 self, *, grad_placements: Optional[Sequence[Placement]] = None
471 placements: Optional[Sequence[Placement]] = None,
541 self, *, grad_placements: Optional[Sequence[Placement]] = None
626 placements: Optional[Sequence[Placement]] = None,
913 placements: Optional[Sequence[Placement]] = None,
978 placements: Optional[Sequence[Placement]] = None,
1021 placements: Optional[Sequence[Placement]] = None,
[all …]
H A D_utils.py19 global_shape: ShapeType, mesh: DeviceMesh, placements: Sequence[Placement]
50 global_shape: ShapeType, mesh: DeviceMesh, placements: Sequence[Placement]
199 tensor: torch.Tensor, mesh: DeviceMesh, placements: Sequence[Placement]
279 global_stride: ShapeType, mesh: DeviceMesh, placements: Sequence[Placement]
H A Dplacement_types.py23 class Placement: class
608 shard_spec: Placement,
H A D_redistribute.py281 placements: Tuple[Placement, ...],
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_matrix_ops.py104 placements1: List[Placement], placements2: List[Placement]
169 tensor_placements: List[Placement],
170 batch_1_placements: List[Placement],
171 batch_2_placements: List[Placement],
235 placements1: List[Placement],
236 placements2: List[Placement],
H A Dtest_pointwise_ops.py31 placements: Sequence[Placement],
74 placements: Sequence[Placement],
116 placements: Sequence[Placement],
/aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/
H A D_tensor_ops.py292 placements: Sequence[Placement], dim: int
302 placements: Sequence[Placement], dim: int
439 cur_placement: Placement, new_placement: Placement
490 placements: Sequence[Placement], insert_dim: int = 0
677 def place(vp: Placement, ip: Placement) -> Placement:
H A D_math_ops.py115 shard_spec: Placement,
181 placements: Sequence[Placement], start_dim: int = 0
194 placements: Tuple[Placement, ...], skipped_dim: int
206 placements: Tuple[Placement, ...], reduction_dims: List[int]
223 placements: Tuple[Placement, ...],
H A D_embedding_ops.py137 shard_spec: Placement,
H A Dutils.py179 placements: Tuple[Placement, ...],
H A D_view_ops.py484 input_src_placements: Sequence[Placement],
/aosp_15_r20/external/clang/test/SemaCXX/
H A Dwarn-new-overaligned-3.cpp27 struct Placement { struct
28 Placement() { in Placement() function
31 helper *d;
/aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/
H A D_tp_transform.py133 parameter_placements: Dict[str, Placement], argument
185 parameter_placements: Dict[str, Placement], argument
268 placements: Tuple[Placement, ...],
/aosp_15_r20/packages/apps/Messaging/src/com/android/messaging/ui/
DSnackBar.java87 public static class Placement { class in SnackBar
91 private Placement(@NonNull final View anchorView, final boolean anchorAbove) { in Placement() method in SnackBar.Placement
/aosp_15_r20/external/skia/tools/
H A Dusing_skia_and_harfbuzz.cpp148 class Placement { class
150 Placement(const Config* conf, SkDocument *doc) in Placement() function in Placement
/aosp_15_r20/out/soong/.intermediates/frameworks/base/libs/WindowManager/Shell/WindowManager-Shell/android_common/kapt/gen/stubs/com/android/wm/shell/pip/tv/
DTvPipKeepClearAlgorithm.java324 public static final class Placement { class in TvPipKeepClearAlgorithm
334 public Placement(@org.jetbrains.annotations.NotNull() in Placement() method in TvPipKeepClearAlgorithm.Placement
/aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/
H A D_visualize_sharding.py91 placements: Sequence[Placement],
/aosp_15_r20/external/aws-sdk-java-v2/codegen/src/main/java/software/amazon/awssdk/codegen/model/service/
H A DAuthorizer.java42 public static class Placement { class in Authorizer
/aosp_15_r20/external/googleapis/google/ads/admanager/v1/
H A Dplacement_service.proto56 message Placement { message
/aosp_15_r20/packages/modules/AdServices/sdksandbox/tests/manual-test-apps/SdkSandboxClient/src/com/android/sdksandboxclient/
DBannerOptions.java38 public enum Placement { enum in BannerOptions
/aosp_15_r20/packages/apps/Contacts/src/com/android/contacts/list/
DIndexerListAdapter.java41 public static final class Placement { class in IndexerListAdapter
/aosp_15_r20/frameworks/base/libs/WindowManager/Shell/src/com/android/wm/shell/pip/tv/
H A DTvPipKeepClearAlgorithm.kt56 data class Placement( in <lambda>() class in com.android.wm.shell.pip.tv.TvPipKeepClearAlgorithm
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/
H A Dcommon_dtensor.py521 self, t: torch.Tensor, mesh: DeviceMesh, placements: List[Placement]

12