Home
last modified time | relevance | path

Searched refs:GetComputeFunction (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/utils/
Dprimitive_utils.cc44 py::function GetComputeFunction(const std::string &name) { in GetComputeFunction() function
91 auto func = GetComputeFunction(prim->name()); in RunComputeFunction()
Dprimitive_utils.h32 py::function GetComputeFunction(const std::string &name);
/third_party/mindspore/mindspore/ccsrc/pybind_api/ir/
Dprimitive_py.cc282 py::function PrimitivePy::GetComputeFunction() const { in GetComputeFunction() function in mindspore::PrimitivePy
298 vm_fn = mindspore::GetComputeFunction(Primitive::name()); in GetComputeFunction()
334 auto func = this->GetComputeFunction(); in RunPyComputeFunction()
343 auto func = GetComputeFunction(); in HasComputeFunction()
Dprimitive_py.h79 py::function GetComputeFunction() const;