Home
last modified time | relevance | path

Searched defs:CalculateInputsInMemory (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc175 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 …]