Searched defs:CalculateInputsInMemory (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() function in mindspore::parallel::MatMulCost
272 void BatchNormCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::BatchNormCost
325 void CastCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::CastCost
333 void GeLUCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory() function in mindspore::parallel::GeLUCost
396 void SoftmaxCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::SoftmaxCost
404 void PackCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::PackCost
412 void TileCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory() function in mindspore::parallel::TileCost
433 void BroadcastToCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::BroadcastToCost
438 void ReLU6Cost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory() function in mindspore::parallel::ReLU6Cost
449 void TransposeCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory() function in mindspore::parallel::TransposeCost
[all …]
|
| D | operator_costmodel.h | 1229 void CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) override { in CalculateInputsInMemory() function
|