Home
last modified time | relevance | path

Searched defs:ShapeType (Results 1 – 25 of 37) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py37 ShapeType: TypeAlias = Union[torch.Size, List[int], Tuple[int, ...]] variable
99 def same_shape(a: ShapeType, b: ShapeType, *, allow_rhs_unbacked=False) -> bool:
589 def validate_shape(shape: ShapeType):
830 shape: Union[ShapeType, Tuple[ShapeType]],
860 def infer_size_shapes(a: ShapeType, b: ShapeType) -> Tuple[int, ...]:
885 def infer_size(shape: ShapeType, numel: int) -> Tuple[int, ...]:
1592 shape: ShapeType, row_major: bool = True
1623 def make_channels_last_1d_strides_for(shape: ShapeType) -> Tuple[int, ...]:
1640 def make_channels_last_2d_strides_for(shape: ShapeType) -> Tuple[int, ...]:
1658 def make_channels_last_3d_strides_for(shape: ShapeType) -> Tuple[int, ...]:
[all …]
H A Dwrappers.py160 def _resize_output_check(out: TensorLikeType, shape: ShapeType):
179 shape: ShapeType,
/aosp_15_r20/external/pytorch/torch/_refs/
H A Dfft.py276 input: TensorLikeType, shape: Optional[ShapeType], dim: Optional[DimsType]
362 s: Optional[ShapeType] = None,
375 s: Optional[ShapeType] = None,
388 s: Optional[ShapeType] = None,
407 s: Optional[ShapeType] = None,
440 s: Optional[ShapeType],
469 s: Optional[ShapeType] = None,
486 s: Optional[ShapeType] = None,
507 s: Optional[ShapeType] = None,
518 s: Optional[ShapeType] = None,
[all …]
H A D__init__.py2669 size: ShapeType,
2684 size: ShapeType,
2705 def broadcast_to(a: TensorLikeType, size: ShapeType) -> TensorLikeType:
3192 normalized_shape: ShapeType,
3567 a_shape: ShapeType, a_stride: StrideType, dimension: int, size: int, step: int
3782 def reshape(a: TensorLikeType, *shape: ShapeType) -> TensorLikeType:
3937 def unflatten(a: TensorLikeType, dim: int, sizes: ShapeType) -> TensorLikeType:
4590 def view(a: TensorLikeType, *shape: ShapeType) -> TensorLikeType:
4705 size: ShapeType,
4729 size: ShapeType,
[all …]
/aosp_15_r20/external/pytorch/torch/_prims/
H A D__init__.py226 shape: Optional[ShapeType] = None,
1216 a: TensorLikeType, size: ShapeType, stride: StrideType, storage_offset: int
1236 a: Tensor, size: ShapeType, stride: StrideType, storage_offset: int
1256 a: TensorLikeType, shape: ShapeType, broadcast_dimensions: Sequence[int]
1360 def _collapsed_shape(shape: ShapeType, start: int, end: int) -> Tuple[int, ...]:
1876 size: ShapeType,
1998 def _reshape_meta(a: TensorLikeType, shape: ShapeType):
2012 def _reshape_aten(a: Tensor, shape: ShapeType) -> Tensor:
2282 def _copy_strided_meta(a: TensorLikeType, stride: ShapeType):
2294 def _copy_strided_aten(a: Tensor, stride: ShapeType) -> Tensor:
[all …]
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_utils.py19 global_shape: ShapeType, mesh: DeviceMesh, placements: Sequence[Placement]
50 global_shape: ShapeType, mesh: DeviceMesh, placements: Sequence[Placement]
279 global_stride: ShapeType, mesh: DeviceMesh, placements: Sequence[Placement]
/aosp_15_r20/external/aws-sdk-java-v2/codegen/src/main/java/software/amazon/awssdk/codegen/model/service/
H A DShapeType.java18 public enum ShapeType { enum
28 ShapeType(String name) { in ShapeType() method in ShapeType
/aosp_15_r20/external/aws-sdk-java-v2/codegen/src/main/java/software/amazon/awssdk/codegen/model/intermediate/
H A DShapeType.java18 public enum ShapeType { enum
28 ShapeType(String value) { in ShapeType() method in ShapeType
/aosp_15_r20/frameworks/base/core/java/com/android/internal/widget/remotecompose/core/operations/layout/modifiers/
H A DShapeType.java19 public class ShapeType { class
/aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/
H A D_visualize_sharding.py89 global_shape: ShapeType,
/aosp_15_r20/external/skia/src/gpu/graphite/geom/
H A DAnalyticBlurMask.h34 enum class ShapeType { enum
/aosp_15_r20/external/skia/gm/
H A Ddstreadshuffle.cpp41 enum ShapeType { enum in skiagm::DstReadShuffle
/aosp_15_r20/external/gson/gson/src/test/java/com/google/gson/functional/
H A DRuntimeTypeAdapterFactoryFunctionalTest.java75 public enum ShapeType { enum in RuntimeTypeAdapterFactoryFunctionalTest
/aosp_15_r20/external/skia/modules/skottie/src/layers/shapelayer/
H A DShapeLayer.cpp88 enum class ShapeType { enum
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Ddebug_utils.py481 shape: "torch._prims_common.ShapeType",
/aosp_15_r20/external/pytorch/torch/_refs/nn/functional/
H A D__init__.py348 normalized_shape: ShapeType,
/aosp_15_r20/external/skia/tests/
H A DGrClipStackTest.cpp1630 enum ShapeType { kRect, kRRect, kConvex }; in DEF_TEST() enum
/aosp_15_r20/out/soong/.intermediates/frameworks/base/framework-minus-apex/android_common/javac/
Dframework.jar27META-INF/ META-INF/MANIFEST.MF com/ com/android/ com/ ...
/aosp_15_r20/out/soong/.intermediates/frameworks/base/framework-minus-apex/android_common/repackaged-jarjar/javac-27/
Dframework.jarcom/android/internal/widget/remotecompose/core/operations/utilities/StringSerializer.class ...
/aosp_15_r20/out/soong/.intermediates/libcore/core-libart/android_common_apex31/turbine/
Dcore-libart.jarjava/net/AddressCache$AddressCacheKey.class <Unknown> package java.net java.net ...
/aosp_15_r20/out/target/common/obj/JAVA_LIBRARIES/core-libart.com.android.art.debug_intermediates/
Dclasses-header.jarandroid/compat/Compatibility$BehaviorChangeDelegate.class <Unknown> package android.compat public abstract android ...
/aosp_15_r20/out/soong/.intermediates/libcore/core-libart/android_common_apex31/turbine-combined/
Dcore-libart.jarandroid/compat/Compatibility$BehaviorChangeDelegate.class <Unknown> package android.compat public abstract android ...
/aosp_15_r20/out/soong/.intermediates/libcore/core-all/android_common/turbine-combined/
Dcore-all.jarandroid/annotation/Discouraged.class <Unknown> package android.annotation public abstract annotation android. ...
/aosp_15_r20/out/soong/.intermediates/libcore/core-all/android_common/turbine/
Dcore-all.jarjava/awt/font/NumericShaper$Range.class <Unknown> package java.awt. ...
/aosp_15_r20/out/soong/.intermediates/frameworks/base/framework-minus-apex-headers/android_common/turbine/
Dframework.jarandroid/annotation/AnimRes.class <Unknown> package android.annotation public abstract annotation android. ...

12