| /third_party/mindspore/mindspore/core/ops/ |
| D | reluv2.cc | 38 if (x_dtype == kUInt8 || x_dtype == kInt8) { in GetOutputMaskShape()
49 if (x_dtype == kUInt8 || x_dtype == kInt8) { in GetOutputMaskShape()
98 auto mask_dtype = kUInt8; in InferType()
|
| D | sigmoid_cross_entropy_with_logits.cc | 43 …et<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUInt, kUInt8, in SigmoidCrossEntropyWithLogitsInfer()
|
| D | roll.cc | 45 const std::set<TypePtr> valid_types = {kFloat32, kFloat16, kInt32, kUInt32, kInt8, kUInt8}; in InferType()
|
| D | reverse_v2.cc | 35 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, kUInt16, in InferType()
|
| D | zeros.cc | 48 const std::set<TypePtr> valid_types = {kBool, kInt8, kInt16, kInt32, kInt64, kUInt8, in ZerosInferType()
|
| D | ones.cc | 46 const std::set<TypePtr> valid_types = {kBool, kInt8, kInt16, kInt32, kInt64, kUInt8, in OnesInferType()
|
| D | op_utils.h | 281 const std::set<TypePtr> common_valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, kUInt1…
285 kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUInt, kUInt8,
|
| D | index_add.cc | 66 …const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kUInt8, kFloat16, kFloat32, kFloat64… in IndexAddInferType()
|
| D | dropout_do_mask.cc | 110 …(void)CheckAndConvertUtils::CheckTensorTypeValid("inputs", input_args[1]->BuildType(), {kUInt8}, o… in InferType()
|
| D | split_v.cc | 88 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, in InferType()
|
| D | dropout_gen_mask.cc | 149 return kUInt8; in InferType()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/ascend310/src/ |
| D | custom_kernel.cc | 158 const auto kUInt8 = DataType::kNumberTypeUInt8; variable
162 REGISTER_CUSTOM_KERNEL(ASCEND310, ACL, kUInt8, ACL, kernel::acl::CustomCreateKernel)
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | sigmoid_cross_entropy_with_logits_grad.cc | 46 …et<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUInt, kUInt8, in SigmoidCrossEntropyWithLogitsGradInfer()
|
| D | smooth_l1_loss_grad.cc | 53 …et<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUInt, kUInt8, in SmoothL1LossGradInfer()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | dropout_unify_mindir.cc | 170 gen_mask_abstract = std::make_shared<abstract::AbstractTensor>(kUInt8, gen_mask_shape); in CreateDropoutGenMaskCNode()
173 gen_mask_abstract = std::make_shared<abstract::AbstractTensor>(kUInt8, gen_mask_shape); in CreateDropoutGenMaskCNode()
386 mask_abstract = std::make_shared<abstract::AbstractTensor>(kUInt8, mask_shape); in Process()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | scalar.h | 198 UInt8Imm() : IntergerImm(kUInt8), v_(0) {} in IMM_TRAITS()
199 explicit UInt8Imm(uint8_t v) : IntergerImm(kUInt8), v_(v) { in IMM_TRAITS()
|
| D | dtype_extends.cc | 68 …{kNumberTypeInt64, kInt64}, {kNumberTypeUInt8, kUInt8}, {kNumberTypeUInt16, kUIn… in TypeIdToType()
|
| /third_party/mindspore/mindspore/core/ir/dtype/ |
| D | number.h | 173 inline const TypePtr kUInt8 = std::make_shared<UInt>(static_cast<int>(BitsNum::eBits8)); variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 777 } else if (type_id == kUInt8->type_id()) { in GetInputTensorValue()
|