Home
last modified time | relevance | path

Searched defs:infer_impl (Results 1 – 25 of 56) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_nonzero.cc51 auto infer_impl = GetOpFrontendFuncImplPtr("NonZero"); in TEST_P() local
62 auto infer_impl = std::make_shared<NonZeroFuncImpl>(); in TEST_P() local
Dtest_ops_cmp_utils.cc48 void TestOpFuncImplWithEltwiseOpParams(const OpFuncImplPtr &infer_impl, const std::string &prim_nam… in TestOpFuncImplWithEltwiseOpParams()
75 void TestOpFuncImplWithMultiInputOpParams(const OpFuncImplPtr &infer_impl, const std::string &prim_… in TestOpFuncImplWithMultiInputOpParams()
Dtest_ops_seq_to_seq.cc51 auto infer_impl = GetOpFrontendFuncImplPtr("TupleToList"); in compare_tuple_to_list() local
75 auto infer_impl = GetOpFrontendFuncImplPtr("ListToTuple"); in compare_list_to_tuple() local
Dtest_ops_bool_not.cc37 auto infer_impl = std::make_shared<BoolNotFuncImpl>(); in TEST_P() local
Dtest_ops_pow.cc39 auto infer_impl = std::make_shared<PowFuncImpl>(); in TEST_P() local
Dtest_ops_minimum.cc39 auto infer_impl = std::make_shared<MinimumFuncImpl>(); in TEST_P() local
Dtest_ops_fast_gelu_grad.cc40 auto infer_impl = std::make_shared<FastGeLUGradFuncImpl>(); in TEST_P() local
Dtest_ops_mul.cc39 auto infer_impl = std::make_shared<MulFuncImpl>(); in TEST_P() local
Dtest_ops_complex.cc38 auto infer_impl = std::make_shared<ComplexFuncImpl>(); in TEST_P() local
Dtest_ops_softmax.cc52 auto infer_impl = std::make_shared<SoftmaxFuncImpl>(); in TEST_P() local
Dtest_ops_rank.cc41 OpFuncImplPtr infer_impl = std::make_shared<RankFuncImpl>(); in TEST_P() local
Dtest_ops_softplus_ext.cc55 auto infer_impl = std::make_shared<SoftplusExtFuncImpl>(); in TEST_P() local
Dtest_ops_split.cc46 auto infer_impl = std::make_shared<SplitFuncImpl>(); in TEST_P() local
Dtest_ops_less_equal.cc45 auto infer_impl = std::make_shared<LessEqualFuncImpl>(); in TEST_P() local
Dtest_ops_less.cc45 auto infer_impl = std::make_shared<LessFuncImpl>(); in TEST_P() local
Dtest_ops_cmp_utils.h32 auto infer_impl = std::make_shared<T>(); in DoFuncImplInferAndCompare() local
Dtest_ops_uniform_ext.cc56 auto infer_impl = std::make_shared<UniformExtFuncImpl>(); in TEST_P() local
Dtest_ops_betainc.cc52 auto infer_impl = std::make_shared<BetaincFuncImpl>(); in TEST_P() local
Dtest_ops_index_select_ext.cc54 auto infer_impl = std::make_shared<IndexSelectFuncImpl>(); in TEST_P() local
Dtest_ops_resize_nearest_neighbor_grad.cc55 auto infer_impl = std::make_shared<ResizeNearestNeighborGradFuncImpl>(); in TEST_P() local
Dtest_ops_avg_pool2d.cc62 auto infer_impl = std::make_shared<AvgPool2DFuncImpl>(); in TEST_P() local
Dtest_ops_resize_nearest_neighbor.cc54 auto infer_impl = std::make_shared<ResizeNearestNeighborFuncImpl>(); in TEST_P() local
Dtest_ops_rms_norm_grad.cc53 auto infer_impl = std::make_shared<RmsNormGradFuncImpl>(); in TEST_P() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.h93 #define REGISTER_PRIMITIVE_FRONT_EVAL_IMPL(name, primitive, infer_impl, infer_value_impl) … argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/expander/core/
Dinfer.cc61 auto &infer_impl = CppInfer::infer_impl_cache()[prim]; in InferAnfnode() local

123