Searched defs:CalculateOutputInMemory (Results 1 – 1 of 1) sorted by relevance
172 void MatMulCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::MatMulCost243 void CastCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::CastCost251 void SqrtCost::CalculateOutputInMemory() { is_output_should_in_memory_ = is_parameter_involve_[0]; } in CalculateOutputInMemory() function in mindspore::parallel::SqrtCost314 void SoftmaxCost::CalculateOutputInMemory() { is_output_should_in_memory_ = is_parameter_involve_[0… in CalculateOutputInMemory() function in mindspore::parallel::SoftmaxCost322 void PackCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::PackCost330 void TileCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::TileCost352 void BroadcastToCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::BroadcastToCost418 void TmpIdentityCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::TmpIdentityCost467 void BatchParallelCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::BatchParallelCost493 void SparseSoftmaxCrossEntropyWithLogitsCost::CalculateOutputInMemory() { in CalculateOutputInMemory() function in mindspore::parallel::SparseSoftmaxCrossEntropyWithLogitsCost[all …]