Home
last modified time | relevance | path

Searched defs:ndarray (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_moe_finalize_routing.py34 def moe_finalize_routing(expanded_permuted_rows: np.ndarray,
35 skip1: np.ndarray,
36 skip2_optional: np.ndarray,
37 bias: np.ndarray,
38 scales: np.ndarray,
39 expanded_src_to_dst_row: np.ndarray,
40 expert_for_source_row: np.ndarray) -> np.ndarray:
/third_party/mindspore/mindspore-src/source/tests/st/scipy_st/
Dutils.py146 def _numpy_to_tensor(arg: List[onp.ndarray]) -> List[Tensor]:
149 def _add_value(arg: List[onp.ndarray], outer, inner, value):
153 def _flatten(arg: List[onp.ndarray]) -> onp.ndarray:
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_matmul.py62 def np_matmul(a: np.ndarray, b: np.ndarray, trans_a: bool, trans_b: bool):
Dtest_shift_op.py65 def numpy_shift(array: np.ndarray, periods: int, axis: int, fill_value=np.nan) -> np.ndarray:
93 def compare(arr: np.ndarray, periods: int, axis: int, fill_value=np.nan):
Dtest_inplace_op.py52 def inplace_op_np(op, x: np.ndarray, v: np.ndarray, indices):
Dtest_rolling_op.py89 def numpy_rolling(array: np.ndarray, window: int, min_periods: int, center: bool, axis: int, closed…
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/profiler/parser/
Dascend_op_generator.py32 self, op_summary: np.ndarray, op_statistic: np.ndarray,
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/mslite_bench/mslite_bench/infer_base/
Dtf_infer_session.py47 def infer(self, input_data_map: Dict[str, np.ndarray]) -> Dict[str, np.ndarray]:
Dpaddle_infer_session.py44 def infer(self, input_data_map: Dict[str, np.ndarray]) -> Dict[str, np.ndarray]:
Donnx_infer_session.py39 def infer(self, input_data_map: Dict[str, np.ndarray]) -> Dict[str, np.ndarray]:
Dabs_infer_session.py54 def infer(self, input_data_map: Dict[str, np.ndarray]):
Dmslite_infer_session.py59 def infer(self, input_data_map: Dict[str, np.ndarray]) -> Dict[str, np.ndarray]:
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/summary/
D_summary_adapter.py281 def _fill_histogram_summary(tag: str, np_value: np.ndarray, summary) -> None:
/third_party/python/Lib/test/
Dtest_buffer.py30 ndarray = None variable