Searched defs:param_state (Results 1 – 8 of 8) sorted by relevance
103 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
86 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
88 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
87 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
436 auto param_state = state_.find(_params.at(0).unsafeGetTensorImpl()); in step() local
107 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
42 DerivedOptimizerParamState param_state; in serialize() local