Searched refs:false_branch (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_control_flow.cc | 75 if_op.false_branch().cloneInto(orig_block->getParent(), in LowerIfOp()
80 Block* false_block = mapper.lookup(&if_op.false_branch().front()); in LowerIfOp()
97 if (failed(ReplaceTerminators(&if_op.false_branch(), tail_block, loc, mapper, in LowerIfOp()
|
| D | sink_constants_to_control_flow.cc | 52 SinkToRegion(&if_op.false_branch()); in runOnFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 116 ImportXlaRegion(op.else_function(), &if_op.false_branch(), loc); in LowerIf()
276 if_op.false_branch().takeBody(op.else_branch()); in LowerIfRegion()
|
| D | legalize_tf_communication.cc | 567 new_if.false_branch().takeBody(region_if.false_branch()); in RewriteRegionIfOp()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 872 false_branch = lambda: self.moving_mean function
873 return control_flow_util.smart_cond(training, true_branch, false_branch)
894 false_branch = lambda: self.moving_variance function
895 return control_flow_util.smart_cond(training, true_branch, false_branch)
|
| /external/llvm-project/mlir/include/mlir/Dialect/SPIRV/ |
| D | SPIRVControlFlowOps.td | 119 spv.BranchConditional %condition, ^true_branch, ^false_branch
120 spv.BranchConditional %condition, ^true_branch(%0: i32), ^false_branch(%1: i32)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 634 xla::XlaComputation false_branch; in ExportXlaOp() local
638 failed(ctx.converter->LowerRegionAsComputation(&op.false_branch(), in ExportXlaOp()
639 &false_branch))) { in ExportXlaOp()
650 xla::Conditional(pred, true_arg, true_branch, false_arg, false_branch); in ExportXlaOp()
|
| D | hlo_function_importer.cc | 525 &op.false_branch())); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 864 HloComputation* false_branch = in TEST_F() local
883 false_branch)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/ |
| D | tf_dialects.md | 73 {cond: @cond_func, true_branch: @func_foo, false_branch: @func_bar}
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 545 AnyRegion:$false_branch);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 2513 def false_branch(): function
2527 false_branch)
|