| /third_party/mindspore/mindspore/core/ops/ |
| D | ceil.cc | 35 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in CeilInfer() local
|
| D | atan.cc | 35 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kInt32}; in AtanInfer() local
|
| D | tan.cc | 41 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kInt32}; in TanInfer() local
|
| D | asin.cc | 38 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kInt32}; in AsinInfer() local
|
| D | sigmoid_cross_entropy_with_logits.cc | 43 …const std::set<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUIn… in SigmoidCrossEntropyWithLogitsInfer() local
|
| D | dtype.cc | 35 const std::set<TypePtr> valid_types = {kTensorType}; in DTypeInferValue() local
|
| D | hshrink.cc | 38 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in InferType() local
|
| D | floor.cc | 35 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64}; in InferType() local
|
| D | logical_and.cc | 40 const std::set<TypePtr> valid_types = {kBool}; in InferType() local
|
| D | squared_difference.cc | 34 const std::set<TypePtr> valid_types = {kInt32, kFloat16, kFloat32}; in InferType() local
|
| D | logical_or.cc | 38 const std::set<TypePtr> valid_types = {kBool}; in InferType() local
|
| D | dropout.cc | 54 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in DropoutInfer() local
|
| D | softmax_cross_entropy_with_logits.cc | 43 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in SoftmaxCrossEntropyWithLogitsInfer() local
|
| D | layer_norm_x_backprop_v2.cc | 40 std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in LayerNormXBackpropV2InferType() local
|
| D | hsigmoid.cc | 38 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}; in InferType() local
|
| D | smooth_l1_loss.cc | 49 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in SmoothL1LossInfer() local
|
| D | LayerNormXBackprop.cc | 39 std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in LayerNormXBackpropInferType() local
|
| D | soft_shrink.cc | 50 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in InferType() local
|
| D | relu6.cc | 44 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in InferType() local
|
| D | roll.cc | 45 const std::set<TypePtr> valid_types = {kFloat32, kFloat16, kInt32, kUInt32, kInt8, kUInt8}; in InferType() local
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | sigmoid_cross_entropy_with_logits_grad.cc | 46 …const std::set<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUIn… in SigmoidCrossEntropyWithLogitsGradInfer() local
|
| D | smooth_l1_loss_grad.cc | 53 …const std::set<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUIn… in SmoothL1LossGradInfer() local
|
| D | soft_shrink_grad.cc | 50 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in SoftShrinkGradInferType() local
|
| D | hshrink_grad.cc | 52 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in HShrinkGradInferType() local
|
| D | soft_margin_loss_grad.cc | 45 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in SoftMarginLossGradInferType() local
|