Searched defs:optim_type (Results 1 – 3 of 3) sorted by relevance
56 AddressPtr OptimizerInfoBuilder::GenInputAddrPtr(const std::string &optim_type, const std::string &… in GenInputAddrPtr()
46 void OptimizerInfo::UpdateOptimInputValue(const std::string &optim_type, const std::string &input_n… in UpdateOptimInputValue()
692 def __init__(self, optim_type='ApplyMomentum', only_shape_indices=None): argument