Searched refs:GatherTreeOp (Results 1 – 1 of 1) sorted by relevance
44 class GatherTreeOp : public OpKernel { class46 explicit GatherTreeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in GatherTreeOp() function in tensorflow::GatherTreeOp98 GatherTreeOp<CPUDevice, T>);189 GatherTreeOp<GPUDevice, T>);