| { | |
| "use_habana_mixed_precision": false, | |
| "hmp_is_verbose": false, | |
| "use_fused_adam": true, | |
| "use_fused_clip_norm": true, | |
| "hmp_bf16_ops": [ | |
| "conv1d", | |
| "conv2d", | |
| "conv3d", | |
| "conv_transpose1d", | |
| "conv_transpose2d", | |
| "conv_transpose3d", | |
| "addmm", | |
| "matmul", | |
| "einsum", | |
| "mm", | |
| "mv", | |
| "linear", | |
| "bmm", | |
| "layer_norm", | |
| "group_norm" | |
| ], | |
| "hmp_fp32_ops": [ | |
| "acos", | |
| "asin", | |
| "cosh", | |
| "erfinv", | |
| "exp", | |
| "expm1", | |
| "log", | |
| "log10", | |
| "log2", | |
| "log1p", | |
| "reciprocal", | |
| "rsqrt", | |
| "sinh", | |
| "tan", | |
| "pow", | |
| "softplus", | |
| "cosine_similarity", | |
| "poisson_nll_loss", | |
| "cosine_embedding_loss", | |
| "nll_loss", | |
| "hinge_embedding_loss", | |
| "kl_div", | |
| "l1_loss", | |
| "smooth_l1_loss", | |
| "huber_loss", | |
| "mse_loss", | |
| "margin_ranking_loss", | |
| "multilabel_margin_loss", | |
| "soft_margin_loss", | |
| "triplet_margin_loss", | |
| "multi_margin_loss", | |
| "binary_cross_entropy_with_logits", | |
| "dist", | |
| "pdist", | |
| "cdist", | |
| "renorm", | |
| "logsumexp" | |
| ] | |
| } |