Home
last modified time | relevance | path

Searched defs:func_map (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend/api/
Dascend_kernel_api.cc23 …std::map<std::string, CreatorFunc> *func_map = new (std::nothrow) std::map<std::string, CreatorFun… in CreateCustomAscendKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dbessel_helper.h86func_map = {{BESSEL_OP_J0, CalBesselJ0<T>}, {BESSEL_OP_J1, CalBesselJ1<T>}, {BESSEL_OP_K0, Cal… in Process() local
Dunary_helper.h118 …std::map<UnaryOptype, std::function<void(const T *, T *, const size_t, cudaStream_t)>> func_map = { in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dreal_div.cc118 std::map<TypeId, std::function<void(void *x, void *y, void *result, size_t size)>> func_map = { member in mindspore::ops::RealDivFrontendFuncImpl
Ddiv.cc118 std::map<TypeId, std::function<void(void *x, void *y, void *result, size_t size)>> func_map = { member in mindspore::ops::DivFrontendFuncImpl
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dreciprocal.cc102 std::map<TypeId, std::function<void(void *origin, void *target, size_t size)>> func_map = { member in mindspore::ops::ReciprocalFrontendFuncImpl
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/sequence/
Dscalar_arithmetic_one_input_cpu_kernel.cc84 std::unordered_map<std::string, MathImplFunc> func_map = { in LaunchKernel() local
Dsequence_less_cpu_kernel.cc101 std::unordered_map<std::string, InequalityImplFunc> func_map = { in LaunchKernel() local
Dscalar_arithmetic_cpu_kernel.cc240 std::unordered_map<std::string, MathImplFunc> func_map = { in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_tensor_scatter_arithmetic.py33 func_map = { variable
Dtest_scatter_func_op.py27 func_map = { variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/
Dpi_jit_config.cc175 py::object func_map = Utils::GetModuleAttr(kModuleName, map_name, true, true); in AddToFuncMap() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_scatter_nd_func_op.py34 func_map = { variable
Dtest_scatter_func_op.py28 func_map = { variable
/third_party/mindspore/mindspore-src/source/tests/ut/python/dataset/
Dtest_filterop.py293 def func_map(data_col1, data_col2): function
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_scatter_nd_arithmetic_op.py29 func_map = { variable
Dtest_tensor_scatter_arithmetic.py27 func_map = { variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/
Dstatistic_kernel.cc322 …std::map<std::string, std::function<TensorPtr(const DeviceContext *, KernelTensor *)>> func_map = { in CalStatistic() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dop_node.cc225 mindspore::HashMap<std::string, std::function<TM(const TM &)>> func_map = { in CalcByOperator() local
255 mindspore::HashMap<std::string, std::function<TM(const TM &, const TM &)>> func_map = { in CalcByOperator() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_capture/
Dspecial_func_infer.cc867 py::object func_map = Utils::GetModuleAttr(kModuleName, kFuncMapName, true, true); in GetFuncKeyMap() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/
Dtrace.cc2293 …py::object func_map = Utils::GetModuleAttr(kFuncWhiteListModuleName, kGuardFuncMapName, true, true… in GetGuardFuncKeyMap() local