Searched refs:SplitOp (Results 1 – 1 of 1) sorted by relevance
32 class SplitOp : public XlaOpKernel { class34 explicit SplitOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SplitOp() function in tensorflow::__anonafb9b3290111::SplitOp97 REGISTER_XLA_OP(Name("Split").CompileTimeConstantInput("split_dim"), SplitOp);