| /external/tensorflow/tensorflow/core/kernels/ |
| D | control_flow_ops.h | 39 class SwitchOp : public OpKernel {
41 explicit SwitchOp(OpKernelConstruction* context) : OpKernel(context) {} in SwitchOp() function
44 ~SwitchOp() override {} in ~SwitchOp()
46 TF_DISALLOW_COPY_AND_ASSIGN(SwitchOp);
|
| D | control_flow_ops.cc | 26 void SwitchOp::Compute(OpKernelContext* context) { in Compute()
56 Name("Switch").Device(DEVICE_DEFAULT).HostMemory("pred"), SwitchOp);
58 Name("Switch").Device(DEVICE_TPU_SYSTEM).HostMemory("pred"), SwitchOp);
69 SwitchOp) \
81 SwitchOp)
88 SwitchOp) \
100 SwitchOp)
131 SwitchOp) \
148 SwitchOp)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_device_propagation.cc | 80 tf_executor::SwitchOp>(op); in IsSupportedExecutorOp()
116 const bool is_switch = llvm::isa<tf_executor::SwitchOp>(op); in FindDeviceFromOperands()
|
| D | shape_inference.cc | 119 tf_executor::SwitchOp, tf_executor::YieldOp>(op); in IsSupportedNonTFOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 53 return mlir::isa<mlir::tf_executor::SwitchOp, mlir::tf_executor::MergeOp, in IsControlFlowV1Op()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 438 void Print(SwitchOp switch_op, OpAsmPrinter &p) { in Print()
|
| /external/llvm/lib/CodeGen/SelectionDAG/ |
| D | SelectionDAGBuilder.cpp | 1964 SDValue SwitchOp = getValue(JTH.SValue); in visitJumpTableHeader() local
1965 EVT VT = SwitchOp.getValueType(); in visitJumpTableHeader()
1966 SDValue Sub = DAG.getNode(ISD::SUB, dl, VT, SwitchOp, in visitJumpTableHeader()
1975 SwitchOp = DAG.getZExtOrTrunc(Sub, dl, TLI.getPointerTy(DAG.getDataLayout())); in visitJumpTableHeader()
1980 JumpTableReg, SwitchOp); in visitJumpTableHeader()
2140 SDValue SwitchOp = getValue(B.SValue); in visitBitTestHeader() local
2141 EVT VT = SwitchOp.getValueType(); in visitBitTestHeader()
2142 SDValue Sub = DAG.getNode(ISD::SUB, dl, VT, SwitchOp, in visitBitTestHeader()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/CodeGen/SelectionDAG/ |
| D | SelectionDAGBuilder.cpp | 2455 SDValue SwitchOp = getValue(JTH.SValue); in visitJumpTableHeader() local
2456 EVT VT = SwitchOp.getValueType(); in visitJumpTableHeader()
2457 SDValue Sub = DAG.getNode(ISD::SUB, dl, VT, SwitchOp, in visitJumpTableHeader()
2466 SwitchOp = DAG.getZExtOrTrunc(Sub, dl, TLI.getPointerTy(DAG.getDataLayout())); in visitJumpTableHeader()
2471 JumpTableReg, SwitchOp); in visitJumpTableHeader()
2654 SDValue SwitchOp = getValue(B.SValue); in visitBitTestHeader() local
2655 EVT VT = SwitchOp.getValueType(); in visitBitTestHeader()
2657 DAG.getNode(ISD::SUB, dl, VT, SwitchOp, DAG.getConstant(B.First, dl, VT)); in visitBitTestHeader()
|
| /external/llvm-project/llvm/lib/CodeGen/SelectionDAG/ |
| D | SelectionDAGBuilder.cpp | 2428 SDValue SwitchOp = getValue(JTH.SValue); in visitJumpTableHeader() local
2429 EVT VT = SwitchOp.getValueType(); in visitJumpTableHeader()
2430 SDValue Sub = DAG.getNode(ISD::SUB, dl, VT, SwitchOp, in visitJumpTableHeader()
2439 SwitchOp = DAG.getZExtOrTrunc(Sub, dl, TLI.getPointerTy(DAG.getDataLayout())); in visitJumpTableHeader()
2444 JumpTableReg, SwitchOp); in visitJumpTableHeader()
2628 SDValue SwitchOp = getValue(B.SValue); in visitBitTestHeader() local
2629 EVT VT = SwitchOp.getValueType(); in visitBitTestHeader()
2631 DAG.getNode(ISD::SUB, dl, VT, SwitchOp, DAG.getConstant(B.First, dl, VT)); in visitBitTestHeader()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1744 return builder_.create<mlir::tf_executor::SwitchOp>(loc, types, operands, in CreateOperation()
|