Searched refs:IfOp (Results 1 – 1 of 1) sorted by relevance
127 class IfOp : public AsyncOpKernel { class129 explicit IfOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in IfOp() function in tensorflow::__anon696c82e80111::IfOp136 ~IfOp() override {} in ~IfOp()166 State(IfOp* kernel, OpKernelContext* ctx, bool cond, FHandle then_handle, in State()202 IfOp* const kernel_;308 REGISTER_KERNEL_BUILDER(Name("_If").Device(DEVICE_CPU), IfOp);310 IfOp);312 REGISTER_KERNEL_BUILDER(Name("If").Device(DEVICE_CPU), IfOp);313 REGISTER_KERNEL_BUILDER(Name("If").Device(DEVICE_GPU).HostMemory("cond"), IfOp);319 REGISTER_KERNEL_BUILDER(Name("StatelessIf").Device(DEVICE_CPU), IfOp);[all …]