Searched defs:CalculateInputsInMemory (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() function in mindspore::parallel::MatMulCost 246 void CastCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::CastCost 254 void GeLUCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory() function in mindspore::parallel::GeLUCost 317 void SoftmaxCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::SoftmaxCost 325 void PackCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::PackCost 333 void TileCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory() function in mindspore::parallel::TileCost 354 void BroadcastToCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::BroadcastToCost 359 void ReLU6Cost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory() function in mindspore::parallel::ReLU6Cost 370 void TransposeCost::CalculateInputsInMemory(const std::map<size_t, bool> &prev_output_in_mem) { in CalculateInputsInMemory() function in mindspore::parallel::TransposeCost 421 void TmpIdentityCost::CalculateInputsInMemory(const std::map<size_t, bool> &) { in CalculateInputsInMemory() function in mindspore::parallel::TmpIdentityCost [all …]
|