Searched refs:GatherNdOp (Results 1 – 4 of 4) sorted by relevance
33 class GatherNdOp : public OpKernel { class35 explicit GatherNdOp(OpKernelConstruction* c) : OpKernel(c) { in GatherNdOp() function in tensorflow::GatherNdOp57 GatherNdOp<dev##Device, type, index_type>)
267 class GatherNdOp : public XlaOpKernel { class269 explicit GatherNdOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in GatherNdOp() function in tensorflow::GatherNdOp302 REGISTER_XLA_OP(Name("GatherNd"), GatherNdOp);
146 TypeID::get<TF::GatherNdOp>(), in IsOpAllowedTf2XlaFallback()
1080 rewriter.replaceOpWithNewOp<TF::GatherNdOp>(gather_op, result_type, operand, in matchAndRewrite()