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