p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!1243 random choice with mask 

Merge pull request !1243 from wuxuejian/wxj_rcwm
This commit is contained in:
mindspore-ci-bot 2020-05-19 16:26:55 +08:00 committed by Gitee
parent 122a6e03a9 390b013e64
commit 75f4c598b0
3 changed files with 39 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/operator/prim_nn.cc
View File

@ -407,7 +407,11 @@ AbstractBasePtr InferImplDropoutGenMask(const AnalysisEnginePtr &, const Primiti
}
// convert to bytes(8 bits) mask, using round up
int bytes_count = (count + 7) / 8;
int n128s = count / 128;
if ((count % 128) != 0) {
n128s++;
}
int bytes_count = n128s * 16;
std::vector<int> shape_y{bytes_count};
primitive->set_attr("T", kInt32);

1
mindspore/ops/_op_impl/aicpu/__init__.py
View File

@ -23,3 +23,4 @@ from .reshape import _reshape_aicpu
from .flatten import _flatten_aicpu
from .squeeze import _squeeze_aicpu
from .expand_dims import _expand_dims_aicpu
from .random_choice_with_mask import _random_choice_with_mask_aicpu

33
mindspore/ops/_op_impl/aicpu/random_choice_with_mask.py Normal file
View File

@ -0,0 +1,33 @@
# Copyright 2020 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
"""RandomChoiceWithMask op"""
from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
random_choice_with_mask_op_info = AiCPURegOp("RandomChoiceWithMask") \
.fusion_type("OPAQUE") \
.input(0, "x", "required") \
.output(0, "y", "required") \
.output(1, "mask", "required") \
.attr("count", "int") \
.attr("seed", "int") \
.attr("seed2", "int") \
.dtype_format(DataType.BOOL_NCHW, DataType.I32_NCHW, DataType.BOOL_NCHW) \
.get_op_info()
@op_info_register(random_choice_with_mask_op_info)
def _random_choice_with_mask_aicpu():
"""RandomChoiceWithMask AiCPU register"""
return