Searched defs:CalculateOutputInMemory (Results 1 – 2 of 2) sorted by relevance
174 void MatMulCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::MatMulCost269 void BatchNormCost::CalculateOutputInMemory() { is_output_should_in_memory_ = true; } in CalculateOutputInMemory() function in mindspore::parallel::BatchNormCost322 void CastCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::CastCost330 void SqrtCost::CalculateOutputInMemory() { is_output_should_in_memory_ = is_parameter_involve_[0]; } in CalculateOutputInMemory() function in mindspore::parallel::SqrtCost393 void SoftmaxCost::CalculateOutputInMemory() { is_output_should_in_memory_ = is_parameter_involve_[0… in CalculateOutputInMemory() function in mindspore::parallel::SoftmaxCost401 void PackCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::PackCost409 void TileCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::TileCost431 void BroadcastToCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::BroadcastToCost497 void TmpIdentityCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::TmpIdentityCost546 void BatchParallelCost::CalculateOutputInMemory() { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function in mindspore::parallel::BatchParallelCost[all …]
1227 void CalculateOutputInMemory() override { is_output_should_in_memory_ = false; } in CalculateOutputInMemory() function