Searched refs:ResourceGatherOp (Results 1 – 2 of 2) sorted by relevance
124 class ResourceGatherOp : public XlaOpKernel { class126 explicit ResourceGatherOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ResourceGatherOp() function in tensorflow::__anond3063dcd0111::ResourceGatherOp145 REGISTER_XLA_OP(Name("ResourceGather"), ResourceGatherOp);
615 class ResourceGatherOp : public OpKernel { class617 explicit ResourceGatherOp(OpKernelConstruction* c) : OpKernel(c) { in ResourceGatherOp() function in tensorflow::ResourceGatherOp739 ResourceGatherOp<dev##Device, type, index_type>)766 ResourceGatherOp<GPUDevice, Variant, int32>)773 ResourceGatherOp<GPUDevice, Variant, int64>)