Searched refs:SkipgramOp (Results 1 – 1 of 1) sorted by relevance
49 class SkipgramOp : public OpKernel { class51 explicit SkipgramOp(OpKernelConstruction* ctx) in SkipgramOp() function in tensorflow::SkipgramOp240 REGISTER_KERNEL_BUILDER(Name("Skipgram").Device(DEVICE_CPU), SkipgramOp);