Searched refs:out_int32 (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_searchsorted_op.py | 29 def __init__(self, out_int32=False, right=False): argument
31 self.searchsorted = P.SearchSorted(out_int32=out_int32, right=right)
46 net = SearchSortedNet(out_int32=True, right=True)
68 net = SearchSortedNet(out_int32=True, right=False)
90 net = SearchSortedNet(out_int32=False, right=True)
109 net = SearchSortedNet(out_int32=False, right=False)
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_search_sorted.py | 26 def __init__(self, right=False, out_int32=True): argument
28 self.search = P.SearchSorted(out_int32=out_int32, right=right)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 6160 def __init__(self, out_int32=False, right=False): argument
6162 self.out_int32 = validator.check_value_type("out_int32", out_int32, [bool], self.name)
6178 … return mstype.tensor_type(mstype.int32) if self.out_int32 else mstype.tensor_type(mstype.int64)
|
| /third_party/mindspore/config/ |
| D | op_info.config | 28 …x": 0, "name": "output", "param_type": "required"}], "attr": [{"name": "out_int32", "type": "bool"…
|