Searched refs:kUInt32 (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | sigmoid_cross_entropy_with_logits.cc | 44 … kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64}; in SigmoidCrossEntropyWithLogitsInfer()
|
| D | roll.cc | 45 const std::set<TypePtr> valid_types = {kFloat32, kFloat16, kInt32, kUInt32, kInt8, kUInt8}; in InferType()
|
| D | reverse_v2.cc | 36 kUInt32, kUInt64, kFloat16, kFloat32, kFloat64, kBool}; in InferType()
|
| D | zeros.cc | 49 kUInt16, kUInt32, kUInt64, kFloat16, kFloat32, kFloat64}; in ZerosInferType()
|
| D | ones.cc | 47 kUInt16, kUInt32, kUInt64, kFloat16, kFloat32, kFloat64}; in OnesInferType()
|
| D | op_utils.h | 282 kUInt32, kUInt64, kFloat16, kFloat32, kFloat64};
286 kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64,
|
| D | split_v.cc | 89 kUInt16, kUInt32, kUInt64, kFloat16, kFloat32}; in InferType()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | sigmoid_cross_entropy_with_logits_grad.cc | 47 … kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64}; in SigmoidCrossEntropyWithLogitsGradInfer()
|
| D | smooth_l1_loss_grad.cc | 54 … kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64}; in SmoothL1LossGradInfer()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | scalar.h | 254 UInt32Imm() : IntergerImm(kUInt32), v_(0) {} in UInt32Imm()
255 explicit UInt32Imm(uint32_t v) : IntergerImm(kUInt32), v_(v) { in UInt32Imm()
|
| D | dtype_extends.cc | 69 …{kNumberTypeUInt32, kUInt32}, {kNumberTypeUInt64, kUInt64}, {kNumberTypeBool, kBool}, in TypeIdToType()
|
| /third_party/mindspore/mindspore/core/ir/dtype/ |
| D | number.h | 175 inline const TypePtr kUInt32 = std::make_shared<UInt>(static_cast<int>(BitsNum::eBits32)); variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 773 } else if (type_id == kUInt32->type_id()) { in GetInputTensorValue()
|