out_nums (reference) in projects: third_party

Searched refs:out_nums (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
D	test_split.py	`26 …def __init__(self, mul_weight, axis=0, out_nums=1, strategy1=None, strategy2=None, strategy3=None): argument 28 self.split = P.Split(axis, out_nums).shard(strategy1) 43 def __init__(self, mul_weight, axis=0, out_nums=1, strategy1=None, strategy2=None): argument 45 self.split = P.Split(axis, out_nums).shard(strategy1) 57 def __init__(self, mul_weight, axis=0, out_nums=1, strategy1=None, strategy2=None): argument 59 self.split = P.Split(axis, out_nums).shard(strategy1)`
/third_party/mindspore/tests/st/ops/gpu/
D	test_split.py	`26 def __init__(self, axis=0, out_nums=1): argument 28 self.split = P.Split(axis, out_nums) 35 def __init__(self, axis=0, out_nums=1): argument 38 self.split = P.Split(axis, out_nums)`
D	test_conv2d_op.py	`172 def __init__(self, axis=0, out_nums=1): argument`