Home
last modified time | relevance | path

Searched refs:SwitchNOp (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcontrol_flow_ops.h54 class SwitchNOp : public OpKernel {
56 explicit SwitchNOp(OpKernelConstruction* context) : OpKernel(context) {} in SwitchNOp() function
59 ~SwitchNOp() override {} in ~SwitchNOp()
61 TF_DISALLOW_COPY_AND_ASSIGN(SwitchNOp);
Dcontrol_flow_ops.cc42 void SwitchNOp::Compute(OpKernelContext* context) { in Compute()
62 SwitchNOp);
74 SwitchNOp)
93 SwitchNOp)
138 SwitchNOp)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor.cc463 LogicalResult Verify(SwitchNOp switchn) { in Verify()
515 void Print(SwitchNOp switchn, OpAsmPrinter &p) { in Print()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dshape_inference.cc118 tf_executor::NextIterationSinkOp, tf_executor::SwitchNOp, in IsSupportedNonTFOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1741 return builder_.create<mlir::tf_executor::SwitchNOp>(loc, types, operands, in CreateOperation()