Home
last modified time | relevance | path

Searched defs:if_op (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dmerge_tf_if_ops.cc117 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
Doptimize_tf_control_flow_side_effect.cc43 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
Dremove_tf_if_const_args.cc55 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()
Dreorder_assert.cc55 if (auto if_op = llvm::dyn_cast<mlir::TF::IfOp>(&op)) { in ProcessFunction() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_regions.cc98 LogicalResult ConvertIfOp(IfOp if_op) { in ConvertIfOp()
144 if (IfOp if_op = llvm::dyn_cast<IfOp>(op)) { in runOnOperation() local
Dmerge_control_flow.cc79 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()
Dtf_saved_model_freeze_variables.cc113 } 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
Dstack_ops_decomposition.cc201 TF::IfOp if_op, ModuleOp module, in HandleIfOp()
509 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(&op)) { in DecomposeStackOpsInternal() local
Dtensor_list_ops_decomposition.cc333 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
Dresource_device_inference.cc285 } else if (auto if_op = dyn_cast<IfOp>(op)) { in runOnOperation() local
Dtensor_array_ops_decomposition.cc496 } 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
Dfunctional_control_flow_to_cfg.cc287 if (IfOp if_op = llvm::dyn_cast<IfOp>(op)) { in runOnOperation() local
Dregion_control_flow_to_functional.cc364 auto if_op = builder.create<IfOp>( in ConvertIfOp() local
Dresource_op_lifting_cleanup.cc424 if (auto if_op = dyn_cast<TF::IfOp>(op)) { in CleanupAndCanonicalize() local
Dcollection_ops_util.cc182 } else if (auto if_op = llvm::dyn_cast<TF::IfOp>(use.getOwner())) { in GetElementTypeFromAccess() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_value_typed_analyzer.cc129 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/
Dlegalize_tf_control_flow.cc139 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/
Dmerge_clusters.cc348 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
Dspmd_expander_common.cc363 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/
Dcontrol_flow_spmd_expander.cc65 auto if_op = llvm::cast<mlir::TF::IfRegionOp>(op); in ExpandOp() local
Dsave_restore_spmd_expander.cc519 auto if_op = builder.create<mlir::TF::IfOp>( in ExpandMergeV2Op() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Dcanonicalize.cc113 LogicalResult matchAndRewrite(scf::IfOp if_op, in matchAndRewrite()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op.cc113 CondBuilder::CondBuilder(Node* if_op, const NameAttrList& then_fn, in CondBuilder()
/external/tensorflow/tensorflow/compiler/jit/
Drearrange_function_argument_pass_test.cc96 auto if_op = ops::If(s.WithOpName("if"), arg1, in TEST() local
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state_test.cc602 auto if_op = in TEST_F() local
669 auto if_op = in TEST_F() local

12