Searched refs:XlaCaseOp (Results 1 – 2 of 2) sorted by relevance
44 class XlaCaseOp : public XlaOpKernel {46 explicit XlaCaseOp(OpKernelConstruction* ctx);51 TF_DISALLOW_COPY_AND_ASSIGN(XlaCaseOp);
30 XlaCaseOp::XlaCaseOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in XlaCaseOp() function in tensorflow::XlaCaseOp46 XlaCaseOp::GetPrunedBranchesAndIndex(XlaOpKernelContext* ctx) { in GetPrunedBranchesAndIndex()66 void XlaCaseOp::Compile(XlaOpKernelContext* ctx) { in Compile()373 XlaCaseOp);375 XlaCaseOp);