Searched defs:prev_output_in_mem (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 177 void MatMulCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
333 void GeLUCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
412 void TileCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
438 void ReLU6Cost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
449 void TransposeCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
548 void BatchParallelCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
649 void PReLUCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
905 void MulCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
935 void DivCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
959 void ModCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory()
[all …]
|
| D | operator_costmodel.h | 1229 void CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) override { in CalculateInputsInMemory()
|