Add cross_entropy to fp32 ops
Browse files- gaudi_config.json +1 -1
gaudi_config.json
CHANGED
@@ -24,6 +24,6 @@
|
|
24 |
"embedding",
|
25 |
"nll_loss",
|
26 |
"log_softmax",
|
27 |
-
"
|
28 |
]
|
29 |
}
|
|
|
24 |
"embedding",
|
25 |
"nll_loss",
|
26 |
"log_softmax",
|
27 |
+
"cross_entropy"
|
28 |
]
|
29 |
}
|