| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | merge_tf_if_ops.cc | 117 auto if_op = llvm::dyn_cast<mlir::TF::IfOp>(&op); in ProcessFunction() local
166 for (auto if_op : if_ops) { in MergeIfOpsWithSameArgs() local
196 for (auto if_op : if_ops) { in MergeIfOpsWithSameArgs() local
227 for (auto if_op : if_ops) { in CreateBranchFunction() local
|
| D | optimize_tf_control_flow_side_effect.cc | 43 if (auto if_op = llvm::dyn_cast<mlir::TF::IfOp>(op)) { in FunctionHasSideEffect() local
109 if (auto if_op = llvm::dyn_cast<mlir::TF::IfOp>(op)) { in runOnOperation() local
|
| D | remove_tf_if_const_args.cc | 55 auto if_op = llvm::dyn_cast<mlir::TF::IfOp>(&op); in ProcessFunction() local
81 void RemoveConstArgsFromTfIfOp(mlir::OpBuilder &builder, mlir::TF::IfOp if_op, in RemoveConstArgsFromTfIfOp()
|
| D | reorder_assert.cc | 55 if (auto if_op = llvm::dyn_cast<mlir::TF::IfOp>(&op)) { in ProcessFunction() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_regions.cc | 98 LogicalResult ConvertIfOp(IfOp if_op) { in ConvertIfOp()
144 if (IfOp if_op = llvm::dyn_cast<IfOp>(op)) { in runOnOperation() local
|
| D | merge_control_flow.cc | 79 llvm::SmallSetVector<Operation*, 4> GetAllOpsFromIf(TF::IfRegionOp if_op) { in GetAllOpsFromIf()
358 for (auto& if_op : if_op_segment) { in GetReturnIndicesToKeep() local
485 auto if_op = index_and_value.value(); in CreateYieldOps() local
498 auto if_op = index_and_value.value(); in CreateYieldOps() local
531 TF::IfRegionOp if_op = if_op_segment[index_and_value.index()]; in MergeIfPerSegment() local
607 block->walk([&](TF::IfRegionOp if_op) { in OptimizeIfRegions()
|
| D | tf_saved_model_freeze_variables.cc | 113 } else if (auto if_op = dyn_cast<TF::IfOp>(user_op)) { in PropagateUsage() local
120 } else if (auto if_op = dyn_cast<TF::IfRegionOp>(user_op)) { in PropagateUsage() local
|
| D | stack_ops_decomposition.cc | 201 TF::IfOp if_op, ModuleOp module, in HandleIfOp()
509 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(&op)) { in DecomposeStackOpsInternal() local
|
| D | tensor_list_ops_decomposition.cc | 333 TF::IfRegionOp if_op, ModuleOp module, in HandleIfRegionOp()
883 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(&op)) { in DecomposeTensorListOpsInternal() local
919 } else if (auto if_op = llvm::dyn_cast<TF::IfRegionOp>(&op)) { in DecomposeTensorListOpsInternal() local
|
| D | resource_device_inference.cc | 285 } else if (auto if_op = dyn_cast<IfOp>(op)) { in runOnOperation() local
|
| D | tensor_array_ops_decomposition.cc | 496 } 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()
902 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(&op)) { in DecomposeTensorArrayOps() local
|
| D | functional_control_flow_to_cfg.cc | 287 if (IfOp if_op = llvm::dyn_cast<IfOp>(op)) { in runOnOperation() local
|
| D | region_control_flow_to_functional.cc | 364 auto if_op = builder.create<IfOp>( in ConvertIfOp() local
|
| D | resource_op_lifting_cleanup.cc | 424 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
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_value_typed_analyzer.cc | 129 if (auto if_op = dyn_cast<TF::IfOp>(op)) { in AnalyzeRegion() local
136 if (auto if_op = dyn_cast<TF::IfRegionOp>(op)) { in AnalyzeRegion() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 139 auto if_op = builder.create<mhlo::IfOp>(loc, op.getResultTypes(), op.cond()); in LowerIf() local
323 auto if_op = builder.create<mhlo::IfOp>(loc, op.getResultTypes(), op.cond()); in LowerIfRegion() local
430 if (auto if_op = dyn_cast<TF::IfOp>(op)) { in runOnOperation() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | merge_clusters.cc | 348 bool IsInsideIfThenBranch(mlir::TF::IfRegionOp if_op, in IsInsideIfThenBranch()
356 mlir::LogicalResult DecomposeIf(mlir::TF::IfRegionOp if_op, in DecomposeIf()
464 if (auto if_op = mlir::dyn_cast<mlir::TF::IfRegionOp>(op)) { in DecomposeControlflow() local
|
| D | spmd_expander_common.cc | 363 if (auto if_op = owner->getParentOfType<mlir::TF::IfRegionOp>()) { in TraceUseToNextTFOp() local
754 if (auto if_op = mlir::dyn_cast<mlir::TF::IfRegionOp>(op)) { in next() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | control_flow_spmd_expander.cc | 65 auto if_op = llvm::cast<mlir::TF::IfRegionOp>(op); in ExpandOp() local
|
| D | save_restore_spmd_expander.cc | 519 auto if_op = builder.create<mlir::TF::IfOp>( in ExpandMergeV2Op() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | canonicalize.cc | 113 LogicalResult matchAndRewrite(scf::IfOp if_op, in matchAndRewrite()
|
| /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
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | tfrt_graph_execution_state_test.cc | 602 auto if_op = in TEST_F() local
669 auto if_op = in TEST_F() local
|