Searched refs:CaseOp (Results 1 – 1 of 1) sorted by relevance
215 class CaseOp : public AsyncOpKernel { class217 explicit CaseOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in CaseOp() function in tensorflow::__anonf4cf7ed10111::CaseOp223 ~CaseOp() override {} in ~CaseOp()256 State(CaseOp* kernel, OpKernelContext* ctx, int branch, in State()295 CaseOp* const kernel_;315 REGISTER_KERNEL_BUILDER(Name("Case").Device(DEVICE_CPU), CaseOp);317 Name("Case").Device(DEVICE_GPU).HostMemory("branch_index"), CaseOp);