Home
last modified time | relevance | path

Searched defs:while_op (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_analysis_test.cc55 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() local
86 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() local
119 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() local
154 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() local
193 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() local
232 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() local
267 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() local
Dwhile_loop_analysis.cc101 const HloInstruction* while_op) { in GetAuxiliaryLoopInductionVars()
245 optional<int64> GetLoopInductionVarTupleIdx(const HloInstruction* while_op) { in GetLoopInductionVarTupleIdx()
381 static optional<int64> PatternMatchLoopTripCount(HloInstruction* while_op, in PatternMatchLoopTripCount()
475 optional<int64> ComputeWhileLoopTripCount(HloInstruction* while_op, in ComputeWhileLoopTripCount()
564 optional<int64> ComputeWhileLoopTripCountUpperBound(HloInstruction* while_op) { in ComputeWhileLoopTripCountUpperBound()
Dwhile_loop_simplifier.cc59 HloInstruction* while_op, absl::flat_hash_set<int64>& used_tuple_indices) { in RemoveDeadTupleIndices()
230 static StatusOr<bool> TryRemoveDeadWhileParams(HloInstruction* while_op) { in TryRemoveDeadWhileParams()
368 HloInstruction* while_op, const int64_t tuple_index, in TryRemoveRepeatedWhileTupleIndicesHelper()
419 HloInstruction* while_op) { in TryRemoveRepeatedWhileTupleIndices()
524 static StatusOr<bool> TryRemoveConstantParams(HloInstruction* while_op) { in TryRemoveConstantParams()
671 static StatusOr<bool> TryRemoveWhileLoop(HloInstruction* while_op) { in TryRemoveWhileLoop()
756 static StatusOr<bool> TryPropagateConstant(HloInstruction* while_op) { in TryPropagateConstant()
892 static StatusOr<bool> TryFlattenNestedTuples(HloInstruction* while_op) { in TryFlattenNestedTuples()
1021 HloInstruction* while_op, PrimitiveType elem_ty) { in TryMergeInductionVariables()
Dwhile_loop_simplifier_test.cc168 auto* while_op = computation->root_instruction(); in TEST_F() local
185 auto* while_op = computation->root_instruction(); in TEST_F() local
201 auto* while_op = computation->root_instruction(); in TEST_F() local
217 auto* while_op = computation->root_instruction(); in TEST_F() local
232 auto* while_op = computation->root_instruction(); in TEST_F() local
Dwhile_loop_concat_code_motion_test.cc93 auto while_op = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_control_flow_flattening_test.cc65 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F() local
119 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F() local
175 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Doptimize_tf_control_flow_side_effect.cc33 if (auto while_op = llvm::dyn_cast<mlir::TF::WhileOp>(op)) { in FunctionHasSideEffect() local
92 if (auto while_op = llvm::dyn_cast<mlir::TF::WhileOp>(op)) { in runOnOperation() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf_while.cc72 void RunOnWhile(TF::WhileOp while_op) { in RunOnWhile()
88 func.getBody().walk([](TF::WhileOp while_op) { RunOnWhile(while_op); }); in RunOnFunction()
Dwhile_loop_outline.cc80 bool IsAlreadyOutlined(WhileOp while_op) { in IsAlreadyOutlined()
199 void WhileOutlinePass::OutlineWhile(WhileOp while_op) { in OutlineWhile()
290 [&](mlir::TFL::WhileOp while_op) { OutlineWhile(while_op); }); in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_saved_model_freeze_variables.cc123 } else if (auto while_op = dyn_cast<TF::WhileOp>(user_op)) { in PropagateUsage() local
129 } else if (auto while_op = dyn_cast<TF::WhileRegionOp>(user_op)) { in PropagateUsage() local
292 T GetUpdatedWhileOp(T while_op, in GetUpdatedWhileOp()
380 } else if (auto while_op = dyn_cast<TF::WhileOp>(items.getFirst())) { in runOnOperation() local
385 } else if (auto while_op = dyn_cast<TF::WhileRegionOp>(items.getFirst())) { in runOnOperation() local
Dfunctional_control_flow_to_regions.cc116 LogicalResult ConvertWhileOp(WhileOp while_op) { in ConvertWhileOp()
147 } else if (auto while_op = llvm::dyn_cast<WhileOp>(op)) { in runOnOperation() local
Dtpu_variable_runtime_reformatting.cc155 TF::WhileRegionOp while_op, tf_device::ReplicateOp replicate, in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
391 void HandleReplicateOp(TF::WhileRegionOp while_op, in HandleReplicateOp()
533 module.walk([&](TF::WhileRegionOp while_op) { in runOnOperation()
Dtensor_list_ops_decomposition.cc148 TF::WhileOp while_op, ModuleOp module, in HandleWhileOp()
266 TF::WhileRegionOp while_op, ModuleOp module, in HandleWhileRegionOp()
872 } else if (auto while_op = llvm::dyn_cast<TF::WhileOp>(&op)) { in DecomposeTensorListOpsInternal() local
908 } else if (auto while_op = llvm::dyn_cast<TF::WhileRegionOp>(&op)) { in DecomposeTensorListOpsInternal() local
Dstack_ops_decomposition.cc167 TF::WhileOp while_op, ModuleOp module, in HandleWhileOp()
536 } else if (auto while_op = llvm::dyn_cast<TF::WhileOp>(&op)) { in DecomposeStackOpsInternal() local
Dtensor_array_ops_decomposition.cc491 } else if (auto while_op = llvm::dyn_cast<TF::WhileOp>(op)) { in AccessedGradients() local
555 LogicalResult HandleWhileOp(TF::WhileOp while_op, ModuleOp module, in HandleWhileOp()
893 } else if (auto while_op = llvm::dyn_cast<TF::WhileOp>(&op)) { in DecomposeTensorArrayOps() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_value_typed_analyzer.cc142 if (auto while_op = dyn_cast<TF::WhileOp>(op)) { in AnalyzeRegion() local
149 if (auto while_op = dyn_cast<TF::WhileRegionOp>(op)) { in AnalyzeRegion() local
Dresource_alias_analysis.cc361 } else if (auto while_op = dyn_cast<WhileOp>(op)) { in ResourceAliasAnalysisInfo() local
466 Operation* while_op, const BacktrackAnalysisInfo& body_info) { in AnalyzeWhileLoop()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc163 auto while_op = builder.create<mhlo::WhileOp>( in LowerWhile() local
337 auto while_op = builder.create<mhlo::WhileOp>( in LowerWhileRegion() local
378 if (auto while_op = dyn_cast<TF::WhileOp>(op)) { in runOnOperation() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dsink_constants_to_control_flow.cc47 if (auto while_op = llvm::dyn_cast<WhileOp>(op)) { in runOnFunction() local
Dlegalize_control_flow.cc109 LogicalResult LowerWhileOp(mlir::mhlo::WhileOp while_op) { in LowerWhileOp()
/external/tensorflow/tensorflow/compiler/jit/
Drearrange_function_argument_pass_test.cc104 auto while_op = in TEST() local
217 auto while_op = ops::While(s.WithOpName("while"), in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op.cc65 static Status Run(Node* while_op, const NameAttrList& cond_fn, in Run()
180 LowerWhileHelper::LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn, in LowerWhileHelper()
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py620 def _get_graph(while_op, func_attr_name, attr_graph_name): argument
643 def _create_grad_func(ys, xs, grads, cond_graph, body_graph, name, while_op, argument
754 def _resolve_grad_captures(body_graph, body_grad_graph, while_op): argument
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc376 auto while_op = in TEST() local
526 auto while_op = in TEST() local
644 auto while_op = in TEST() local
793 auto while_op = in TEST() local
1149 auto while_op = ops::While(scope.WithOpName("outer/LoopCond"), in CheckOuterNodesFunctionalized() local
1204 auto while_op = in CheckOuterNodesFunctionalized() local
Dtf2xla_util_test.cc362 auto while_op = in TEST() local
396 auto while_op = in TEST() local

12