Home
last modified time | relevance | path

Searched defs:InferType (Results 1 – 25 of 1013) sorted by relevance

12345678910>>...41

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dop_node.h195 …std::vector<TypeId> InferType(const NodePtrList &inputs, const DAttrs &) override { return {inputs… in InferType() function
214 …std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &) override { return {TypeId::kNum… in InferType() function
227 …std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &) override { return {TypeId::kNum… in InferType() function
240 std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &attrs) override { in InferType() function
253 …std::vector<TypeId> InferType(const NodePtrList &inputs, const DAttrs &) override { return {inputs… in InferType() function
263 …std::vector<TypeId> InferType(const NodePtrList &inputs, const DAttrs &) override { return {inputs… in InferType() function
317 …std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &) override { return {TypeId::kNum… in InferType() function
327 …std::vector<TypeId> InferType(const NodePtrList &inputs, const DAttrs &) override { return {inputs… in InferType() function
338 …std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &) override { return {TypeId::kNum… in InferType() function
348 …std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &) override { return {TypeId::kNum… in InferType() function
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dlogical_and.cc29 TypePtr LogicalAndFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::LogicalAndFuncImpl
34 TypePtrList LogicalAndFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_… in InferType() function in mindspore::ops::LogicalAndFuncImpl
Delu_grad_ext.cc28 TypePtr EluGradExtFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::EluGradExtFuncImpl
37 TypePtrList EluGradExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_… in InferType() function in mindspore::ops::EluGradExtFuncImpl
Dlogical_or.cc29 TypePtr LogicalOrFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::LogicalOrFuncImpl
34 TypePtrList LogicalOrFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_v… in InferType() function in mindspore::ops::LogicalOrFuncImpl
Dtanh_grad.cc30 TypePtr TanhGradFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::TanhGradFuncImpl
39 TypePtrList TanhGradFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_va… in InferType() function in mindspore::ops::TanhGradFuncImpl
Dmax.cc30 TypePtr MaxFuncImpl::InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &i… in InferType() function in mindspore::ops::MaxFuncImpl
33 TypePtrList MaxFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_values)… in InferType() function in mindspore::ops::MaxFuncImpl
Dlogical_not.cc29 TypePtr LogicalNotFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::LogicalNotFuncImpl
34 TypePtrList LogicalNotFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_… in InferType() function in mindspore::ops::LogicalNotFuncImpl
Delu_ext.cc28 TypePtr EluExtFuncImpl::InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr>… in InferType() function in mindspore::ops::EluExtFuncImpl
38 TypePtrList EluExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_valu… in InferType() function in mindspore::ops::EluExtFuncImpl
Dfloor.cc28 TypePtr FloorFuncImpl::InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> … in InferType() function in mindspore::ops::FloorFuncImpl
32 TypePtrList FloorFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_value… in InferType() function in mindspore::ops::FloorFuncImpl
Dmean_ext.cc29 TypePtr MeanExtFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::MeanExtFuncImpl
43 TypePtrList MeanExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_val… in InferType() function in mindspore::ops::MeanExtFuncImpl
Drelu.cc30 TypePtr ReLUFuncImpl::InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &… in InferType() function in mindspore::ops::ReLUFuncImpl
38 TypePtrList ReLUFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_values… in InferType() function in mindspore::ops::ReLUFuncImpl
Dmul.cc30 TypePtr MulFuncImpl::InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &i… in InferType() function in mindspore::ops::MulFuncImpl
39 TypePtrList MulFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_values)… in InferType() function in mindspore::ops::MulFuncImpl
Dzeros_like_ext.cc24 TypePtr ZerosLikeExtFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::ZerosLikeExtFuncImpl
41 TypePtrList ZerosLikeExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &inpu… in InferType() function in mindspore::ops::ZerosLikeExtFuncImpl
Dreduce_all.cc45 TypePtr ReduceAllFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::ReduceAllFuncImpl
56 TypePtrList ReduceAllFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_v… in InferType() function in mindspore::ops::ReduceAllFuncImpl
Dless_equal.cc46 TypePtr LessEqualFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::LessEqualFuncImpl
61 TypePtrList LessEqualFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_v… in InferType() function in mindspore::ops::LessEqualFuncImpl
Dones_like_ext.cc24 TypePtr OnesLikeExtFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::OnesLikeExtFuncImpl
41 TypePtrList OnesLikeExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input… in InferType() function in mindspore::ops::OnesLikeExtFuncImpl
Dless.cc36 TypePtr LessFuncImpl::InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &… in InferType() function in mindspore::ops::LessFuncImpl
50 TypePtrList LessFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_values… in InferType() function in mindspore::ops::LessFuncImpl
Dleaky_relu_grad_ext.cc30 TypePtr LeakyReLUGradExtFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::LeakyReLUGradExtFuncImpl
35 TypePtrList LeakyReLUGradExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &… in InferType() function in mindspore::ops::LeakyReLUGradExtFuncImpl
Dbinary_cross_entropy_with_logits_backward.cc29 TypePtr BinaryCrossEntropyWithLogitsBackwardFuncImpl::InferType( in InferType() function in mindspore::ops::BinaryCrossEntropyWithLogitsBackwardFuncImpl
50 TypePtrList BinaryCrossEntropyWithLogitsBackwardFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::BinaryCrossEntropyWithLogitsBackwardFuncImpl
Dbinary_cross_entropy_grad.cc29 TypePtr BinaryCrossEntropyGradFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::BinaryCrossEntropyGradFuncImpl
44 TypePtrList BinaryCrossEntropyGradFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::BinaryCrossEntropyGradFuncImpl
Duniform_ext.cc36 TypePtr UniformExtFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::UniformExtFuncImpl
46 TypePtrList UniformExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input_… in InferType() function in mindspore::ops::UniformExtFuncImpl
Dsoftplus_ext.cc35 TypePtr SoftplusExtFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::SoftplusExtFuncImpl
45 TypePtrList SoftplusExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input… in InferType() function in mindspore::ops::SoftplusExtFuncImpl
Dclamp_scalar.cc26 TypePtr ClampScalarFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::ClampScalarFuncImpl
45 TypePtrList ClampScalarFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &input… in InferType() function in mindspore::ops::ClampScalarFuncImpl
Dleaky_relu_ext.cc30 TypePtr LeakyReLUExtFuncImpl::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::LeakyReLUExtFuncImpl
38 TypePtrList LeakyReLUExtFuncImpl::InferType(const PrimitivePtr &primitive, const ValuePtrList &inpu… in InferType() function in mindspore::ops::LeakyReLUExtFuncImpl
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dfused_matmul_elemwise.cc48 TypePtr FusedMatMulElemBinary::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::FusedMatMulElemBinary
54 TypePtr FusedMatMulElemUnary::InferType(const PrimitivePtr &primitive, in InferType() function in mindspore::ops::FusedMatMulElemUnary

12345678910>>...41