Searched defs:get_scale (Results 1 – 7 of 7) sorted by relevance
25 def get_scale(self, match: InternalMatch) -> Optional[float]: member in ConvertToSDPAPass
213 def get_scale(x: torch.Tensor) -> torch.Tensor: function
51 get_scale(nir_builder *b, lower_drawpixels_state *state) in get_scale() function
329 inline double get_scale() const { in get_scale() function
536 def get_scale(self) -> float: member in GradScaler
175 float get_scale(const Tensor& scale, size_t channel_ix) { in get_scale() function
139 pub fn get_scale(&self) -> (i32, i32) { in get_scale() method