Searched defs:get_weights (Results 1 – 10 of 10) sorted by relevance
56 ITensor *get_weights() override in get_weights() function
22 def get_weights(self): member in SimpleMegatronLM
124 ICLTensor *get_weights() override in get_weights() function
140 def get_weights(self): member in Optimizer
752 def get_weights(self): member in LossScaleOptimizer
1083 def get_weights(self): member in OptimizerV2
1306 def get_weights(self): member in Layer
2072 def get_weights(self): member in Model
1808 def get_weights(self): member in Layer
168 def get_weights(self): member in Model