Searched full:force_amp_for_fp16_bf16_models (Results 1 – 4 of 4) sorted by relevance
112 FORCE_AMP_FOR_FP16_BF16_MODELS = { variable217 def force_amp_for_fp16_bf16_models(self): member in TimmRunner218 return FORCE_AMP_FOR_FP16_BF16_MODELS
173 def force_amp_for_fp16_bf16_models(self): member in TorchBenchmarkRunner174 return self._config["dtype"]["force_amp_for_fp16_bf16_models"]
106 force_amp_for_fp16_bf16_models:
2178 def force_amp_for_fp16_bf16_models(self): member in BenchmarkRunner2248 if self.args.only in self.force_amp_for_fp16_bf16_models:2258 if self.args.only in self.force_amp_for_fp16_bf16_models: