Skip to content

Commit 0599eb6

Browse files
co63oclixcli
authored andcommitted
【Hackathon 6th Fundable Projects 3 No.252】 nce (PaddlePaddle#65897)
* Fix * Fix
1 parent bab1629 commit 0599eb6

File tree

11 files changed

+771
-467
lines changed

11 files changed

+771
-467
lines changed

paddle/fluid/operators/nce_op.cc

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,10 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
See the License for the specific language governing permissions and
1313
limitations under the License. */
1414

15-
#include "paddle/fluid/operators/nce_op.h"
16-
1715
#include <memory>
1816
#include <string>
1917
#include <vector>
18+
#include "paddle/fluid/framework/op_registry.h"
2019

2120
namespace paddle {
2221
namespace operators {
@@ -321,8 +320,3 @@ REGISTER_OPERATOR(nce_grad,
321320
ops::NCEOpGrad,
322321
ops::NCEOpGradVarTypeInference,
323322
ops::NCEGradOpNoNeedBufferVarInferer);
324-
325-
PD_REGISTER_STRUCT_KERNEL(nce, CPU, ALL_LAYOUT, ops::NCEKernel, float, double) {
326-
}
327-
PD_REGISTER_STRUCT_KERNEL(
328-
nce_grad, CPU, ALL_LAYOUT, ops::NCEGradKernel, float, double) {}

0 commit comments

Comments
 (0)