Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcostmodel_context.h139 bool elementwise_stra_follow() const { return elementwise_stra_follow_; } in elementwise_stra_follow() function
Dstep_auto_parallel.cc611 const auto stra_follow = CostModelContext::GetInstance()->elementwise_stra_follow(); in CreateEdgeBetweenTwoOps()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dedge_costmodel.cc94 const auto stra_follow = CostModelContext::GetInstance()->elementwise_stra_follow(); in InitEdgeCost()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc287 .def("get_elementwise_op_strategy_follow", &CostModelContext::elementwise_stra_follow, in PYBIND11_MODULE()