Searched defs:GetForwardComputationCost (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 129 double MatMulCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, in GetForwardComputationCost() function in mindspore::parallel::MatMulCost
252 double BatchNormCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::v… in GetForwardComputationCost() function in mindspore::parallel::BatchNormCost
307 double CastCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vector… in GetForwardComputationCost() function in mindspore::parallel::CastCost
373 double SoftmaxCost::GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<Te… in GetForwardComputationCost() function in mindspore::parallel::SoftmaxCost
483 double TmpIdentityCost::GetForwardComputationCost(const std::vector<mindspore::parallel::TensorInfo… in GetForwardComputationCost() function in mindspore::parallel::TmpIdentityCost
504 double BatchParallelCost::GetForwardComputationCost(const std::vector<mindspore::parallel::TensorIn… in GetForwardComputationCost() function in mindspore::parallel::BatchParallelCost
610 double PReLUCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vecto… in GetForwardComputationCost() function in mindspore::parallel::PReLUCost
677 double OneHotCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vect… in GetForwardComputationCost() function in mindspore::parallel::OneHotCost
718 double SoftmaxCrossEntropyWithLogitsCost::GetForwardComputationCost(const std::vector<TensorInfo> &… in GetForwardComputationCost() function in mindspore::parallel::SoftmaxCrossEntropyWithLogitsCost
784 double ReshapeCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, in GetForwardComputationCost() function in mindspore::parallel::ReshapeCost
[all …]
|
| D | operator_costmodel.h | 451 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost() function
485 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost() function
876 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost() function
|