Home
last modified time | relevance | path

Searched defs:while_op (Results 1 – 25 of 41) 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.cc97 const HloInstruction* while_op) { in GetAuxiliaryLoopInductionVars()
241 optional<int64> GetLoopInductionVarTupleIdx(const HloInstruction* while_op) { in GetLoopInductionVarTupleIdx()
377 static optional<int64> PatternMatchLoopTripCount(HloInstruction* while_op, in PatternMatchLoopTripCount()
471 optional<int64> ComputeWhileLoopTripCount(HloInstruction* while_op, in ComputeWhileLoopTripCount()
560 optional<int64> ComputeWhileLoopTripCountUpperBound(HloInstruction* while_op) { in ComputeWhileLoopTripCountUpperBound()
Dwhile_loop_simplifier.cc44 HloInstruction* while_op, absl::flat_hash_set<int64>& used_tuple_indices) { in RemoveDeadTupleIndices()
213 static StatusOr<bool> TryRemoveDeadWhileParams(HloInstruction* while_op) { in TryRemoveDeadWhileParams()
335 HloInstruction* while_op, const int64 tuple_index, in TryRemoveRepeatedWhileTupleIndicesHelper()
386 HloInstruction* while_op) { in TryRemoveRepeatedWhileTupleIndices()
491 static StatusOr<bool> TryRemoveConstantParams(HloInstruction* while_op) { in TryRemoveConstantParams()
637 static StatusOr<bool> TryRemoveWhileLoop(HloInstruction* while_op) { in TryRemoveWhileLoop()
718 static StatusOr<bool> TryPropagateConstant(HloInstruction* while_op) { in TryPropagateConstant()
854 static StatusOr<bool> TryFlattenNestedTuples(HloInstruction* while_op) { in TryFlattenNestedTuples()
980 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
Dhlo_live_range_test.cc318 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf_while.cc49 void RunOnWhile(TF::WhileOp while_op) { in RunOnWhile()
76 func.getBody().walk([](TF::WhileOp while_op) { RunOnWhile(while_op); }); in RunOnFunction()
Dwhile_loop_outline.cc65 bool IsAlreadyOutlined(WhileOp while_op) { in IsAlreadyOutlined()
99 void WhileOutlinePass::OutlineWhile(WhileOp while_op) { in OutlineWhile()
267 [&](mlir::TFL::WhileOp while_op) { OutlineWhile(while_op); }); in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
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.cc146 TF::WhileRegionOp while_op, tf_device::ReplicateOp replicate, in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
382 void HandleReplicateOp(TF::WhileRegionOp while_op, in HandleReplicateOp()
524 module.walk([&](TF::WhileRegionOp while_op) { in runOnOperation()
Dtensor_list_ops_decomposition.cc164 TF::WhileOp while_op, ModuleOp module, in HandleWhileOp()
282 TF::WhileRegionOp while_op, ModuleOp module, in HandleWhileRegionOp()
888 } else if (auto while_op = llvm::dyn_cast<TF::WhileOp>(&op)) { in DecomposeTensorListOpsInternal() local
924 } else if (auto while_op = llvm::dyn_cast<TF::WhileRegionOp>(&op)) { in DecomposeTensorListOpsInternal() local
Dstack_ops_decomposition.cc160 TF::WhileOp while_op, ModuleOp module, in HandleWhileOp()
529 } else if (auto while_op = llvm::dyn_cast<TF::WhileOp>(&op)) { in DecomposeStackOpsInternal() local
Doptimize_global_tensors.cc116 if (auto while_op = dyn_cast<TF::WhileOp>(op)) { in AnalyzeFunc() local
Dresource_device_inference.cc283 if (auto while_op = dyn_cast<WhileOp>(op)) { in runOnOperation() local
Dtensor_array_ops_decomposition.cc492 } else if (auto while_op = llvm::dyn_cast<TF::WhileOp>(op)) { in AccessedGradients() local
556 LogicalResult HandleWhileOp(TF::WhileOp while_op, ModuleOp module, in HandleWhileOp()
894 } else if (auto while_op = llvm::dyn_cast<TF::WhileOp>(&op)) { in DecomposeTensorArrayOps() local
Dresource_op_lifting.cc662 if (auto while_op = dyn_cast<TF::WhileRegionOp>(op)) in ReplaceOpWithNewOp() local
897 LogicalResult HandleWhileLoop(TF::WhileOp while_op, FuncOp body, FuncOp cond) { in HandleWhileLoop()
1237 if (auto while_op = llvm::dyn_cast<TF::WhileOp>(&op)) { in HoistForControlFlow() local
Dfunctional_control_flow_to_cfg.cc291 if (WhileOp while_op = llvm::dyn_cast<WhileOp>(op)) { in runOnFunction() 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.cc108 LogicalResult LowerWhileOp(mlir::mhlo::WhileOp while_op) { in LowerWhileOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc162 auto while_op = builder.create<mhlo::WhileOp>( in LowerWhile() local
333 auto while_op = builder.create<mhlo::WhileOp>( in LowerWhileRegion() local
371 if (auto while_op = dyn_cast<TF::WhileOp>(op)) { in runOnOperation() local
/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()
177 LowerWhileHelper::LowerWhileHelper(Node* while_op, const NameAttrList& cond_fn, in LowerWhileHelper()
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py585 def _get_graph(while_op, func_attr_name, attr_graph_name): argument
608 def _create_grad_func(ys, xs, grads, cond_graph, body_graph, name, while_op, argument
717 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
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.cc316 } else if (auto while_op = dyn_cast<WhileOp>(op)) { in ResourceAliasAnalysisInfo() local
399 Operation* while_op, const BacktrackAnalysisInfo& body_info) { in AnalyzeWhileLoop()

12