Searched refs:mask_dtype (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | reluv2.cc | 98 auto mask_dtype = kUInt8; in InferType() local
99 return std::make_shared<Tuple>(std::vector<TypePtr>{x_type, mask_dtype}); in InferType()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | derelu_fusion.cc | 52 TypeId mask_dtype = kNumberTypeUInt8; in CreateReluV2() local
77 auto types = {AnfAlgo::GetOutputInferDataType(relu, 0), mask_dtype}; in CreateReluV2()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 1532 … dy_dtype, dh_dtype, update_dtype, reset_dtype, new_dtype, hnew_dtype, seq_dtype, mask_dtype): argument
1544 if mask_dtype is not None:
1545 validator.check_tensor_dtype_valid("mask_dtype", mask_dtype, valid_types, self.name)
|
| D | nn_ops.py | 7353 mask_dtype = mstype.tensor_type(mstype.bool_)
7354 return x_dtype, mask_dtype
7410 mask_dtype = mstype.tensor_type(mstype.bool_)
7411 return x_dtype, mask_dtype
|
| D | array_ops.py | 6118 def check_dtype(self, x_dtype, mask_dtype): argument
6119 validator.check_tensor_dtype_valid('mask', mask_dtype, [mstype.bool_], self.name)
|