| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_regions.cc | 96 LogicalResult ConvertIfOp(IfOp if_op) { in ConvertIfOp()
142 if (IfOp if_op = llvm::dyn_cast<IfOp>(op)) { in runOnOperation() local
|
| D | stack_ops_decomposition.cc | 228 TF::IfOp if_op, ModuleOp module, in HandleIfOp()
534 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(&op)) { in DecomposeStackOpsInternal() local
|
| D | tensor_list_ops_decomposition.cc | 346 TF::IfRegionOp if_op, ModuleOp module, in HandleIfRegionOp()
893 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(&op)) { in DecomposeTensorListOpsInternal() local
929 } else if (auto if_op = llvm::dyn_cast<TF::IfRegionOp>(&op)) { in DecomposeTensorListOpsInternal() local
|
| D | optimize_global_tensors.cc | 110 if (auto if_op = dyn_cast<TF::IfOp>(op)) { in AnalyzeFunc() local
|
| D | merge_control_flow.cc | 291 block->walk([&](TF::IfRegionOp if_op) { in OptimizeIfRegions()
|
| D | resource_device_inference.cc | 289 } else if (auto if_op = dyn_cast<IfOp>(op)) { in runOnOperation() local
|
| D | tensor_array_ops_decomposition.cc | 497 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(op)) { in AccessedGradients() local
640 LogicalResult HandleIfOp(TF::IfOp if_op, ModuleOp module, in HandleIfOp()
899 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(&op)) { in DecomposeTensorArrayOps() local
|
| D | functional_control_flow_to_cfg.cc | 285 if (IfOp if_op = llvm::dyn_cast<IfOp>(op)) { in runOnFunction() local
|
| D | region_control_flow_to_functional.cc | 345 auto if_op = builder.create<IfOp>( in ConvertIfOp() local
|
| D | resource_op_lifting_cleanup.cc | 419 if (auto if_op = dyn_cast<TF::IfOp>(op)) { in CleanupAndCanonicalize() local
|
| D | collection_ops_util.cc | 182 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(use.getOwner())) { in GetElementTypeFromAccess() local
|
| D | tpu_extract_outside_compilation.cc | 361 if (auto if_op = llvm::dyn_cast<TF::IfRegionOp>(op)) { in DecomposeControlFlow() local
|
| D | shape_inference.cc | 1066 if (auto if_op = dyn_cast<IfOp>(op)) return InferShapeForIf(if_op); in InferShapeForSingleOperation() local
1325 if (auto if_op = dyn_cast<TF::IfOp>(op)) { in PropagateShapeIntoAttachedFunctions() local
|
| D | resource_op_lifting.cc | 1246 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(&op)) { in HoistForControlFlow() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | sink_constants_to_control_flow.cc | 50 } else if (auto if_op = llvm::dyn_cast<IfOp>(op)) { in runOnFunction() local
|
| D | legalize_control_flow.cc | 63 LogicalResult LowerIfOp(mlir::mhlo::IfOp if_op) { in LowerIfOp()
|
| D | transform_unranked_hlo.cc | 237 auto if_op = rewriter.create<scf::IfOp>( in matchAndRewrite() local
432 scf::IfOp if_op = createIfOpForRankSpecializedBroadcastAndOp( in HandleBroadcastAndOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 109 auto if_op = builder.create<mhlo::IfOp>(loc, result_type, op.cond(), in LowerIf() local
273 auto if_op = builder.create<mhlo::IfOp>(loc, result_type, op.cond(), in LowerIfRegion() local
379 if (auto if_op = dyn_cast<TF::IfOp>(op)) { in runOnOperation() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.cc | 113 CondBuilder::CondBuilder(Node* if_op, const NameAttrList& then_fn, in CondBuilder()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | rearrange_function_argument_pass_test.cc | 96 auto if_op = ops::If(s.WithOpName("if"), arg1, in TEST() local
|
| D | extract_outside_compilation_pass_test.cc | 472 auto if_op = ops::If(s.WithOpName("if"), cond, in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 326 } else if (auto if_op = dyn_cast<IfOp>(op)) { in ResourceAliasAnalysisInfo() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_test.cc | 156 auto if_op = in CheckGraphDef() local
|