| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_analysis_test.cc | 55 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
|
| D | while_loop_analysis.cc | 99 optional<int64> GetLoopInductionVarTupleIdx(const HloInstruction* while_op) { in GetLoopInductionVarTupleIdx()
235 static optional<int64> PatternMatchLoopTripCount(HloInstruction* while_op, in PatternMatchLoopTripCount()
329 optional<int64> ComputeWhileLoopTripCount(HloInstruction* while_op, in ComputeWhileLoopTripCount()
418 optional<int64> ComputeWhileLoopTripCountUpperBound(HloInstruction* while_op) { in ComputeWhileLoopTripCountUpperBound()
|
| D | while_loop_simplifier.cc | 43 static StatusOr<bool> TryRemoveDeadWhileParams(HloInstruction* while_op) { in TryRemoveDeadWhileParams()
310 static StatusOr<bool> TryRemoveConstantParams(HloInstruction* while_op) { in TryRemoveConstantParams()
456 static StatusOr<bool> TryRemoveWhileLoop(HloInstruction* while_op) { in TryRemoveWhileLoop()
506 static StatusOr<bool> TryPropagateConstant(HloInstruction* while_op) { in TryPropagateConstant()
642 static StatusOr<bool> TryFlattenNestedTuples(HloInstruction* while_op) { in TryFlattenNestedTuples()
768 HloInstruction* while_op, PrimitiveType elem_ty) { in TryMergeInductionVariables()
|
| D | while_loop_simplifier_test.cc | 168 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
219 auto* while_op = computation->root_instruction(); in TEST_F() local
|
| D | heap_simulator_test.cc | 80 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
721 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
|
| D | buffer_assignment_test.cc | 864 auto while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
1760 auto while_op = builder.AddInstruction( in TEST_F() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 396 def _get_graph(while_op, func_attr_name): argument
423 def _create_grad_func(ys, xs, grads, cond_graph, body_graph, name, while_op, argument
514 def _resolve_grad_captures(body_graph, body_grad_graph, while_op): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 55 static Status Run(Node* while_op, const string& cond_fn_name, in Run()
152 LowerWhileHelper::LowerWhileHelper(Node* while_op, const string& cond_fn_name, in LowerWhileHelper()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_test.cc | 268 auto while_op = in TEST() local
420 auto while_op = in TEST() local
540 auto while_op = in TEST() local
689 auto while_op = in TEST() local
939 auto while_op = ops::While(scope.WithOpName("outer/LoopCond"), in TEST() local
995 auto while_op = in TEST() local
|
| D | tf2xla_util_test.cc | 357 auto while_op = in TEST() local
391 auto while_op = in TEST() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 329 def GetAccumulatorForInputAtIndex(while_op, idx): argument
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 736 auto while_op = in TEST_F() local
|