Searched refs:MergeOp (Results 1 – 3 of 3) sorted by relevance
215 MergeOp::MergeOp(OpKernelConstruction* context) : OpKernel(context) { in MergeOp() function in tensorflow::MergeOp222 void MergeOp::Compute(OpKernelContext* context) { in Compute()246 REGISTER_KERNEL_BUILDER(Name("Merge").Device(DEVICE_CPU), MergeOp);247 REGISTER_KERNEL_BUILDER(Name("RefMerge").Device(DEVICE_CPU), MergeOp);254 MergeOp);261 MergeOp);281 MergeOp);290 MergeOp);308 MergeOp); \315 MergeOp)[all …]
52 class MergeOp : public OpKernel {54 explicit MergeOp(OpKernelConstruction* context);57 ~MergeOp() override {} in ~MergeOp()59 TF_DISALLOW_COPY_AND_ASSIGN(MergeOp);
181 Name("Merge").Device(DEVICE).HostMemory("value_index"), MergeOp); \