Searched defs:GetForwardComputationCost (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 128 double MatMulCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, in GetForwardComputationCost() function in mindspore::parallel::MatMulCost
228 double CastCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vector… in GetForwardComputationCost() function in mindspore::parallel::CastCost
294 double SoftmaxCost::GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<Te… in GetForwardComputationCost() function in mindspore::parallel::SoftmaxCost
404 double TmpIdentityCost::GetForwardComputationCost(const std::vector<mindspore::parallel::TensorInfo… in GetForwardComputationCost() function in mindspore::parallel::TmpIdentityCost
425 double BatchParallelCost::GetForwardComputationCost(const std::vector<mindspore::parallel::TensorIn… in GetForwardComputationCost() function in mindspore::parallel::BatchParallelCost
531 double PReLUCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vecto… in GetForwardComputationCost() function in mindspore::parallel::PReLUCost
598 double OneHotCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vect… in GetForwardComputationCost() function in mindspore::parallel::OneHotCost
639 double SoftmaxCrossEntropyWithLogitsCost::GetForwardComputationCost(const std::vector<TensorInfo> &… in GetForwardComputationCost() function in mindspore::parallel::SoftmaxCrossEntropyWithLogitsCost
705 double ReshapeCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, in GetForwardComputationCost() function in mindspore::parallel::ReshapeCost
734 double SubCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<… in GetForwardComputationCost() function in mindspore::parallel::SubCost
[all …]
|
| D | operator_costmodel.h | 403 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost() function
437 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost() function
804 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost() function
|