Searched refs:NegTrainOp (Results 1 – 1 of 1) sorted by relevance
242 class NegTrainOp : public OpKernel { class244 explicit NegTrainOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in NegTrainOp() function in tensorflow::NegTrainOp261 ~NegTrainOp() override { delete sampler_; } in ~NegTrainOp()354 REGISTER_KERNEL_BUILDER(Name("NegTrain").Device(DEVICE_CPU), NegTrainOp);