Searched defs:ndarray (Results 26 – 31 of 31) sorted by relevance
12
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _ndarray.py | 185 class ndarray: class
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 11352 def reference_one_hot(a: np.ndarray, num_classes: int = -1) -> np.ndarray:
11372 def reference_layer_norm(inp: np.ndarray, normalized_shape: Tuple[int], weight=None, bias=None, eps…
11376 def reference_native_layer_norm(inp: np.ndarray, normalized_shape: Tuple[int], weight, bias, eps):
11393 def reference_rms_norm(inp: np.ndarray, normalized_shape: Tuple[int], weight=None, eps=None):
11405 def reference_group_norm(inp: np.ndarray, num_groups: int, weight=None, bias=None, eps=1e-5):
11491 def wrapper(x: np.ndarray, *args, **kwargs):
|
| /aosp_15_r20/external/python/cpython3/Lib/test/ |
| D | test_buffer.py | 30 ndarray = None variable
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | common.py | 1353 def adapt_onnx_outputs_to_pt(self, onnx_outputs: List[np.ndarray]) -> Any:
|
| /aosp_15_r20/external/bazelbuild-rules_rust/crate_universe/test_data/metadata/crate_optional_deps_disabled_build_dep_enabled/ |
| H A D | metadata.json | 2068 "ndarray": [ array
|
| /aosp_15_r20/external/bazelbuild-rules_rust/crate_universe/test_data/metadata/aliases/ |
| H A D | metadata.json | 47487 "ndarray": [ array
|
12