Searched refs:MatMulBase (Results 1 – 2 of 2) sorted by relevance
33 class MatMulBase : public OperatorInfo {35 MatMulBase(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MatMulBase() function38 ~MatMulBase() override = default;71 class MatMul : public MatMulBase {74 : MatMulBase(name, inputs_shape, outputs_shape, attrs) {} in MatMul()
70 Status MatMulBase::GetAttrs() { in GetAttrs()206 Status MatMulBase::InferDevMatrixShape() { in InferDevMatrixShape()216 Status MatMulBase::InferForwardCommunication() { in InferForwardCommunication()248 Status MatMulBase::InferTensorMap() { in InferTensorMap()314 Status MatMulBase::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout) { in InferTensorLayout()343 Status MatMulBase::InferTensorInfo() { in InferTensorInfo()363 Status MatMulBase::Init(const StrategyPtr &strategy) { in Init()378 Status MatMulBase::InitForCostModel(const StrategyPtr &strategy) { in InitForCostModel()388 Status MatMulBase::SwapLastTwoElements(mindspore::parallel::Shape *const input) { in SwapLastTwoElements()402 Status MatMulBase::GenerateStrategies(int64_t stage_id) { in GenerateStrategies()[all …]