Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor.cc463 LogicalResult Verify(SwitchNOp switchn) { in Verify() argument
464 IntegerAttr num_outs = switchn->getAttrOfType<IntegerAttr>("num_outs"); in Verify()
466 return switchn.emitOpError() << "expects a `num_outs` integer attribute"; in Verify()
469 if (switchn.getNumResults() != num_outs.getInt() + 1) in Verify()
470 return switchn.emitOpError() in Verify()
472 << (switchn.getNumResults() - 1); in Verify()
475 auto operand0_type = switchn.getOperand(0).getType(); in Verify()
478 return switchn.emitOpError() in Verify()
482 for (Type output_type : switchn.getResultTypes()) { in Verify()
487 return switchn.emitOpError() in Verify()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dbreakup-islands.mlir343 %switchn:4 = tf_executor._SwitchN %island#0, %arg1 of 3: tensor<*xi32>
344 tf_executor.fetch %switchn#0 : tensor<*xi32>
Dshape_inference.mlir360 …%switchn:3 = "tf_executor._SwitchN"(%island#0, %arg3) {num_outs = 2} : (tensor<?x?x?xf32>, tensor<…