Searched refs:kComplex64 (Results 1 – 7 of 7) 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 | rfft.cc | 39 return kComplex64; in InferType()
|
| D | op_utils.h | 286 kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64,
|
| /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 | dtype_extends.cc | 66 …{kNumberTypeFloat64, kFloat64}, {kNumberTypeComplex64, kComplex64}, {kNumberTypeInt8, kInt8}, in TypeIdToType()
70 …{kNumberTypeComplex64, kComplex64}, {kNumberTypeComplex128, kComplex128}, {kMetaTypeExternal, kTyp… in TypeIdToType()
|
| /third_party/mindspore/mindspore/core/ir/dtype/ |
| D | number.h | 184 inline const TypePtr kComplex64 = std::make_shared<Complex>(static_cast<int>(BitsNum::eBits64)); variable
|