Searched refs:InferTensorInfo (Results 1 – 16 of 16) sorted by relevance
54 Status InferTensorInfo() override;89 Status InferTensorInfo() override;
52 Status InferTensorInfo() override; // the inputs of GatherD is [tensor, int, tensor]
59 Status InferTensorInfo() override;
48 …Status InferTensorInfo() override; // The GetNext() has not input, so need override the InferTens…
54 …Status InferTensorInfo() override; // the forward_reduce_scatter mode need to override this funct…
78 Status InferTensorInfo() override;
57 Status InferTensorInfo() override;
112 Status GatherDInfo::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::GatherDInfo
347 Status ReduceMethod::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::ReduceMethod476 Status ArgMaxWithValueInfo::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::ArgMaxWithValueInfo
71 Status GetNextInfo::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::GetNextInfo
161 Status GatherInfo::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::GatherInfo
200 virtual Status InferTensorInfo();
241 Status ReshapeInfo::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::ReshapeInfo
202 Status OperatorInfo::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::OperatorInfo765 if (InferTensorInfo() != SUCCESS) { in InitForCostModelWithAutoRepeatCalc()813 if (InferTensorInfo() != SUCCESS) { in InitForCostModelWithManualRepeatCalc()
343 Status MatMulBase::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::MatMulBase
563 Status GatherPInfo::InferTensorInfo() { in InferTensorInfo() function in mindspore::parallel::GatherPInfo