Searched defs:prev_output_in_mem (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 175 void MatMulCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
254 void GeLUCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
333 void TileCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
359 void ReLU6Cost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
370 void TransposeCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
469 void BatchParallelCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
570 void PReLUCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
822 void MulCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
852 void DivCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
876 void ModCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
[all …]
|