Searched refs:WhileOp (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 323 class WhileOp : public AsyncOpKernel { class
325 explicit WhileOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in WhileOp() function in tensorflow::__anonf4cf7ed10111::WhileOp
330 ~WhileOp() override {} in ~WhileOp()
357 State(WhileOp* kernel, OpKernelContext* ctx, FHandle cond_handle, in State()
376 WhileOp* const kernel_;
478 REGISTER_KERNEL_BUILDER(Name("_While").Device(DEVICE_CPU), WhileOp);
479 REGISTER_KERNEL_BUILDER(Name("_While").Device(DEVICE_GPU), WhileOp);
481 REGISTER_KERNEL_BUILDER(Name("While").Device(DEVICE_CPU), WhileOp);
482 REGISTER_KERNEL_BUILDER(Name("While").Device(DEVICE_GPU), WhileOp);
484 REGISTER_KERNEL_BUILDER(Name("StatelessWhile").Device(DEVICE_CPU), WhileOp);
[all …]
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 83 if isinstance(op, WhileOp):
95 class WhileOp(object): class
1203 while_op = WhileOp(
1303 assert not isinstance(y_op, WhileOp)
|