Searched defs:InferTensorLayout (Results 1 – 5 of 5) sorted by relevance
62 Status GetNextInfo::InferTensorLayout(TensorLayouts *outputs_layout) { in InferTensorLayout() function in mindspore::parallel::GetNextInfo
471 Status QuantBatchMatmulInfo::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs… in InferTensorLayout() function in mindspore::parallel::QuantBatchMatmulInfo
481 Status WeightQuantBatchMatmulInfo::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *o… in InferTensorLayout() function in mindspore::parallel::WeightQuantBatchMatmulInfo
530 Status ReshapeInfo::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout) { in InferTensorLayout() function in mindspore::parallel::ReshapeInfo
912 Status MatMulBase::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout) { in InferTensorLayout() function in mindspore::parallel::MatMulBase