Home
last modified time | relevance | path

Searched refs:UNSORTEDSEGMENTSUM_INPUTS_SIZE (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc1676 …is_inputs_should_in_memory_[UNSORTEDSEGMENTSUM_INPUTS_SIZE - 1] = is_parameter_[UNSORTEDSEGMENTSUM… in CalculateInputsInMemory()
1752 if ((prev_output_in_mem.find(UNSORTEDSEGMENTSUM_INPUTS_SIZE - 1) == prev_output_in_mem.end()) || in CalculateInputsInMemory()
1753 (!prev_output_in_mem.at(UNSORTEDSEGMENTSUM_INPUTS_SIZE - 1))) { in CalculateInputsInMemory()
1754 is_inputs_should_in_memory_[UNSORTEDSEGMENTSUM_INPUTS_SIZE - 1] = true; in CalculateInputsInMemory()
1760 if (!is_inputs_should_in_memory_[UNSORTEDSEGMENTSUM_INPUTS_SIZE - 1]) { in CalculateInputsInMemory()
1761 …is_inputs_should_in_memory_[UNSORTEDSEGMENTSUM_INPUTS_SIZE - 1] = is_parameter_[UNSORTEDSEGMENTSUM… in CalculateInputsInMemory()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dops_utils.h48 constexpr size_t UNSORTEDSEGMENTSUM_INPUTS_SIZE = 3; variable