| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | leaky_relu_ext.cc | 32 static const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64, kBFloat16}; in InferType() local
43 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64, kBFloat16}; in InferType() local
|
| D | gelu.cc | 42 const std::set<TypePtr> valid_types = {kBFloat16, kFloat16, kFloat32, kFloat64}; in InferType() local
51 const std::set<TypePtr> valid_types = {kBFloat16, kFloat16, kFloat32, kFloat64}; in InferType() local
|
| D | ceil.cc | 44 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64, kBFloat16}; in InferType() local
53 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64, kBFloat16}; in InferType() local
|
| D | fft_with_size.cc | 200 auto valid_types = get_input_types(kRfftTypes); in InferType() local
204 auto valid_types = get_input_types(kIrfftTypes); in InferType() local
209 auto valid_types = get_input_types(kFftTypes); in InferType() local
|
| D | addmm.cc | 72 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kBFloat16}; in InferType() local
83 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kBFloat16}; in InferType() local
|
| D | softmax.cc | 37 const std::set<TypePtr> valid_types{kBFloat16, kFloat16, kFloat32, kFloat64}; in InferType() local
68 const std::set<TypePtr> valid_types{kBFloat16, kFloat16, kFloat32, kFloat64}; in InferType() local
|
| D | binary_cross_entropy_with_logits.cc | 37 std::set<TypePtr> valid_types = {kFloat32, kFloat16, kBFloat16}; in InferType() local
137 std::set<TypePtr> valid_types = {kFloat32, kFloat16, kBFloat16}; in InferType() local
|
| D | hsigmoid.cc | 33 …const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32, kFloat64… in InferType() local
|
| D | relu6.cc | 34 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in InferType() local
|
| D | ffn_ext.cc | 36 const std::set<TypePtr> valid_types = {kFloat16, kBFloat16, kInt8}; in InferType() local
|
| D | hshrink.cc | 42 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in InferType() local
|
| D | fast_gelu.cc | 30 std::vector<TypeId> valid_types = {kNumberTypeFloat16, kNumberTypeFloat32, kNumberTypeBFloat16}; in InferType() local
|
| D | fast_gelu_grad.cc | 30 std::vector<TypeId> valid_types = {kNumberTypeFloat16, kNumberTypeFloat32, kNumberTypeBFloat16}; in InferType() local
|
| D | hshrink_grad.cc | 42 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in InferType() local
|
| D | right_shift.cc | 44 …const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, kUInt16, kUInt32, kU… in InferType() local
|
| D | trace.cc | 49 …const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32, … in InferType() local
|
| D | dropout.cc | 43 const std::set<TypePtr> valid_types = {kFloat16, kBFloat16, kFloat32, kFloat64}; in InferType() local
|
| D | scalar_to_tensor.cc | 44 …const std::set<TypePtr> valid_types = {kBool, kInt8, kInt16, kInt32, kInt64, kUInt8, … in InferType() local
|
| D | npu_clear_float_status_v2.cc | 56 std::set<TypePtr> valid_types = {kInt32}; in InferType() local
|
| D | npu_get_float_status_v2.cc | 57 std::set<TypePtr> valid_types = {kInt32}; in InferType() local
|
| D | embedding_apply_ada_grad.cc | 48 const std::set<TypePtr> valid_types = {kFloat16, kFloat32}; in InferType() local
|
| D | exp.cc | 32 const std::set<TypePtr> valid_types = {kInt64, kBool, kFloat16, kFloat32, in InferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | dtype.cc | 65 const std::set<TypePtr> valid_types = {kTensorType}; in InferValue() local
69 const std::set<TypePtr> valid_types = {kCSRTensorType, kCOOTensorType}; in InferValue() local
|
| D | nth_element.cc | 65 const std::set<TypePtr> valid_types = {kInt32}; in NthElementInferShape() local
103 …const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kInt64, kInt32, kInt16, kInt8, kUInt8, … in NthElementInferType() local
|
| D | fmin.cc | 53 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64, kInt32, kInt64}; in InferType() local
|