Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.h407 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
442 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
738 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
809 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
843 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
876 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
909 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
936 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
966 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost() function
1026 …double GetBackwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorI… in GetBackwardComputationCost() function
[all …]
Doperator_costmodel.cc147 double MatMulCost::GetBackwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vec… in GetBackwardComputationCost() function in mindspore::parallel::MatMulCost
237 double CastCost::GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<Tens… in GetBackwardComputationCost() function in mindspore::parallel::CastCost
308 double SoftmaxCost::GetBackwardComputationCost(const std::vector<mindspore::parallel::TensorInfo> &, in GetBackwardComputationCost() function in mindspore::parallel::SoftmaxCost
411 double TmpIdentityCost::GetBackwardComputationCost(const std::vector<mindspore::parallel::TensorInf… in GetBackwardComputationCost() function in mindspore::parallel::TmpIdentityCost
435 double BatchParallelCost::GetBackwardComputationCost(const std::vector<mindspore::parallel::TensorI… in GetBackwardComputationCost() function in mindspore::parallel::BatchParallelCost
543 double PReLUCost::GetBackwardComputationCost(const std::vector<mindspore::parallel::TensorInfo> &in… in GetBackwardComputationCost() function in mindspore::parallel::PReLUCost
607 double OneHotCost::GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<Te… in GetBackwardComputationCost() function in mindspore::parallel::OneHotCost
651 double SoftmaxCrossEntropyWithLogitsCost::GetBackwardComputationCost(const std::vector<TensorInfo> … in GetBackwardComputationCost() function in mindspore::parallel::SoftmaxCrossEntropyWithLogitsCost
722 double ReshapeCost::GetBackwardComputationCost(const std::vector<mindspore::parallel::TensorInfo> &, in GetBackwardComputationCost() function in mindspore::parallel::ReshapeCost
741 double SubCost::GetBackwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vector… in GetBackwardComputationCost() function in mindspore::parallel::SubCost
[all …]