Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h171 int64_t is_output_critical() const { return is_output_critical_; } in is_output_critical()
172 void mark_output_critical() { is_output_critical_ = 1; } in mark_output_critical()
173 void mark_output_not_critical() { is_output_critical_ = 0; } in mark_output_not_critical()
272 int64_t is_output_critical_ = -1; variable
Doperator_info.cc1537 if (is_output_critical_ == -1) { in CalculateMemoryCostForInference()
1541 operator_cost()->set_output_critical(is_output_critical_); in CalculateMemoryCostForInference()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dedge_costmodel.h145 void mark_output_critical() { is_output_critical_ = 1; } in mark_output_critical()
174 int64_t is_output_critical_ = 0; variable
Dedge_costmodel.cc320 if ((is_output_critical_ != 0) && (is_output_critical_ != 1)) { in CalculateMemoryCostForInference()
321 MS_LOG(ERROR) << "Failure: unexpected output critical flag value: " << is_output_critical_; in CalculateMemoryCostForInference()