Searched refs:kUInt16 (Results 1 – 12 of 12) 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 | reverse_v2.cc | 35 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, kUInt16, 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 | 281 …nst std::set<TypePtr> common_valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, kUInt16,
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 | 226 UInt16Imm() : IntergerImm(kUInt16), v_(0) {} in UInt16Imm()
227 explicit UInt16Imm(uint16_t v) : IntergerImm(kUInt16), v_(v) { in UInt16Imm()
|
| D | dtype_extends.cc | 68 …mberTypeInt64, kInt64}, {kNumberTypeUInt8, kUInt8}, {kNumberTypeUInt16, kUInt16}, in TypeIdToType()
|
| /third_party/mindspore/mindspore/core/ir/dtype/ |
| D | number.h | 174 inline const TypePtr kUInt16 = std::make_shared<UInt>(static_cast<int>(BitsNum::eBits16)); variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 775 } else if (type_id == kUInt16->type_id()) { in GetInputTensorValue()
|