Searched refs:GetComputeFunction (Results 1 – 4 of 4) sorted by relevance
44 py::function GetComputeFunction(const std::string &name) { in GetComputeFunction() function91 auto func = GetComputeFunction(prim->name()); in RunComputeFunction()
32 py::function GetComputeFunction(const std::string &name);
282 py::function PrimitivePy::GetComputeFunction() const { in GetComputeFunction() function in mindspore::PrimitivePy298 vm_fn = mindspore::GetComputeFunction(Primitive::name()); in GetComputeFunction()334 auto func = this->GetComputeFunction(); in RunPyComputeFunction()343 auto func = GetComputeFunction(); in HasComputeFunction()
79 py::function GetComputeFunction() const;