Home
last modified time | relevance | path

Searched refs:LoopCondOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcontrol_flow_ops.h104 class LoopCondOp : public OpKernel {
106 explicit LoopCondOp(OpKernelConstruction* context);
107 ~LoopCondOp() override;
113 TF_DISALLOW_COPY_AND_ASSIGN(LoopCondOp);
Dcontrol_flow_ops.cc606 LoopCondOp::LoopCondOp(OpKernelConstruction* context) : OpKernel(context) {} in LoopCondOp() function in tensorflow::LoopCondOp
607 LoopCondOp::~LoopCondOp() = default;
609 void LoopCondOp::Compute(OpKernelContext* context) { in Compute()
620 bool LoopCondOp::IsExpensive() { return false; } in IsExpensive()
622 REGISTER_KERNEL_BUILDER(Name("LoopCond").Device(DEVICE_CPU), LoopCondOp);
627 LoopCondOp);
634 LoopCondOp);
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h190 LoopCondOp); \