Searched refs:ResourceGatherOp (Results 1 – 2 of 2) sorted by relevance
123 class ResourceGatherOp : public XlaOpKernel { class125 explicit ResourceGatherOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in ResourceGatherOp() function in tensorflow::__anon5691afd60111::ResourceGatherOp147 REGISTER_XLA_OP(Name("ResourceGather"), ResourceGatherOp);
586 class ResourceGatherOp : public OpKernel { class612 explicit ResourceGatherOp(OpKernelConstruction* c) : OpKernel(c) { in ResourceGatherOp() function in tensorflow::ResourceGatherOp711 ResourceGatherOp<dev##Device, type, index_type>)737 ResourceGatherOp<GPUDevice, Variant, int32>)744 ResourceGatherOp<GPUDevice, Variant, int64>)